Frank
dc0e23a48c
spätere Fortsetzung soll bis zu 14 Tage lang möglich sein. Rückwärts navigation durch die einzelenen Schritte sollen möglich sein
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"log"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/sqlite"
|
|
"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
|
|
}
|