Add callback and update user role

2024-06-05 00:24:53 +07:00
parent 4d402e32c9
commit 04a89c5508
27 changed files with 388 additions and 49 deletions
@@ -25,6 +25,7 @@ type UserDB struct {
 	RoleID        int64                 `gorm:"column:role_id" json:"role_id"`
 	RoleName      string                `gorm:"column:role_name" json:"role_name"`
 	PartnerID     *int64                `gorm:"column:partner_id" json:"partner_id"`
+	SiteID        *int64                `gorm:"column:site_id" json:"site_id"`
 	PartnerName   string                `gorm:"column:partner_name" json:"partner_name"`
 	PartnerStatus string                `gorm:"column:partner_status" json:"partner_status"`
 	CreatedAt     time.Time             `gorm:"column:created_at" json:"created_at"`
@@ -50,6 +51,7 @@ func (u *UserDB) ToUser() *User {
 		RoleName:    u.RoleName,
 		PartnerID:   u.PartnerID,
 		PartnerName: u.PartnerName,
+		SiteID:      u.SiteID,
 	}

 	return userEntity
@@ -67,6 +69,7 @@ func (u *UserDB) ToUserRoleDB() *UserRoleDB {
 		PartnerID: u.PartnerID,
 		CreatedAt: u.CreatedAt,
 		UpdatedAt: u.UpdatedAt,
+		SiteID:    u.SiteID,
 	}

 	return userRole