paadi/dbHandler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

combine get with relation request

Browse Source
This commit is contained in:
Adrian Zürcher
2025-11-19 14:10:56 +01:00
parent 02a97b9762
commit 5ff6b2b148
2 changed files with 65 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
dbHandler.go
View File

@@ -129,14 +129,22 @@ func (dH *DBHandler) AddNewColum(model any) error {
//
// Returns:
// - error: Any query error or “not found” message.
func (dH *DBHandler) GetById(model any, id uint) error {
func (dH *DBHandler) GetById(model any, relation string, id uint) (err error) {
dH.logger.Debug("getById", "find id "+fmt.Sprint(id))
if id == 0 {
if relation != "" {
return dH.db.Preload(relation).Find(model).Error
}
return dH.db.Find(model).Error
}
err := dH.db.First(model, id).Error
if relation != "" {
err = dH.db.Preload(relation).First(model, id).Error
} else {
err = dH.db.First(model, id).Error
}
if errors.Is(err, gorm.ErrRecordNotFound) {
return fmt.Errorf("no record found for id: %v", id)
} else if err != nil {
@@ -163,14 +171,20 @@ func (dH *DBHandler) GetById(model any, id uint) error {
//
// Returns:
// - error: Any database query error.
func (dH *DBHandler) GetByKey(model any, key string, value any, LikeSearch bool) error {
func (dH *DBHandler) GetByKey(model any, relation string, key string, value any, LikeSearch bool) error {
if LikeSearch {
value = strings.ReplaceAll(fmt.Sprint(value), "*", "%")
dH.logger.Debug("getByKey", "find like key "+key+" value "+fmt.Sprint(value))
if relation != "" {
return dH.db.Preload(relation).Where(key+" LIKE ?", value).Delete(model).Error
}
return dH.db.Where(key+" LIKE ?", value).Find(model).Error
}
dH.logger.Debug("getByKey", "find equal key "+key+" value "+fmt.Sprint(value))
if relation != "" {
return dH.db.Preload(relation).Where(key+" = ?", value).Delete(model).Error
}
return dH.db.Find(model, key+" = ?", value).Error
}
@@ -191,7 +205,7 @@ func (dH *DBHandler) GetByKey(model any, key string, value any, LikeSearch bool)
//
// Returns:
// - error: If the model is invalid or query/update fails.
func (dH *DBHandler) UpdateValuesById(model any, id uint) error {
func (dH *DBHandler) UpdateValuesById(model any, relation string, id uint) error {
dH.logger.Debug("updateValuesById", "model"+fmt.Sprint(model))
modelType := reflect.TypeOf(model)
if modelType.Kind() != reflect.Ptr {
@@ -199,9 +213,12 @@ func (dH *DBHandler) UpdateValuesById(model any, id uint) error {
}
lookUpModel := reflect.New(modelType.Elem()).Interface()
if err := dH.GetById(lookUpModel, id); err != nil {
if err := dH.GetById(lookUpModel, relation, id); err != nil {
return err
}
if relation != "" {
return dH.db.Preload(relation).Model(lookUpModel).Updates(model).Error
}
return dH.db.Model(lookUpModel).Updates(model).Error
}
@@ -223,7 +240,7 @@ func (dH *DBHandler) UpdateValuesById(model any, id uint) error {
//
// Returns:
// - error: Any query or update error.
func (dH *DBHandler) UpdateValuesByKey(model any, key string, value any) error {
func (dH *DBHandler) UpdateValuesByKey(model any, relation string, key string, value any) error {
dH.logger.Debug("updateValuesByKey", "model"+fmt.Sprint(model))
modelType := reflect.TypeOf(model)
if modelType.Kind() != reflect.Ptr {
@@ -231,9 +248,12 @@ func (dH *DBHandler) UpdateValuesByKey(model any, key string, value any) error {
}
lookUpModel := reflect.New(modelType.Elem()).Interface()
if err := dH.GetByKey(lookUpModel, key, value, false); err != nil {
if err := dH.GetByKey(lookUpModel, "", key, value, false); err != nil {
return err
}
if relation != "" {
return dH.db.Preload(relation).Model(lookUpModel).Updates(model).Error
}
return dH.db.Model(lookUpModel).Updates(model).Error
}
@@ -322,23 +342,6 @@ func (dH *DBHandler) Exists(model any, key string, value any, likeSearch bool) (
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)
}