Reconnect on a few more gocql errors (#4132)

temporalio · Apr 17, 2023 · 16d53cb · 16d53cb
1 parent 8f178a3
commit 16d53cb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/common/persistence/nosql/nosqlplugin/cassandra/gocql/session.go b/common/persistence/nosql/nosqlplugin/cassandra/gocql/session.go
@@ -28,6 +28,7 @@ import (
 	"context"
 	"sync"
 	"sync/atomic"
+	"syscall"
 	"time"
 
 	"github.com/gocql/gocql"
@@ -181,7 +182,10 @@ func (s *session) handleError(
 	err error,
 ) {
 	switch err {
-	case gocql.ErrNoConnections:
+	case gocql.ErrNoConnections,
+		gocql.ErrSessionClosed,
+		gocql.ErrConnectionClosed,
+		syscall.ECONNRESET:
 		s.refresh()
 	default:
 		// noop