system_service.go 20 KB

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