photoprism/internal/entity/entity_save.go

36 lines
718 B
Go
Raw Normal View History

package entity
import (
"fmt"
"strings"
)
// Save updates a record in the database, or inserts
// if it does not exist.
func Save(m interface{}, keyNames ...string) (err error) {
defer func() {
if r := recover(); r != nil {
err = fmt.Errorf("save: %s (panic)", r)
}
}()
// Try a regular update first.
if err = Update(m, keyNames...); err == nil {
return nil
}
// Automatically insert/update record as needed.
if err = UnscopedDb().Save(m).Error; err == nil {
return nil
}
// Try again if database was locked, return otherwise.
if !strings.Contains(strings.ToLower(err.Error()), "lock") {
return err
} else if err = UnscopedDb().Save(m).Error; err == nil {
return nil
}
return err
}