grpc/proxy: Redial if connection unexpectedly shuts down

oasisprotocol · Mar 31, 2020 · 6e591d1 · 6e591d1
1 parent 06289ef
commit 6e591d1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/go/common/grpc/proxy/proxy.go b/go/common/grpc/proxy/proxy.go
@@ -9,6 +9,7 @@ import (
 
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/connectivity"
 	"google.golang.org/grpc/metadata"
 	"google.golang.org/grpc/status"
 
@@ -74,6 +75,12 @@ func (p *proxy) handler(srv interface{}, stream grpc.ServerStream) error {
 	// Pass subject header upstream.
 	upstreamCtx = metadata.AppendToOutgoingContext(upstreamCtx, policy.ForwardedSubjectMD, sub)
 
+	// Check if upstream connection was disconnected.
+	if p.upstreamConn != nil && p.upstreamConn.GetState() == connectivity.Shutdown {
+		// We need to redial if the connection was shut down.
+		p.upstreamConn = nil
+	}
+
 	// Dial upstream if necessary.
 	if p.upstreamConn == nil {
 		var grr error