Add wait for shards to be ready after update (#44)

rlmanrique · web-flow · commit 9f98220a30f0 · 2025-02-11T17:09:04.000+10:00
diff --git a/benchmarker/cmd/ann_benchmark.go b/benchmarker/cmd/ann_benchmark.go
@@ -1081,6 +1081,10 @@ var annBenchmarkCommand = &cobra.Command{
 						log.Fatalf("Error waiting for tombstones to be empty: %v", err)
 					}
 				}
+				if !cfg.SkipAsyncReady {
+					startTime := time.Now()
+					waitReady(&cfg, client, startTime, 30*time.Minute, 1000)
+				}
 
 				runQueries(&cfg, importTime, testData, neighbors, testFilters)
 

Original file line number	Diff line number	Diff line change
`@@ -1081,6 +1081,10 @@ var annBenchmarkCommand = &cobra.Command{`
`1081`	`1081`	`log.Fatalf("Error waiting for tombstones to be empty: %v", err)`
`1082`	`1082`	`}`
`1083`	`1083`	`}`
	`1084`	`+ if !cfg.SkipAsyncReady {`
	`1085`	`+ startTime := time.Now()`
	`1086`	`+ waitReady(&cfg, client, startTime, 30*time.Minute, 1000)`
	`1087`	`+ }`
`1084`	`1088`
`1085`	`1089`	`runQueries(&cfg, importTime, testData, neighbors, testFilters)`
`1086`	`1090`