system.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. package services
  2. import (
  3. "eta_gn/eta_api/models/system"
  4. "eta_gn/eta_api/utils"
  5. )
  6. func GetRoleTypeCode(roleType string) string {
  7. var roleTypeCode string
  8. switch roleType {
  9. case utils.ROLE_TYPE_ADMIN:
  10. roleTypeCode = utils.ROLE_TYPE_CODE_ADMIN
  11. case utils.ROLE_TYPE_FICC_ADMIN:
  12. roleTypeCode = utils.ROLE_TYPE_CODE_FICC_ADMIN
  13. case utils.ROLE_TYPE_FICC_SELLER:
  14. roleTypeCode = utils.ROLE_TYPE_CODE_FICC_SELLER
  15. case utils.ROLE_TYPE_RAI_ADMIN:
  16. roleTypeCode = utils.ROLE_TYPE_CODE_RAI_ADMIN
  17. case utils.ROLE_TYPE_FICC_GROUP, utils.ROLE_TYPE_FICC_MANAGER:
  18. roleTypeCode = utils.ROLE_TYPE_CODE_FICC_GROUP
  19. case utils.ROLE_TYPE_RAI_GROUP:
  20. roleTypeCode = utils.ROLE_TYPE_CODE_RAI_GROUP
  21. case utils.ROLE_TYPE_FICC_DEPARTMENT:
  22. roleTypeCode = utils.ROLE_TYPE_CODE_FICC_DEPARTMENT
  23. case utils.ROLE_TYPE_RAI_DEPARTMENT:
  24. roleTypeCode = utils.ROLE_TYPE_CODE_RAI_DEPARTMENT
  25. case utils.ROLE_TYPE_FICC_RESEARCHR:
  26. roleTypeCode = utils.ROLE_TYPE_CODE_FICC_RESEARCHR
  27. case utils.ROLE_TYPE_RAI_RESEARCHR:
  28. roleTypeCode = utils.ROLE_TYPE_CODE_RAI_RESEARCHR
  29. default:
  30. roleTypeCode = utils.ROLE_TYPE_ADMIN
  31. }
  32. return roleTypeCode
  33. }
  34. func GetMenuTreeRecursive(list []*system.SysMenuItem, parentId int) []*system.SysMenuItem {
  35. res := make([]*system.SysMenuItem, 0)
  36. for _, v := range list {
  37. if v.ParentId == parentId {
  38. v.Children = GetMenuTreeRecursive(list, v.MenuId)
  39. res = append(res, v)
  40. }
  41. }
  42. return res
  43. }