system_service.go 23 KB

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