Add BeforeSave to card
This commit is contained in:
parent
0c9bde7c5e
commit
40fa706514
28
db/card.go
28
db/card.go
|
@ -1,6 +1,10 @@
|
||||||
package db
|
package db
|
||||||
|
|
||||||
import "gorm.io/gorm"
|
import (
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
// Card can be either card or wallet
|
// Card can be either card or wallet
|
||||||
type Card struct {
|
type Card struct {
|
||||||
|
@ -12,3 +16,25 @@ type Card struct {
|
||||||
UserID uint
|
UserID uint
|
||||||
User *User
|
User *User
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
ERROR_CARD_NAME_EMPTY = errors.New("Card's name can't be empty")
|
||||||
|
ERROR_CARD_NAME_NOT_UNIQUE = errors.New("Card's name have to be unique for user")
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *Card) BeforeSave(tx *gorm.DB) error {
|
||||||
|
if c.Name == "" {
|
||||||
|
return ERROR_CARD_NAME_EMPTY
|
||||||
|
}
|
||||||
|
|
||||||
|
var dup Card
|
||||||
|
if err := tx.Find(&dup, Card{Name: c.Name}).Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if dup.ID != 0 {
|
||||||
|
return ERROR_CARD_NAME_NOT_UNIQUE
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user