fix race condition when context is canceled (#1562)

methane · web-flow · commit d86c4527bae9 · 2024-03-17T13:30:21.000+09:00
Fix #1559.
diff --git a/connection.go b/connection.go
@@ -138,7 +138,7 @@ func (mc *mysqlConn) Close() (err error) {
 	}
 
 	mc.cleanup()
-
+	mc.clearResult()
 	return
 }
 
@@ -153,13 +153,16 @@ func (mc *mysqlConn) cleanup() {
 
 	// Makes cleanup idempotent
 	close(mc.closech)
-	if mc.netConn == nil {
+	nc := mc.netConn
+	if nc == nil {
 		return
 	}
-	if err := mc.netConn.Close(); err != nil {
+	if err := nc.Close(); err != nil {
 		mc.log(err)
 	}
-	mc.clearResult()
+	// This function can be called from multiple goroutines.
+	// So we can not mc.clearResult() here.
+	// Caller should do it if they are in safe goroutine.
 }
 
 func (mc *mysqlConn) error() error {
