fix: shared sql log panic

GaleHuang · Oct 25, 2024 · 5b1787a · 5b1787a
1 parent c143943
commit 5b1787a
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/conn_pool.go b/conn_pool.go
@@ -41,8 +41,10 @@ func (pool ConnPool) ExecContext(ctx context.Context, query string, args ...any)
 			if r.DoubleWrite {
 				pool.sharding.Logger.Trace(ctx, curTime, func() (sql string, rowsAffected int64) {
 					result, _ := pool.ConnPool.ExecContext(ctx, ftQuery, args...)
-					rowsAffected, _ = result.RowsAffected()
-					return fmt.Sprintf("[Shard] %s", pool.sharding.Explain(ftQuery, args...)), rowsAffected
+					if result != nil {
+						rowsAffected, _ = result.RowsAffected()
+					}
+					return fmt.Sprintf("[Shard DoubleWrite] %s", pool.sharding.Explain(ftQuery, args...)), rowsAffected
 				}, pool.sharding.Error)
 			}
 		}
@@ -51,7 +53,9 @@ func (pool ConnPool) ExecContext(ctx context.Context, query string, args ...any)
 	var result sql.Result
 	result, err = pool.ConnPool.ExecContext(ctx, stQuery, args...)
 	pool.sharding.Logger.Trace(ctx, curTime, func() (sql string, rowsAffected int64) {
-		rowsAffected, _ = result.RowsAffected()
+		if result != nil {
+			rowsAffected, _ = result.RowsAffected()
+		}
 		return fmt.Sprintf("[Shard] %s", pool.sharding.Explain(stQuery, args...)), rowsAffected
 	}, pool.sharding.Error)