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"
|
"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 {
|
||||||
|
|||||||
Reference in New Issue
Block a user