This commit is contained in:
suyl
2021-07-19 10:55:03 +08:00
parent d946c5018a
commit 91abfc8835

View File

@@ -43,7 +43,7 @@ func AddMenu(ctx *jxcontext.Context, menu *model.Menu) (err error) {
if len(menus) > 0 { if len(menus) > 0 {
return fmt.Errorf("添加失败!已存在相同名称的 menu name : %v", menu.Name) return fmt.Errorf("添加失败!已存在相同名称的 menu name : %v", menu.Name)
} }
txDB , _ := dao.Begin(db) txDB, _ := dao.Begin(db)
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
dao.Rollback(db, txDB) dao.Rollback(db, txDB)
@@ -66,7 +66,7 @@ func UpdateMenu(ctx *jxcontext.Context, menuID int, payload map[string]interface
if err != nil { if err != nil {
return 0, err return 0, err
} }
txDB , _ := dao.Begin(db) txDB, _ := dao.Begin(db)
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
dao.Rollback(db, txDB) dao.Rollback(db, txDB)
@@ -120,7 +120,7 @@ func AddRole(ctx *jxcontext.Context, name string) (err error) {
role := &model.Role{ role := &model.Role{
Name: name, Name: name,
} }
txDB , _ := dao.Begin(db) txDB, _ := dao.Begin(db)
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
dao.Rollback(db, txDB) dao.Rollback(db, txDB)
@@ -154,7 +154,7 @@ func UpdateRole(ctx *jxcontext.Context, roleID int, name string, isDelete bool,
if err != nil { if err != nil {
return 0, err return 0, err
} }
txDB , _ := dao.Begin(db) txDB, _ := dao.Begin(db)
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
dao.Rollback(db, txDB) dao.Rollback(db, txDB)
@@ -216,7 +216,7 @@ func UpdateUserRole(ctx *jxcontext.Context, userIDs []string, roleIDs []int) (er
} }
} }
} }
txDB , _ := dao.Begin(db) txDB, _ := dao.Begin(db)
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
dao.Rollback(db, txDB) dao.Rollback(db, txDB)
@@ -296,7 +296,7 @@ func UpdateRoleMenu(ctx *jxcontext.Context, roleIDs, menuIDs []int) (err error)
} }
} }
} }
txDB , _ := dao.Begin(db) txDB, _ := dao.Begin(db)
defer func() { defer func() {
if r := recover(); r != nil { if r := recover(); r != nil {
dao.Rollback(db, txDB) dao.Rollback(db, txDB)
@@ -339,7 +339,12 @@ func GetUserStoresResultMap(userID string) (resultMap map[int]int, err error) {
// storeIDMap = make(map[int]int) // storeIDMap = make(map[int]int)
) )
resultMap = make(map[int]int) resultMap = make(map[int]int)
userRoles, err := dao.GetUserRole2(db, []string{userID}, nil) user, _ := dao.GetUserByID(db, "user_id", userID)
if user.BindStoreID != 0 {
resultMap[user.BindStoreID] = user.BindStoreID
} else {
userRoles, err2 := dao.GetUserRole2(db, []string{userID}, nil)
err = err2
for _, v := range userRoles { for _, v := range userRoles {
var ( var (
brandIDs, cityCodes, storeIDs []int brandIDs, cityCodes, storeIDs []int
@@ -397,7 +402,7 @@ func GetUserStoresResultMap(userID string) (resultMap map[int]int, err error) {
// if stores3, err := dao.GetStoreList(db, storeIDs, nil, nil, nil, nil, ""); len(stores3) > 0 && err == nil { // if stores3, err := dao.GetStoreList(db, storeIDs, nil, nil, nil, nil, ""); len(stores3) > 0 && err == nil {
// stores = append(stores, stores3...) // stores = append(stores, stores3...)
// } // }
}
return resultMap, err return resultMap, err
} }