mirror of
https://github.com/gravitl/netmaker.git
synced 2024-09-21 15:56:07 +08:00
Merge pull request #2010 from gravitl/GRA-1153-database-sync
Gra 1153 database sync
This commit is contained in:
commit
7c7528eeae
|
@ -4,6 +4,7 @@ import (
|
|||
"crypto/rand"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
|
@ -84,6 +85,8 @@ const (
|
|||
isConnected = "isconnected"
|
||||
)
|
||||
|
||||
var dbMutex sync.RWMutex
|
||||
|
||||
func getCurrentDB() map[string]interface{} {
|
||||
switch servercfg.GetDB() {
|
||||
case "rqlite":
|
||||
|
@ -150,6 +153,8 @@ func IsJSONString(value string) bool {
|
|||
|
||||
// Insert - inserts object into db
|
||||
func Insert(key string, value string, tableName string) error {
|
||||
dbMutex.Lock()
|
||||
defer dbMutex.Unlock()
|
||||
if key != "" && value != "" && IsJSONString(value) {
|
||||
return getCurrentDB()[INSERT].(func(string, string, string) error)(key, value, tableName)
|
||||
} else {
|
||||
|
@ -159,6 +164,8 @@ func Insert(key string, value string, tableName string) error {
|
|||
|
||||
// InsertPeer - inserts peer into db
|
||||
func InsertPeer(key string, value string) error {
|
||||
dbMutex.Lock()
|
||||
defer dbMutex.Unlock()
|
||||
if key != "" && value != "" && IsJSONString(value) {
|
||||
return getCurrentDB()[INSERT_PEER].(func(string, string) error)(key, value)
|
||||
} else {
|
||||
|
@ -168,11 +175,15 @@ func InsertPeer(key string, value string) error {
|
|||
|
||||
// DeleteRecord - deletes a record from db
|
||||
func DeleteRecord(tableName string, key string) error {
|
||||
dbMutex.Lock()
|
||||
defer dbMutex.Unlock()
|
||||
return getCurrentDB()[DELETE].(func(string, string) error)(tableName, key)
|
||||
}
|
||||
|
||||
// DeleteAllRecords - removes a table and remakes
|
||||
func DeleteAllRecords(tableName string) error {
|
||||
dbMutex.Lock()
|
||||
defer dbMutex.Unlock()
|
||||
err := getCurrentDB()[DELETE_ALL].(func(string) error)(tableName)
|
||||
if err != nil {
|
||||
return err
|
||||
|
@ -198,6 +209,8 @@ func FetchRecord(tableName string, key string) (string, error) {
|
|||
|
||||
// FetchRecords - fetches all records in given table
|
||||
func FetchRecords(tableName string) (map[string]string, error) {
|
||||
dbMutex.RLock()
|
||||
defer dbMutex.RUnlock()
|
||||
return getCurrentDB()[FETCH_ALL].(func(string) (map[string]string, error))(tableName)
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue