Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2b18549f02 | ||
|
|
d2a4a3d548 |
69
dbHandler.go
69
dbHandler.go
@@ -234,17 +234,25 @@ func (dH *DBHandler) UpdateValuesByKey(model any, key string, value any) error {
|
||||
//
|
||||
// Returns:
|
||||
// - error: Any deletion error.
|
||||
func (dH *DBHandler) DeleteById(model any, id ...uint) error {
|
||||
if id[0] == 0 {
|
||||
func (dH *DBHandler) DeleteById(model any, ids ...uint) error {
|
||||
if len(ids) == 0 {
|
||||
return fmt.Errorf("no IDs provided")
|
||||
}
|
||||
|
||||
if ids[0] == 0 {
|
||||
dH.logger.Debug("deleteById", "delete all")
|
||||
return dH.db.Where("1 = 1").Delete(model).Error
|
||||
}
|
||||
|
||||
dH.logger.Debug("deleteById", "delete ids"+fmt.Sprint(id))
|
||||
if err := dH.Exists(model, "id", id, false); err != nil {
|
||||
return err
|
||||
dH.logger.Debug("deleteById", "delete ids "+fmt.Sprint(ids))
|
||||
err := dH.Exists(model, "id", ids, false)
|
||||
|
||||
// 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
|
||||
@@ -294,20 +302,43 @@ func (dH *DBHandler) DeleteByKey(model any, key string, value any, LikeSearch bo
|
||||
//
|
||||
// Returns:
|
||||
// - error: “no record found” or DB error.
|
||||
func (dH *DBHandler) Exists(model any, key string, value any, LikeSearch bool) (err error) {
|
||||
if LikeSearch {
|
||||
value = strings.ReplaceAll(fmt.Sprint(value), "*", "%")
|
||||
key = key + " LIKE ?"
|
||||
} else {
|
||||
key = key + " = ?"
|
||||
func (dH *DBHandler) Exists(model any, key string, value any, likeSearch bool) error {
|
||||
var query string
|
||||
var args any
|
||||
|
||||
// Detect LIKE and IN behavior
|
||||
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))
|
||||
err = dH.db.Where(key, value).First(model).Error
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return fmt.Errorf("no record found for %s %v", key[:len(key)-1], value)
|
||||
} else if err != nil {
|
||||
return fmt.Errorf("query failed: %w", err)
|
||||
dH.logger.Debug("exists", "checking existence for key "+query+" value "+fmt.Sprint(args))
|
||||
|
||||
var count int64
|
||||
tx := dH.db.Model(model).Where(query, args).Count(&count)
|
||||
if tx.Error != nil {
|
||||
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