mirror of
https://github.com/gravitl/netmaker.git
synced 2024-11-10 17:48:25 +08:00
112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/gravitl/netmaker/servercfg"
|
|
"github.com/rqlite/gorqlite"
|
|
)
|
|
|
|
// RQliteDatabase - the rqlite db connection
|
|
var RQliteDatabase gorqlite.Connection
|
|
|
|
// RQLITE_FUNCTIONS - all the functions to run with rqlite
|
|
var RQLITE_FUNCTIONS = map[string]interface{}{
|
|
INIT_DB: initRqliteDatabase,
|
|
CREATE_TABLE: rqliteCreateTable,
|
|
INSERT: rqliteInsert,
|
|
INSERT_PEER: rqliteInsertPeer,
|
|
DELETE: rqliteDeleteRecord,
|
|
DELETE_ALL: rqliteDeleteAllRecords,
|
|
FETCH_ALL: rqliteFetchRecords,
|
|
CLOSE_DB: rqliteCloseDB,
|
|
isConnected: rqliteConnected,
|
|
}
|
|
|
|
func initRqliteDatabase() error {
|
|
|
|
conn, err := gorqlite.Open(servercfg.GetSQLConn())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
RQliteDatabase = *conn
|
|
RQliteDatabase.SetConsistencyLevel(gorqlite.ConsistencyLevelStrong)
|
|
return nil
|
|
}
|
|
|
|
func rqliteCreateTable(tableName string) error {
|
|
_, err := RQliteDatabase.WriteOne("CREATE TABLE IF NOT EXISTS " + tableName + " (key TEXT NOT NULL UNIQUE PRIMARY KEY, value TEXT)")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func rqliteInsert(key string, value string, tableName string) error {
|
|
if key != "" && value != "" && IsJSONString(value) {
|
|
_, err := RQliteDatabase.WriteOne("INSERT OR REPLACE INTO " + tableName + " (key, value) VALUES ('" + key + "', '" + value + "')")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
return errors.New("invalid insert " + key + " : " + value)
|
|
}
|
|
|
|
func rqliteInsertPeer(key string, value string) error {
|
|
if key != "" && value != "" && IsJSONString(value) {
|
|
_, err := RQliteDatabase.WriteOne("INSERT OR REPLACE INTO " + PEERS_TABLE_NAME + " (key, value) VALUES ('" + key + "', '" + value + "')")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
return errors.New("invalid peer insert " + key + " : " + value)
|
|
}
|
|
|
|
func rqliteDeleteRecord(tableName string, key string) error {
|
|
_, err := RQliteDatabase.WriteOne("DELETE FROM " + tableName + " WHERE key = \"" + key + "\"")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func rqliteDeleteAllRecords(tableName string) error {
|
|
_, err := RQliteDatabase.WriteOne("DELETE TABLE " + tableName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = rqliteCreateTable(tableName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func rqliteFetchRecords(tableName string) (map[string]string, error) {
|
|
row, err := RQliteDatabase.QueryOne("SELECT * FROM " + tableName + " ORDER BY key")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
records := make(map[string]string)
|
|
for row.Next() { // Iterate and fetch the records from result cursor
|
|
var key string
|
|
var value string
|
|
row.Scan(&key, &value)
|
|
records[key] = value
|
|
}
|
|
if len(records) == 0 {
|
|
return nil, errors.New(NO_RECORDS)
|
|
}
|
|
return records, nil
|
|
}
|
|
|
|
func rqliteCloseDB() {
|
|
RQliteDatabase.Close()
|
|
}
|
|
|
|
func rqliteConnected() bool {
|
|
leader, err := RQliteDatabase.Leader()
|
|
return err == nil && len(leader) > 0
|
|
}
|