change custom error to ErrGRPCUnhealthy

Signed-off-by: Nic <qianyong@api7.ai>
k3s-io · Dec 6, 2024 · f23c1dc · f23c1dc
1 parent 5e1d414
commit f23c1dc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pkg/logstructured/logstructured.go b/pkg/logstructured/logstructured.go
@@ -437,7 +437,7 @@ func (l *LogStructured) Watch(ctx context.Context, prefix string, revision int64
 				wr.CompactRevision = compact
 				wr.CurrentRevision = rev
 			} else {
-				errc <- errors.New("failed to execute query in database")
+				errc <- server.ErrGRPCUnhealthy
 			}
 		}
 		cancel()

diff --git a/pkg/server/types.go b/pkg/server/types.go
@@ -12,9 +12,10 @@ import (
 var (
 	ErrNotSupported = status.New(codes.InvalidArgument, "etcdserver: unsupported operations in txn request").Err()
 
-	ErrKeyExists = rpctypes.ErrGRPCDuplicateKey
-	ErrCompacted = rpctypes.ErrGRPCCompacted
-	ErrFutureRev = rpctypes.ErrGRPCFutureRev
+	ErrKeyExists     = rpctypes.ErrGRPCDuplicateKey
+	ErrCompacted     = rpctypes.ErrGRPCCompacted
+	ErrFutureRev     = rpctypes.ErrGRPCFutureRev
+	ErrGRPCUnhealthy = rpctypes.ErrGRPCUnhealthy
 )
 
 type Backend interface {