fix types

Author: DifferentialOrange

connection.go

@@ -28,6 +28,11 @@ const (
 	connClosed       = 2
 )
 
+const (
+	shutdownNotInProgress = 0
+	shutdownInProgress    = 1
+)
+
 const (
 	connTransportNone = ""
 	connTransportSsl  = "ssl"
@@ -163,10 +168,9 @@ type Connection struct {
 	// watchMap is a map of key -> chan watchState.
 	watchMap sync.Map
 
-	// shutdownOn is true if shutdown is processed now.
-	shutdownOn      atomic.Bool
-	shutdownWatcher Watcher
-	shutdownWg      sync.WaitGroup
+	shutdownInProgress uint32
+	shutdownWatcher    Watcher
+	shutdownWg         sync.WaitGroup
 }
 
 var _ = Connector(&Connection{}) // Check compatibility with connector interface.
@@ -390,7 +394,7 @@ func Connect(addr string, opts Opts) (conn *Connection, err error) {
 		conn.opts.Logger = defaultLogger{}
 	}
 
-	conn.shutdownOn.Store(false)
+	atomic.StoreUint32(&(conn.shutdownInProgress), shutdownNotInProgress)
 
 	if err = conn.createConnection(false); err != nil {
 		ter, ok := err.(Error)
@@ -456,8 +460,8 @@ func (conn *Connection) shutdownOnServerEvent() {
 	conn.mutex.Lock()
 
 	fmt.Printf("starting shutdown... (%v)\n", conn.UniqueId())
-	conn.shutdownOn.Store(true)
-	defer conn.shutdownOn.Store(false)
+	atomic.StoreUint32(&(conn.shutdownInProgress), shutdownInProgress)
+	defer atomic.StoreUint32(&(conn.shutdownInProgress), shutdownNotInProgress)
 
 	fmt.Printf("wait for futures to finish... (%v)\n", conn.UniqueId())
 	conn.shutdownWg.Wait()
@@ -1167,7 +1171,7 @@ func (conn *Connection) send(req Request, streamId uint64) *Future {
 		}
 	}
 
-	if conn.shutdownOn.Load() {
+	if atomic.LoadUint32(&(conn.shutdownInProgress)) == shutdownInProgress {
 		conn.cancelFuture(fut, fmt.Errorf("instance shutdown in process"))
 		return fut
 	}