Merge pull request #1 from ggriffiths/exit_on_csiconn_loss

Exit on CSI gRPC conn loss
libopenstorage · Sep 16, 2019 · 8e9fc4e · 8e9fc4e
2 parents 07a4fd9 + 94c9ac0
commit 8e9fc4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/csi/client.go b/pkg/csi/client.go
@@ -52,7 +52,7 @@ type Client interface {
 
 // New creates a new CSI client.
 func New(address string, timeout time.Duration) (Client, error) {
-	conn, err := connection.Connect(address)
+	conn, err := connection.Connect(address, connection.OnConnectionLoss(connection.ExitOnConnectionLoss()))
 	if err != nil {
 		return nil, fmt.Errorf("failed to connect to CSI driver: %v", err)
 	}