Merge remote-tracking branch 'refs/remotes/origin/feat/read-only-repl…

…ica' into feat/read-only-replica
uptrace · Jan 22, 2025 · a95c99a · a95c99a
2 parents 9f5e8b1 + 06dd779
commit a95c99a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/db.go b/db.go
@@ -95,7 +95,9 @@ func (db *DB) String() string {
 }
 
 func (db *DB) Close() error {
-	db.closed.Store(true)
+	if db.closed.Swap(true) {
+		return nil
+	}
 
 	firstErr := db.DB.Close()
 
@@ -786,7 +788,9 @@ func WithReadOnlyReplica(dbs ...*sql.DB) ReadWriteConnResolverOption {
 }
 
 func (r *ReadWriteConnResolver) Close() error {
-	r.closed.Store(true)
+	if r.closed.Swap(true) {
+		return nil
+	}
 
 	var firstErr error
 	for _, db := range r.replicas {