system_service.go 20 KB

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