pasture: 集团账号下发

models/menu_pasture.go

@@ -0,0 +1,18 @@
+package models
+
+type MenuPasture struct {
+	Id        int64 `xorm:"id"`
+	PastureId int64 `xorm:"pastureid"`
+	IsGroups  int32 `xorm:"isgroups"`
+}
+
+func (m *MenuPasture) TableName() string {
+	return "menu_pasture"
+}
+
+func NewMenuPasture(pastureId int64) *MenuPasture {
+	return &MenuPasture{
+		PastureId: pastureId,
+		IsGroups:  0,
+	}
+}

module/group.go

@@ -174,6 +174,7 @@ func AccountDistribution(req *models.AccountDistributionRequest) error {
 		return err
 	}
 
+	// 查看牧场是否存在
 	pastureData := &models.Pasture{}
 	if _, err = session.Table(new(models.Pasture).TableName()).Where("pastureid = ?", req.PastureId).Get(pastureData); err != nil {
 		return err
@@ -186,6 +187,19 @@ func AccountDistribution(req *models.AccountDistributionRequest) error {
 		}
 	}
 
+	// 查询牧场对应菜单是否存在
+	menuPasture := &models.MenuPasture{}
+	if _, err = session.Table(new(models.MenuPasture).TableName()).Where("pastureid = ?", req.PastureId).Get(menuPasture); err != nil {
+		return err
+	}
+
+	if menuPasture.PastureId <= 0 {
+		newMenuPasture := models.NewMenuPasture(int64(req.PastureId))
+		if _, err = session.Table(new(models.MenuPasture).TableName()).Insert(newMenuPasture); err != nil {
+			return err
+		}
+	}
+
 	// 创建新的超级管理员角色
 	newRole := models.NewDefaultRole(int64(req.PastureId))
 	if _, err = session.Table(new(models.Role).TableName()).Insert(newRole); err != nil {