add new functions and create folder of db path if nonexisting

This commit is contained in:
Adrian Zürcher
2025-11-19 12:02:10 +01:00
parent c6157bb573
commit 02a97b9762

View File

@@ -3,6 +3,7 @@ package dbHandler
import ( import (
"errors" "errors"
"fmt" "fmt"
"os"
"path/filepath" "path/filepath"
"reflect" "reflect"
"strings" "strings"
@@ -57,12 +58,30 @@ func NewDBHandler(name, path string, logger *logging.Logger) (dH *DBHandler, err
if filepath.Ext(name) == "" { if filepath.Ext(name) == "" {
name += ".db" name += ".db"
} }
if path != "" {
if _, err := os.Stat(path); err != nil {
if err := os.MkdirAll(path, 0666); err != err {
return nil, err
}
}
}
dbPath := filepath.Join(path, name) dbPath := filepath.Join(path, name)
logger.Debug("NewDBHandler", "open database "+dbPath) logger.Debug("NewDBHandler", "open database "+dbPath)
dH.db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{}) dH.db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
return return
} }
// close DB
func (dH *DBHandler) Close() error {
sqlDB, err := dH.db.DB()
if err != nil {
return err
}
return sqlDB.Close()
}
// addNewTable // addNewTable
// //
// Description: // Description:
@@ -302,3 +321,24 @@ func (dH *DBHandler) Exists(model any, key string, value any, likeSearch bool) (
} }
return tx.RowsAffected > 0 return tx.RowsAffected > 0
} }
func (dH *DBHandler) GetByIdWithRelation(model any, relation string, id uint) error {
if id == 0 {
return dH.db.Preload(relation).Find(model).Error
}
// Load member AND its events
err := dH.db.Preload(relation).First(model, id).Error
if errors.Is(err, gorm.ErrRecordNotFound) {
return fmt.Errorf("no record found for id: %v", id)
} else if err != nil {
return fmt.Errorf("query failed: %w", err)
}
return nil
}
func (dH *DBHandler) AddRelation(model, relation any, relationName string) error {
return dH.db.Model(model).Association(relationName).Append(relation)
}