gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-02-27 09:36:08 +08:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #905 from gravitl/feature_v0.12.1_db_sync

Browse source 
added mutex for sqlite
This commit is contained in:
dcarns 2022-03-15 22:38:30 -04:00 committed by GitHub
commit 07fb4a679a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
database/sqlite.go
View file

@ -5,6 +5,7 @@ import (
"errors"
"os"
"path/filepath"
"sync"
_ "github.com/mattn/go-sqlite3" // need to blank import this package
)
@ -27,6 +28,8 @@ var SQLITE_FUNCTIONS = map[string]interface{}{
CLOSE_DB: sqliteCloseDB,
}
var mutex sync.Mutex
func initSqliteDB() error {
// == create db file if not present ==
if _, err := os.Stat("data"); os.IsNotExist(err) {
@ -60,6 +63,8 @@ func sqliteCreateTable(tableName string) error {
func sqliteInsert(key string, value string, tableName string) error {
if key != "" && value != "" && IsJSONString(value) {
mutex.Lock()
defer mutex.Unlock()
insertSQL := "INSERT OR REPLACE INTO " + tableName + " (key, value) VALUES (?, ?)"
statement, err := SqliteDB.Prepare(insertSQL)
if err != nil {
@ -77,6 +82,8 @@ func sqliteInsert(key string, value string, tableName string) error {
func sqliteInsertPeer(key string, value string) error {
if key != "" && value != "" && IsJSONString(value) {
mutex.Lock()
defer mutex.Unlock()
err := sqliteInsert(key, value, PEERS_TABLE_NAME)
if err != nil {
return err
@ -87,6 +94,8 @@ func sqliteInsertPeer(key string, value string) error {
}
func sqliteDeleteRecord(tableName string, key string) error {
mutex.Lock()
defer mutex.Unlock()
deleteSQL := "DELETE FROM " + tableName + " WHERE key = \"" + key + "\""
statement, err := SqliteDB.Prepare(deleteSQL)
if err != nil {
@ -100,6 +109,8 @@ func sqliteDeleteRecord(tableName string, key string) error {
}
func sqliteDeleteAllRecords(tableName string) error {
mutex.Lock()
defer mutex.Unlock()
deleteSQL := "DELETE FROM " + tableName
statement, err := SqliteDB.Prepare(deleteSQL)
if err != nil {