Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2b18549f02 | ||
|
|
d2a4a3d548 | ||
|
|
e65fb8fdf5 |
71
dbHandler.go
71
dbHandler.go
@@ -1,4 +1,4 @@
|
|||||||
package AccessHandler
|
package dbHandler
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
@@ -234,17 +234,25 @@ func (dH *DBHandler) UpdateValuesByKey(model any, key string, value any) error {
|
|||||||
//
|
//
|
||||||
// Returns:
|
// Returns:
|
||||||
// - error: Any deletion error.
|
// - error: Any deletion error.
|
||||||
func (dH *DBHandler) DeleteById(model any, id ...uint) error {
|
func (dH *DBHandler) DeleteById(model any, ids ...uint) error {
|
||||||
if id[0] == 0 {
|
if len(ids) == 0 {
|
||||||
|
return fmt.Errorf("no IDs provided")
|
||||||
|
}
|
||||||
|
|
||||||
|
if ids[0] == 0 {
|
||||||
dH.logger.Debug("deleteById", "delete all")
|
dH.logger.Debug("deleteById", "delete all")
|
||||||
return dH.db.Where("1 = 1").Delete(model).Error
|
return dH.db.Where("1 = 1").Delete(model).Error
|
||||||
}
|
}
|
||||||
|
|
||||||
dH.logger.Debug("deleteById", "delete ids"+fmt.Sprint(id))
|
dH.logger.Debug("deleteById", "delete ids "+fmt.Sprint(ids))
|
||||||
if err := dH.Exists(model, "id", id, false); err != nil {
|
err := dH.Exists(model, "id", ids, false)
|
||||||
return err
|
|
||||||
|
// Always use Model() to avoid GORM’s “limit 1” safeguard
|
||||||
|
if err := dH.db.Model(model).Where("id IN ?", ids).Delete(nil).Error; err != nil {
|
||||||
|
return fmt.Errorf("delete failed: %w", err)
|
||||||
}
|
}
|
||||||
return dH.db.Delete(model, id).Error
|
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// deleteByKey
|
// deleteByKey
|
||||||
@@ -294,20 +302,43 @@ func (dH *DBHandler) DeleteByKey(model any, key string, value any, LikeSearch bo
|
|||||||
//
|
//
|
||||||
// Returns:
|
// Returns:
|
||||||
// - error: “no record found” or DB error.
|
// - error: “no record found” or DB error.
|
||||||
func (dH *DBHandler) Exists(model any, key string, value any, LikeSearch bool) (err error) {
|
func (dH *DBHandler) Exists(model any, key string, value any, likeSearch bool) error {
|
||||||
if LikeSearch {
|
var query string
|
||||||
value = strings.ReplaceAll(fmt.Sprint(value), "*", "%")
|
var args any
|
||||||
key = key + " LIKE ?"
|
|
||||||
} else {
|
// Detect LIKE and IN behavior
|
||||||
key = key + " = ?"
|
val := reflect.ValueOf(value)
|
||||||
|
isSlice := val.Kind() == reflect.Slice
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case isSlice:
|
||||||
|
if val.Len() == 0 {
|
||||||
|
return fmt.Errorf("empty slice for %s", key)
|
||||||
|
}
|
||||||
|
query = fmt.Sprintf("%s IN ?", key)
|
||||||
|
args = value
|
||||||
|
|
||||||
|
case likeSearch:
|
||||||
|
str := strings.ReplaceAll(fmt.Sprint(value), "*", "%")
|
||||||
|
query = fmt.Sprintf("%s LIKE ?", key)
|
||||||
|
args = str
|
||||||
|
|
||||||
|
default:
|
||||||
|
query = fmt.Sprintf("%s = ?", key)
|
||||||
|
args = value
|
||||||
}
|
}
|
||||||
|
|
||||||
dH.logger.Debug("exists", "check if exists key "+key+" value "+fmt.Sprint(value))
|
dH.logger.Debug("exists", "checking existence for key "+query+" value "+fmt.Sprint(args))
|
||||||
err = dH.db.Where(key, value).First(model).Error
|
|
||||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
var count int64
|
||||||
return fmt.Errorf("no record found for %s %v", key[:len(key)-1], value)
|
tx := dH.db.Model(model).Where(query, args).Count(&count)
|
||||||
} else if err != nil {
|
if tx.Error != nil {
|
||||||
return fmt.Errorf("query failed: %w", err)
|
return fmt.Errorf("query failed: %w", tx.Error)
|
||||||
}
|
}
|
||||||
return
|
|
||||||
|
if count == 0 {
|
||||||
|
return fmt.Errorf("no record found for %s %v", key, value)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user