89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"log"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var DB *gorm.DB
|
|
|
|
// getBackendDir returns the absolute path to the backend directory
|
|
func getBackendDir() string {
|
|
_, filename, _, _ := runtime.Caller(0)
|
|
// This file is in backend/database/config.go
|
|
// So we need to go up one level to get to backend/
|
|
return filepath.Dir(filepath.Dir(filename))
|
|
}
|
|
|
|
// Test database configuration paths
|
|
var (
|
|
// PreparedTestDB is the path to the prepared test database file
|
|
// This file contains a snapshot of test data that can be loaded into test databases
|
|
// Usage: database.PreparedTestDB to access from any package
|
|
PreparedTestDB = filepath.Join(getBackendDir(), "testdata", "prepared_test_data.db")
|
|
|
|
// TestDataDir is the directory for maintenance test data
|
|
// This directory contains temporary test data files during test execution
|
|
// Usage: database.TestDataDir to access from any package
|
|
TestDataDir = filepath.Join(getBackendDir(), "maintenance", "testdata")
|
|
)
|
|
|
|
func ConnectDatabase() *gorm.DB {
|
|
SetupTestDB()
|
|
/*
|
|
db, err := gorm.Open(sqlite.Open(PreparedTestDB), &gorm.Config{})
|
|
if err != nil {
|
|
log.Fatal("Failed to connect to database:", err)
|
|
}
|
|
DB = db
|
|
*/
|
|
return DB
|
|
}
|
|
func ConnectDatabaseOrig() *gorm.DB {
|
|
dsn := "bamort:bG4)efozrc@tcp(192.168.0.5:3306)/bamort?charset=utf8mb4&parseTime=True&loc=Local"
|
|
database, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
log.Fatal("Failed to connect to database:", err)
|
|
}
|
|
DB = database
|
|
return DB
|
|
}
|
|
|
|
func GetDB() *gorm.DB {
|
|
if DB == nil {
|
|
DB = ConnectDatabase()
|
|
}
|
|
return DB
|
|
}
|
|
|
|
// StringArray is a custom type for []string
|
|
type StringArray []string
|
|
|
|
// Value implements the driver.Valuer interface for database storage
|
|
func (s StringArray) Value() (driver.Value, error) {
|
|
return json.Marshal(s) // Serialize []string to JSON
|
|
}
|
|
|
|
// Scan implements the sql.Scanner interface for database retrieval
|
|
func (s *StringArray) Scan(value interface{}) error {
|
|
if value == nil {
|
|
*s = []string{}
|
|
return nil
|
|
}
|
|
|
|
bytes, ok := value.([]byte)
|
|
if !ok {
|
|
return errors.New("failed to convert database value to []byte")
|
|
}
|
|
|
|
return json.Unmarshal(bytes, s) // Deserialize JSON to []string
|
|
}
|