system_service.go 20 KB

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