mirror of
https://github.com/gravitl/netmaker.git
synced 2026-04-22 16:07:11 +08:00
v1.5.1: add mutex for sqlite write ops
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gravitl/netmaker/db"
|
||||
@@ -13,6 +14,9 @@ import (
|
||||
// SqliteDB is the db object for sqlite database connections
|
||||
var SqliteDB *sql.DB
|
||||
|
||||
// sqliteWriteMu serializes SQLite write operations to reduce lock contention.
|
||||
var sqliteWriteMu sync.Mutex
|
||||
|
||||
// SQLITE_FUNCTIONS - contains a map of the functions for sqlite
|
||||
var SQLITE_FUNCTIONS = map[string]interface{}{
|
||||
INIT_DB: initSqliteDB,
|
||||
@@ -40,6 +44,9 @@ func initSqliteDB() error {
|
||||
}
|
||||
|
||||
func sqliteCreateTable(tableName string) error {
|
||||
sqliteWriteMu.Lock()
|
||||
defer sqliteWriteMu.Unlock()
|
||||
|
||||
statement, err := SqliteDB.Prepare("CREATE TABLE IF NOT EXISTS " + tableName + " (key TEXT NOT NULL UNIQUE PRIMARY KEY, value TEXT)")
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -54,6 +61,9 @@ func sqliteCreateTable(tableName string) error {
|
||||
|
||||
func sqliteInsert(key string, value string, tableName string) error {
|
||||
if key != "" && value != "" {
|
||||
sqliteWriteMu.Lock()
|
||||
defer sqliteWriteMu.Unlock()
|
||||
|
||||
insertSQL := "INSERT OR REPLACE INTO " + tableName + " (key, value) VALUES (?, ?)"
|
||||
statement, err := SqliteDB.Prepare(insertSQL)
|
||||
if err != nil {
|
||||
@@ -81,6 +91,9 @@ func sqliteInsertPeer(key string, value string) error {
|
||||
}
|
||||
|
||||
func sqliteDeleteRecord(tableName string, key string) error {
|
||||
sqliteWriteMu.Lock()
|
||||
defer sqliteWriteMu.Unlock()
|
||||
|
||||
deleteSQL := "DELETE FROM " + tableName + " WHERE key = ?"
|
||||
statement, err := SqliteDB.Prepare(deleteSQL)
|
||||
if err != nil {
|
||||
@@ -94,6 +107,9 @@ func sqliteDeleteRecord(tableName string, key string) error {
|
||||
}
|
||||
|
||||
func sqliteDeleteAllRecords(tableName string) error {
|
||||
sqliteWriteMu.Lock()
|
||||
defer sqliteWriteMu.Unlock()
|
||||
|
||||
deleteSQL := "DELETE FROM " + tableName
|
||||
statement, err := SqliteDB.Prepare(deleteSQL)
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user