fix: add error check for flushing

Signed-off-by: hlts2 <[email protected]>
vdaas · Sep 9, 2024 · 42d0553 · 42d0553
1 parent 3433f28
commit 42d0553
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/agent/core/ngt/handler/grpc/index.go b/pkg/agent/core/ngt/handler/grpc/index.go
@@ -64,6 +64,9 @@ func (s *server) CreateIndex(
 					},
 				}, info.Get())...)
 			code = codes.FailedPrecondition
+		case errors.Is(err, errors.ErrFlushingIsInProgress):
+			err = status.WrapWithAborted("CreateIndex API aborted to process create indexes request due to flushing indices is in progress", err, details...)
+			code = codes.Aborted
 		case errors.Is(err, context.Canceled):
 			err = status.WrapWithCanceled(fmt.Sprintf("CreateIndex API canceled to create indexes pool_size = %d, error: %v", c.GetPoolSize(), err), err, details...)
 			code = codes.Canceled
@@ -149,6 +152,9 @@ func (s *server) CreateAndSaveIndex(
 					},
 				}, info.Get())...)
 			code = codes.FailedPrecondition
+		case errors.Is(err, errors.ErrFlushingIsInProgress):
+			err = status.WrapWithAborted("CreateIndex API aborted to process create indexes request due to flushing indices is in progress", err, details...)
+			code = codes.Aborted
 		case errors.Is(err, context.Canceled):
 			err = status.WrapWithCanceled(fmt.Sprintf("CreateAndSaveIndex API canceled to create indexes pool_size = %d, error: %v", c.GetPoolSize(), err), err, details...)
 			code = codes.Canceled