qowevisa/fin-check-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes

Browse Source
This commit is contained in:
qowevisa 2024-08-03 10:57:13 +03:00
parent 5ce7223b92
commit 218ef61537
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
db/category.go
View File

@ -19,17 +19,21 @@ type Category struct {
var ( var (
ERROR_CATEGORY_PARENT_NOT_FOUND = errors.New("ParentID is invalid for user") 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") ERROR_CATEGORY_NAME_NOT_UNIQUE = errors.New("Name for Category have to be unique for user")
ERROR_CATEGORY_USER_ID_NOT_EQUAL = errors.New("ParentID is invalid for user")
) )
func (c *Category) BeforeSave(tx *gorm.DB) error { func (c *Category) BeforeSave(tx *gorm.DB) error {
if c.ParentID != 0 { if c.ParentID != 0 {
var parent Category var parent Category
if err := tx.Find(&parent, Category{ParentID: c.ParentID, UserID: c.UserID}).Error; err != nil { if err := tx.Find(&parent, c.ID).Error; err != nil {
return err return err
} }
if parent.ID == 0 { if parent.ID == 0 {
return ERROR_CATEGORY_PARENT_NOT_FOUND return ERROR_CATEGORY_PARENT_NOT_FOUND
} }
if parent.UserID != c.UserID {
return ERROR_CATEGORY_USER_ID_NOT_EQUAL
}
} }
var dup Category var dup Category
if err := tx.Find(&dup, Category{Name: c.Name, UserID: c.UserID}).Error; err != nil { if err := tx.Find(&dup, Category{Name: c.Name, UserID: c.UserID}).Error; err != nil {