system_service.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682
  1. package backend
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "kpt-tmr-group/model"
  7. "kpt-tmr-group/pkg/jwt"
  8. "kpt-tmr-group/pkg/logger/zaplog"
  9. "kpt-tmr-group/pkg/tool"
  10. "kpt-tmr-group/pkg/xerr"
  11. operationPb "kpt-tmr-group/proto/go/backend/operation"
  12. "net/http"
  13. "strconv"
  14. "strings"
  15. "go.uber.org/zap"
  16. "gorm.io/gorm"
  17. )
  18. // Auth 用户登录
  19. func (s *StoreEntry) Auth(ctx context.Context, auth *operationPb.UserAuthData) (*operationPb.SystemToken, error) {
  20. systemUser := &model.SystemUser{}
  21. if err := s.DB.Where("name = ?", auth.UserName).Find(systemUser).Error; err != nil {
  22. return nil, xerr.WithStack(err)
  23. }
  24. if systemUser.Password != auth.Password {
  25. return nil, xerr.Customf("密码错误,来自用户:%s", auth.UserName)
  26. }
  27. if systemUser.IsShow == operationPb.IsShow_NO {
  28. return nil, xerr.Customf("该账号已被禁用,请联系管理员")
  29. }
  30. jwtToken := jwt.NewJWTTokenGen(s.Cfg.AppName, s.Cfg.JwtTokenKeyConfig.PrivateKey)
  31. token, err := jwtToken.GenerateToken(systemUser.Name, s.Cfg.JwtExpireTime)
  32. if err != nil {
  33. return nil, xerr.WithStack(err)
  34. }
  35. if token == "" {
  36. return nil, xerr.Custom("获取token错误")
  37. }
  38. return &operationPb.SystemToken{
  39. Code: http.StatusOK,
  40. Msg: "ok",
  41. Data: &operationPb.TokenData{Token: token},
  42. }, nil
  43. }
  44. func (s *StoreEntry) GetUserName(ctx context.Context) (string, error) {
  45. userNameInter := ctx.Value("userName")
  46. if userNameInter == nil {
  47. return "", xerr.Customf("cannot userName")
  48. }
  49. if userName, ok := userNameInter.(string); ok {
  50. return userName, nil
  51. } else {
  52. return "", xerr.Customf("waring userName")
  53. }
  54. }
  55. // GetUserInfo 获取用户信息
  56. func (s *StoreEntry) GetUserInfo(ctx context.Context, token string) (*operationPb.UserAuth, error) {
  57. systemUser := &model.SystemUser{}
  58. userName, err := s.GetUserName(ctx)
  59. if err != nil {
  60. return nil, xerr.WithStack(err)
  61. }
  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: req.CreateUser,
  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, token string) (*operationPb.SystemUserMenuPermissions, error) {
  226. // 解析token
  227. /*claims, err := jwt.ParseToken(token)
  228. if err != nil {
  229. return nil, xerr.WithStack(err)
  230. }
  231. if err = claims.Valid(); err != nil {
  232. return nil, xerr.WithStack(err)
  233. }
  234. // 根据用户token获取用户数据
  235. systemUser := &model.SystemUser{Name: claims.Username}
  236. if err = s.DB.Where("name = ?", claims.Username).First(systemUser).Error; err != nil {
  237. if errors.Is(err, gorm.ErrRecordNotFound) {
  238. return nil, xerr.Custom("该用户数据不存在")
  239. }
  240. return nil, xerr.WithStack(err)
  241. }
  242. roleIds := systemUser.SystemUserRoleToSlice()
  243. // 获取用户角色数据
  244. systemRoles := make([]*model.SystemRole, 0)
  245. if err = s.DB.Where("is_show = ?", operationPb.IsShow_OK).Find(&systemRoles, roleIds).Error; err != nil {
  246. return nil, xerr.WithStack(err)
  247. }*/
  248. systemRoles := make([]*model.SystemRole, 0)
  249. systemAllPermissionsList := &SystemAllPermissionsList{
  250. PastureList: make([]*model.SystemGroupPasturePermissions, 0),
  251. MenuList: make([]*model.SystemMenuPermissions, 0),
  252. MobileList: make([]*model.SystemMobilePermissions, 0),
  253. }
  254. for _, role := range systemRoles {
  255. item := s.GetSystemAllPermissionsList(ctx, role.Id)
  256. systemAllPermissionsList.PastureList = append(systemAllPermissionsList.PastureList, item.PastureList...)
  257. systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)
  258. systemAllPermissionsList.MobileList = append(systemAllPermissionsList.MobileList, item.MobileList...)
  259. }
  260. systemAllPermissionsList.SystemUserMenuPermissionsUnDuplicate()
  261. groupPastureList, err := s.GetPastureList(ctx, systemAllPermissionsList.PastureList)
  262. if err != nil {
  263. return nil, xerr.WithStack(err)
  264. }
  265. menuList, err := s.GetMenuList(ctx, systemAllPermissionsList.MenuList)
  266. if err != nil {
  267. return nil, xerr.WithStack(err)
  268. }
  269. mobileList, err := s.GetMobileList(ctx, systemAllPermissionsList.MobileList)
  270. if err != nil {
  271. return nil, xerr.WithStack(err)
  272. }
  273. return s.SystemPermissionsFormatPb(groupPastureList, mobileList, menuList), nil
  274. }
  275. // CreateSystemRole 添加角色
  276. func (s *StoreEntry) CreateSystemRole(ctx context.Context, req *operationPb.AddRoleRequest) error {
  277. if err := s.DB.Transaction(func(tx *gorm.DB) error {
  278. // 创建角色数据
  279. role := model.NewSystemRole(req)
  280. if err := tx.Create(role).Error; err != nil {
  281. return xerr.WithStack(err)
  282. }
  283. // 创建角色菜单权限数据
  284. if len(req.MenuId) > 0 {
  285. menuPermissions := model.NewSystemMenuPermissions(role.Id, req.MenuId)
  286. if err := tx.Create(menuPermissions).Error; err != nil {
  287. return xerr.WithStack(err)
  288. }
  289. }
  290. // 创建角色移动端权限数据
  291. if len(req.MobileId) > 0 {
  292. mobilePermissions := model.NewSystemMobilePermissions(role.Id, req.MobileId)
  293. if err := tx.Create(mobilePermissions).Error; err != nil {
  294. return xerr.WithStack(err)
  295. }
  296. }
  297. // 创建角色牧场端权限数据
  298. if len(req.PastureId) > 0 {
  299. pasturePermissions := model.NewSystemGroupPasturePermissions(role.Id, req.PastureId)
  300. if err := tx.Create(pasturePermissions).Error; err != nil {
  301. return xerr.WithStack(err)
  302. }
  303. }
  304. return nil
  305. }); err != nil {
  306. return xerr.WithStack(err)
  307. }
  308. return nil
  309. }
  310. // EditSystemRole 编辑角色
  311. func (s *StoreEntry) EditSystemRole(ctx context.Context, req *operationPb.AddRoleRequest) error {
  312. role := &model.SystemRole{Id: int64(req.Id)}
  313. if err := s.DB.First(role).Error; err != nil {
  314. if errors.Is(err, gorm.ErrRecordNotFound) {
  315. return xerr.Custom("该数据不存在")
  316. }
  317. return xerr.WithStack(err)
  318. }
  319. if err := s.DB.Transaction(func(tx *gorm.DB) error {
  320. updateSystemRole := model.NewSystemRole(req)
  321. if err := tx.Omit("is_show").
  322. Where("id = ?", role.Id).
  323. Updates(updateSystemRole).Error; err != nil {
  324. return xerr.WithStack(err)
  325. }
  326. // 更新牧场权限关系表
  327. pastureIds := req.PastureId
  328. if err := tx.Model(new(model.SystemGroupPasturePermissions)).
  329. Where("role_id = ?", req.Id).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  330. return xerr.WithStack(err)
  331. }
  332. newSystemGroupPasturePermissions := model.NewSystemGroupPasturePermissions(int64(req.Id), pastureIds)
  333. if err := tx.Create(newSystemGroupPasturePermissions).Error; err != nil {
  334. return xerr.WithStack(err)
  335. }
  336. // 更新菜单权限关系表
  337. menuIds := req.MenuId
  338. if err := tx.Model(new(model.SystemMenuPermissions)).
  339. Where("role_id = ?", req.Id).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  340. return xerr.WithStack(err)
  341. }
  342. newSystemMenuPermissions := model.NewSystemMenuPermissions(int64(req.Id), menuIds)
  343. if err := tx.Create(newSystemMenuPermissions).Error; err != nil {
  344. return xerr.WithStack(err)
  345. }
  346. // 更新移动端权限关系表
  347. mobileIds := req.MobileId
  348. if err := tx.Model(new(model.SystemMobilePermissions)).
  349. Where("role_id = ?", req.Id).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  350. return xerr.WithStack(err)
  351. }
  352. newSystemMobilePermissions := model.NewSystemMobilePermissions(int64(req.Id), mobileIds)
  353. if err := tx.Create(newSystemMobilePermissions).Error; err != nil {
  354. return xerr.WithStack(err)
  355. }
  356. return nil
  357. }); err != nil {
  358. return xerr.WithStack(err)
  359. }
  360. return nil
  361. }
  362. // DeleteSystemRole 删除系统角色
  363. func (s *StoreEntry) DeleteSystemRole(ctx context.Context, roleId int64) error {
  364. systemRole := &model.SystemRole{
  365. Id: roleId,
  366. }
  367. if err := s.DB.First(systemRole).Error; err != nil {
  368. if errors.Is(err, gorm.ErrRecordNotFound) {
  369. return xerr.Custom("该数据不存在")
  370. }
  371. return xerr.WithStack(err)
  372. }
  373. if err := s.DB.Model(systemRole).Update("is_show", operationPb.IsShow_NO).Error; err != nil {
  374. return xerr.WithStack(err)
  375. }
  376. return nil
  377. }
  378. // SearchSystemRoleList 查询系统角色
  379. func (s *StoreEntry) SearchSystemRoleList(ctx context.Context, req *operationPb.SearchRoleRequest) (*operationPb.SearchRoleResponse, error) {
  380. systemRoleList := make([]*model.SystemRole, 0)
  381. var count int64 = 0
  382. pref := s.DB.Model(new(model.SystemRole)).Where("is_show = ?", operationPb.IsShow_OK)
  383. if req.Name != "" {
  384. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  385. }
  386. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).
  387. Offset(int(req.Pagination.PageOffset)).Find(&systemRoleList).Error; err != nil {
  388. return nil, xerr.WithStack(err)
  389. }
  390. permissionsPastureMap := make(map[int64][]*model.GroupPasture, 0)
  391. permissionsMenuMap := make(map[int64][]*model.SystemMenu, 0)
  392. for _, role := range systemRoleList {
  393. groupPastureList := make([]*model.GroupPasture, 0)
  394. if err := s.DB.Table(new(model.GroupPasture).TableName()).Select(fmt.Sprintf("%s.*", new(model.GroupPasture).TableName())).
  395. Joins(fmt.Sprintf("left join %s on %s = %s and %s", new(model.SystemGroupPasturePermissions).TableName(), fmt.Sprintf("%s.pasture_id", new(model.SystemGroupPasturePermissions).TableName()),
  396. fmt.Sprintf("%s.id", new(model.GroupPasture).TableName()), fmt.Sprintf("%s.is_show = %d", new(model.SystemGroupPasturePermissions).TableName(), operationPb.IsShow_OK))).
  397. Where(fmt.Sprintf("%s.role_id = ?", new(model.SystemGroupPasturePermissions).TableName()), role.Id).
  398. Find(&groupPastureList).Error; err != nil {
  399. zaplog.Error("SearchSystemRoleList", zap.Any("SystemGroupPasturePermissions", err))
  400. }
  401. permissionsPastureMap[role.Id] = groupPastureList
  402. systemMenuList := make([]*model.SystemMenu, 0)
  403. if err := s.DB.Table(new(model.SystemMenu).TableName()).Select(fmt.Sprintf("%s.*", new(model.SystemMenu).TableName())).
  404. Joins(fmt.Sprintf("left join %s on %s = %s and %s", new(model.SystemMenuPermissions).TableName(), fmt.Sprintf("%s.menu_id", new(model.SystemMenuPermissions).TableName()),
  405. fmt.Sprintf("%s.id", new(model.SystemMenu).TableName()), fmt.Sprintf("%s.is_show = %d", new(model.SystemMenuPermissions).TableName(), operationPb.IsShow_OK))).
  406. Where(fmt.Sprintf("%s.role_id = ?", new(model.SystemMenuPermissions).TableName()), role.Id).Where(fmt.Sprintf("%s.level = ?", new(model.SystemMenu).TableName()), model.Level2).
  407. Find(&systemMenuList).Error; err != nil {
  408. zaplog.Error("SearchSystemRoleList", zap.Any("SystemMenuPermissions", err))
  409. }
  410. permissionsMenuMap[role.Id] = systemMenuList
  411. }
  412. return &operationPb.SearchRoleResponse{
  413. Code: http.StatusOK,
  414. Msg: "ok",
  415. Data: &operationPb.SearchRoleData{
  416. Page: req.Pagination.Page,
  417. Total: int32(count),
  418. PageSize: req.Pagination.PageSize,
  419. List: model.SystemRoleSlice(systemRoleList).ToPB(permissionsPastureMap, permissionsMenuMap),
  420. },
  421. }, nil
  422. }
  423. // SearchSystemRoleListByIds 根据id查询角色列表
  424. func (s *StoreEntry) SearchSystemRoleListByIds(ctx context.Context, ids []int64) ([]*model.SystemRole, error) {
  425. systemRoleList := make([]*model.SystemRole, 0)
  426. if err := s.DB.Model(new(model.SystemRole)).Where("is_show = ?", operationPb.IsShow_OK).Find(&systemRoleList, ids).Error; err != nil {
  427. return nil, xerr.WithStack(err)
  428. }
  429. return systemRoleList, nil
  430. }
  431. // GetRolePermissions 查询系统角色权限
  432. func (s *StoreEntry) GetRolePermissions(ctx context.Context, roleId int64) (*operationPb.RolePermissionsList, error) {
  433. systemRole := &model.SystemRole{
  434. Id: roleId,
  435. }
  436. if err := s.DB.First(systemRole).Error; err != nil {
  437. if errors.Is(err, gorm.ErrRecordNotFound) {
  438. return nil, xerr.Custom("该数据不存在")
  439. }
  440. return nil, xerr.WithStack(err)
  441. }
  442. systemAllPermissionsList := &SystemAllPermissionsList{
  443. PastureList: make([]*model.SystemGroupPasturePermissions, 0),
  444. MenuList: make([]*model.SystemMenuPermissions, 0),
  445. MobileList: make([]*model.SystemMobilePermissions, 0),
  446. }
  447. item := s.GetSystemAllPermissionsList(ctx, systemRole.Id)
  448. systemAllPermissionsList.PastureList = append(systemAllPermissionsList.PastureList, item.PastureList...)
  449. systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)
  450. systemAllPermissionsList.MobileList = append(systemAllPermissionsList.MobileList, item.MobileList...)
  451. return s.AllPermissionsListToRolePermissions(systemAllPermissionsList), nil
  452. }
  453. // CreateSystemMenu 添加系统菜单权限
  454. func (s *StoreEntry) CreateSystemMenu(ctx context.Context, req *operationPb.AddMenuRequest) error {
  455. systemMenu := model.NewSystemMenu(req)
  456. if err := s.DB.Create(systemMenu).Error; err != nil {
  457. return xerr.WithStack(err)
  458. }
  459. return nil
  460. }
  461. // EditSystemMenu 编辑系统菜单权限
  462. func (s *StoreEntry) EditSystemMenu(ctx context.Context, req *operationPb.AddMenuRequest) error {
  463. systemMenu := &model.SystemMenu{Id: int64(req.Id)}
  464. if err := s.DB.First(systemMenu).Error; err != nil {
  465. if errors.Is(err, gorm.ErrRecordNotFound) {
  466. return xerr.Custom("该数据不存在!")
  467. }
  468. return xerr.WithStack(err)
  469. }
  470. updateData := &model.SystemMenu{
  471. Name: req.Name,
  472. MenuType: req.MenuType,
  473. Title: req.Title,
  474. Path: req.Path,
  475. Component: req.Component,
  476. Icon: req.Icon,
  477. Sort: req.Sort,
  478. Redirect: req.Redirect,
  479. ParentId: int64(req.ParentId),
  480. }
  481. if err := s.DB.Model(new(model.SystemMenu)).Omit("is_show").
  482. Where("id = ?", systemMenu.Id).
  483. Updates(updateData).Error; err != nil {
  484. return xerr.WithStack(err)
  485. }
  486. return nil
  487. }
  488. // IsShowSystemMenu 菜单是否启用
  489. func (s *StoreEntry) IsShowSystemMenu(ctx context.Context, req *operationPb.IsShowSystemMenuRequest) error {
  490. systemMenu := &model.SystemMenu{Id: int64(req.MenuId)}
  491. if err := s.DB.First(systemMenu).Error; err != nil {
  492. if errors.Is(err, gorm.ErrRecordNotFound) {
  493. return xerr.Custom("该数据不存在")
  494. }
  495. return xerr.WithStack(err)
  496. }
  497. if err := s.DB.Model(systemMenu).Update("is_show", req.IsShow).Error; err != nil {
  498. return xerr.WithStack(err)
  499. }
  500. return nil
  501. }
  502. // SearchSystemMenuList 菜单列表查询
  503. func (s *StoreEntry) SearchSystemMenuList(ctx context.Context, req *operationPb.SearchMenuRequest) (*operationPb.SearchMenuResponse, error) {
  504. systemMenuLevel1 := make([]*model.SystemMenu, 0)
  505. var count int64 = 0
  506. pref := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? and level = ?", operationPb.IsShow_OK, model.Level1)
  507. if req.Name != "" {
  508. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  509. }
  510. if err := pref.Order("sort desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  511. Find(&systemMenuLevel1).Debug().Error; err != nil {
  512. return nil, xerr.WithStack(err)
  513. }
  514. systemMenuLevel1 = append(systemMenuLevel1, s.searchMenuLevel23ByLevel1(ctx, systemMenuLevel1)...)
  515. return &operationPb.SearchMenuResponse{
  516. Code: http.StatusOK,
  517. Msg: "ok",
  518. Data: &operationPb.SearchMenuData{
  519. Page: req.Pagination.Page,
  520. Total: int32(count),
  521. List: model.SystemMenuSlice(systemMenuLevel1).ToPB(),
  522. },
  523. }, nil
  524. }
  525. // searchMenuLevel23ByLevel1 根据一级菜单返回对应二三级菜单
  526. func (s *StoreEntry) searchMenuLevel23ByLevel1(ctx context.Context, res []*model.SystemMenu) []*model.SystemMenu {
  527. systemMenuLevel23 := make([]*model.SystemMenu, 0)
  528. if len(res) <= 0 {
  529. return systemMenuLevel23
  530. }
  531. ids1 := make([]int64, 0)
  532. for _, v := range res {
  533. ids1 = append(ids1, v.Id)
  534. }
  535. if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? and level = ?", operationPb.IsShow_OK, model.Level2).
  536. Where("parent_id IN ?", ids1).Order("sort desc").
  537. Find(&systemMenuLevel23).Error; err != nil {
  538. return systemMenuLevel23
  539. }
  540. ids2 := make([]int64, 0)
  541. for _, v := range systemMenuLevel23 {
  542. ids2 = append(ids2, v.Id)
  543. }
  544. systemMenuLevel3 := make([]*model.SystemMenu, 0)
  545. if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? and level = ?", operationPb.IsShow_OK, model.Level3).
  546. Where("parent_id IN ?", ids2).Order("sort desc").
  547. Find(&systemMenuLevel3).Error; err != nil {
  548. return systemMenuLevel23
  549. }
  550. if len(systemMenuLevel3) > 0 {
  551. systemMenuLevel23 = append(systemMenuLevel23, systemMenuLevel3...)
  552. }
  553. return systemMenuLevel23
  554. }
  555. // DeleteSystemMenu 删除系统菜单
  556. func (s *StoreEntry) DeleteSystemMenu(ctx context.Context, menuId int64) error {
  557. systemMenu := &model.SystemMenu{Id: menuId}
  558. if err := s.DB.First(systemMenu).Error; err != nil {
  559. if errors.Is(err, gorm.ErrRecordNotFound) {
  560. return xerr.Custom("该数据不存在")
  561. }
  562. return xerr.WithStack(err)
  563. }
  564. if err := s.DB.Model(systemMenu).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
  565. return xerr.WithStack(err)
  566. }
  567. return nil
  568. }
  569. // SearchMobileList 查询移动端角色
  570. func (s *StoreEntry) SearchMobileList(ctx context.Context, req *operationPb.SearchMobileRequest) (*operationPb.SearchMobileResponse, error) {
  571. systemMobile := make([]*model.SystemMobile, 0)
  572. var count int64 = 0
  573. pref := s.DB.Model(new(model.SystemMobile)).Where("is_show = ?", operationPb.IsShow_OK)
  574. if req.Name != "" {
  575. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  576. }
  577. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  578. Find(&systemMobile).Debug().Error; err != nil {
  579. return nil, xerr.WithStack(err)
  580. }
  581. return &operationPb.SearchMobileResponse{
  582. Code: http.StatusOK,
  583. Msg: "ok",
  584. Data: &operationPb.SearchMobileData{
  585. Page: req.Pagination.Page,
  586. Total: int32(count),
  587. PageSize: req.Pagination.PageSize,
  588. List: model.SystemMobileSlice(systemMobile).ToPB(),
  589. },
  590. }, nil
  591. }