feat: add save
This commit is contained in:
@@ -40,3 +40,5 @@ field_relate:
|
||||
table: users
|
||||
pivot: tenant_users
|
||||
json: users
|
||||
join_foreign_key: tenant_id
|
||||
join_references: user_id
|
||||
|
||||
@@ -28,7 +28,7 @@ type Tenant struct {
|
||||
ExpiredAt time.Time `gorm:"column:expired_at;type:timestamp with time zone" json:"expired_at"`
|
||||
CreatedAt time.Time `gorm:"column:created_at;type:timestamp with time zone;not null;default:now()" json:"created_at"`
|
||||
UpdatedAt time.Time `gorm:"column:updated_at;type:timestamp with time zone;not null;default:now()" json:"updated_at"`
|
||||
Users []*User `gorm:"many2many:tenant_users" json:"users,omitempty"`
|
||||
Users []*User `gorm:"joinForeignKey:TenantID;joinReferences:UserID;many2many:tenant_users" json:"users,omitempty"`
|
||||
}
|
||||
|
||||
// Quick operations without importing query package
|
||||
|
||||
12
backend/database/models/tenants.go
Normal file
12
backend/database/models/tenants.go
Normal file
@@ -0,0 +1,12 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"go.ipao.vip/gen/types"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// BeforeCreate
|
||||
func (m *Tenant) BeforeCreate(tx *gorm.DB) error {
|
||||
m.UUID = types.NewUUIDv4()
|
||||
return nil
|
||||
}
|
||||
@@ -15,20 +15,17 @@ func (m *User) ComparePassword(ctx context.Context, password string) bool {
|
||||
return err == nil
|
||||
}
|
||||
|
||||
func (m *User) EncryptPassword(ctx context.Context) error {
|
||||
bytes, err := bcrypt.GenerateFromPassword([]byte(m.Password), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
m.Password = string(bytes)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *User) HasRole(ctx context.Context, role consts.Role) bool {
|
||||
return lo.Contains(m.Roles, role)
|
||||
}
|
||||
|
||||
// BeforeCreate
|
||||
func (m *User) BeforeCreate(tx *gorm.DB) error {
|
||||
return m.EncryptPassword(tx.Statement.Context)
|
||||
bytes, err := bcrypt.GenerateFromPassword([]byte(m.Password), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
m.Password = string(bytes)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user