From faaf92c6a94274fc045140ea7a56c4b06d0de004 Mon Sep 17 00:00:00 2001 From: Matt Braymer-Hayes Date: Sat, 26 Jun 2021 00:34:56 -0400 Subject: [PATCH] Handle gRPC deadline exceeded as well. --- server.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/server.go b/server.go index 233d4ba..4d5e21d 100644 --- a/server.go +++ b/server.go @@ -16,6 +16,8 @@ import ( etcdv3 "go.etcd.io/etcd/client/v3" netcontext "golang.org/x/net/context" "google.golang.org/grpc" + "google.golang.org/grpc/codes" + "google.golang.org/grpc/status" ) const ( @@ -431,7 +433,7 @@ func (s *Server) startActorC(c context.Context, start *ActorStart) error { defer cancel() switch err := s.registry.Register(timeout, nsName); { case err == nil: - case errors.Is(err, context.DeadlineExceeded): + case errors.Is(err, context.DeadlineExceeded), status.Code(err) == codes.DeadlineExceeded: s.deregisterActor(nsName) default: return fmt.Errorf("registering actor %q: %w", nsName, err)