Add Site

internal/entity/auth.go

@@ -26,6 +26,7 @@ type UserDB struct {
 	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"`
+	SiteName      string                `gorm:"column:name" json:"site_name"`
 	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"`
@@ -52,6 +53,7 @@ func (u *UserDB) ToUser() *User {
 		PartnerID:   u.PartnerID,
 		PartnerName: u.PartnerName,
 		SiteID:      u.SiteID,
+		SiteName:    u.SiteName,
 	}
 
 	return userEntity
@@ -88,6 +90,8 @@ func (u *UserDB) ToUserAuthenticate(signedToken string) *AuthenticateUser {
 		PartnerID:     u.PartnerID,
 		PartnerName:   u.PartnerName,
 		PartnerStatus: u.PartnerStatus,
+		SiteID:        u.SiteID,
+		SiteName:      u.SiteName,
 	}
 }