feat(go): use single db handle; use connection pool;

This commit is contained in:
Vishal Dalwadi
2025-06-06 15:26:23 +05:30
parent 0b03b2cdbf
commit 48957c02fa
3 changed files with 32 additions and 25 deletions

View File

@@ -1,10 +1,11 @@
package database
import (
"context"
"database/sql"
"errors"
"os"
"path/filepath"
"github.com/gravitl/netmaker/db"
"time"
_ "github.com/mattn/go-sqlite3" // need to blank import this package
)
@@ -29,21 +30,17 @@ var SQLITE_FUNCTIONS = map[string]interface{}{
}
func initSqliteDB() error {
// == create db file if not present ==
if _, err := os.Stat("data"); os.IsNotExist(err) {
os.Mkdir("data", 0700)
}
dbFilePath := filepath.Join("data", dbFilename)
if _, err := os.Stat(dbFilePath); os.IsNotExist(err) {
os.Create(dbFilePath)
}
// == "connect" the database ==
gormDB := db.FromContext(db.WithContext(context.TODO()))
var dbOpenErr error
SqliteDB, dbOpenErr = sql.Open("sqlite3", dbFilePath)
SqliteDB, dbOpenErr = gormDB.DB()
if dbOpenErr != nil {
return dbOpenErr
}
SqliteDB.SetMaxOpenConns(1)
SqliteDB.SetMaxOpenConns(5)
SqliteDB.SetConnMaxLifetime(time.Hour)
return nil
}
@@ -134,7 +131,7 @@ func sqliteFetchRecords(tableName string) (map[string]string, error) {
}
func sqliteCloseDB() {
SqliteDB.Close()
//SqliteDB.Close()
}
func sqliteConnected() bool {