mirror of
https://github.com/hanzoai/migrate.git
synced 2026-09-30 22:55:10 +00:00
bef0635743
* 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]>
34 lines
879 B
Go
34 lines
879 B
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"go.uber.org/atomic"
|
|
"hash/crc32"
|
|
"strings"
|
|
)
|
|
|
|
const advisoryLockIDSalt uint = 1486364155
|
|
|
|
// GenerateAdvisoryLockId inspired by rails migrations, see https://goo.gl/8o9bCT
|
|
func GenerateAdvisoryLockId(databaseName string, additionalNames ...string) (string, error) { // nolint: golint
|
|
if len(additionalNames) > 0 {
|
|
databaseName = strings.Join(append(additionalNames, databaseName), "\x00")
|
|
}
|
|
sum := crc32.ChecksumIEEE([]byte(databaseName))
|
|
sum = sum * uint32(advisoryLockIDSalt)
|
|
return fmt.Sprint(sum), nil
|
|
}
|
|
|
|
// CasRestoreOnErr CAS wrapper to automatically restore the lock state on error
|
|
func CasRestoreOnErr(lock *atomic.Bool, o, n bool, casErr error, f func() error) error {
|
|
if !lock.CAS(o, n) {
|
|
return casErr
|
|
}
|
|
if err := f(); err != nil {
|
|
// Automatically unlock/lock on error
|
|
lock.Store(o)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|