used uber atomic bool instead standard in lock/unlock db (#580)

* used uber atomic bool instead standard in lock/unlock db * fixing local lock optimization implementation * fixing mysql local lock optimization implementation * fixing cockroachdb local lock optimization implementation * CAS wrapper to automatically restore * added test for cas with restore * added atomic lock to sqlite & mongo * added atomic lock to sqlite & mongo * fix/refactor test cas restore function * disable lock for mongo * mongo local lock before acquiring the db lock Co-authored-by: a.prinkov <[email protected]>
2026-09-30 22:55:10 +00:00 · 2021-06-23 14:40:50 -07:00
parent b37eea94ca
commit bef0635743
19 changed files with 359 additions and 311 deletions
@@ -3,6 +3,7 @@ package sqlcipher
 import (
 	"database/sql"
 	"fmt"
+	"go.uber.org/atomic"
 	"io"
 	"io/ioutil"
 	nurl "net/url"
@@ -34,7 +35,7 @@ type Config struct {

 type Sqlite struct {
 	db       *sql.DB
-	isLocked bool
+	isLocked atomic.Bool

 	config *Config
 }
@@ -177,18 +178,16 @@ func (m *Sqlite) Drop() (err error) {
 }

 func (m *Sqlite) Lock() error {
-	if m.isLocked {
+	if !m.isLocked.CAS(false, true) {
 		return database.ErrLocked
 	}
-	m.isLocked = true
 	return nil
 }

 func (m *Sqlite) Unlock() error {
-	if !m.isLocked {
-		return nil
+	if !m.isLocked.CAS(true, false) {
+		return database.ErrNotLocked
 	}
-	m.isLocked = false
 	return nil
 }