system_service.go 20 KB

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