edb_info_share.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990
  1. package data_manage
  2. import (
  3. "encoding/json"
  4. "eta_gn/eta_api/cache"
  5. "eta_gn/eta_api/controllers"
  6. "eta_gn/eta_api/models"
  7. "eta_gn/eta_api/models/data_manage"
  8. "eta_gn/eta_api/models/data_manage/request"
  9. "eta_gn/eta_api/models/data_manage/response"
  10. "eta_gn/eta_api/models/system"
  11. "eta_gn/eta_api/services/data"
  12. "eta_gn/eta_api/services/elastic"
  13. "eta_gn/eta_api/utils"
  14. "fmt"
  15. "github.com/rdlucklib/rdluck_tools/paging"
  16. "strconv"
  17. "strings"
  18. )
  19. // TODO 取消共享的时候,需要清除对应用户的收藏状态
  20. // EdbInfoShareController 数据管理
  21. type EdbInfoShareController struct {
  22. controllers.BaseAuthController
  23. }
  24. // UserList
  25. // @Title 获取指标设置共享的详情
  26. // @Description 获取指标详情接口
  27. // @Param EdbInfoId query int true "指标id"
  28. // @Success 200 {object} response.EdbInfoShareUserResp
  29. // @router /edb_info/share/user_list [get]
  30. func (c *EdbInfoShareController) UserList() {
  31. br := new(models.BaseResponse).Init()
  32. defer func() {
  33. c.Data["json"] = br
  34. c.ServeJSON()
  35. }()
  36. sysUser := c.SysUser
  37. if sysUser == nil {
  38. br.Msg = "请登录"
  39. br.ErrMsg = "请登录,SysUser Is Empty"
  40. br.Ret = 408
  41. return
  42. }
  43. edbInfoId, _ := c.GetInt("EdbInfoId")
  44. if edbInfoId <= 0 {
  45. br.Msg = "参数错误"
  46. br.ErrMsg = "参数错误"
  47. return
  48. }
  49. edbInfo, err := data_manage.GetEdbInfoById(edbInfoId)
  50. if err != nil {
  51. br.Msg = "获取失败"
  52. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  53. return
  54. }
  55. if edbInfo.SysUserId != sysUser.AdminId {
  56. br.Msg = "您没有权限设置共享"
  57. br.ErrMsg = "您没有权限设置共享,EdbInfoId:" + strconv.Itoa(edbInfo.EdbInfoId)
  58. br.IsSendEmail = false
  59. return
  60. }
  61. resp := response.EdbInfoShareUserResp{}
  62. obj := data_manage.EdbInfoShare{}
  63. list, err := obj.GetListByEdbInfoId(edbInfoId)
  64. if err != nil {
  65. br.Msg = `获取失败`
  66. br.ErrMsg = `获取失败:` + err.Error()
  67. return
  68. }
  69. resp.List = list
  70. br.Ret = 200
  71. br.Success = true
  72. br.Msg = "获取成功"
  73. br.Data = resp
  74. }
  75. // Save
  76. // @Title 单个指标设置共享
  77. // @Description 单个指标设置共享
  78. // @Param request body request.SetEdbInfoShareReq true "type json string"
  79. // @Success 200 {object} data_manage.EdbInfo
  80. // @router /edb_info/share/save [post]
  81. func (c *EdbInfoShareController) Save() {
  82. br := new(models.BaseResponse).Init()
  83. defer func() {
  84. c.Data["json"] = br
  85. c.ServeJSON()
  86. }()
  87. sysUser := c.SysUser
  88. if sysUser == nil {
  89. br.Msg = "请登录"
  90. br.ErrMsg = "请登录,SysUser Is Empty"
  91. br.Ret = 408
  92. return
  93. }
  94. var req request.SetEdbInfoShareReq
  95. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  96. if err != nil {
  97. br.Msg = "参数解析异常!"
  98. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  99. return
  100. }
  101. if req.EdbInfoId <= 0 {
  102. br.Msg = `请选择指标`
  103. br.IsSendEmail = false
  104. return
  105. }
  106. edbItem, err := data_manage.GetEdbInfoById(req.EdbInfoId)
  107. if err != nil {
  108. br.Msg = "查找指标失败"
  109. br.ErrMsg = "查找指标失败,Err:" + err.Error()
  110. return
  111. }
  112. if edbItem.EdbInfoId <= 0 {
  113. br.Msg = "指标不存在"
  114. br.ErrMsg = "指标不存在,ChartInfoId:" + strconv.Itoa(req.EdbInfoId)
  115. br.IsSendEmail = false
  116. return
  117. }
  118. if edbItem.SysUserId != c.SysUser.AdminId {
  119. br.Msg = "您没有权限设置共享"
  120. br.ErrMsg = "您没有权限设置共享,EdbInfoId:" + strconv.Itoa(req.EdbInfoId)
  121. br.IsSendEmail = false
  122. return
  123. }
  124. obj := data_manage.EdbInfoShare{}
  125. // 如果指标未公开,则判断是否被其他用户使用
  126. if edbItem.PublicStatus != utils.DataPublicSuccess {
  127. checkUserIdList := []int{sysUser.AdminId}
  128. if len(req.UserIdList) > 0 {
  129. checkUserIdList = append(checkUserIdList, req.UserIdList...)
  130. }
  131. {
  132. userIdList, err := obj.GetAllUserIdListByEdbInfoIdAndNoShareType(req.EdbInfoId, req.ShareType)
  133. if err != nil {
  134. br.Msg = "设置失败!"
  135. br.ErrMsg = "设置失败,Err:" + err.Error()
  136. return
  137. }
  138. if len(userIdList) > 0 {
  139. checkUserIdList = append(checkUserIdList, userIdList...)
  140. }
  141. }
  142. // 校验是否被其他用户关联
  143. noReferenced, otherUserIdList, err, tips := data.CheckEdbReferencedByOthers(edbItem.EdbInfoType, []int{edbItem.EdbInfoId}, checkUserIdList)
  144. if err != nil {
  145. br.Msg = "保存失败"
  146. if tips != "" {
  147. br.Msg = tips
  148. }
  149. br.ErrMsg = "设置指标共享失败,查找指标的关联用户失败,Err:" + err.Error()
  150. return
  151. }
  152. if !noReferenced {
  153. userNameStr := ``
  154. if len(otherUserIdList) > 0 {
  155. userNameList := make([]string, 0)
  156. userList, err := system.GetAdminListByIdList(otherUserIdList)
  157. if err != nil {
  158. br.Msg = "保存失败"
  159. if tips != "" {
  160. br.Msg = tips
  161. }
  162. br.ErrMsg = "设置指标共享失败,查找用户信息失败,Err:" + err.Error()
  163. return
  164. }
  165. for _, userInfo := range userList {
  166. userNameList = append(userNameList, userInfo.RealName)
  167. }
  168. userNameStr = strings.Join(userNameList, "、")
  169. }
  170. msg := `指标被其他用户使用,无法设置共享`
  171. if userNameStr != "" {
  172. msg = fmt.Sprintf(`指标被用户%s使用,无法设置共享`, userNameStr)
  173. }
  174. if tips != "" {
  175. msg = fmt.Sprintf(`%s,用户为:%s,无法设置共享`, tips, userNameStr)
  176. }
  177. br.Msg = msg
  178. br.ErrMsg = "指标被其他用户共享,无法设置共享,EdbInfoId:" + strconv.Itoa(req.EdbInfoId)
  179. br.IsSendEmail = false
  180. return
  181. }
  182. }
  183. // 修改
  184. err = obj.SaveEdbInfoShare([]int{req.EdbInfoId}, req.UserIdList, req.ShareType)
  185. if err != nil {
  186. br.Msg = `保存失败`
  187. br.ErrMsg = `保存失败:` + err.Error()
  188. return
  189. }
  190. // 更新es
  191. data.AddOrEditEdbInfoToEs(req.EdbInfoId)
  192. // 指标收藏操作处理
  193. go cache.AddDataCollectHandler(req.EdbInfoId, utils.DataCollectTypeEdb)
  194. br.Ret = 200
  195. br.Success = true
  196. br.Msg = "保存成功"
  197. }
  198. // Tree
  199. // @Title 获取共享指标的分类/指标树
  200. // @Description 获取共享指标的分类/指标树
  201. // @Success 200 {object} response.EdbShareListResp
  202. // @router /edb_info/share/tree [get]
  203. func (c *EdbInfoShareController) Tree() {
  204. br := new(models.BaseResponse).Init()
  205. defer func() {
  206. c.Data["json"] = br
  207. c.ServeJSON()
  208. }()
  209. sysUser := c.SysUser
  210. if sysUser == nil {
  211. br.Msg = "请登录"
  212. br.ErrMsg = "请登录,SysUser Is Empty"
  213. br.Ret = 408
  214. return
  215. }
  216. resp := response.EdbShareListResp{}
  217. // 我共享的
  218. {
  219. sendList, err := data.GetAllShareEdbListByFromUserId(sysUser.AdminId)
  220. if err != nil {
  221. br.Msg = "获取失败"
  222. br.ErrMsg = "获取我共享的指标列表信息失败,Err:" + err.Error()
  223. return
  224. }
  225. nodeList, err := data.GetEdbClassifyItemListByShareEdbInfoQueryList(sendList)
  226. if err != nil {
  227. br.Msg = "获取失败"
  228. br.ErrMsg = "获取我共享出去的指标列表分类结构信息失败,Err:" + err.Error()
  229. return
  230. }
  231. resp.Send = nodeList
  232. }
  233. // 我收到的
  234. {
  235. sendList, err := data.GetAllShareEdbListByReceivedUserId(sysUser.AdminId)
  236. if err != nil {
  237. br.Msg = "获取失败"
  238. br.ErrMsg = "获取我共享的指标列表信息失败,Err:" + err.Error()
  239. return
  240. }
  241. nodeList, err := data.GetEdbClassifyItemListByShareEdbInfoQueryList(sendList)
  242. if err != nil {
  243. br.Msg = "获取失败"
  244. br.ErrMsg = "获取我收到共享的指标列表分类结构信息失败,Err:" + err.Error()
  245. return
  246. }
  247. receivedMap := make(map[int]int)
  248. var receivedList data_manage.UserShareEdbClassifyItemList
  249. for _, v := range nodeList {
  250. index, ok := receivedMap[v.SysUserId]
  251. if !ok {
  252. index = len(receivedMap)
  253. item := &data_manage.UserShareEdbClassifyItem{
  254. UserId: v.SysUserId,
  255. UserName: v.SysUserRealName,
  256. EdbInfoId: 0,
  257. ClassifyId: v.SysUserId,
  258. ClassifyName: v.SysUserRealName,
  259. ClassifyType: 0,
  260. ClassifyNameEn: "",
  261. ParentId: 0,
  262. RootId: 0,
  263. Level: 0,
  264. Sort: 0,
  265. UniqueCode: utils.MD5(fmt.Sprint(v.SysUserId, "_", v.SysUserRealName)),
  266. Source: 0,
  267. SourceName: "",
  268. SysUserId: v.SysUserId,
  269. SysUserRealName: v.SysUserRealName,
  270. StartDate: "",
  271. EdbCode: "",
  272. EdbType: 0,
  273. Children: []*data_manage.UserShareEdbClassifyItem{},
  274. Button: data_manage.EdbClassifyItemsButton{},
  275. IsJoinPermission: 0,
  276. HaveOperaAuth: true,
  277. ClassifyIdPath: "",
  278. }
  279. receivedList = append(receivedList, item)
  280. }
  281. receivedList[index].Children = append(receivedList[index].Children, v)
  282. receivedMap[v.SysUserId] = index
  283. }
  284. resp.Received = receivedList
  285. }
  286. br.Ret = 200
  287. br.Success = true
  288. br.Msg = "获取成功"
  289. br.Data = resp
  290. }
  291. // List
  292. // @Title 指标列表接口
  293. // @Description 指标列表接口
  294. // @Param PageSize query int true "每页数据条数"
  295. // @Param CurrentIndex query int true "当前页页码,从1开始"
  296. // @Param ClassifyId query int true "分类id"
  297. // @Param FilterSource query int false "共享指标的列表数据来源,0:我共享的;1:别人共享给我的"
  298. // @Success 200 {object} response.EdbInfoChartListResp
  299. // @router /edb_info/share/list [get]
  300. func (c *EdbInfoShareController) List() {
  301. br := new(models.BaseResponse).Init()
  302. defer func() {
  303. if br.ErrMsg == "" {
  304. br.IsSendEmail = false
  305. }
  306. c.Data["json"] = br
  307. c.ServeJSON()
  308. }()
  309. sysUser := c.SysUser
  310. if sysUser == nil {
  311. br.Msg = "请登录"
  312. br.ErrMsg = "请登录,SysUser Is Empty"
  313. br.Ret = 408
  314. return
  315. }
  316. // 分页
  317. pageSize, _ := c.GetInt("PageSize")
  318. currentIndex, _ := c.GetInt("CurrentIndex")
  319. filterSource, _ := c.GetInt("FilterSource")
  320. var total int
  321. page := paging.GetPaging(currentIndex, pageSize, total)
  322. var startSize int
  323. if pageSize <= 0 {
  324. pageSize = utils.PageSize20
  325. }
  326. if currentIndex <= 0 {
  327. currentIndex = 1
  328. }
  329. startSize = paging.StartIndex(currentIndex, pageSize)
  330. // 基础指标
  331. var condition string
  332. var pars []interface{}
  333. condition += ` AND edb_info_type = ? `
  334. pars = append(pars, 0)
  335. // 指标类型
  336. edbType, _ := c.GetInt("EdbType", 0)
  337. if edbType > 0 {
  338. condition += ` AND edb_type = ? `
  339. pars = append(pars, edbType)
  340. }
  341. // 用户筛选
  342. userId, _ := c.GetInt("UserId")
  343. if userId > 0 {
  344. condition += ` AND b.sys_user_id = ? `
  345. pars = append(pars, userId)
  346. }
  347. // 分类筛选
  348. classifyId, _ := c.GetInt("ClassifyId")
  349. if classifyId > 0 {
  350. childClassify, e, _ := data.GetChildClassifyByClassifyId(classifyId)
  351. if e != nil && !utils.IsErrNoRow(e) {
  352. br.Msg = "获取失败"
  353. br.ErrMsg = "获取分类信息失败, GetEdbClassify,Err:" + e.Error()
  354. return
  355. }
  356. var classifyIds []int
  357. for _, v := range childClassify {
  358. classifyIds = append(classifyIds, v.ClassifyId)
  359. }
  360. condition += fmt.Sprintf(` AND b.classify_id IN (%s) `, utils.GetOrmInReplace(len(classifyIds)))
  361. pars = append(pars, classifyIds)
  362. }
  363. switch filterSource {
  364. case 1: // 别人共享给我的
  365. condition += ` AND a.sys_user_id = ? AND b.sys_user_id != ? `
  366. pars = append(pars, sysUser.AdminId, sysUser.AdminId)
  367. default: // 我共享的
  368. condition += ` AND b.sys_user_id = ? `
  369. pars = append(pars, sysUser.AdminId)
  370. }
  371. obj := data_manage.EdbInfoShare{}
  372. dataCount, respList, err := obj.GetShareEdbInfoListPageList(condition, pars, startSize, pageSize)
  373. if err != nil {
  374. br.Msg = "获取失败"
  375. br.ErrMsg = fmt.Sprintf("获取分享指标列表失败, %v", err)
  376. return
  377. }
  378. editShareEdbInfoIdMap, err := data.GetAllEditSharedEdbInfoIdMapByReceivedUserId(c.SysUser.AdminId)
  379. if err != nil {
  380. br.Msg = "获取失败"
  381. br.ErrMsg = "获取所有有编辑权限的分享指标失败,Err:" + err.Error()
  382. return
  383. }
  384. for _, edbInfo := range respList {
  385. edbInfo.HaveOperaAuth = true
  386. edbInfo.Button = data.GetEdbOpButton(c.SysUser, edbInfo.SysUserId, edbInfo.EdbInfoId, edbInfo.EdbType, edbInfo.EdbInfoType, edbInfo.HaveOperaAuth, editShareEdbInfoIdMap)
  387. }
  388. page = paging.GetPaging(currentIndex, pageSize, int(dataCount))
  389. resp := response.EdbInfoChartListResp{
  390. Paging: page,
  391. List: respList,
  392. }
  393. br.Ret = 200
  394. br.Success = true
  395. br.Msg = "获取成功"
  396. br.Data = resp
  397. }
  398. // ListByEs
  399. // @Title 指标筛选接口
  400. // @Description 指标筛选接口
  401. // @Success 200 {object} data_manage.EdbInfoList
  402. // @Param request body request.SearchEdbInfoShareReq true "type json string"
  403. // @Success 200 {object} data_manage.EdbInfoFilterDataResp
  404. // @router /edb_info/share/list/es [post]
  405. func (c *EdbInfoShareController) ListByEs() {
  406. br := new(models.BaseResponse).Init()
  407. defer func() {
  408. c.Data["json"] = br
  409. c.ServeJSON()
  410. }()
  411. var req request.SearchEdbInfoShareReq
  412. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  413. if err != nil {
  414. br.Msg = "参数解析异常!"
  415. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  416. return
  417. }
  418. pageSize := req.PageSize
  419. currentIndex := req.CurrentIndex
  420. var total int64
  421. var startSize int
  422. if pageSize <= 0 {
  423. pageSize = utils.PageSize20
  424. }
  425. if currentIndex <= 0 {
  426. currentIndex = 1
  427. }
  428. startSize = paging.StartIndex(currentIndex, pageSize)
  429. keyword := req.Keyword
  430. keyword = strings.TrimSpace(keyword) //移除字符串首尾空格
  431. //指标来源
  432. sourceList := req.SourceList
  433. edbInfoType := 0 // 普通指标
  434. edbTypeList := []int{utils.EdbTypeCalculate} // 普通指标中的计算指标
  435. // 指标分享状态:1-未共享;2-已共享。可多选,多选用英文,隔开;默认是未全部
  436. edbShareList := req.EdbShareList
  437. if len(edbShareList) <= 0 {
  438. edbShareList = []int{}
  439. }
  440. edbShare := 0 // 0:全部,1:未共享,2:已共享
  441. lenEdbShareList := len(edbShareList)
  442. if len(edbShareList) > 0 {
  443. if lenEdbShareList > 1 {
  444. edbShare = 0
  445. } else {
  446. edbShare = edbShareList[0]
  447. }
  448. }
  449. // 直指标所属分类id
  450. edbClassifyIdList := req.ClassifyIdList
  451. edbAuth := 1 // 选择范围是:只有我的指标
  452. var edbInfoList []*data_manage.EdbInfoList
  453. sortMap := make(map[string]string)
  454. // 如果没有搜索关键词,则默认根据指标编码倒序排序
  455. if keyword == `` {
  456. sortMap["EdbInfoId"] = `desc`
  457. }
  458. total, edbInfoList, err = elastic.SearchEdbInfoDataByShared(keyword, startSize, pageSize, edbShare, sourceList, edbClassifyIdList, edbTypeList, edbInfoType, edbAuth, c.SysUser.AdminId, sortMap)
  459. if err != nil {
  460. edbInfoList = make([]*data_manage.EdbInfoList, 0)
  461. }
  462. page := paging.GetPaging(currentIndex, pageSize, int(total))
  463. edbInfoListLen := len(edbInfoList)
  464. // 因为是ES查找的,所以需要重新查一下指标的信息,主要是为了把是否授权字段找出来
  465. if len(edbInfoList) > 0 {
  466. sysUserIdList := make([]int, 0)
  467. edbInfoIdList := make([]int, 0)
  468. for _, v := range edbInfoList {
  469. v.ConvertToResp()
  470. v.EdbNameAlias = v.EdbName
  471. v.HaveOperaAuth = true
  472. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  473. if v.SharedUserIdList != nil && len(v.SharedUserIdList) > 0 {
  474. sysUserIdList = append(sysUserIdList, v.SharedUserIdList...)
  475. }
  476. }
  477. // 获取指标数据
  478. tmpEdbList, err := data_manage.GetEdbInfoByIdList(edbInfoIdList)
  479. if err != nil {
  480. br.Msg = "获取失败"
  481. br.ErrMsg = "获取所有有权限的指标失败,Err:" + err.Error()
  482. return
  483. }
  484. edbInfoMap := make(map[int]*data_manage.EdbInfo)
  485. for _, v := range tmpEdbList {
  486. edbInfoMap[v.EdbInfoId] = v
  487. }
  488. // 获取用户数据
  489. sysUserMap := make(map[int]string)
  490. if len(sysUserIdList) > 0 {
  491. sysUserList, err := system.GetAdminListByIdList(sysUserIdList)
  492. if err != nil {
  493. br.Msg = "获取失败"
  494. br.ErrMsg = "获取分享用户列表失败,Err:" + err.Error()
  495. return
  496. }
  497. for _, v := range sysUserList {
  498. sysUserMap[v.AdminId] = v.RealName
  499. }
  500. }
  501. for _, v := range edbInfoList {
  502. tmpEdbInfo, ok := edbInfoMap[v.EdbInfoId]
  503. if !ok {
  504. continue
  505. }
  506. v.IsJoinPermission = tmpEdbInfo.IsJoinPermission
  507. // 分享人
  508. shareUserName := ``
  509. shareUserNameList := make([]string, 0)
  510. shareUserNameMap := make(map[int]bool)
  511. for _, userId := range v.SharedUserIdList {
  512. userName, ok := sysUserMap[userId]
  513. if !ok {
  514. continue
  515. }
  516. if _, ok = shareUserNameMap[userId]; ok {
  517. continue
  518. }
  519. shareUserNameMap[userId] = true
  520. shareUserNameList = append(shareUserNameList, userName)
  521. }
  522. if len(shareUserNameList) > 0 {
  523. shareUserName = strings.Join(shareUserNameList, `,`)
  524. }
  525. v.SharedUserName = shareUserName
  526. }
  527. }
  528. for i := 0; i < edbInfoListLen; i++ {
  529. for j := 0; j < edbInfoListLen; j++ {
  530. if (edbInfoList[i].EdbNameAlias == edbInfoList[j].EdbNameAlias) &&
  531. (edbInfoList[i].EdbInfoId != edbInfoList[j].EdbInfoId) &&
  532. !(strings.Contains(edbInfoList[i].EdbName, edbInfoList[i].SourceName)) {
  533. edbInfoList[i].EdbName = edbInfoList[i].EdbName + "(" + edbInfoList[i].SourceName + ")"
  534. }
  535. }
  536. }
  537. resp := data_manage.EdbInfoFilterDataResp{
  538. Paging: page,
  539. List: edbInfoList,
  540. }
  541. br.Ret = 200
  542. br.Success = true
  543. br.Msg = "获取成功"
  544. br.Data = resp
  545. }
  546. // BatchSave
  547. // @Title 批量指标设置共享
  548. // @Description 批量指标设置共享
  549. // @Param request body request.SetEdbSharePermissionReq true "type json string"
  550. // @Success 200 {object} data_manage.EdbInfo
  551. // @router /edb_info/share/batch_save [post]
  552. func (c *EdbInfoShareController) BatchSave() {
  553. br := new(models.BaseResponse).Init()
  554. defer func() {
  555. c.Data["json"] = br
  556. c.ServeJSON()
  557. }()
  558. sysUser := c.SysUser
  559. if sysUser == nil {
  560. br.Msg = "请登录"
  561. br.ErrMsg = "请登录,SysUser Is Empty"
  562. br.Ret = 408
  563. return
  564. }
  565. var req request.SetEdbSharePermissionReq
  566. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  567. if err != nil {
  568. br.Msg = "参数解析异常!"
  569. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  570. return
  571. }
  572. // 选中的指标id列表
  573. selectEdbInfoIdList := make([]int, 0)
  574. // 选择所有指标,所以需要通过es获取数据
  575. if req.IsSelectAll {
  576. allEsEdbInfoIdList, err := getAllEdbInfoIdListByShared(req, c.SysUser.AdminId)
  577. if err != nil {
  578. br.Msg = "设置失败!"
  579. br.ErrMsg = "设置失败,Err:" + err.Error()
  580. return
  581. }
  582. if len(req.NoEdbIdList) <= 0 {
  583. selectEdbInfoIdList = allEsEdbInfoIdList
  584. } else {
  585. noEdbIdMap := make(map[int]bool)
  586. for _, v := range req.NoEdbIdList {
  587. noEdbIdMap[v] = true
  588. }
  589. for _, v := range allEsEdbInfoIdList {
  590. if _, ok := noEdbIdMap[v]; !ok {
  591. // 如果不在未选中的指标id列表中,那么就加入到选中的指标id列表
  592. selectEdbInfoIdList = append(selectEdbInfoIdList, v)
  593. }
  594. }
  595. }
  596. } else {
  597. selectEdbInfoIdList = req.EdbIdList
  598. }
  599. num := len(selectEdbInfoIdList)
  600. if num <= 0 {
  601. br.Msg = `请选择指标`
  602. br.IsSendEmail = false
  603. return
  604. }
  605. if num > 30 {
  606. br.Msg = `指标数量不能超过30个`
  607. br.IsSendEmail = false
  608. return
  609. }
  610. edbInfoList, err := data_manage.GetEdbInfoListByEdbInfoId(selectEdbInfoIdList)
  611. if err != nil {
  612. br.Msg = "设置失败!"
  613. br.ErrMsg = "设置失败,Err:" + err.Error()
  614. return
  615. }
  616. obj := data_manage.EdbInfoShare{}
  617. {
  618. //checkBaseEdbInfoIdList := make([]int, 0)
  619. //checkCalculateEdbInfoIdList := make([]int, 0)
  620. //
  621. //for _, edbItem := range edbInfoList {
  622. // // 设置公开的可以忽略
  623. // if edbItem.PublicStatus == utils.DataPublicSuccess {
  624. // continue
  625. // }
  626. //
  627. // if edbItem.EdbInfoType == utils.EDB_INFO_TYPE {
  628. // checkBaseEdbInfoIdList = append(checkBaseEdbInfoIdList, edbItem.EdbInfoId)
  629. // } else if edbItem.EdbInfoType == utils.PREDICT_EDB_INFO_TYPE {
  630. // checkCalculateEdbInfoIdList = append(checkCalculateEdbInfoIdList, edbItem.EdbInfoId)
  631. // }
  632. //}
  633. //
  634. //checkUserIdList := []int{sysUser.AdminId}
  635. //if len(req.UserIdList) > 0 {
  636. // checkUserIdList = append(checkUserIdList, req.UserIdList...)
  637. //}
  638. //
  639. //// 普通指标
  640. //{
  641. // noReferenced, otherUserIdList, err, tips := data.CheckEdbReferencedByOthers(utils.EDB_INFO_TYPE, checkBaseEdbInfoIdList, checkUserIdList)
  642. // if err != nil {
  643. // br.Msg = "保存失败"
  644. // if tips != "" {
  645. // br.Msg = tips
  646. // }
  647. // br.ErrMsg = "设置指标共享失败,查找指标的关联用户失败,Err:" + err.Error()
  648. // return
  649. // }
  650. //
  651. // if !noReferenced {
  652. // userNameStr := ``
  653. // if len(otherUserIdList) > 0 {
  654. // userNameList := make([]string, 0)
  655. // userList, err := system.GetAdminListByIdList(otherUserIdList)
  656. // if err != nil {
  657. // br.Msg = "保存失败"
  658. // if tips != "" {
  659. // br.Msg = tips
  660. // }
  661. // br.ErrMsg = "设置指标共享失败,查找用户信息失败,Err:" + err.Error()
  662. // return
  663. // }
  664. // for _, userInfo := range userList {
  665. // userNameList = append(userNameList, userInfo.RealName)
  666. // }
  667. // userNameStr = strings.Join(userNameList, "、")
  668. // }
  669. // msg := `指标被其他用户使用,无法设置共享`
  670. // if userNameStr != "" {
  671. // msg = fmt.Sprintf(`指标被用户%s使用,无法设置共享`, userNameStr)
  672. // }
  673. // if tips != "" {
  674. // msg = fmt.Sprintf(`%s,用户为:%s,无法设置共享`, tips, userNameStr)
  675. // }
  676. // br.Msg = msg
  677. // br.ErrMsg = "指标被其他用户共享,无法设置共享"
  678. // br.IsSendEmail = false
  679. // return
  680. // }
  681. //}
  682. //
  683. //// 预测指标
  684. //{
  685. // noReferenced, otherUserIdList, err, tips := data.CheckEdbReferencedByOthers(utils.PREDICT_EDB_INFO_TYPE, checkCalculateEdbInfoIdList, checkUserIdList)
  686. // if err != nil {
  687. // br.Msg = "保存失败"
  688. // if tips != "" {
  689. // br.Msg = tips
  690. // }
  691. // br.ErrMsg = "设置指标共享失败,查找指标的关联用户失败,Err:" + err.Error()
  692. // return
  693. // }
  694. //
  695. // if !noReferenced {
  696. // userNameStr := ``
  697. // if len(otherUserIdList) > 0 {
  698. // userNameList := make([]string, 0)
  699. // userList, err := system.GetAdminListByIdList(otherUserIdList)
  700. // if err != nil {
  701. // br.Msg = "保存失败"
  702. // if tips != "" {
  703. // br.Msg = tips
  704. // }
  705. // br.ErrMsg = "设置指标共享失败,查找用户信息失败,Err:" + err.Error()
  706. // return
  707. // }
  708. // for _, userInfo := range userList {
  709. // userNameList = append(userNameList, userInfo.RealName)
  710. // }
  711. // userNameStr = strings.Join(userNameList, "、")
  712. // }
  713. // msg := `指标被其他用户使用,无法设置共享`
  714. // if userNameStr != "" {
  715. // msg = fmt.Sprintf(`指标被用户%s使用,无法设置共享`, userNameStr)
  716. // }
  717. // if tips != "" {
  718. // msg = fmt.Sprintf(`%s,用户为:%s,无法设置共享`, tips, userNameStr)
  719. // }
  720. // br.Msg = msg
  721. // br.ErrMsg = "指标被其他用户共享,无法设置共享"
  722. // br.IsSendEmail = false
  723. // return
  724. // }
  725. //}
  726. for _, edbItem := range edbInfoList {
  727. // 如果指标已公开,则不判断是否被其他用户使用
  728. if edbItem.PublicStatus == utils.DataPublicSuccess {
  729. continue
  730. }
  731. if edbItem.PublicStatus != utils.DataPublicSuccess {
  732. checkUserIdList := []int{sysUser.AdminId}
  733. if len(req.UserIdList) > 0 {
  734. checkUserIdList = append(checkUserIdList, req.UserIdList...)
  735. }
  736. {
  737. userIdList, err := obj.GetAllUserIdListByEdbInfoIdAndNoShareType(edbItem.EdbInfoId, req.ShareType)
  738. if err != nil {
  739. br.Msg = "设置失败!"
  740. br.ErrMsg = "设置失败,Err:" + err.Error()
  741. return
  742. }
  743. if len(userIdList) > 0 {
  744. checkUserIdList = append(checkUserIdList, userIdList...)
  745. }
  746. }
  747. // 校验是否被其他用户关联
  748. noReferenced, otherUserIdList, err, tips := data.CheckEdbReferencedByOthers(edbItem.EdbInfoType, []int{edbItem.EdbInfoId}, checkUserIdList)
  749. if err != nil {
  750. br.Msg = "保存失败"
  751. if tips != "" {
  752. br.Msg = tips
  753. }
  754. br.ErrMsg = "设置指标共享失败,查找指标的关联用户失败,Err:" + err.Error()
  755. return
  756. }
  757. if !noReferenced {
  758. userNameStr := ``
  759. if len(otherUserIdList) > 0 {
  760. userNameList := make([]string, 0)
  761. userList, err := system.GetAdminListByIdList(otherUserIdList)
  762. if err != nil {
  763. br.Msg = "保存失败"
  764. if tips != "" {
  765. br.Msg = tips
  766. }
  767. br.ErrMsg = "设置指标共享失败,查找用户信息失败,Err:" + err.Error()
  768. return
  769. }
  770. for _, userInfo := range userList {
  771. userNameList = append(userNameList, userInfo.RealName)
  772. }
  773. userNameStr = strings.Join(userNameList, "、")
  774. }
  775. msg := fmt.Sprintf(`指标《%s》被其他用户使用,无法设置共享`, edbItem.EdbName)
  776. if userNameStr != "" {
  777. msg = fmt.Sprintf(`指标《%s》被用户%s使用,无法设置共享`, edbItem.EdbName, userNameStr)
  778. }
  779. if tips != "" {
  780. msg = fmt.Sprintf(`%s,用户为:%s,无法设置共享`, tips, userNameStr)
  781. }
  782. br.Msg = msg
  783. br.ErrMsg = "指标被其他用户共享,无法设置共享,EdbInfoId:" + strconv.Itoa(edbItem.EdbInfoId)
  784. br.IsSendEmail = false
  785. return
  786. }
  787. }
  788. }
  789. }
  790. // 设置
  791. err = obj.SaveEdbInfoShare(selectEdbInfoIdList, req.UserIdList, req.ShareType)
  792. if err != nil {
  793. br.Msg = `保存失败`
  794. br.ErrMsg = `保存失败:` + err.Error()
  795. return
  796. }
  797. // 更新es
  798. for _, edbInfoId := range selectEdbInfoIdList {
  799. data.AddOrEditEdbInfoToEs(edbInfoId)
  800. // 指标收藏操作处理
  801. go cache.AddDataCollectHandler(edbInfoId, utils.DataCollectTypeEdb)
  802. }
  803. br.Ret = 200
  804. br.Success = true
  805. br.Msg = "保存成功"
  806. }
  807. // getAllEdbInfoIdListByShared
  808. // @Description: 获取所有的指标id列表
  809. // @author: Roc
  810. // @datetime 2024-12-04 15:43:14
  811. // @param req request.SetEdbChartPermissionReq
  812. // @param userId int
  813. // @return edbInfoIdList []int
  814. // @return err error
  815. func getAllEdbInfoIdListByShared(req request.SetEdbSharePermissionReq, userId int) (edbInfoIdList []int, err error) {
  816. keyword := req.Keyword
  817. keyword = strings.TrimSpace(keyword) //移除字符串首尾空格
  818. //指标来源
  819. sourceList := req.SourceList
  820. edbInfoType := 0 // 普通指标
  821. edbTypeList := []int{utils.EdbTypeCalculate} // 普通指标中的计算指标
  822. // 指标分享状态:1-未共享;2-已共享。可多选,多选用英文,隔开;默认是未共享
  823. edbShareList := req.EdbShareList
  824. if len(edbShareList) <= 0 {
  825. edbShareList = []int{1}
  826. }
  827. edbShare := 1 // 0:全部,1:未共享,2:已共享
  828. lenEdbShareList := len(edbShareList)
  829. if len(edbShareList) > 0 {
  830. if lenEdbShareList > 1 {
  831. edbShare = 0
  832. } else {
  833. edbShare = edbShareList[0]
  834. }
  835. }
  836. // 指标分享状态:1-未共享;2-已共享。可多选,多选用英文,隔开;默认是未共享
  837. edbClassifyIdList := req.ClassifyIdList
  838. edbAuth := 1 // 选择范围是:只有我的指标
  839. sortMap := make(map[string]string)
  840. // 如果没有搜索关键词,则默认根据指标编码倒序排序
  841. if keyword == `` {
  842. sortMap["EdbInfoId"] = `desc`
  843. }
  844. _, edbInfoList, err := getAllEdbInfoDataByShared(keyword, 1, edbShare, sourceList, edbClassifyIdList, edbTypeList, edbInfoType, edbAuth, userId, sortMap)
  845. if err != nil {
  846. return
  847. }
  848. // 返回指标id列表
  849. for _, v := range edbInfoList {
  850. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  851. }
  852. return
  853. }
  854. // getAllEdbInfoDataByShared
  855. // @Description: 获取所有的指标列表(设置共享的时候)
  856. // @author: Roc
  857. // @datetime 2024-12-04 15:27:53
  858. // @param keyword string
  859. // @param currPage int
  860. // @param edbShare int
  861. // @param sourceList []int
  862. // @param classifyIdList []int
  863. // @param edbTypeList []int
  864. // @param edbInfoType int
  865. // @param edbAuth int
  866. // @param sysUserId int
  867. // @param sortMap map[string]string
  868. // @return total int64
  869. // @return list []*data_manage.EdbInfoList
  870. // @return err error
  871. func getAllEdbInfoDataByShared(keyword string, currPage, edbShare int, sourceList, classifyIdList, edbTypeList []int, edbInfoType, edbAuth, sysUserId int, sortMap map[string]string) (total int64, list []*data_manage.EdbInfoList, err error) {
  872. // 每页获取数据的数量
  873. pageSize := 5000
  874. var startSize int
  875. if currPage <= 0 {
  876. currPage = 1
  877. }
  878. startSize = paging.StartIndex(currPage, pageSize)
  879. total, list, err = elastic.SearchEdbInfoDataByShared(keyword, startSize, pageSize, edbShare, sourceList, classifyIdList, edbTypeList, edbInfoType, edbAuth, sysUserId, sortMap)
  880. if err != nil {
  881. return
  882. }
  883. page := paging.GetPaging(currPage, pageSize, int(total))
  884. if !page.IsEnd {
  885. _, nextList, tmpErr := getAllEdbInfoDataByShared(keyword, page.NextIndex, edbShare, sourceList, classifyIdList, edbTypeList, edbInfoType, edbAuth, sysUserId, sortMap)
  886. if tmpErr != nil {
  887. err = tmpErr
  888. return
  889. }
  890. list = append(list, nextList...)
  891. }
  892. return
  893. }