system_service.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654
  1. package backend
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "kpt-pasture/model"
  7. "kpt-pasture/util"
  8. "net/http"
  9. "time"
  10. pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
  11. operationPb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/operation"
  12. "gitee.com/xuyiping_admin/pkg/jwt"
  13. "gitee.com/xuyiping_admin/pkg/xerr"
  14. "gorm.io/gorm"
  15. )
  16. const CurrentUserName = "userName"
  17. // Login 用户登录
  18. func (s *StoreEntry) Login(ctx context.Context, req *pasturePb.SearchUserRequest) (*pasturePb.SystemUserResponse, error) {
  19. systemUser := &model.SystemUser{}
  20. if err := s.DB.Where("name = ?", req.Name).Find(systemUser).Error; err != nil {
  21. return nil, xerr.WithStack(err)
  22. }
  23. if systemUser.Password != req.Password {
  24. return nil, xerr.Customf("密码错误,来自用户:%s", req.Name)
  25. }
  26. if systemUser.IsShow == pasturePb.IsShow_No {
  27. return nil, xerr.Customf("该账号已被禁用,请联系管理员")
  28. }
  29. jwtToken := jwt.NewJWTTokenGen(s.Cfg.AppName, s.Cfg.JwtTokenKeyConfig.PrivateKey)
  30. token, err := jwtToken.GenerateToken(systemUser.Name, s.Cfg.JwtExpireTime)
  31. if err != nil {
  32. return nil, xerr.WithStack(err)
  33. }
  34. if token == "" {
  35. return nil, xerr.Custom("获取token错误")
  36. }
  37. expires := time.Now().Add(time.Duration(s.Cfg.JwtExpireTime) * time.Second).Format(util.LayoutTime)
  38. return &pasturePb.SystemUserResponse{
  39. Code: http.StatusOK,
  40. Message: "ok",
  41. Data: &pasturePb.SystemUserData{
  42. AccessToken: token,
  43. Expires: expires,
  44. RefreshToken: expires,
  45. Username: systemUser.Name,
  46. Roles: []string{"admin"},
  47. Avatar: systemUser.Avatar,
  48. },
  49. }, nil
  50. }
  51. // SearchSystemUserList 查询系统用户
  52. func (s *StoreEntry) SearchSystemUserList(ctx context.Context, req *pasturePb.SearchUserRequest) (*pasturePb.SearchUserResponse, error) {
  53. systemUserList := make([]*model.SystemUser, 0)
  54. var count int64 = 0
  55. pref := s.DB.Model(new(model.SystemUser)).Where("is_delete = ?", operationPb.IsShow_OK)
  56. if req.Name != "" {
  57. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  58. }
  59. if req.Mobile != "" {
  60. pref.Where("mobile = ?", req.Mobile)
  61. }
  62. if req.DeptId > 0 {
  63. pref.Where("dept_id = ?", req.DeptId)
  64. }
  65. if req.IsShow > 0 {
  66. pref.Where("is_show = ?", req.IsShow)
  67. }
  68. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  69. Find(&systemUserList).Debug().Error; err != nil {
  70. return nil, xerr.WithStack(err)
  71. }
  72. roleList, err := s.SearchSystemRoleListByIds(ctx, []int64{})
  73. if err != nil {
  74. return nil, xerr.WithStack(err)
  75. }
  76. deptList, err := s.SearchSystemDeptListByIds(ctx, []int64{})
  77. if err != nil {
  78. return nil, xerr.WithStack(err)
  79. }
  80. return &pasturePb.SearchUserResponse{
  81. Code: http.StatusOK,
  82. Message: "ok",
  83. Data: &pasturePb.SearchUserData{
  84. List: model.SystemUserSlice(systemUserList).ToPB(deptList, roleList),
  85. Total: int32(count),
  86. PageSize: req.Pagination.PageSize,
  87. Page: req.Pagination.Page,
  88. },
  89. }, nil
  90. }
  91. // DeleteSystemUser 删除系统用户
  92. func (s *StoreEntry) DeleteSystemUser(ctx context.Context, userId int64) error {
  93. systemUser := &model.SystemUser{
  94. Id: userId,
  95. }
  96. if err := s.DB.First(systemUser).Error; err != nil {
  97. if errors.Is(err, gorm.ErrRecordNotFound) {
  98. return xerr.Custom("该用户不存在")
  99. }
  100. return xerr.WithStack(err)
  101. }
  102. if err := s.DB.Model(systemUser).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
  103. return xerr.WithStack(err)
  104. }
  105. return nil
  106. }
  107. // IsShowSystemUser 用户是否启用
  108. func (s *StoreEntry) IsShowSystemUser(ctx context.Context, userId int64) error {
  109. systemUser := &model.SystemUser{
  110. Id: userId,
  111. }
  112. if err := s.DB.First(systemUser).Error; err != nil {
  113. if errors.Is(err, gorm.ErrRecordNotFound) {
  114. return xerr.Custom("该用户不存在")
  115. }
  116. return xerr.WithStack(err)
  117. }
  118. isShow := pasturePb.IsShow_No
  119. if systemUser.IsShow == pasturePb.IsShow_No {
  120. isShow = pasturePb.IsShow_Ok
  121. }
  122. if err := s.DB.Model(systemUser).Update("is_show", isShow).Error; err != nil {
  123. return xerr.WithStack(err)
  124. }
  125. return nil
  126. }
  127. // GetCurrentUserName 获取当前用户名
  128. func (s *StoreEntry) GetCurrentUserName(ctx context.Context) (string, error) {
  129. userNameInter := ctx.Value(CurrentUserName)
  130. if userNameInter == nil {
  131. return "", xerr.Customf("cannot userName")
  132. }
  133. if userName, ok := userNameInter.(string); ok {
  134. return userName, nil
  135. } else {
  136. return "", xerr.Customf("waring userName")
  137. }
  138. }
  139. // GetUserInfo 获取用户信息
  140. func (s *StoreEntry) GetUserInfo(ctx context.Context, token string) (*operationPb.UserAuth, error) {
  141. systemUser := &model.SystemUser{}
  142. userName, err := s.GetCurrentUserName(ctx)
  143. if err != nil {
  144. return nil, xerr.WithStack(err)
  145. }
  146. if err = s.DB.Where("name = ?", userName).First(systemUser).Error; err != nil {
  147. return nil, xerr.WithStack(err)
  148. }
  149. systemRole := make([]*model.SystemRole, 0)
  150. return systemUser.SystemUserFormat(systemRole, nil), nil
  151. }
  152. // SystemUserCreateOrUpdate 创建或者更新系统用户
  153. func (s *StoreEntry) SystemUserCreateOrUpdate(ctx context.Context, req *pasturePb.SearchUserRequest) error {
  154. if req.Id > 0 {
  155. systemUser := &model.SystemUser{}
  156. if err := s.DB.First(systemUser).Error; err != nil {
  157. return xerr.WithStack(err)
  158. }
  159. req.Password = systemUser.Password
  160. }
  161. if err := s.DB.Model(&model.SystemUser{}).Where(map[string]interface{}{
  162. "id": req.Id,
  163. }).Assign(map[string]interface{}{
  164. "name": req.Name,
  165. "nick_name": req.NickName,
  166. "mobile": req.Mobile,
  167. "gender": req.Gender,
  168. "password": req.Password,
  169. "dept_id": req.DeptId,
  170. "remarks": req.Remarks,
  171. "is_delete": operationPb.IsShow_OK,
  172. "is_show": req.IsShow,
  173. }).FirstOrCreate(&model.SystemUser{}).Error; err != nil {
  174. return xerr.WithStack(err)
  175. }
  176. return nil
  177. }
  178. // ResetPasswordSystemUser 重置系统用户密码
  179. func (s *StoreEntry) ResetPasswordSystemUser(ctx context.Context, req *pasturePb.ResetUserPasswordRequest) error {
  180. systemUser := &model.SystemUser{
  181. Id: int64(req.Id),
  182. }
  183. if err := s.DB.First(systemUser).Error; err != nil {
  184. if errors.Is(err, gorm.ErrRecordNotFound) {
  185. return xerr.Custom("该用户不存在")
  186. }
  187. return xerr.WithStack(err)
  188. }
  189. if err := s.DB.Model(systemUser).Update("password", req.Password).Error; err != nil {
  190. return xerr.WithStack(err)
  191. }
  192. return nil
  193. }
  194. // DetailsSystemUser 系统用户详情
  195. func (s *StoreEntry) DetailsSystemUser(ctx context.Context, userId int64) (*operationPb.UserDetails, error) {
  196. systemUser := &model.SystemUser{
  197. Id: userId,
  198. }
  199. if err := s.DB.First(systemUser).Error; err != nil {
  200. if errors.Is(err, gorm.ErrRecordNotFound) {
  201. return nil, xerr.Custom("该用户不存在")
  202. }
  203. return nil, xerr.WithStack(err)
  204. }
  205. return &operationPb.UserDetails{
  206. Code: http.StatusOK,
  207. Msg: "ok",
  208. Data: systemUser.ToPb(),
  209. }, nil
  210. }
  211. // GetSystemUserPermissions 返回系统用户相关菜单权限
  212. func (s *StoreEntry) GetSystemUserPermissions(ctx context.Context, token string) (*operationPb.SystemUserMenuPermissions, error) {
  213. // 解析token
  214. userName, err := s.GetCurrentUserName(ctx)
  215. if err != nil {
  216. return nil, xerr.WithStack(err)
  217. }
  218. // 根据用户token获取用户数据
  219. systemUser := &model.SystemUser{Name: userName}
  220. if err = s.DB.Where("name = ?", userName).First(systemUser).Error; err != nil {
  221. if errors.Is(err, gorm.ErrRecordNotFound) {
  222. return nil, xerr.Custom("该用户数据不存在")
  223. }
  224. return nil, xerr.WithStack(err)
  225. }
  226. /*roleIds := systemUser.SystemUserRoleToSlice()*/
  227. roleIds := []int64{1}
  228. // 获取用户角色数据
  229. systemRoles := make([]*model.SystemRole, 0)
  230. if err = s.DB.Where("is_show = ?", operationPb.IsShow_OK).Find(&systemRoles, roleIds).Error; err != nil {
  231. return nil, xerr.WithStack(err)
  232. }
  233. systemAllPermissionsList := &SystemAllPermissionsList{
  234. MenuList: make([]*model.SystemMenuPermissions, 0),
  235. }
  236. for _, role := range systemRoles {
  237. item := s.GetSystemAllPermissionsList(ctx, role.Id)
  238. systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)
  239. }
  240. systemAllPermissionsList.SystemUserMenuPermissionsUnDuplicate()
  241. menuList, err := s.GetMenuList(ctx, systemAllPermissionsList.MenuList)
  242. if err != nil {
  243. return nil, xerr.WithStack(err)
  244. }
  245. return s.SystemPermissionsFormatPb(menuList), nil
  246. }
  247. func (s *StoreEntry) SystemPermissionsFormatPb(menuList []*model.SystemMenu) *operationPb.SystemUserMenuPermissions {
  248. systemUserMenuPermissions := &operationPb.SystemUserMenuPermissions{
  249. Code: http.StatusOK,
  250. Msg: "ok",
  251. Data: &operationPb.SystemUserMenuData{
  252. PastureList: make([]*operationPb.AddPastureRequest, 0),
  253. MenuList: make([]*operationPb.AddMenuRequest, 0),
  254. MobileList: make([]*operationPb.AddMobileRequest, 0),
  255. MenuButtonsPath: make([]*operationPb.MenuButtonsPath, 0),
  256. },
  257. }
  258. /*level := make(map[int32][]*operationPb.AddMenuRequest, 0)
  259. for _, menu := range menuList {
  260. if _, ok := level[menu.Level]; !ok {
  261. level[menu.Level] = make([]*operationPb.AddMenuRequest, 0)
  262. }
  263. level[menu.Level] = append(level[menu.Level], &operationPb.AddMenuRequest{
  264. Id: int32(menu.Id),
  265. Name: menu.Name,
  266. ParentId: int32(menu.ParentId),
  267. MenuType: menu.MenuType,
  268. Title: menu.Title,
  269. Path: menu.Path,
  270. IsShow: menu.IsShow,
  271. Component: menu.Component,
  272. Icon: menu.Icon,
  273. Sort: menu.Sort,
  274. Redirect: menu.Redirect,
  275. CreatedAt: int32(menu.CreatedAt),
  276. CreatedAtFormat: time.Unix(menu.CreatedAt, 0).Format(model.LayoutTime),
  277. Level: menu.Level,
  278. Affix: true,
  279. KeepAlive: true,
  280. Children: make([]*operationPb.AddMenuRequest, 0),
  281. })
  282. if menu.Level == model.Level3 {
  283. systemUserMenuPermissions.Data.MenuButtonsPath = append(systemUserMenuPermissions.Data.MenuButtonsPath, &operationPb.MenuButtonsPath{
  284. Path: menu.Path,
  285. MenuId: int32(menu.Id),
  286. })
  287. }
  288. }
  289. level2IsShow := make([]int32, 0)
  290. for _, leve2Data := range level[model.Level2] {
  291. for _, leve3Data := range level[model.Level3] {
  292. if leve3Data.ParentId == leve2Data.Id && strings.Contains(leve3Data.Path, ":page") {
  293. level2IsShow = append(level2IsShow, leve2Data.Id)
  294. }
  295. if leve3Data.ParentId == leve2Data.Id {
  296. if leve2Data.Children == nil {
  297. leve2Data.Children = make([]*operationPb.AddMenuRequest, 0)
  298. }
  299. leve2Data.Children = append(leve2Data.Children, leve3Data)
  300. }
  301. }
  302. }
  303. sort.SliceStable(level[model.Level2], func(i, j int) bool {
  304. return level[model.Level2][i].Sort > level[model.Level2][j].Sort
  305. })
  306. for _, leve2Data := range level[model.Level2] {
  307. var info bool
  308. for _, v := range level2IsShow {
  309. if v == leve2Data.Id {
  310. info = true
  311. }
  312. }
  313. if !info {
  314. continue
  315. }
  316. for _, leve1Data := range level[model.Level1] {
  317. if leve2Data.ParentId == leve1Data.Id {
  318. if leve1Data.Children == nil {
  319. leve1Data.Children = make([]*operationPb.AddMenuRequest, 0)
  320. }
  321. leve1Data.Children = append(leve1Data.Children, leve2Data)
  322. }
  323. }
  324. }
  325. sort.SliceStable(level[model.Level1], func(i, j int) bool {
  326. return level[model.Level1][i].Sort > level[model.Level1][j].Sort
  327. })
  328. systemUserMenuPermissions.Data.MenuList = level[model.Level1]*/
  329. return systemUserMenuPermissions
  330. }
  331. // SystemRoleCreateOrUpdate 添加角色
  332. func (s *StoreEntry) SystemRoleCreateOrUpdate(ctx context.Context, req *pasturePb.SearchRoleRequest) error {
  333. isShow := pasturePb.IsShow_Ok
  334. if req.Id > 0 {
  335. isShow = req.IsShow
  336. }
  337. if err := s.DB.Model(&model.SystemRole{}).Where(map[string]interface{}{
  338. "id": req.Id,
  339. }).Assign(map[string]interface{}{
  340. "name": req.Name,
  341. "remarks": req.Remarks,
  342. "is_delete": operationPb.IsShow_OK,
  343. "is_show": isShow,
  344. }).FirstOrCreate(&model.SystemRole{}).Error; err != nil {
  345. return xerr.WithStack(err)
  346. }
  347. return nil
  348. }
  349. // DeleteSystemRole 删除系统角色
  350. func (s *StoreEntry) DeleteSystemRole(ctx context.Context, roleId int64) error {
  351. systemRole := &model.SystemRole{
  352. Id: roleId,
  353. }
  354. if err := s.DB.First(systemRole).Error; err != nil {
  355. if errors.Is(err, gorm.ErrRecordNotFound) {
  356. return xerr.Custom("该数据不存在")
  357. }
  358. return xerr.WithStack(err)
  359. }
  360. if err := s.DB.Model(systemRole).Update("is_delete", pasturePb.IsShow_No).Error; err != nil {
  361. return xerr.WithStack(err)
  362. }
  363. return nil
  364. }
  365. // IsShowSystemRole 角色是否启用
  366. func (s *StoreEntry) IsShowSystemRole(ctx context.Context, roleId int64) error {
  367. systemRole := &model.SystemRole{
  368. Id: roleId,
  369. }
  370. if err := s.DB.First(systemRole).Error; err != nil {
  371. if errors.Is(err, gorm.ErrRecordNotFound) {
  372. return xerr.Custom("该角色不存在")
  373. }
  374. return xerr.WithStack(err)
  375. }
  376. isShow := pasturePb.IsShow_No
  377. if systemRole.IsShow == pasturePb.IsShow_No {
  378. isShow = pasturePb.IsShow_Ok
  379. }
  380. if err := s.DB.Model(systemRole).Update("is_show", isShow).Error; err != nil {
  381. return xerr.WithStack(err)
  382. }
  383. return nil
  384. }
  385. // SearchSystemRoleList 查询系统角色
  386. func (s *StoreEntry) SearchSystemRoleList(ctx context.Context, req *pasturePb.SearchRoleRequest) (*pasturePb.SearchRoleResponse, error) {
  387. systemRoleList := make([]*model.SystemRole, 0)
  388. var count int64 = 0
  389. pref := s.DB.Model(new(model.SystemRole)).Where("is_delete = ?", operationPb.IsShow_OK)
  390. if req.Name != "" {
  391. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  392. }
  393. if req.IsShow > 0 {
  394. pref.Where("is_show = ?", req.IsShow)
  395. }
  396. if err := pref.Order("id desc").Count(&count).Limit(int(req.Pagination.PageSize)).
  397. Offset(int(req.Pagination.PageOffset)).Find(&systemRoleList).Error; err != nil {
  398. return nil, xerr.WithStack(err)
  399. }
  400. /*permissionsMenuMap := make(map[int64][]*model.SystemMenu, 0)
  401. for _, role := range systemRoleList {
  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 &pasturePb.SearchRoleResponse{
  413. Code: http.StatusOK,
  414. Message: "ok",
  415. Data: &pasturePb.SearchRoleData{
  416. Page: req.Pagination.Page,
  417. Total: int32(count),
  418. PageSize: req.Pagination.PageSize,
  419. List: model.SystemRoleSlice(systemRoleList).ToPB(nil),
  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)).
  427. Where("is_show = ?", operationPb.IsShow_OK).
  428. Find(&systemRoleList, ids).Error; err != nil {
  429. return nil, xerr.WithStack(err)
  430. }
  431. return systemRoleList, nil
  432. }
  433. // GetRolePermissions 查询系统角色权限
  434. func (s *StoreEntry) GetRolePermissions(ctx context.Context, roleId int64) (*operationPb.RolePermissionsList, error) {
  435. systemRole := &model.SystemRole{
  436. Id: roleId,
  437. }
  438. if err := s.DB.First(systemRole).Error; err != nil {
  439. if errors.Is(err, gorm.ErrRecordNotFound) {
  440. return nil, xerr.Custom("该数据不存在")
  441. }
  442. return nil, xerr.WithStack(err)
  443. }
  444. systemAllPermissionsList := &SystemAllPermissionsList{
  445. MenuList: make([]*model.SystemMenuPermissions, 0),
  446. }
  447. item := s.GetSystemAllPermissionsList(ctx, systemRole.Id)
  448. systemAllPermissionsList.MenuList = append(systemAllPermissionsList.MenuList, item.MenuList...)
  449. return s.AllPermissionsListToRolePermissions(systemAllPermissionsList), nil
  450. }
  451. // CreateOrUpdateSystemMenu 添加或者更新系统菜单权限
  452. func (s *StoreEntry) CreateOrUpdateSystemMenu(ctx context.Context, req *pasturePb.SearchMenuRequest) error {
  453. if req.Id > 0 {
  454. }
  455. if err := s.DB.Model(&model.SystemMenu{}).Where(map[string]interface{}{
  456. "id": req.Id,
  457. }).Assign(map[string]interface{}{
  458. "name": req.Name,
  459. "path": req.Path,
  460. "title": req.Title,
  461. "menu_type": req.MenuType,
  462. "parent_id": req.ParentId,
  463. "frame_src": req.FrameSrc,
  464. "frame_loading": model.BooleanToIsShow(req.FrameLoading),
  465. "keepalive": model.BooleanToIsShow(req.Keepalive),
  466. "hidden_tag": model.BooleanToIsShow(req.HiddenTag),
  467. "show_link": model.BooleanToIsShow(req.ShowLink),
  468. "show_parent": model.BooleanToIsShow(req.ShowParent),
  469. "icon": req.Icon,
  470. "component": req.Component,
  471. "redirect": req.Redirect,
  472. "auths": req.Auths,
  473. "rank": req.Rank,
  474. "extra_icon": req.ExtraIcon,
  475. "enter_transition": req.EnterTransition,
  476. "leave_transition": req.LeaveTransition,
  477. "active_path": req.ActivePath,
  478. "is_delete": pasturePb.IsShow_Ok,
  479. "is_show": pasturePb.IsShow_Ok,
  480. }).FirstOrCreate(&model.SystemMenu{}).Error; err != nil {
  481. return xerr.WithStack(err)
  482. }
  483. return nil
  484. }
  485. // SearchSystemMenuList 菜单列表查询
  486. func (s *StoreEntry) SearchSystemMenuList(ctx context.Context, req *pasturePb.SearchMenuRequest) (*pasturePb.SearchMenuResponse, error) {
  487. systemMenuList := make([]*model.SystemMenu, 0)
  488. var count int64 = 0
  489. if err := s.DB.Model(new(model.SystemMenu)).Where("is_delete = ? ", operationPb.IsShow_OK).Order("parent_id").Count(&count).Find(&systemMenuList).Error; err != nil {
  490. return nil, xerr.WithStack(err)
  491. }
  492. return &pasturePb.SearchMenuResponse{
  493. Code: http.StatusOK,
  494. Message: "ok",
  495. Data: &pasturePb.SearchMenuData{
  496. Page: req.Pagination.Page,
  497. Total: int32(count),
  498. PageSize: req.Pagination.PageSize,
  499. List: model.SystemMenuSlice(systemMenuList).ToPB(),
  500. },
  501. }, nil
  502. }
  503. // DeleteSystemMenu 删除系统菜单
  504. func (s *StoreEntry) DeleteSystemMenu(ctx context.Context, menuId int64) error {
  505. systemMenu := &model.SystemMenu{Id: menuId}
  506. if err := s.DB.First(systemMenu).Error; err != nil {
  507. if errors.Is(err, gorm.ErrRecordNotFound) {
  508. return xerr.Custom("该数据不存在")
  509. }
  510. return xerr.WithStack(err)
  511. }
  512. if err := s.DB.Model(systemMenu).Update("is_delete", pasturePb.IsShow_No).Error; err != nil {
  513. return xerr.WithStack(err)
  514. }
  515. return nil
  516. }
  517. func (s *StoreEntry) SearchSystemDeptList(ctx context.Context, req *pasturePb.SearchDeptRequest) (*pasturePb.SearchDeptResponse, error) {
  518. deptList := make([]*model.SystemDept, 0)
  519. var count int64 = 0
  520. pref := s.DB.Model(new(model.SystemDept)).Where("is_delete = ?", operationPb.IsShow_OK)
  521. if req.Name != "" {
  522. pref.Where("name like ?", fmt.Sprintf("%s%s%s", "%", req.Name, "%"))
  523. }
  524. if req.IsShow > 0 {
  525. pref.Where("is_show = ?", req.IsShow)
  526. }
  527. if err := pref.Order("sort desc").Count(&count).Limit(int(req.Pagination.PageSize)).Offset(int(req.Pagination.PageOffset)).
  528. Find(&deptList).Error; err != nil {
  529. return nil, xerr.WithStack(err)
  530. }
  531. return &pasturePb.SearchDeptResponse{
  532. Code: http.StatusOK,
  533. Message: "ok",
  534. Data: model.SystemDeptSlice(deptList).ToPB(),
  535. }, nil
  536. }
  537. func (s *StoreEntry) SystemDepDelete(ctx context.Context, id int64) error {
  538. dept := &model.SystemDept{Id: id}
  539. if err := s.DB.First(dept).Error; err != nil {
  540. return xerr.WithStack(err)
  541. }
  542. if dept.IsShow == pasturePb.IsShow_No {
  543. return nil
  544. }
  545. if err := s.DB.Model(dept).Update("is_delete", operationPb.IsShow_NO).Error; err != nil {
  546. return xerr.WithStack(err)
  547. }
  548. return nil
  549. }
  550. func (s *StoreEntry) SystemDeptCreateOrUpdate(ctx context.Context, req *pasturePb.SearchDeptRequest) error {
  551. if err := s.DB.Model(&model.SystemDept{}).Where(map[string]interface{}{
  552. "id": req.Id,
  553. }).Assign(map[string]interface{}{
  554. "name": req.Name,
  555. "sort": req.Sort,
  556. "parent_id": req.ParentId,
  557. "remarks": req.Remarks,
  558. "is_delete": operationPb.IsShow_OK,
  559. "is_show": req.IsShow,
  560. "dept_type": pasturePb.Depth_Department,
  561. "leader_id": req.LeaderId,
  562. }).FirstOrCreate(&model.SystemDept{}).Error; err != nil {
  563. return xerr.WithStack(err)
  564. }
  565. return nil
  566. }
  567. // SearchSystemDeptListByIds 根据id查询部门列表
  568. func (s *StoreEntry) SearchSystemDeptListByIds(ctx context.Context, ids []int64) ([]*model.SystemDept, error) {
  569. systemDeptList := make([]*model.SystemDept, 0)
  570. if err := s.DB.Model(new(model.SystemDept)).
  571. Where("is_show = ?", operationPb.IsShow_OK).
  572. Find(&systemDeptList, ids).Error; err != nil {
  573. return nil, xerr.WithStack(err)
  574. }
  575. return systemDeptList, nil
  576. }