system_service.go 20 KB

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