system_service.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676
  1. package backend
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "kpt-tmr-group/model"
  7. "net/http"
  8. "strconv"
  9. "strings"
  10. operationPb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/operation"
  11. "gitee.com/xuyiping_admin/pkg/jwt"
  12. "gitee.com/xuyiping_admin/pkg/logger/zaplog"
  13. "gitee.com/xuyiping_admin/pkg/tool"
  14. "gitee.com/xuyiping_admin/pkg/xerr"
  15. "go.uber.org/zap"
  16. "gorm.io/gorm"
  17. )
  18. const CurrentUserName = "userName"
  19. // Auth 用户登录
  20. func (s *StoreEntry) Auth(ctx context.Context, auth *operationPb.UserAuthData) (*operationPb.SystemToken, error) {
  21. systemUser := &model.SystemUser{}
  22. if err := s.DB.Where("name = ?", auth.UserName).Find(systemUser).Error; err != nil {
  23. return nil, xerr.WithStack(err)
  24. }
  25. if systemUser.Password != auth.Password {
  26. return nil, xerr.Customf("密码错误,来自用户:%s", auth.UserName)
  27. }
  28. if systemUser.IsShow == operationPb.IsShow_NO {
  29. return nil, xerr.Customf("该账号已被禁用,请联系管理员")
  30. }
  31. jwtToken := jwt.NewJWTTokenGen(s.Cfg.AppName, s.Cfg.JwtTokenKeyConfig.PrivateKey)
  32. token, err := jwtToken.GenerateToken(systemUser.Name, s.Cfg.JwtExpireTime)
  33. if err != nil {
  34. return nil, xerr.WithStack(err)
  35. }
  36. if token == "" {
  37. return nil, xerr.Custom("获取token错误")
  38. }
  39. return &operationPb.SystemToken{
  40. Code: http.StatusOK,
  41. Msg: "ok",
  42. Data: &operationPb.TokenData{Token: token},
  43. }, nil
  44. }
  45. func (s *StoreEntry) GetCurrentUserName(ctx context.Context) string {
  46. userNameInter := ctx.Value(CurrentUserName)
  47. if userNameInter == nil {
  48. zaplog.Error("GetCurrentUserName", zap.String("userNameInter", "cannot userName"))
  49. return ""
  50. }
  51. if userName, ok := userNameInter.(string); ok {
  52. return userName
  53. } else {
  54. zaplog.Error("GetCurrentUserName", zap.String("userNameInter", "waring userName"))
  55. return ""
  56. }
  57. }
  58. // GetUserInfo 获取用户信息
  59. func (s *StoreEntry) GetUserInfo(ctx context.Context) (*operationPb.UserAuth, error) {
  60. systemUser := &model.SystemUser{}
  61. userName := s.GetCurrentUserName(ctx)
  62. if err := s.DB.Where("name = ?", userName).First(systemUser).Error; err != nil {
  63. return nil, xerr.WithStack(err)
  64. }
  65. systemRole := make([]*model.SystemRole, 0)
  66. pastureList := make([]*operationPb.UserPasture, 0)
  67. roleIdStr := strings.Split(systemUser.RoleIds, ",")
  68. if len(roleIdStr) > 0 {
  69. roleIds := make([]int, 0)
  70. for _, v := range roleIdStr {
  71. roleId, _ := strconv.Atoi(v)
  72. roleIds = append(roleIds, roleId)
  73. }
  74. if err := s.DB.Model(new(model.SystemGroupPasturePermissions)).Select("DISTINCT(group_pasture.id) AS did, group_pasture.id as id,group_pasture.name as name").
  75. Where("system_group_pasture_permissions.role_id IN ?", roleIds).Where("system_group_pasture_permissions.is_show = ?", operationPb.IsShow_OK).
  76. Joins("right join group_pasture on system_group_pasture_permissions.pasture_id = group_pasture.id").Find(&pastureList).Debug().Error; err != nil {
  77. }
  78. if err := s.DB.Find(&systemRole, roleIds).Error; err != nil {
  79. return nil, xerr.WithStack(err)
  80. }
  81. }
  82. return systemUser.SystemUserFormat(systemRole, pastureList), nil
  83. }
  84. // CreateSystemUser 创建系统用户
  85. func (s *StoreEntry) CreateSystemUser(ctx context.Context, req *operationPb.AddSystemUser) error {
  86. systemUsers := &model.SystemUser{
  87. Name: req.Name,
  88. EmployeeName: req.EmployeeName,
  89. Phone: req.Phone,
  90. Password: tool.Md5String(model.InitManagerPassword),
  91. CreateUser: s.GetCurrentUserName(ctx),
  92. IsShow: operationPb.IsShow_OK,
  93. IsDelete: operationPb.IsShow_OK,
  94. }
  95. systemUsers.SystemUserRoleFormat(req)
  96. if err := s.DB.Create(systemUsers).Error; err != nil {
  97. return xerr.WithStack(err)
  98. }
  99. return nil
  100. }
  101. // SearchSystemUserList 查询系统用户
  102. func (s *StoreEntry) SearchSystemUserList(ctx context.Context, req *operationPb.SearchUserRequest) (*operationPb.SearchUserResponse, error) {
  103. systemUserList := make([]*model.SystemUser, 0)
  104. var count int64 = 0
  105. pref := s.DB.Model(new(model.SystemUser)).Where("is_delete = ?", operationPb.IsShow_OK)
  106. if req.Name != "" {
  107. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  108. }
  109. if req.EmployeeName != "" {
  110. pref.Where("employee_name like ?", fmt.Sprintf("%s%s%s", "%", req.EmployeeName, "%"))
  111. }
  112. if req.IsShow > 0 {
  113. pref.Where("is_show = ?", req.IsShow)
  114. }
  115. if req.CreatedStartTime > 0 && req.CreatedEndTime > 0 && req.CreatedStartTime <= req.CreatedEndTime {
  116. pref.Where("created_at BETWEEN ? AND ?", req.CreatedStartTime, req.CreatedEndTime)
  117. }
  118. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  119. Find(&systemUserList).Debug().Error; err != nil {
  120. return nil, xerr.WithStack(err)
  121. }
  122. roleList, err := s.SearchSystemRoleListByIds(ctx, []int64{})
  123. if err != nil {
  124. return nil, xerr.WithStack(err)
  125. }
  126. return &operationPb.SearchUserResponse{
  127. Code: http.StatusOK,
  128. Msg: "ok",
  129. Data: &operationPb.SearchUserData{
  130. Page: req.Pagination.Page,
  131. Total: int32(count),
  132. PageSize: req.Pagination.PageSize,
  133. List: model.SystemUserSlice(systemUserList).ToPB(roleList),
  134. },
  135. }, nil
  136. }
  137. // EditSystemUser 编辑用户
  138. func (s *StoreEntry) EditSystemUser(ctx context.Context, req *operationPb.AddSystemUser) error {
  139. systemUser := &model.SystemUser{Id: int64(req.Id)}
  140. if err := s.DB.First(systemUser).Error; err != nil {
  141. if errors.Is(err, gorm.ErrRecordNotFound) {
  142. return xerr.Custom("该数据不存在!")
  143. }
  144. return xerr.WithStack(err)
  145. }
  146. updateData := &model.SystemUser{
  147. Name: req.Name,
  148. EmployeeName: req.EmployeeName,
  149. Phone: req.Phone,
  150. }
  151. updateData.SystemUserRoleFormat(req)
  152. if err := s.DB.Model(new(model.SystemUser)).Omit("is_show", "password", "is_delete", "create_user").
  153. Where("id = ?", systemUser.Id).
  154. Updates(updateData).Error; err != nil {
  155. return xerr.WithStack(err)
  156. }
  157. return nil
  158. }
  159. // DeleteSystemUser 删除系统用户
  160. func (s *StoreEntry) DeleteSystemUser(ctx context.Context, userId int64) error {
  161. systemUser := &model.SystemUser{
  162. Id: userId,
  163. }
  164. if err := s.DB.First(systemUser).Error; err != nil {
  165. if errors.Is(err, gorm.ErrRecordNotFound) {
  166. return xerr.Custom("该用户不存在")
  167. }
  168. return xerr.WithStack(err)
  169. }
  170. if err := s.DB.Model(systemUser).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
  171. return xerr.WithStack(err)
  172. }
  173. return nil
  174. }
  175. // ResetPasswordSystemUser 重置系统用户密码
  176. func (s *StoreEntry) ResetPasswordSystemUser(ctx context.Context, userId int64) error {
  177. systemUser := &model.SystemUser{
  178. Id: userId,
  179. }
  180. if err := s.DB.First(systemUser).Error; err != nil {
  181. if errors.Is(err, gorm.ErrRecordNotFound) {
  182. return xerr.Custom("该用户不存在")
  183. }
  184. return xerr.WithStack(err)
  185. }
  186. if err := s.DB.Model(systemUser).Update("password", tool.Md5String(model.InitManagerPassword)).Error; err != nil {
  187. return xerr.WithStack(err)
  188. }
  189. return nil
  190. }
  191. // DetailsSystemUser 系统用户详情
  192. func (s *StoreEntry) DetailsSystemUser(ctx context.Context, userId int64) (*operationPb.UserDetails, error) {
  193. systemUser := &model.SystemUser{
  194. Id: userId,
  195. }
  196. if err := s.DB.First(systemUser).Error; err != nil {
  197. if errors.Is(err, gorm.ErrRecordNotFound) {
  198. return nil, xerr.Custom("该用户不存在")
  199. }
  200. return nil, xerr.WithStack(err)
  201. }
  202. return &operationPb.UserDetails{
  203. Code: http.StatusOK,
  204. Msg: "ok",
  205. Data: systemUser.ToPb(),
  206. }, nil
  207. }
  208. // IsShowSystemUser 用户是否启用
  209. func (s *StoreEntry) IsShowSystemUser(ctx context.Context, req *operationPb.IsShowSystemUserRequest) error {
  210. systemUser := &model.SystemUser{
  211. Id: int64(req.UserId),
  212. }
  213. if err := s.DB.First(systemUser).Error; err != nil {
  214. if errors.Is(err, gorm.ErrRecordNotFound) {
  215. return xerr.Custom("该用户不存在")
  216. }
  217. return xerr.WithStack(err)
  218. }
  219. if err := s.DB.Model(systemUser).Update("is_show", req.IsShow).Error; err != nil {
  220. return xerr.WithStack(err)
  221. }
  222. return nil
  223. }
  224. // GetSystemUserPermissions 返回系统用户相关菜单权限
  225. func (s *StoreEntry) GetSystemUserPermissions(ctx context.Context) (*operationPb.SystemUserMenuPermissions, error) {
  226. // 解析token
  227. userName := s.GetCurrentUserName(ctx)
  228. // 根据用户token获取用户数据
  229. systemUser := &model.SystemUser{Name: userName}
  230. if err := s.DB.Where("name = ?", userName).First(systemUser).Error; err != nil {
  231. if errors.Is(err, gorm.ErrRecordNotFound) {
  232. return nil, xerr.Custom("该用户数据不存在")
  233. }
  234. return nil, xerr.WithStack(err)
  235. }
  236. roleIds := systemUser.SystemUserRoleToSlice()
  237. // 获取用户角色数据
  238. systemRoles := make([]*model.SystemRole, 0)
  239. if err := s.DB.Where("is_show = ?", operationPb.IsShow_OK).Find(&systemRoles, roleIds).Error; err != nil {
  240. return nil, xerr.WithStack(err)
  241. }
  242. systemAllPermissionsList := &SystemAllPermissionsList{
  243. PastureList: make([]*model.SystemGroupPasturePermissions, 0),
  244. MenuList: make([]*model.SystemMenuPermissions, 0),
  245. MobileList: make([]*model.SystemMobilePermissions, 0),
  246. }
  247. for _, role := range systemRoles {
  248. item := s.GetSystemAllPermissionsList(ctx, role.Id)
  249. systemAllPermissionsList.PastureList = append(systemAllPermissionsList.PastureList, item.PastureList...)
  250. systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)
  251. systemAllPermissionsList.MobileList = append(systemAllPermissionsList.MobileList, item.MobileList...)
  252. }
  253. systemAllPermissionsList.SystemUserMenuPermissionsUnDuplicate()
  254. groupPastureList, err := s.GetPastureList(ctx, systemAllPermissionsList.PastureList)
  255. if err != nil {
  256. return nil, xerr.WithStack(err)
  257. }
  258. menuList, err := s.GetMenuList(ctx, systemAllPermissionsList.MenuList)
  259. if err != nil {
  260. return nil, xerr.WithStack(err)
  261. }
  262. mobileList, err := s.GetMobileList(ctx, systemAllPermissionsList.MobileList)
  263. if err != nil {
  264. return nil, xerr.WithStack(err)
  265. }
  266. return s.SystemPermissionsFormatPb(groupPastureList, mobileList, menuList), nil
  267. }
  268. // CreateSystemRole 添加角色
  269. func (s *StoreEntry) CreateSystemRole(ctx context.Context, req *operationPb.AddRoleRequest) error {
  270. if err := s.DB.Transaction(func(tx *gorm.DB) error {
  271. // 创建角色数据
  272. req.CreateUser = s.GetCurrentUserName(ctx)
  273. role := model.NewSystemRole(req)
  274. if err := tx.Create(role).Error; err != nil {
  275. return xerr.WithStack(err)
  276. }
  277. // 创建角色菜单权限数据
  278. if len(req.MenuId) > 0 {
  279. menuPermissions := model.NewSystemMenuPermissions(role.Id, req.MenuId)
  280. if err := tx.Create(menuPermissions).Error; err != nil {
  281. return xerr.WithStack(err)
  282. }
  283. }
  284. // 创建角色移动端权限数据
  285. if len(req.MobileId) > 0 {
  286. mobilePermissions := model.NewSystemMobilePermissions(role.Id, req.MobileId)
  287. if err := tx.Create(mobilePermissions).Error; err != nil {
  288. return xerr.WithStack(err)
  289. }
  290. }
  291. // 创建角色牧场端权限数据
  292. if len(req.PastureId) > 0 {
  293. pasturePermissions := model.NewSystemGroupPasturePermissions(role.Id, req.PastureId)
  294. if err := tx.Create(pasturePermissions).Error; err != nil {
  295. return xerr.WithStack(err)
  296. }
  297. }
  298. return nil
  299. }); err != nil {
  300. return xerr.WithStack(err)
  301. }
  302. return nil
  303. }
  304. // EditSystemRole 编辑角色
  305. func (s *StoreEntry) EditSystemRole(ctx context.Context, req *operationPb.AddRoleRequest) error {
  306. role := &model.SystemRole{Id: int64(req.Id)}
  307. if err := s.DB.First(role).Error; err != nil {
  308. if errors.Is(err, gorm.ErrRecordNotFound) {
  309. return xerr.Custom("该数据不存在")
  310. }
  311. return xerr.WithStack(err)
  312. }
  313. if err := s.DB.Transaction(func(tx *gorm.DB) error {
  314. updateSystemRole := model.NewSystemRole(req)
  315. if err := tx.Omit("is_show").
  316. Where("id = ?", role.Id).
  317. Updates(updateSystemRole).Error; err != nil {
  318. return xerr.WithStack(err)
  319. }
  320. // 更新牧场权限关系表
  321. pastureIds := req.PastureId
  322. if err := tx.Model(new(model.SystemGroupPasturePermissions)).
  323. Where("role_id = ?", req.Id).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  324. return xerr.WithStack(err)
  325. }
  326. newSystemGroupPasturePermissions := model.NewSystemGroupPasturePermissions(int64(req.Id), pastureIds)
  327. if err := tx.Create(newSystemGroupPasturePermissions).Error; err != nil {
  328. return xerr.WithStack(err)
  329. }
  330. // 更新菜单权限关系表
  331. menuIds := req.MenuId
  332. if err := tx.Model(new(model.SystemMenuPermissions)).
  333. Where("role_id = ?", req.Id).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  334. return xerr.WithStack(err)
  335. }
  336. newSystemMenuPermissions := model.NewSystemMenuPermissions(int64(req.Id), menuIds)
  337. if err := tx.Create(newSystemMenuPermissions).Error; err != nil {
  338. return xerr.WithStack(err)
  339. }
  340. // 更新移动端权限关系表
  341. mobileIds := req.MobileId
  342. if err := tx.Model(new(model.SystemMobilePermissions)).
  343. Where("role_id = ?", req.Id).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  344. return xerr.WithStack(err)
  345. }
  346. newSystemMobilePermissions := model.NewSystemMobilePermissions(int64(req.Id), mobileIds)
  347. if err := tx.Create(newSystemMobilePermissions).Error; err != nil {
  348. return xerr.WithStack(err)
  349. }
  350. return nil
  351. }); err != nil {
  352. return xerr.WithStack(err)
  353. }
  354. return nil
  355. }
  356. // DeleteSystemRole 删除系统角色
  357. func (s *StoreEntry) DeleteSystemRole(ctx context.Context, roleId int64) error {
  358. systemRole := &model.SystemRole{
  359. Id: roleId,
  360. }
  361. if err := s.DB.First(systemRole).Error; err != nil {
  362. if errors.Is(err, gorm.ErrRecordNotFound) {
  363. return xerr.Custom("该数据不存在")
  364. }
  365. return xerr.WithStack(err)
  366. }
  367. if err := s.DB.Model(systemRole).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  368. return xerr.WithStack(err)
  369. }
  370. return nil
  371. }
  372. // SearchSystemRoleList 查询系统角色
  373. func (s *StoreEntry) SearchSystemRoleList(ctx context.Context, req *operationPb.SearchRoleRequest) (*operationPb.SearchRoleResponse, error) {
  374. systemRoleList := make([]*model.SystemRole, 0)
  375. var count int64 = 0
  376. pref := s.DB.Model(new(model.SystemRole)).Where("is_show = ?", operationPb.IsShow_OK)
  377. if req.Name != "" {
  378. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  379. }
  380. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).
  381. Offset(int(req.Pagination.PageOffset)).Find(&systemRoleList).Error; err != nil {
  382. return nil, xerr.WithStack(err)
  383. }
  384. permissionsPastureMap := make(map[int64][]*model.GroupPasture, 0)
  385. permissionsMenuMap := make(map[int64][]*model.SystemMenu, 0)
  386. for _, role := range systemRoleList {
  387. groupPastureList := make([]*model.GroupPasture, 0)
  388. if err := s.DB.Table(new(model.GroupPasture).TableName()).Select(fmt.Sprintf("%s.*", new(model.GroupPasture).TableName())).
  389. Joins(fmt.Sprintf("left join %s on %s = %s and %s", new(model.SystemGroupPasturePermissions).TableName(), fmt.Sprintf("%s.pasture_id", new(model.SystemGroupPasturePermissions).TableName()),
  390. fmt.Sprintf("%s.id", new(model.GroupPasture).TableName()), fmt.Sprintf("%s.is_show = %d", new(model.SystemGroupPasturePermissions).TableName(), operationPb.IsShow_OK))).
  391. Where(fmt.Sprintf("%s.role_id = ?", new(model.SystemGroupPasturePermissions).TableName()), role.Id).
  392. Find(&groupPastureList).Error; err != nil {
  393. zaplog.Error("SearchSystemRoleList", zap.Any("SystemGroupPasturePermissions", err))
  394. }
  395. permissionsPastureMap[role.Id] = groupPastureList
  396. systemMenuList := make([]*model.SystemMenu, 0)
  397. if err := s.DB.Table(new(model.SystemMenu).TableName()).Select(fmt.Sprintf("%s.*", new(model.SystemMenu).TableName())).
  398. Joins(fmt.Sprintf("left join %s on %s = %s and %s", new(model.SystemMenuPermissions).TableName(), fmt.Sprintf("%s.menu_id", new(model.SystemMenuPermissions).TableName()),
  399. fmt.Sprintf("%s.id", new(model.SystemMenu).TableName()), fmt.Sprintf("%s.is_show = %d", new(model.SystemMenuPermissions).TableName(), operationPb.IsShow_OK))).
  400. Where(fmt.Sprintf("%s.role_id = ?", new(model.SystemMenuPermissions).TableName()), role.Id).Where(fmt.Sprintf("%s.level = ?", new(model.SystemMenu).TableName()), model.Level2).
  401. Find(&systemMenuList).Error; err != nil {
  402. zaplog.Error("SearchSystemRoleList", zap.Any("SystemMenuPermissions", err))
  403. }
  404. permissionsMenuMap[role.Id] = systemMenuList
  405. }
  406. return &operationPb.SearchRoleResponse{
  407. Code: http.StatusOK,
  408. Msg: "ok",
  409. Data: &operationPb.SearchRoleData{
  410. Page: req.Pagination.Page,
  411. Total: int32(count),
  412. PageSize: req.Pagination.PageSize,
  413. List: model.SystemRoleSlice(systemRoleList).ToPB(permissionsPastureMap, permissionsMenuMap),
  414. },
  415. }, nil
  416. }
  417. // SearchSystemRoleListByIds 根据id查询角色列表
  418. func (s *StoreEntry) SearchSystemRoleListByIds(ctx context.Context, ids []int64) ([]*model.SystemRole, error) {
  419. systemRoleList := make([]*model.SystemRole, 0)
  420. if err := s.DB.Model(new(model.SystemRole)).Where("is_show = ?", operationPb.IsShow_OK).Find(&systemRoleList, ids).Error; err != nil {
  421. return nil, xerr.WithStack(err)
  422. }
  423. return systemRoleList, nil
  424. }
  425. // GetRolePermissions 查询系统角色权限
  426. func (s *StoreEntry) GetRolePermissions(ctx context.Context, roleId int64) (*operationPb.RolePermissionsList, error) {
  427. systemRole := &model.SystemRole{
  428. Id: roleId,
  429. }
  430. if err := s.DB.First(systemRole).Error; err != nil {
  431. if errors.Is(err, gorm.ErrRecordNotFound) {
  432. return nil, xerr.Custom("该数据不存在")
  433. }
  434. return nil, xerr.WithStack(err)
  435. }
  436. systemAllPermissionsList := &SystemAllPermissionsList{
  437. PastureList: make([]*model.SystemGroupPasturePermissions, 0),
  438. MenuList: make([]*model.SystemMenuPermissions, 0),
  439. MobileList: make([]*model.SystemMobilePermissions, 0),
  440. }
  441. item := s.GetSystemAllPermissionsList(ctx, systemRole.Id)
  442. systemAllPermissionsList.PastureList = append(systemAllPermissionsList.PastureList, item.PastureList...)
  443. systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)
  444. systemAllPermissionsList.MobileList = append(systemAllPermissionsList.MobileList, item.MobileList...)
  445. return s.AllPermissionsListToRolePermissions(systemAllPermissionsList), nil
  446. }
  447. // CreateSystemMenu 添加系统菜单权限
  448. func (s *StoreEntry) CreateSystemMenu(ctx context.Context, req *operationPb.AddMenuRequest) error {
  449. systemMenu := model.NewSystemMenu(req)
  450. if err := s.DB.Create(systemMenu).Error; err != nil {
  451. return xerr.WithStack(err)
  452. }
  453. return nil
  454. }
  455. // EditSystemMenu 编辑系统菜单权限
  456. func (s *StoreEntry) EditSystemMenu(ctx context.Context, req *operationPb.AddMenuRequest) error {
  457. systemMenu := &model.SystemMenu{Id: int64(req.Id)}
  458. if err := s.DB.First(systemMenu).Error; err != nil {
  459. if errors.Is(err, gorm.ErrRecordNotFound) {
  460. return xerr.Custom("该数据不存在!")
  461. }
  462. return xerr.WithStack(err)
  463. }
  464. updateData := &model.SystemMenu{
  465. Name: req.Name,
  466. MenuType: req.MenuType,
  467. Title: req.Title,
  468. Path: req.Path,
  469. Component: req.Component,
  470. Icon: req.Icon,
  471. Sort: req.Sort,
  472. Redirect: req.Redirect,
  473. ParentId: int64(req.ParentId),
  474. }
  475. if err := s.DB.Model(new(model.SystemMenu)).Omit("is_show").
  476. Where("id = ?", systemMenu.Id).
  477. Updates(updateData).Error; err != nil {
  478. return xerr.WithStack(err)
  479. }
  480. return nil
  481. }
  482. // IsShowSystemMenu 菜单是否启用
  483. func (s *StoreEntry) IsShowSystemMenu(ctx context.Context, req *operationPb.IsShowSystemMenuRequest) error {
  484. systemMenu := &model.SystemMenu{Id: int64(req.MenuId)}
  485. if err := s.DB.First(systemMenu).Error; err != nil {
  486. if errors.Is(err, gorm.ErrRecordNotFound) {
  487. return xerr.Custom("该数据不存在")
  488. }
  489. return xerr.WithStack(err)
  490. }
  491. if err := s.DB.Model(systemMenu).Update("is_show", req.IsShow).Error; err != nil {
  492. return xerr.WithStack(err)
  493. }
  494. return nil
  495. }
  496. // SearchSystemMenuList 菜单列表查询
  497. func (s *StoreEntry) SearchSystemMenuList(ctx context.Context, req *operationPb.SearchMenuRequest) (*operationPb.SearchMenuResponse, error) {
  498. systemMenuLevel1 := make([]*model.SystemMenu, 0)
  499. var count int64 = 0
  500. pref := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? and level = ?", operationPb.IsShow_OK, model.Level1)
  501. if req.Name != "" {
  502. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  503. }
  504. if err := pref.Order("sort desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  505. Find(&systemMenuLevel1).Debug().Error; err != nil {
  506. return nil, xerr.WithStack(err)
  507. }
  508. systemMenuLevel1 = append(systemMenuLevel1, s.searchMenuLevel23ByLevel1(ctx, systemMenuLevel1)...)
  509. return &operationPb.SearchMenuResponse{
  510. Code: http.StatusOK,
  511. Msg: "ok",
  512. Data: &operationPb.SearchMenuData{
  513. Page: req.Pagination.Page,
  514. Total: int32(count),
  515. List: model.SystemMenuSlice(systemMenuLevel1).ToPB(),
  516. },
  517. }, nil
  518. }
  519. // searchMenuLevel23ByLevel1 根据一级菜单返回对应二三级菜单
  520. func (s *StoreEntry) searchMenuLevel23ByLevel1(ctx context.Context, res []*model.SystemMenu) []*model.SystemMenu {
  521. systemMenuLevel23 := make([]*model.SystemMenu, 0)
  522. if len(res) <= 0 {
  523. return systemMenuLevel23
  524. }
  525. ids1 := make([]int64, 0)
  526. for _, v := range res {
  527. ids1 = append(ids1, v.Id)
  528. }
  529. if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? and level = ?", operationPb.IsShow_OK, model.Level2).
  530. Where("parent_id IN ?", ids1).Order("sort desc").
  531. Find(&systemMenuLevel23).Error; err != nil {
  532. return systemMenuLevel23
  533. }
  534. ids2 := make([]int64, 0)
  535. for _, v := range systemMenuLevel23 {
  536. ids2 = append(ids2, v.Id)
  537. }
  538. systemMenuLevel3 := make([]*model.SystemMenu, 0)
  539. if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? and level = ?", operationPb.IsShow_OK, model.Level3).
  540. Where("parent_id IN ?", ids2).Order("sort desc").
  541. Find(&systemMenuLevel3).Error; err != nil {
  542. return systemMenuLevel23
  543. }
  544. if len(systemMenuLevel3) > 0 {
  545. systemMenuLevel23 = append(systemMenuLevel23, systemMenuLevel3...)
  546. }
  547. return systemMenuLevel23
  548. }
  549. // DeleteSystemMenu 删除系统菜单
  550. func (s *StoreEntry) DeleteSystemMenu(ctx context.Context, menuId int64) error {
  551. systemMenu := &model.SystemMenu{Id: menuId}
  552. if err := s.DB.First(systemMenu).Error; err != nil {
  553. if errors.Is(err, gorm.ErrRecordNotFound) {
  554. return xerr.Custom("该数据不存在")
  555. }
  556. return xerr.WithStack(err)
  557. }
  558. if err := s.DB.Model(systemMenu).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
  559. return xerr.WithStack(err)
  560. }
  561. return nil
  562. }
  563. // SearchMobileList 查询移动端角色
  564. func (s *StoreEntry) SearchMobileList(ctx context.Context, req *operationPb.SearchMobileRequest) (*operationPb.SearchMobileResponse, error) {
  565. systemMobile := make([]*model.SystemMobile, 0)
  566. var count int64 = 0
  567. pref := s.DB.Model(new(model.SystemMobile)).Where("is_show = ?", operationPb.IsShow_OK)
  568. if req.Name != "" {
  569. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  570. }
  571. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  572. Find(&systemMobile).Debug().Error; err != nil {
  573. return nil, xerr.WithStack(err)
  574. }
  575. return &operationPb.SearchMobileResponse{
  576. Code: http.StatusOK,
  577. Msg: "ok",
  578. Data: &operationPb.SearchMobileData{
  579. Page: req.Pagination.Page,
  580. Total: int32(count),
  581. PageSize: req.Pagination.PageSize,
  582. List: model.SystemMobileSlice(systemMobile).ToPB(),
  583. },
  584. }, nil
  585. }