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

added mutex for sqlite

Browse source
This commit is contained in:
0xdcarns 2022-03-15 16:40:46 -04:00
parent 5c54671810
commit 9d0d3ea732
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 {