base_from_sci_hq.go 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579
  1. package data
  2. import (
  3. "errors"
  4. "eta/eta_api/models/data_manage"
  5. "eta/eta_api/utils"
  6. "fmt"
  7. "time"
  8. )
  9. func AddSciHqClassify(classifyName, lang string, parentId int) (ok bool, err error) {
  10. var count int
  11. switch lang {
  12. case utils.EnLangVersion:
  13. count, err = data_manage.GetBaseFromSciHqClassifyEnCount(classifyName, parentId)
  14. default:
  15. count, err = data_manage.GetBaseFromSciHqClassifyCount(classifyName, parentId)
  16. }
  17. if err != nil {
  18. return
  19. }
  20. if count > 0 {
  21. return false, errors.New("分类已存在")
  22. }
  23. sort, err := data_manage.GetChildSciHqClassifyMaxSortById(parentId)
  24. if err != nil {
  25. return false, err
  26. }
  27. if parentId > 0 {
  28. count, err = data_manage.GetSciHqClassifyCountById(parentId)
  29. if err != nil {
  30. return false, err
  31. }
  32. if count == 0 {
  33. return false, errors.New("父分类不存在")
  34. }
  35. sciHqClassify := &data_manage.BaseFromSciHqClassify{
  36. ClassifyName: classifyName,
  37. ClassifyNameEn: classifyName,
  38. ParentId: parentId,
  39. Level: 2,
  40. Sort: sort + 1,
  41. ModifyTime: time.Now(),
  42. CreateTime: time.Now(),
  43. }
  44. _, err = sciHqClassify.Add()
  45. if err != nil {
  46. return
  47. }
  48. } else {
  49. sciHqClassify := &data_manage.BaseFromSciHqClassify{
  50. ClassifyName: classifyName,
  51. ClassifyNameEn: classifyName,
  52. ParentId: 0,
  53. Level: 1,
  54. Sort: sort + 1,
  55. ModifyTime: time.Now(),
  56. CreateTime: time.Now(),
  57. }
  58. insertId, er := sciHqClassify.Add()
  59. if er != nil {
  60. return false, er
  61. }
  62. sciHqClassifyList := []*data_manage.BaseFromSciHqClassify{
  63. {
  64. ClassifyName: "价格",
  65. ClassifyNameEn: "价格",
  66. ParentId: int(insertId),
  67. Level: 2,
  68. Sort: 1,
  69. ModifyTime: time.Now(),
  70. CreateTime: time.Now(),
  71. },
  72. {
  73. ClassifyName: "供应",
  74. ClassifyNameEn: "供应",
  75. ParentId: int(insertId),
  76. Level: 2,
  77. Sort: 2,
  78. ModifyTime: time.Now(),
  79. CreateTime: time.Now(),
  80. },
  81. {
  82. ClassifyName: "需求",
  83. ClassifyNameEn: "需求",
  84. ParentId: int(insertId),
  85. Level: 2,
  86. Sort: 3,
  87. ModifyTime: time.Now(),
  88. CreateTime: time.Now(),
  89. },
  90. {
  91. ClassifyName: "库存",
  92. ClassifyNameEn: "库存",
  93. ParentId: int(insertId),
  94. Level: 2,
  95. Sort: 4,
  96. ModifyTime: time.Now(),
  97. CreateTime: time.Now(),
  98. },
  99. }
  100. err = data_manage.BatchAddSciHqClassify(sciHqClassifyList)
  101. if err != nil {
  102. return
  103. }
  104. }
  105. return true, nil
  106. }
  107. func DelSciHqClassify(classifyId int) (err error) {
  108. classify, err := data_manage.GetSciHqClassifyById(classifyId)
  109. if err != nil {
  110. return
  111. }
  112. var classifyIds []int
  113. classifyIds = append(classifyIds, classify.ClassifyId)
  114. if classify.ParentId == 0 {
  115. tmpClassifyIds, er := data_manage.GetChildSciHqClassifyIdsById(classify.ClassifyId)
  116. if er != nil {
  117. err = er
  118. return
  119. }
  120. classifyIds = append(classifyIds, tmpClassifyIds...)
  121. }
  122. // 获取分类下的所有指标
  123. count, err := data_manage.GetSciHqIndexCountByClassifyIds(classifyIds)
  124. if err != nil {
  125. err = errors.New("获取分类下的指标信息失败,Err:" + err.Error())
  126. return
  127. }
  128. if count > 0 {
  129. err = errors.New("该分类下有指标,不可删除")
  130. return
  131. }
  132. // 删除对应的分类
  133. err = data_manage.DeleteSciHqClassifyByClassifyId(classifyIds)
  134. if err != nil {
  135. return
  136. }
  137. return
  138. }
  139. // ResetSciHqIndex 指标数据清除分类接口
  140. func ResetSciHqIndex(indexId int) (err error) {
  141. sciHqIndex, err := data_manage.GetSciHqIndexByIndexId(indexId)
  142. if err != nil {
  143. return
  144. }
  145. // 更新指标在未分类下的排序,永远排在未分类的最后一个
  146. //移动排序
  147. classifyId := 0
  148. updateCol := make([]string, 0)
  149. var currentSort, prevSort int
  150. currentSort = sciHqIndex.Sort
  151. //未分类的最大的sort值
  152. prevSort, err = data_manage.GetSciHqIndexMaxSortByClassifyId(classifyId)
  153. if err != nil {
  154. err = errors.New("获取上一个兄弟节点指标信息失败,Err:" + err.Error())
  155. return
  156. }
  157. //如果前面一个节点的sort值比当前大,则当前节点往下移动
  158. if prevSort >= currentSort {
  159. //往下移动
  160. err = data_manage.MoveDownSciIndexBySort(classifyId, prevSort, currentSort)
  161. if err != nil {
  162. err = errors.New("向下移动出错:" + err.Error())
  163. return
  164. }
  165. sciHqIndex.Sort = prevSort
  166. }
  167. //更新
  168. sciHqIndex.ModifyTime = time.Now()
  169. sciHqIndex.ClassifyId = classifyId
  170. updateCol = append(updateCol, "Sort", "ModifyTime", "ClassifyId")
  171. err = sciHqIndex.Update(updateCol)
  172. if err != nil {
  173. err = errors.New("移动失败,Err:" + err.Error())
  174. return
  175. }
  176. return
  177. }
  178. // 批量删除卓创红期指标
  179. func BatchDelSciHqData(indexIds []int) (existIndex []*data_manage.BaseFromSciHqIndex, err error) {
  180. sciHqIndexList, err := data_manage.GetSciHqIndexListByIndexIds(indexIds)
  181. if err != nil {
  182. return
  183. }
  184. indexCodes := make([]string, 0)
  185. codeToIndex := make(map[string]*data_manage.BaseFromSciHqIndex)
  186. for _, v := range sciHqIndexList {
  187. indexCodes = append(indexCodes, v.IndexCode)
  188. codeToIndex[v.IndexCode] = v
  189. }
  190. edbInfoList, err := data_manage.GetEdbInfoListByEdbCodes(utils.DATA_SOURCE_SCI_HQ, indexCodes)
  191. if err != nil {
  192. return
  193. }
  194. for _, v := range edbInfoList {
  195. if index, ok := codeToIndex[v.EdbCode]; ok {
  196. existIndex = append(existIndex, index)
  197. delete(codeToIndex, v.EdbCode)
  198. }
  199. }
  200. // 删除对应的指标
  201. err = data_manage.DeleteSciHqIndexByIds(indexIds)
  202. if err != nil {
  203. return
  204. }
  205. return
  206. }
  207. // DelSciHqData 删除卓创红期指标
  208. func DelSciHqData(indexId int) (err error, errMsg string) {
  209. baseFromSciHqIndex, err := data_manage.GetSciHqIndexByIndexId(indexId)
  210. if err != nil {
  211. errMsg = `获取数据失败`
  212. if err.Error() == utils.ErrNoRow() {
  213. errMsg = `该指标未入库`
  214. err = nil
  215. }
  216. return
  217. }
  218. // 获取已经加入到EDB指标库的Sci指标
  219. edbInfo, err := data_manage.GetEdbInfoByEdbCode(utils.DATA_SOURCE_SCI_HQ, baseFromSciHqIndex.IndexCode)
  220. if err != nil && err.Error() != utils.ErrNoRow() {
  221. errMsg = "删除失败"
  222. err = errors.New("获取分类下的指标信息失败,Err:" + err.Error())
  223. return
  224. }
  225. if edbInfo != nil {
  226. errMsg = "当前指标已被引用,不可删除"
  227. err = nil
  228. return
  229. }
  230. // 删除对应的指标
  231. err = data_manage.DeleteSciHqIndexById(indexId)
  232. if err != nil {
  233. return
  234. }
  235. return
  236. }
  237. // EditSciHqIndex 编辑卓创红期指标
  238. func EditSciHqIndex(indexId, classifyId int) (baseFromSciHqIndex *data_manage.BaseFromSciHqIndex, errMsg string, err error) {
  239. baseFromSciHqIndex, err = data_manage.GetSciHqIndexByIndexId(indexId)
  240. if err != nil {
  241. if err.Error() == utils.ErrNoRow() {
  242. errMsg = `该指标未入库`
  243. err = nil
  244. }
  245. return
  246. }
  247. classify, err := data_manage.GetSciHqClassifyById(classifyId)
  248. if err != nil {
  249. if err.Error() == utils.ErrNoRow() {
  250. errMsg = `该分类不存在或已删除`
  251. err = nil
  252. }
  253. return
  254. }
  255. if classify.Level == 1 { // 顶级分类
  256. errMsg = `不合法的编辑`
  257. err = nil
  258. return
  259. }
  260. maxSort, err := data_manage.GetChildSciHqClassifyMaxSortById(classifyId)
  261. if err != nil {
  262. return
  263. }
  264. baseFromSciHqIndex.ClassifyId = classifyId
  265. baseFromSciHqIndex.Sort = maxSort + 1
  266. baseFromSciHqIndex.ModifyTime = time.Now()
  267. err = baseFromSciHqIndex.Update([]string{"classify_id", "sort", "modify_time"})
  268. return
  269. }
  270. type SciHqIndexSource2EdbReq struct {
  271. EdbCode string
  272. EdbName string
  273. Frequency string
  274. Unit string
  275. ClassifyId int
  276. AdminId int
  277. AdminRealName string
  278. }
  279. // SciHqIndexSource2Edb 新增卓创红期数据从数据源到指标库
  280. func SciHqIndexSource2Edb(req SciHqIndexSource2EdbReq, lang string) (edb *data_manage.EdbInfo, errMsg string, skip bool, err error) {
  281. if req.EdbCode == "" {
  282. err = fmt.Errorf("指标ID为空")
  283. return
  284. }
  285. defer func() {
  286. if err != nil {
  287. tips := fmt.Sprintf("SciHqIndexSource2Edb新增失败, Err: %s", err.Error())
  288. fmt.Println(tips)
  289. utils.FileLog.Info(tips)
  290. }
  291. }()
  292. source := utils.DATA_SOURCE_SCI_HQ
  293. // 是否新增过指标
  294. exist, e := data_manage.GetEdbInfoByEdbCode(source, req.EdbCode)
  295. if e != nil && e.Error() != utils.ErrNoRow() {
  296. err = fmt.Errorf("获取指标是否存在失败, err: %s", e.Error())
  297. return
  298. }
  299. if exist != nil {
  300. skip = true
  301. return
  302. }
  303. // 开始结束时间
  304. var startDate, endDate string
  305. // 新增指标库
  306. edbInfo, e, msg, _ := EdbInfoAdd(source, utils.DATA_SUB_SOURCE_EDB, req.ClassifyId, req.EdbCode, req.EdbName, req.Frequency, req.Unit, startDate, endDate, req.AdminId, req.AdminRealName, lang)
  307. if e != nil {
  308. errMsg = msg
  309. err = fmt.Errorf("EdbInfo: 新增指标失败, err: %s", e.Error())
  310. return
  311. }
  312. edb = edbInfo
  313. return
  314. }
  315. // MoveSciHqClassify 移动卓创红期分类
  316. func MoveSciHqClassify(classifyId, parentClassifyId, prevClassifyId, nextClassifyId int) (err error, errMsg string) {
  317. //判断分类是否存在
  318. classifyInfo, err := data_manage.GetSciHqClassifyById(classifyId)
  319. if err != nil {
  320. errMsg = "移动失败"
  321. err = errors.New("获取分类信息失败,Err:" + err.Error())
  322. return
  323. }
  324. updateCol := make([]string, 0)
  325. //判断上级id是否一致,如果不一致的话,那么需要移动该分类层级
  326. if classifyInfo.ParentId != parentClassifyId && parentClassifyId != 0 {
  327. parentClassifyInfo, tmpErr := data_manage.GetSciHqClassifyById(parentClassifyId)
  328. if tmpErr != nil {
  329. errMsg = "移动失败"
  330. err = errors.New("获取上级分类信息失败,Err:" + tmpErr.Error())
  331. return
  332. }
  333. classifyInfo.ParentId = parentClassifyInfo.ClassifyId
  334. classifyInfo.Level = parentClassifyInfo.Level + 1
  335. updateCol = append(updateCol, "ParentId", "Level")
  336. }
  337. var currentSort, prevSort, nextSort int
  338. currentSort = classifyInfo.Sort
  339. var prevClassify *data_manage.BaseFromSciHqClassify
  340. var nextClassify *data_manage.BaseFromSciHqClassify
  341. //如果有传入 上一个兄弟节点分类id
  342. if prevClassifyId > 0 {
  343. prevClassify, err = data_manage.GetSciHqClassifyById(prevClassifyId)
  344. if err != nil {
  345. if err.Error() == utils.ErrNoRow() {
  346. errMsg = "移动失败"
  347. err = errors.New("上一个兄弟节点分类信息不存在" + err.Error())
  348. return
  349. }
  350. errMsg = "移动失败"
  351. err = errors.New("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
  352. return
  353. }
  354. prevSort = prevClassify.Sort
  355. }
  356. if nextClassifyId > 0 {
  357. //下一个兄弟节点
  358. nextClassify, err = data_manage.GetSciHqClassifyById(nextClassifyId)
  359. if err != nil {
  360. if err.Error() == utils.ErrNoRow() {
  361. errMsg = "移动失败"
  362. err = errors.New("下一个兄弟节点分类信息不存在" + err.Error())
  363. return
  364. }
  365. errMsg = "移动失败"
  366. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
  367. return
  368. }
  369. nextSort = nextClassify.Sort
  370. }
  371. // 修改历史数据中的排序为0的情况
  372. if prevSort == 0 && nextSort == 0 { //目标是让当前分类处于目录的最顶部
  373. //更新为0排序的数据为当前最小排序
  374. if nextClassify != nil {
  375. minSort, e := data_manage.GetSciHqIndexClassifyMinSort(parentClassifyId)
  376. if e != nil {
  377. errMsg = "移动失败"
  378. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + e.Error())
  379. return
  380. }
  381. if minSort > 1 {
  382. minSort -= 1
  383. }
  384. nextClassify.Sort = minSort
  385. err = nextClassify.Update([]string{"sort"})
  386. if err != nil {
  387. errMsg = "移动失败"
  388. err = errors.New("更新下一个兄弟节点分类信息失败,Err:" + err.Error())
  389. return
  390. }
  391. nextSort = minSort
  392. }
  393. }
  394. //移到两个排序值中间操作
  395. if prevSort >= currentSort {
  396. //往下移动
  397. err = data_manage.MoveDownSciHqIndexClassifyBySort(parentClassifyId, prevSort, currentSort)
  398. if err != nil {
  399. err = errors.New("向下移动出错:" + err.Error())
  400. return
  401. }
  402. classifyInfo.Sort = prevSort
  403. } else if nextSort <= currentSort && nextSort != 0 {
  404. //往上移动
  405. err = data_manage.MoveUpSciHqIndexClassifyBySort(parentClassifyId, nextSort, currentSort)
  406. if err != nil {
  407. err = errors.New("向上移动出错:" + err.Error())
  408. return
  409. }
  410. classifyInfo.Sort = nextSort
  411. }
  412. classifyInfo.ModifyTime = time.Now()
  413. updateCol = append(updateCol, "Sort", "ModifyTime")
  414. err = classifyInfo.Update(updateCol)
  415. if err != nil {
  416. errMsg = "移动失败"
  417. err = errors.New("修改失败,Err:" + err.Error())
  418. return
  419. }
  420. return
  421. }
  422. // MoveSciHqData 移动卓创红期指标
  423. func MoveSciHqData(indexId, classifyId, prevIndexId, nextIndexId int) (err error, errMsg string) {
  424. //分类信息
  425. sciHqIndex, err := data_manage.GetSciHqIndexByIndexId(indexId)
  426. if err != nil {
  427. errMsg = `获取数据失败`
  428. if err.Error() == utils.ErrNoRow() {
  429. errMsg = `该指标未入库`
  430. err = nil
  431. }
  432. return
  433. }
  434. //判断分类是否存在
  435. if classifyId > 0 {
  436. _, err = data_manage.GetSciHqClassifyById(classifyId)
  437. if err != nil {
  438. errMsg = "移动失败"
  439. err = errors.New("获取分类信息失败,Err:" + err.Error())
  440. if err.Error() == utils.ErrNoRow() {
  441. errMsg = "找不到该分类"
  442. err = nil
  443. }
  444. return
  445. }
  446. }
  447. //如果改变了分类,那么移动该图表数据
  448. if sciHqIndex.ClassifyId != classifyId {
  449. tmpErr := data_manage.MoveSciHqIndex(indexId, classifyId)
  450. if tmpErr != nil {
  451. errMsg = "移动失败"
  452. err = errors.New("移动失败,Err:" + tmpErr.Error())
  453. return
  454. }
  455. }
  456. //移动排序
  457. updateCol := make([]string, 0)
  458. var currentSort, prevSort, nextSort int
  459. currentSort = sciHqIndex.Sort
  460. //如果有传入 上一个兄弟节点分类id
  461. var prevIndex *data_manage.BaseFromSciHqIndex
  462. var nextIndex *data_manage.BaseFromSciHqIndex
  463. if prevIndexId > 0 {
  464. prevIndex, err = data_manage.GetSciHqIndexByIndexId(prevIndexId)
  465. if err != nil {
  466. if err.Error() == utils.ErrNoRow() {
  467. errMsg = "移动失败"
  468. err = errors.New("上一个兄弟节点指标信息不存在" + err.Error())
  469. return
  470. }
  471. errMsg = "移动失败"
  472. err = errors.New("获取上一个兄弟节点指标信息失败,Err:" + err.Error())
  473. return
  474. }
  475. prevSort = prevIndex.Sort
  476. }
  477. if nextIndexId > 0 {
  478. //下一个兄弟节点
  479. nextIndex, err = data_manage.GetSciHqIndexByIndexId(nextIndexId)
  480. if err != nil {
  481. if err.Error() == utils.ErrNoRow() {
  482. errMsg = "移动失败"
  483. err = errors.New("下一个兄弟节点指标信息不存在" + err.Error())
  484. return
  485. }
  486. errMsg = "移动失败"
  487. err = errors.New("获取下一个兄弟节点指标信息失败,Err:" + err.Error())
  488. return
  489. }
  490. nextSort = nextIndex.Sort
  491. }
  492. // 修改历史数据中的排序为0的情况
  493. if prevSort == 0 && nextSort == 0 { //目标是让当前分类处于目录的最顶部, 但是历史数据里下个节点的sort为0的情况
  494. //更新为0排序的数据为当前最小排序
  495. if nextIndex != nil {
  496. minSort, e := data_manage.GetSciHqIndexMinSortByClassifyId(classifyId)
  497. if e != nil {
  498. errMsg = "移动失败"
  499. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + e.Error())
  500. return
  501. }
  502. if minSort > 1 {
  503. minSort -= 1
  504. }
  505. nextIndex.Sort = minSort
  506. err = nextIndex.Update([]string{"Sort"})
  507. if err != nil {
  508. errMsg = "移动失败"
  509. err = errors.New("更新下一个兄弟节点分类信息失败,Err:" + err.Error())
  510. return
  511. }
  512. nextSort = minSort
  513. }
  514. }
  515. //移到两个排序值中间操作
  516. if prevSort >= currentSort {
  517. //往下移动
  518. err = data_manage.MoveDownSciHqIndexBySort(classifyId, prevSort, currentSort)
  519. if err != nil {
  520. err = errors.New("向下移动出错:" + err.Error())
  521. return
  522. }
  523. sciHqIndex.Sort = prevSort
  524. } else if nextSort <= currentSort && nextSort != 0 {
  525. //往上移动
  526. err = data_manage.MoveUpSciHqIndexBySort(classifyId, nextSort, currentSort)
  527. if err != nil {
  528. err = errors.New("向上移动出错:" + err.Error())
  529. return
  530. }
  531. sciHqIndex.Sort = nextSort
  532. }
  533. //更新
  534. sciHqIndex.ModifyTime = time.Now()
  535. updateCol = append(updateCol, "Sort", "ModifyTime")
  536. err = sciHqIndex.Update(updateCol)
  537. if err != nil {
  538. errMsg = "移动失败"
  539. err = errors.New("修改失败,Err:" + err.Error())
  540. return
  541. }
  542. return
  543. }