Add Category.BeforeSave func

This commit is contained in:
qowevisa 2024-08-03 10:49:46 +03:00
parent f59b57f147
commit 12fd07c0a3

View File

@ -1,6 +1,10 @@
package db package db
import "gorm.io/gorm" import (
"errors"
"gorm.io/gorm"
)
type Category struct { type Category struct {
gorm.Model gorm.Model
@ -11,3 +15,26 @@ type Category struct {
UserID uint UserID uint
User *User User *User
} }
var (
ERROR_CATEGORY_PARENT_NOT_FOUND = errors.New("ParentID is invalid for user")
ERROR_CATEGORY_NAME_NOT_UNIQUE = errors.New("Name for Category have to be unique for user")
)
func (c *Category) BeforeSave(tx *gorm.DB) error {
var parent Category
if err := tx.Find(&parent, Category{ParentID: c.ParentID, UserID: c.UserID}).Error; err != nil {
return err
}
if parent.ID == 0 {
return ERROR_CATEGORY_PARENT_NOT_FOUND
}
var dup Category
if err := tx.Find(&dup, Category{Name: c.Name, UserID: c.UserID}).Error; err != nil {
return err
}
if dup.ID != 0 {
return ERROR_CATEGORY_NAME_NOT_UNIQUE
}
return nil
}