fix CAS

ydb-platform · Nov 7, 2023 · ce10101 · ce10101
1 parent 4ec2ee9
commit ce10101
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/internal/conn/conn.go b/internal/conn/conn.go
@@ -140,8 +140,7 @@ func (c *conn) SetState(s State) State {
 }
 
 func (c *conn) setState(s State) State {
-	state := c.state.Load()
-	if c.state.CompareAndSwap(state, uint32(s)) {
+	if state := State(c.state.Swap(uint32(s))); state != s {
 		trace.DriverOnConnStateChange(
 			c.config.Trace(),
 			c.endpoint.Copy(),