v1.5.1: add mutex for sqlite write ops

This commit is contained in:
abhishek9686
2026-04-08 19:05:49 +05:30
parent 32ca51db92
commit 109dab9055
+16
View File
@@ -4,6 +4,7 @@ import (
"context" "context"
"database/sql" "database/sql"
"errors" "errors"
"sync"
"time" "time"
"github.com/gravitl/netmaker/db" "github.com/gravitl/netmaker/db"
@@ -13,6 +14,9 @@ import (
// SqliteDB is the db object for sqlite database connections // SqliteDB is the db object for sqlite database connections
var SqliteDB *sql.DB 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 // SQLITE_FUNCTIONS - contains a map of the functions for sqlite
var SQLITE_FUNCTIONS = map[string]interface{}{ var SQLITE_FUNCTIONS = map[string]interface{}{
INIT_DB: initSqliteDB, INIT_DB: initSqliteDB,
@@ -40,6 +44,9 @@ func initSqliteDB() error {
} }
func sqliteCreateTable(tableName string) 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)") statement, err := SqliteDB.Prepare("CREATE TABLE IF NOT EXISTS " + tableName + " (key TEXT NOT NULL UNIQUE PRIMARY KEY, value TEXT)")
if err != nil { if err != nil {
return err return err
@@ -54,6 +61,9 @@ func sqliteCreateTable(tableName string) error {
func sqliteInsert(key string, value string, tableName string) error { func sqliteInsert(key string, value string, tableName string) error {
if key != "" && value != "" { if key != "" && value != "" {
sqliteWriteMu.Lock()
defer sqliteWriteMu.Unlock()
insertSQL := "INSERT OR REPLACE INTO " + tableName + " (key, value) VALUES (?, ?)" insertSQL := "INSERT OR REPLACE INTO " + tableName + " (key, value) VALUES (?, ?)"
statement, err := SqliteDB.Prepare(insertSQL) statement, err := SqliteDB.Prepare(insertSQL)
if err != nil { if err != nil {
@@ -81,6 +91,9 @@ func sqliteInsertPeer(key string, value string) error {
} }
func sqliteDeleteRecord(tableName string, key string) error { func sqliteDeleteRecord(tableName string, key string) error {
sqliteWriteMu.Lock()
defer sqliteWriteMu.Unlock()
deleteSQL := "DELETE FROM " + tableName + " WHERE key = ?" deleteSQL := "DELETE FROM " + tableName + " WHERE key = ?"
statement, err := SqliteDB.Prepare(deleteSQL) statement, err := SqliteDB.Prepare(deleteSQL)
if err != nil { if err != nil {
@@ -94,6 +107,9 @@ func sqliteDeleteRecord(tableName string, key string) error {
} }
func sqliteDeleteAllRecords(tableName string) error { func sqliteDeleteAllRecords(tableName string) error {
sqliteWriteMu.Lock()
defer sqliteWriteMu.Unlock()
deleteSQL := "DELETE FROM " + tableName deleteSQL := "DELETE FROM " + tableName
statement, err := SqliteDB.Prepare(deleteSQL) statement, err := SqliteDB.Prepare(deleteSQL)
if err != nil { if err != nil {