sys_department.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta_gn/eta_api/models"
  5. "eta_gn/eta_api/models/system"
  6. "eta_gn/eta_api/utils"
  7. "fmt"
  8. "strings"
  9. "time"
  10. )
  11. // 系统设置
  12. type SysDepartmentController struct {
  13. BaseAuthController
  14. }
  15. // @Title 新增部门
  16. // @Description 新增部门接口
  17. // @Param request body system.SysDepartmentAddReq true "type json string"
  18. // @Success 200 新增成功
  19. // @router /department/add [post]
  20. func (this *SysDepartmentController) Add() {
  21. br := new(models.BaseResponse).Init()
  22. defer func() {
  23. this.Data["json"] = br
  24. this.ServeJSON()
  25. }()
  26. var req system.SysDepartmentAddReq
  27. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  28. if err != nil {
  29. br.Msg = "参数解析异常!"
  30. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  31. return
  32. }
  33. if req.DepartmentName == "" {
  34. br.Msg = "部门名称不能为空"
  35. return
  36. }
  37. count, err := system.GetSysDepartmentCount(req.DepartmentName)
  38. if err != nil {
  39. br.Msg = "获取数据失败"
  40. br.ErrMsg = "获取数据失败"
  41. return
  42. }
  43. if count > 0 {
  44. br.Msg = "部门名称已存在,请重新输入"
  45. return
  46. }
  47. item := new(system.SysDepartment)
  48. item.DepartmentName = req.DepartmentName
  49. item.CreateTime = time.Now()
  50. departmentId, err := system.AddSysDepartment(item)
  51. if err != nil {
  52. br.Msg = "新增失败"
  53. br.ErrMsg = "新增失败,Err:" + err.Error()
  54. return
  55. }
  56. // 同步部门缓存
  57. if utils.BusinessCode == utils.BusinessCodeRelease {
  58. var syncData system.SyncDepartmentData
  59. syncData.Source = utils.SOURCE_ETA_FLAG
  60. syncData.DepartmentId = int(departmentId)
  61. _ = utils.Rc.LPush(utils.CACHE_SYNC_DEPARTMENT, syncData)
  62. }
  63. br.Ret = 200
  64. br.Success = true
  65. br.Msg = "新增成功"
  66. }
  67. // @Title 修改部门
  68. // @Description 修改部门接口
  69. // @Param request body system.SysDepartmentEditReq true "type json string"
  70. // @Success 200 修改成功
  71. // @router /department/edit [post]
  72. func (this *SysDepartmentController) Edit() {
  73. br := new(models.BaseResponse).Init()
  74. defer func() {
  75. this.Data["json"] = br
  76. this.ServeJSON()
  77. }()
  78. var req system.SysDepartmentEditReq
  79. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  80. if err != nil {
  81. br.Msg = "参数解析异常!"
  82. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  83. return
  84. }
  85. if req.DepartmentId <= 0 {
  86. br.Msg = "参数错误"
  87. br.ErrMsg = "参数错误,DepartmentId 小于等于0 "
  88. return
  89. }
  90. if req.DepartmentName == "" {
  91. br.Msg = "部门名称不能为空"
  92. return
  93. }
  94. item, err := system.GetSysDepartmentByName(req.DepartmentName)
  95. if err != nil && !utils.IsErrNoRow(err) {
  96. br.Msg = "获取数据失败"
  97. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  98. return
  99. }
  100. if item != nil && item.DepartmentId > 0 {
  101. if item.DepartmentId != req.DepartmentId {
  102. br.Msg = "名称已存在,请重新输入"
  103. return
  104. }
  105. }
  106. err = system.ModifySysDepartment(req.DepartmentName, req.DepartmentId)
  107. if err != nil {
  108. br.Msg = "修改失败"
  109. br.ErrMsg = "修改失败,Err:" + err.Error()
  110. return
  111. }
  112. // 同步部门缓存
  113. if utils.BusinessCode == utils.BusinessCodeRelease {
  114. var syncData system.SyncDepartmentData
  115. syncData.Source = utils.SOURCE_ETA_FLAG
  116. syncData.DepartmentId = req.DepartmentId
  117. _ = utils.Rc.LPush(utils.CACHE_SYNC_DEPARTMENT, syncData)
  118. }
  119. br.Ret = 200
  120. br.Success = true
  121. br.Msg = "修改成功"
  122. }
  123. // @Title 删除部门
  124. // @Description 删除部门接口
  125. // @Param request body system.SysDepartmentDeleteReq true "type json string"
  126. // @Success 200 删除成功
  127. // @router /department/delete [post]
  128. func (this *SysDepartmentController) Delete() {
  129. br := new(models.BaseResponse).Init()
  130. defer func() {
  131. this.Data["json"] = br
  132. this.ServeJSON()
  133. }()
  134. var req system.SysDepartmentDeleteReq
  135. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  136. if err != nil {
  137. br.Msg = "参数解析异常!"
  138. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  139. return
  140. }
  141. if req.DepartmentId <= 0 {
  142. br.Msg = "参数错误"
  143. br.ErrMsg = "参数错误,DepartmentId 小于等于0 "
  144. return
  145. }
  146. err = system.DeleteSysDepartment(req.DepartmentId)
  147. if err != nil {
  148. br.Msg = "删除失败"
  149. br.ErrMsg = "删除失败,Err:" + err.Error()
  150. return
  151. }
  152. // 同步部门缓存
  153. if utils.BusinessCode == utils.BusinessCodeRelease {
  154. var syncData system.SyncDepartmentData
  155. syncData.Source = utils.SOURCE_ETA_FLAG
  156. syncData.DepartmentId = req.DepartmentId
  157. _ = utils.Rc.LPush(utils.CACHE_SYNC_DEPARTMENT, syncData)
  158. }
  159. br.Ret = 200
  160. br.Success = true
  161. br.Msg = "删除成功"
  162. }
  163. // @Title 获取部门列表
  164. // @Description 获取部门列表接口
  165. // @Success 200 {object} system.SysDepartmentListResp
  166. // @router /department/list [get]
  167. func (this *SysDepartmentController) ListDepartment() {
  168. br := new(models.BaseResponse).Init()
  169. defer func() {
  170. this.Data["json"] = br
  171. this.ServeJSON()
  172. }()
  173. sysUser := this.SysUser
  174. if sysUser == nil {
  175. br.Msg = "请登录"
  176. br.ErrMsg = "请登录,SysUser Is Empty"
  177. return
  178. }
  179. //list, err := system.GetDepartmentList()
  180. //if err != nil {
  181. // br.Msg = "获取失败"
  182. // br.ErrMsg = "获取失败,Err:" + err.Error()
  183. // return
  184. //}
  185. //lenList := len(list)
  186. //for i := 0; i < lenList; i++ {
  187. // departmentId := list[i].DepartmentId
  188. // groupList, err := system.GetSysGroupByDepartmentId(departmentId)
  189. // if err != nil {
  190. // br.Msg = "获取数据失败"
  191. // br.ErrMsg = "获取数据失败,Err:" + err.Error()
  192. // return
  193. // }
  194. // list[i].Child = groupList
  195. // list[i].IsDepartment = true
  196. // for i2, sysGroup := range groupList {
  197. // teamList, err := system.GetSysTeamByDepartmentId(sysGroup.GroupId)
  198. // if err != nil {
  199. // br.Msg = "获取数据失败"
  200. // br.ErrMsg = "获取数据失败,Err:" + err.Error()
  201. // return
  202. // }
  203. // groupList[i2].Child = teamList
  204. // groupList[i2].IsGroup = true
  205. // }
  206. //}
  207. // 获取部门
  208. departments, e := system.GetDepartmentList()
  209. if e != nil {
  210. br.Msg = "获取失败"
  211. br.ErrMsg = fmt.Sprintf("获取部门失败, %v", e)
  212. return
  213. }
  214. // 获取分组和小组
  215. groups, e := system.GetSysGroupList()
  216. if e != nil {
  217. br.Msg = "获取失败"
  218. br.ErrMsg = fmt.Sprintf("获取分组失败, %v", e)
  219. return
  220. }
  221. departmentGroup := make(map[int][]*system.SysGroupList)
  222. groupTeam := make(map[int][]*system.SysTeamList)
  223. for _, v := range groups {
  224. if departmentGroup[v.DepartmentId] == nil {
  225. departmentGroup[v.DepartmentId] = make([]*system.SysGroupList, 0)
  226. }
  227. if v.ParentId == 0 {
  228. departmentGroup[v.DepartmentId] = append(departmentGroup[v.DepartmentId], v)
  229. }
  230. if v.ParentId > 0 {
  231. if groupTeam[v.ParentId] == nil {
  232. groupTeam[v.ParentId] = make([]*system.SysTeamList, 0)
  233. }
  234. groupTeam[v.ParentId] = append(groupTeam[v.ParentId], &system.SysTeamList{
  235. GroupId: v.GroupId,
  236. ParentId: v.ParentId,
  237. GroupName: v.GroupName,
  238. CreateTime: v.CreateTime,
  239. })
  240. }
  241. }
  242. // 数据重组
  243. for _, groups := range departmentGroup {
  244. for _, g := range groups {
  245. g.IsGroup = true
  246. if groupTeam[g.GroupId] != nil {
  247. g.Child = groupTeam[g.GroupId]
  248. } else {
  249. g.Child = make([]*system.SysTeamList, 0)
  250. }
  251. }
  252. }
  253. for _, v := range departments {
  254. v.IsDepartment = true
  255. if departmentGroup[v.DepartmentId] != nil {
  256. v.Child = departmentGroup[v.DepartmentId]
  257. } else {
  258. v.Child = make([]*system.SysGroupList, 0)
  259. }
  260. }
  261. //lenList := len(list)
  262. //for i := 0; i < lenList; i++ {
  263. // departmentId := list[i].DepartmentId
  264. // groupList, err := system.GetSysGroupByDepartmentId(departmentId)
  265. // if err != nil {
  266. // br.Msg = "获取数据失败"
  267. // br.ErrMsg = "获取数据失败,Err:" + err.Error()
  268. // return
  269. // }
  270. // list[i].Child = groupList
  271. // list[i].IsDepartment = true
  272. // for i2, sysGroup := range groupList {
  273. // teamList, err := system.GetSysTeamByDepartmentId(sysGroup.GroupId)
  274. // if err != nil {
  275. // br.Msg = "获取数据失败"
  276. // br.ErrMsg = "获取数据失败,Err:" + err.Error()
  277. // return
  278. // }
  279. // groupList[i2].Child = teamList
  280. // groupList[i2].IsGroup = true
  281. // }
  282. //}
  283. resp := new(system.SysDepartmentListResp)
  284. resp.List = departments
  285. br.Ret = 200
  286. br.Success = true
  287. br.Msg = "获取成功"
  288. br.Data = resp
  289. }
  290. // DepartmentUserTree
  291. // @Title 获取部门及分组用户树
  292. // @Description 获取部门列表接口
  293. // @Success 200 {object} system.DepartmentUserTree
  294. // @router /department/user_tree [get]
  295. func (this *SysDepartmentController) DepartmentUserTree() {
  296. br := new(models.BaseResponse).Init()
  297. defer func() {
  298. if br.ErrMsg == "" {
  299. br.IsSendEmail = false
  300. }
  301. this.Data["json"] = br
  302. this.ServeJSON()
  303. }()
  304. sysUser := this.SysUser
  305. if sysUser == nil {
  306. br.Msg = "请登录"
  307. br.ErrMsg = "请登录,SysUser Is Empty"
  308. return
  309. }
  310. keywords := this.GetString("Keywords", "")
  311. keywords = strings.TrimSpace(keywords)
  312. // 获取部门/分组/用户
  313. departments, e := system.GetDepartmentList()
  314. if e != nil {
  315. br.Msg = "获取失败"
  316. br.ErrMsg = "获取部门失败,Err:" + e.Error()
  317. return
  318. }
  319. groups, e := system.GetFullGroup()
  320. if e != nil {
  321. br.Msg = "获取失败"
  322. br.ErrMsg = "获取分组失败,Err:" + e.Error()
  323. return
  324. }
  325. cond := ` AND enabled = 1`
  326. pars := make([]interface{}, 0)
  327. if keywords != "" {
  328. kw := fmt.Sprint("%", keywords, "%")
  329. cond += ` AND real_name LIKE ?`
  330. pars = append(pars, kw)
  331. }
  332. admins, e := system.GetSysAdminList(cond, pars, []string{}, "")
  333. if e != nil {
  334. br.Msg = "获取失败"
  335. br.ErrMsg = "获取用户失败,Err:" + e.Error()
  336. return
  337. }
  338. // 用户map
  339. departmentAdmins := make(map[int][]*system.DepartmentUserTree, 0)
  340. groupAdmins := make(map[int][]*system.DepartmentUserTree, 0)
  341. for _, v := range admins {
  342. t := new(system.DepartmentUserTree)
  343. t.NodeId = v.AdminId
  344. t.NodeType = 3
  345. t.NodeName = v.RealName
  346. t.Children = make([]*system.DepartmentUserTree, 0)
  347. if groupAdmins[v.GroupId] == nil {
  348. groupAdmins[v.GroupId] = make([]*system.DepartmentUserTree, 0)
  349. }
  350. groupAdmins[v.GroupId] = append(groupAdmins[v.GroupId], t)
  351. // 直属于部门
  352. if v.GroupId == 0 {
  353. if departmentAdmins[v.DepartmentId] == nil {
  354. departmentAdmins[v.DepartmentId] = make([]*system.DepartmentUserTree, 0)
  355. }
  356. departmentAdmins[v.DepartmentId] = append(departmentAdmins[v.DepartmentId], t)
  357. }
  358. }
  359. // 小组
  360. groupTeams := make(map[int][]*system.DepartmentUserTree, 0)
  361. for _, v := range groups {
  362. if v.ParentId == 0 {
  363. continue
  364. }
  365. // 关键词查询时不显示空组
  366. if keywords != "" && groupAdmins[v.GroupId] == nil {
  367. continue
  368. }
  369. t := new(system.DepartmentUserTree)
  370. t.NodeId = v.GroupId
  371. t.NodeName = v.GroupName
  372. t.NodeType = 2
  373. t.Children = make([]*system.DepartmentUserTree, 0)
  374. t.Children = groupAdmins[v.GroupId]
  375. if groupTeams[v.ParentId] == nil {
  376. groupTeams[v.ParentId] = make([]*system.DepartmentUserTree, 0)
  377. }
  378. groupTeams[v.ParentId] = append(groupTeams[v.ParentId], t)
  379. }
  380. // 大组
  381. departmentGroups := make(map[int][]*system.DepartmentUserTree, 0)
  382. for _, v := range groups {
  383. if v.ParentId > 0 {
  384. continue
  385. }
  386. // 关键词查询时不显示空组
  387. if keywords != "" && groupAdmins[v.GroupId] == nil && groupTeams[v.GroupId] == nil {
  388. continue
  389. }
  390. t := new(system.DepartmentUserTree)
  391. t.NodeId = v.GroupId
  392. t.NodeName = v.GroupName
  393. t.NodeType = 2
  394. t.Children = make([]*system.DepartmentUserTree, 0)
  395. if groupTeams[v.GroupId] != nil {
  396. t.Children = append(t.Children, groupTeams[v.GroupId]...)
  397. }
  398. if groupAdmins[v.GroupId] != nil {
  399. t.Children = append(t.Children, groupAdmins[v.GroupId]...)
  400. }
  401. if departmentGroups[v.DepartmentId] == nil {
  402. departmentGroups[v.DepartmentId] = make([]*system.DepartmentUserTree, 0)
  403. }
  404. departmentGroups[v.DepartmentId] = append(departmentGroups[v.DepartmentId], t)
  405. }
  406. // 部门
  407. list := make([]*system.DepartmentUserTree, 0)
  408. for _, v := range departments {
  409. // 关键词查询时不显示空部门
  410. if keywords != "" && departmentGroups[v.DepartmentId] == nil && departmentAdmins[v.DepartmentId] == nil {
  411. continue
  412. }
  413. t := new(system.DepartmentUserTree)
  414. t.NodeId = v.DepartmentId
  415. t.NodeType = 1
  416. t.NodeName = v.DepartmentName
  417. t.Children = make([]*system.DepartmentUserTree, 0)
  418. if departmentGroups[v.DepartmentId] != nil {
  419. t.Children = append(t.Children, departmentGroups[v.DepartmentId]...)
  420. }
  421. if departmentAdmins[v.DepartmentId] != nil {
  422. t.Children = append(t.Children, departmentAdmins[v.DepartmentId]...)
  423. }
  424. list = append(list, t)
  425. }
  426. br.Ret = 200
  427. br.Success = true
  428. br.Msg = "获取成功"
  429. br.Data = list
  430. }