task.go 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. package services
  2. import (
  3. "eta/eta_mini_crm/models"
  4. "eta/eta_mini_crm/utils"
  5. "fmt"
  6. )
  7. func InitTask() {
  8. }
  9. func InitSystemUserPermission() {
  10. menuList, err := models.GetSysMenuItemsByCondition("", []interface{}{}, []string{"sys_menu_id"}, "")
  11. if err != nil {
  12. fmt.Println("获取系统菜单失败, Err:", err)
  13. utils.FileLog.Info("获取系统菜单失败, Err:", err)
  14. return
  15. }
  16. roleMenuList, err := models.GetSysRoleMenuByRoleId(1)
  17. if err != nil {
  18. if err.Error() == utils.ErrNoRow() {
  19. fmt.Println("系统角色不存在, 请先创建系统角色")
  20. utils.FileLog.Info("系统角色不存在, 请先创建系统角色")
  21. }
  22. fmt.Println("获取系统角色菜单失败, Err:", err)
  23. utils.FileLog.Info("获取系统角色菜单失败, Err:", err)
  24. return
  25. }
  26. menuMap := make(map[int]struct{})
  27. for _, v := range menuList {
  28. menuMap[v.SysMenuId] = struct{}{}
  29. }
  30. for _, v := range roleMenuList {
  31. delete(menuMap, v.SysMenuId)
  32. }
  33. insertMenu := make([]*models.SysRoleMenuMapping, 0)
  34. for k, _ := range menuMap {
  35. t := new(models.SysRoleMenuMapping)
  36. t.SysMenuId = k
  37. t.SysRoleId = 1
  38. t.Type = 0
  39. insertMenu = append(insertMenu, t)
  40. }
  41. if len(insertMenu) > 0 {
  42. err = models.BatchAddSysRoleMenu(insertMenu)
  43. if err != nil {
  44. fmt.Println("初始化系统用户权限失败, Err:", err)
  45. utils.FileLog.Info("初始化系统用户权限失败, Err:", err)
  46. return
  47. }
  48. }
  49. fmt.Println("初始化系统用户权限完成")
  50. }