manual.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990
  1. package data
  2. import (
  3. "errors"
  4. "eta/eta_api/models"
  5. "eta/eta_api/models/data_manage"
  6. "eta/eta_api/models/system"
  7. "eta/eta_api/services/alarm_msg"
  8. "eta/eta_api/utils"
  9. "fmt"
  10. "github.com/shopspring/decimal"
  11. "github.com/tealeg/xlsx"
  12. "strconv"
  13. "strings"
  14. "time"
  15. )
  16. func GetManualSysUser(keyWord string) (list []*data_manage.ManualSysUser, err error) {
  17. //departmentId := 1
  18. list = make([]*data_manage.ManualSysUser, 0)
  19. departmentItems, err := system.GetSysDepartmentAll()
  20. if err != nil {
  21. return list, err
  22. }
  23. for _, dv := range departmentItems {
  24. department := new(data_manage.ManualSysUser)
  25. department.ItemId = dv.DepartmentId * 10000
  26. department.ItemName = dv.DepartmentName
  27. fmt.Println(department.ItemId, department.ItemName)
  28. //GetSysuserList
  29. var condition string
  30. var pars []interface{}
  31. if keyWord != "" {
  32. condition += ` AND (real_name LIKE ? OR admin_name LIKE ? OR mobile LIKE ? ) `
  33. pars = utils.GetLikeKeywordPars(pars, keyWord, 3)
  34. }
  35. sysUsers, err := system.GetSysUserItems(condition, pars)
  36. if err != nil {
  37. return list, err
  38. }
  39. groups, err := system.GetSysGroupByDepartmentId(dv.DepartmentId)
  40. if err != nil {
  41. return list, err
  42. }
  43. dg := make([]*data_manage.ManualSysUser, 0)
  44. for _, v := range groups {
  45. group := new(data_manage.ManualSysUser)
  46. group.ItemId = v.DepartmentId * 100000
  47. group.ItemName = v.GroupName
  48. for _, sv := range sysUsers {
  49. user := new(data_manage.ManualSysUser)
  50. user.ItemId = sv.AdminId
  51. user.ItemName = sv.RealName
  52. if sv.GroupId == v.GroupId {
  53. group.Children = append(group.Children, user)
  54. }
  55. }
  56. if len(group.Children) > 0 {
  57. dg = append(dg, group)
  58. }
  59. }
  60. if len(groups) <= 0 {
  61. group := new(data_manage.ManualSysUser)
  62. group.ItemId = dv.DepartmentId * 100000
  63. group.ItemName = "无分组"
  64. for _, sv := range sysUsers {
  65. user := new(data_manage.ManualSysUser)
  66. user.ItemId = sv.AdminId
  67. user.ItemName = sv.RealName
  68. if sv.DepartmentId == dv.DepartmentId && sv.GroupId == 0 {
  69. group.Children = append(group.Children, user)
  70. }
  71. }
  72. if len(group.Children) > 0 {
  73. dg = append(dg, group)
  74. }
  75. }
  76. if len(dg) > 0 {
  77. department.Children = dg
  78. list = append(list, department)
  79. }
  80. }
  81. return
  82. }
  83. // GetManualEdbClassifyListByAdminId
  84. // @Description: 根据账户类型获取手工指标分类ID集合
  85. // @author: Roc
  86. // @datetime 2024-07-16 13:18:39
  87. // @param adminId int64
  88. // @return classifyIdList []int
  89. // @return err error
  90. func GetManualEdbClassifyListByAdminId(adminId int64) (classifyIdList []int, err error) {
  91. var list []*models.EdbdataClassify
  92. if adminId <= 0 {
  93. list, err = models.GetAllChildManualEdbClassify()
  94. } else {
  95. userClassifyList, _ := models.GetManualUserClassify(int(adminId))
  96. var userClassifyIdList []int
  97. for _, v := range userClassifyList {
  98. userClassifyIdList = append(userClassifyIdList, v.ClassifyId)
  99. }
  100. list, err = models.GetChildManualEdbClassifyByIdList(userClassifyIdList)
  101. }
  102. if err != nil {
  103. return
  104. }
  105. for _, classify := range list {
  106. classifyIdList = append(classifyIdList, classify.ClassifyId)
  107. }
  108. return
  109. }
  110. type ManualIndexSource2EdbReq struct {
  111. EdbCode string
  112. EdbName string
  113. Frequency string
  114. Unit string
  115. ClassifyId int
  116. AdminId int
  117. AdminRealName string
  118. }
  119. // ManualIndexSource2Edb
  120. // @Description: 新增手工数据源到指标库
  121. // @author: Roc
  122. // @datetime 2024-07-26 13:23:19
  123. // @param req ManualIndexSource2EdbReq
  124. // @param lang string
  125. // @return edb *data_manage.EdbInfo
  126. // @return err error
  127. // @return errMsg string
  128. // @return skip bool
  129. func ManualIndexSource2Edb(req ManualIndexSource2EdbReq, lang string) (edb *data_manage.EdbInfo, err error, errMsg string, skip bool) {
  130. if req.EdbCode == "" {
  131. err = fmt.Errorf("指标ID为空")
  132. return
  133. }
  134. defer func() {
  135. if err != nil {
  136. tips := fmt.Sprintf("ManualIndexSource2Edb新增失败, Err: %s", err.Error())
  137. fmt.Println(tips)
  138. utils.FileLog.Info(tips)
  139. }
  140. }()
  141. source := utils.DATA_SOURCE_MANUAL
  142. // 是否已有指标数据
  143. dataList, e := data_manage.GetEdbDataAllByEdbCode(req.EdbCode, source, utils.DATA_SUB_SOURCE_EDB, utils.EDB_DATA_LIMIT)
  144. if e != nil {
  145. err = fmt.Errorf("获取指标数据失败, Err: %s", e.Error())
  146. return
  147. }
  148. // 新增指标数据
  149. if len(dataList) == 0 {
  150. res, e := AddEdbData(source, req.EdbCode, req.Frequency)
  151. if e != nil {
  152. err = fmt.Errorf("index_lib: 新增指标数据失败, Err: %s", e.Error())
  153. return
  154. }
  155. if res == nil {
  156. err = fmt.Errorf("index_lib: 新增指标数据失败, res nil")
  157. return
  158. }
  159. if res.Ret != 200 {
  160. err = fmt.Errorf("index_lib: 新增指标数据失败, Ret: %d", res.Ret)
  161. return
  162. }
  163. }
  164. // 是否新增过指标
  165. exist, e := data_manage.GetEdbInfoByEdbCode(source, req.EdbCode)
  166. if e != nil && e.Error() != utils.ErrNoRow() {
  167. err = fmt.Errorf("获取指标是否存在失败, err: %s", e.Error())
  168. return
  169. }
  170. if exist != nil {
  171. skip = true
  172. return
  173. }
  174. // 开始结束时间
  175. var startDate, endDate string
  176. minMax, e := data_manage.GetEdbInfoMaxAndMinInfo(source, utils.DATA_SUB_SOURCE_EDB, req.EdbCode)
  177. if e != nil && e.Error() != utils.ErrNoRow() {
  178. err = fmt.Errorf("MinMax: 获取指标极值失败, err: %s", e.Error())
  179. return
  180. }
  181. if minMax != nil {
  182. startDate = minMax.MinDate
  183. endDate = minMax.MaxDate
  184. }
  185. // 新增指标库
  186. 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)
  187. if e != nil {
  188. errMsg = msg
  189. err = fmt.Errorf("EdbInfo: 新增指标失败, err: %s", e.Error())
  190. return
  191. }
  192. edb = edbInfo
  193. // 新增es
  194. go AddOrEditEdbInfoToEs(edbInfo.EdbInfoId)
  195. return
  196. }
  197. func ImportManualData(path string, sysUser *system.Admin) (successCount, failCount int, err error, errMsg string) {
  198. errMsg = `导入失败`
  199. xlFile, err := xlsx.OpenFile(path)
  200. if err != nil {
  201. fmt.Println(err.Error())
  202. return
  203. }
  204. if len(xlFile.Sheets) <= 0 {
  205. errMsg = "导入模板异常"
  206. err = errors.New(errMsg)
  207. return
  208. }
  209. //导入失败数据
  210. failDataList := make([]*models.EdbdataImportFail, 0)
  211. var indexDataList []ImportManualIndexData
  212. // 指标名称列表
  213. indexNameList := make([]string, 0)
  214. // 模板校验,然后处理成标准化格式
  215. for _, sheet := range xlFile.Sheets {
  216. var tmpIndexDataList []ImportManualIndexData
  217. var tmpFailDataList []*models.EdbdataImportFail
  218. rowList := sheet.Rows
  219. if len(rowList) <= 0 {
  220. errMsg = sheet.Name + "页异常"
  221. err = errors.New(errMsg)
  222. return
  223. }
  224. headerCell := rowList[0].Cells
  225. if len(headerCell) < 7 {
  226. errMsg = sheet.Name + "页模板异常"
  227. err = errors.New(errMsg)
  228. return
  229. }
  230. templateName := headerCell[6].String()
  231. switch templateName {
  232. case "导入模板2/Import Template 2":
  233. // 模板2需要走对应的取数逻辑
  234. tmpIndexDataList, tmpFailDataList, err, errMsg = getDataByTemplate2(sheet, sysUser.AdminId)
  235. default:
  236. // 模板1需要走对应的取数逻辑
  237. tmpIndexDataList, tmpFailDataList, err, errMsg = getDataByTemplate1(sheet, sysUser.AdminId)
  238. }
  239. indexDataList = append(indexDataList, tmpIndexDataList...)
  240. failDataList = append(failDataList, tmpFailDataList...)
  241. }
  242. var indexDataListMap map[string]ImportManualIndexData
  243. for _, v := range indexDataList {
  244. indexData, ok := indexDataListMap[v.IndexName]
  245. if !ok {
  246. // 没有就赋值
  247. indexDataListMap[v.IndexName] = v
  248. indexNameList = append(indexNameList, v.IndexName)
  249. continue
  250. }
  251. indexData.Unit = v.Unit
  252. indexData.ClassName = v.ClassName
  253. indexData.Frequency = v.Frequency
  254. indexData.DataMap = v.DataMap
  255. }
  256. if len(indexNameList) <= 0 {
  257. return
  258. }
  259. //超管账号可以查看分类下的所有频度数据
  260. userId := sysUser.AdminId
  261. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN {
  262. userId = 0
  263. }
  264. //获取账户所拥有权限的分类id集合
  265. classifyNameStrList, edbDataClassifyMap, err := GetEdbClassifyNameListByAdminId(int64(userId))
  266. if err != nil {
  267. errMsg = "获取分类数据失败"
  268. return
  269. }
  270. //指标map
  271. targetMap := make(map[string]*models.Edbinfo)
  272. defer func() {
  273. for _, target := range targetMap {
  274. //结束后,清除掉对应的缓存
  275. key := "import:edbinfo:data:" + target.TradeCode
  276. utils.Rc.Delete(key)
  277. //将该指标的code加入到 “手工数据导入后刷新” 缓存
  278. if utils.Re == nil {
  279. err := utils.Rc.LPush(utils.CACHE_IMPORT_MANUAL_DATA, target.TradeCode)
  280. if err != nil {
  281. fmt.Println("CACHE_IMPORT_MANUAL_DATA LPush Err:" + err.Error())
  282. }
  283. }
  284. }
  285. }()
  286. // 所有的指标数据map
  287. edbCodeDataMap := make(map[string]map[string]string)
  288. {
  289. // 获取指标信息
  290. manualIndexList, tmpErr := models.GetEdbinfoListBySecNameList(indexNameList)
  291. if tmpErr != nil {
  292. err = tmpErr
  293. errMsg = "获取指标信息失败"
  294. return
  295. }
  296. tradeCodeList := make([]string, 0)
  297. for _, v := range manualIndexList {
  298. targetMap[v.SecName] = v
  299. tradeCodeList = append(tradeCodeList, v.TradeCode)
  300. }
  301. // 获取明细数据
  302. dataList, tmpErr := models.GetTargetsDataListByCodeList(tradeCodeList)
  303. if tmpErr != nil {
  304. err = tmpErr
  305. errMsg = "获取指标明细数据失败"
  306. return
  307. }
  308. for _, tmpData := range dataList {
  309. //tmpDataMap,ok := edbCodeDataMap[tmpData.TradeCode]
  310. //if !ok{
  311. // tmpDataMap = make(map[string]string)
  312. //}
  313. edbCodeDataMap[tmpData.TradeCode][tmpData.Dt] = tmpData.Close
  314. }
  315. }
  316. //// TODO 成功数量超过20个指标,那就不导入了
  317. //if len(targetMap) >= 150 {
  318. // failItem := new(models.EdbdataImportFail)
  319. // failItem.SysUserId = strconv.Itoa(sysUser.AdminId)
  320. // failItem.ClassifyName = classifyName
  321. // failItem.CreateDate = createDate
  322. // failItem.SecName = secName
  323. // failItem.Close = closeVal
  324. // failItem.Remark = "导入指标数量过多"
  325. // failItem.Frequency = frequency
  326. // failItem.Unit = unit
  327. // failDataList = append(failDataList, failItem)
  328. // //go utils.SendEmail(utils.APPNAME+"失败提醒", "导入数据 获取分类:Err:"+err.Error(), utils.EmailSendToUsers)
  329. // continue
  330. //}
  331. errMsgList := make([]string, 0)
  332. addDataList := make([]*models.Edbdata, 0)
  333. for _, v := range indexDataListMap {
  334. if len(v.DataMap) <= 0 {
  335. continue
  336. }
  337. var tmpDate, tmpValue string
  338. for date, val := range v.DataMap {
  339. tmpDate = date
  340. tmpValue = val
  341. break
  342. }
  343. if !strings.Contains(strings.Join(classifyNameStrList, ","), v.ClassName) {
  344. failItem := &models.EdbdataImportFail{
  345. ClassifyName: v.ClassName,
  346. CreateDate: tmpDate,
  347. SecName: v.IndexName,
  348. Close: tmpValue,
  349. Remark: "没有该品种分类权限",
  350. SysUserId: fmt.Sprint(sysUser.AdminId),
  351. Frequency: v.Frequency,
  352. Unit: v.Unit,
  353. }
  354. failDataList = append(failDataList, failItem)
  355. continue
  356. }
  357. //获取指标分类信息
  358. classify, ok := edbDataClassifyMap[v.ClassName]
  359. if !ok {
  360. failItem := &models.EdbdataImportFail{
  361. ClassifyName: v.ClassName,
  362. CreateDate: tmpDate,
  363. SecName: v.IndexName,
  364. Close: tmpValue,
  365. Remark: "指标分类不存在",
  366. SysUserId: fmt.Sprint(sysUser.AdminId),
  367. Frequency: v.Frequency,
  368. Unit: v.Unit,
  369. }
  370. failDataList = append(failDataList, failItem)
  371. continue
  372. }
  373. target, ok := targetMap[v.IndexName]
  374. if !ok {
  375. if v.Frequency == "" {
  376. failItem := &models.EdbdataImportFail{
  377. ClassifyName: v.ClassName,
  378. CreateDate: tmpDate,
  379. SecName: v.IndexName,
  380. Close: tmpValue,
  381. Remark: "新增指标失败,频度字段为空",
  382. SysUserId: fmt.Sprint(sysUser.AdminId),
  383. Frequency: v.Frequency,
  384. Unit: v.Unit,
  385. }
  386. failDataList = append(failDataList, failItem)
  387. continue
  388. }
  389. if v.Unit == "" {
  390. failItem := &models.EdbdataImportFail{
  391. ClassifyName: v.ClassName,
  392. CreateDate: tmpDate,
  393. SecName: v.IndexName,
  394. Close: tmpValue,
  395. Remark: "新增指标失败,单位字段为空",
  396. SysUserId: fmt.Sprint(sysUser.AdminId),
  397. Frequency: v.Frequency,
  398. Unit: v.Unit,
  399. }
  400. failDataList = append(failDataList, failItem)
  401. continue
  402. }
  403. tmpErr := AddEdbInfo(v.IndexName, v.Unit, v.Frequency, "", sysUser.Mobile, classify.ClassifyId, sysUser.AdminId)
  404. if tmpErr != nil {
  405. fmt.Println("line 158")
  406. failItem := &models.EdbdataImportFail{
  407. ClassifyName: v.ClassName,
  408. CreateDate: tmpDate,
  409. SecName: v.IndexName,
  410. Close: tmpValue,
  411. Remark: "新增指标失败",
  412. SysUserId: fmt.Sprint(sysUser.AdminId),
  413. Frequency: v.Frequency,
  414. Unit: v.Unit,
  415. }
  416. failDataList = append(failDataList, failItem)
  417. continue
  418. }
  419. tmpTarget, tmpErr := models.GetTargetBySecName(v.IndexName)
  420. target = tmpTarget
  421. targetMap[v.IndexName] = target
  422. }
  423. if target == nil {
  424. fmt.Println("指标不存在")
  425. failItem := &models.EdbdataImportFail{
  426. ClassifyName: v.ClassName,
  427. CreateDate: tmpDate,
  428. SecName: v.IndexName,
  429. Close: tmpValue,
  430. Remark: "指标不存在",
  431. SysUserId: fmt.Sprint(sysUser.AdminId),
  432. Frequency: v.Frequency,
  433. Unit: v.Unit,
  434. }
  435. failDataList = append(failDataList, failItem)
  436. continue
  437. }
  438. //设置10分钟缓存,不允许其他地方删除
  439. key := "import:edbinfo:data:" + target.TradeCode
  440. utils.Rc.SetNX(key, 1, time.Second*600)
  441. //更新指标信息
  442. {
  443. updateCols := make([]string, 0)
  444. //更新指标分类
  445. if target.ClassifyId <= 0 && classify.ClassifyId > 0 {
  446. target.ClassifyId = classify.ClassifyId
  447. updateCols = append(updateCols, "ClassifyId")
  448. }
  449. if target.Frequency != v.Frequency {
  450. target.Frequency = v.Frequency
  451. target.NoticeTime = ""
  452. updateCols = append(updateCols, "Frequency", "NoticeTime")
  453. }
  454. if target.Unit != v.Unit {
  455. target.Unit = v.Unit
  456. updateCols = append(updateCols, "Unit")
  457. }
  458. if len(updateCols) > 0 {
  459. _ = target.Update(updateCols)
  460. }
  461. }
  462. {
  463. // 判断指标数据列表是否已经存在
  464. tmpDataMap, ok := edbCodeDataMap[target.TradeCode]
  465. if !ok {
  466. tmpDataMap = make(map[string]string)
  467. }
  468. for createDate, closeVal := range v.DataMap {
  469. //判断数据是否已经存在
  470. tmpVal, ok := tmpDataMap[createDate]
  471. //数据已存在,进行更新操作
  472. if ok {
  473. if tmpVal != closeVal {
  474. edbCodeDataMap[target.TradeCode][createDate] = closeVal
  475. tmpErr := models.ModifyTargetsDataByImport(target.TradeCode, createDate, closeVal)
  476. if tmpErr != nil {
  477. errMsgList = append(errMsgList, fmt.Sprintf("%s修改数据失败,日期:%s,值:%v,Err:%s", target.TradeCode, createDate, closeVal, tmpErr.Error()))
  478. }
  479. }
  480. } else { //数据不存在,进行新增操作
  481. if target.TradeCode != "" && createDate != "" && closeVal != "" {
  482. addDataList = append(addDataList, &models.Edbdata{
  483. TradeCode: target.TradeCode,
  484. Dt: createDate,
  485. Close: closeVal,
  486. ModifyTime: time.Now(),
  487. })
  488. edbCodeDataMap[target.TradeCode][createDate] = closeVal
  489. }
  490. }
  491. successCount++
  492. }
  493. }
  494. }
  495. // 批量添加明细数据
  496. if len(addDataList) > 0 {
  497. tmpErr := models.OnlyMultiAddEdbdata(addDataList)
  498. if tmpErr != nil {
  499. fmt.Println("line 221")
  500. errMsgList = append(errMsgList, fmt.Sprintf("批量添加明细数据失败,Err:%s", tmpErr.Error()))
  501. }
  502. }
  503. // 失败数量
  504. failCount = len(failDataList)
  505. //fmt.Println("failDataList:", len(failDataList))
  506. if failCount > 0 {
  507. //先删除导入失败记录
  508. _ = models.DelEdbDataImportFail(sysUser.AdminId)
  509. // 批量添加导入失败记录
  510. err = models.MultiAddEdbdataImportFail(failDataList)
  511. if err != nil {
  512. go alarm_msg.SendAlarmMsg("导入数据 新增失败记录失败,Err:"+err.Error(), 3)
  513. //go utils.SendEmail(utils.APPNAME+"失败提醒", "导入数据 新增失败记录失败:Err:"+err.Error(), utils.EmailSendToUsers)
  514. }
  515. {
  516. //错误信息字符串切片,最后作为发送邮件通知使用
  517. failContents := make([]string, 0)
  518. for _, v := range failDataList {
  519. failContents = append(failContents, fmt.Sprint(v.SecName, "导入失败:", v.Remark))
  520. }
  521. utils.FileLog.Info("导入数据 存在部分数据导入失败:" + strings.Join(failContents, ";"))
  522. //导入失败的话,最后统一邮件提醒就好啦,不需要每次都去提醒
  523. go alarm_msg.SendAlarmMsg("导入数据 存在部分数据导入失败:"+strings.Join(failContents, ";"), 3)
  524. }
  525. }
  526. return
  527. }
  528. // ImportManualIndexData
  529. // @Description: excel模板后的内容
  530. type ImportManualIndexData struct {
  531. IndexName string `description:"指标名称"`
  532. Unit string `description:"单位"`
  533. ClassName string `description:"所属品种"`
  534. Frequency string `description:"频度"`
  535. DataMap map[string]string `description:"时间数据"`
  536. }
  537. // TODO 模板数据获取异常的结果记录
  538. // getDataByTemplate1
  539. // @Description: 根据模板1获取数据
  540. // @author: Roc
  541. // @datetime 2024-07-24 16:17:45
  542. // @param sheet *xlsx.Sheet
  543. // @return indexDataList []ImportManualIndexData
  544. // @return err error
  545. // @return errMsg string
  546. func getDataByTemplate1(sheet *xlsx.Sheet, sysUserId int) (indexDataList []ImportManualIndexData, failDataList []*models.EdbdataImportFail, err error, errMsg string) {
  547. fmt.Println("sheet name: ", sheet.Name)
  548. indexDataList = make([]ImportManualIndexData, 0)
  549. indexDataListMap := make(map[string]ImportManualIndexData, 0)
  550. failDataList = make([]*models.EdbdataImportFail, 0)
  551. //遍历行读取
  552. maxRow := sheet.MaxRow
  553. fmt.Println("maxRow:", maxRow)
  554. // 表头信息
  555. if maxRow <= 2 {
  556. errMsg = "模板异常1"
  557. err = errors.New(errMsg)
  558. return
  559. }
  560. // 表头校验
  561. {
  562. headerRow := sheet.Row(1)
  563. cells := headerRow.Cells
  564. if len(cells) < 6 {
  565. errMsg = "导入文件异常,请下载最新导入模板文件"
  566. err = errors.New(errMsg)
  567. return
  568. }
  569. templateFail := false
  570. if cells[0].Value != "品种/Variety" {
  571. templateFail = true
  572. }
  573. if cells[1].Value != "指标名称/Indicator Name" {
  574. templateFail = true
  575. }
  576. if cells[2].Value != "指标日期/Indicator Date" {
  577. templateFail = true
  578. }
  579. if cells[3].Value != "值/Value" {
  580. templateFail = true
  581. }
  582. if cells[4].Value != "频度/Frequency" {
  583. templateFail = true
  584. }
  585. if cells[5].Value != "单位/Unit" {
  586. templateFail = true
  587. }
  588. if templateFail {
  589. errMsg = "导入文件异常,请下载最新导入模板文件"
  590. err = errors.New(errMsg)
  591. return
  592. }
  593. }
  594. for i := 2; i < maxRow; i++ {
  595. row := sheet.Row(i)
  596. cells := row.Cells
  597. if len(cells) < 6 {
  598. errMsg = "导入文件异常,请下载最新导入模板文件"
  599. err = errors.New(errMsg)
  600. return
  601. }
  602. classifyName := strings.TrimSpace(cells[0].Value) //分类
  603. if classifyName == "" { //过滤空白行
  604. continue
  605. }
  606. // 指标名称
  607. cell1 := cells[1].Value
  608. indexName := strings.TrimSpace(cell1)
  609. if indexName == "" { //过滤空白行
  610. continue
  611. }
  612. //createDate := utils.ConvertToFormatDay(cell1) //录入日期
  613. createDate := cells[2].Value //指标日期
  614. frequency := strings.TrimSpace(cells[4].String()) //频度
  615. unit := strings.TrimSpace(cells[5].String()) //单位
  616. closeVal := cells[3].Value //值
  617. if strings.Contains(closeVal, "#N/A") {
  618. continue
  619. }
  620. currDate, tmpErr := getExcelDate(createDate)
  621. if tmpErr != nil {
  622. failDataList = append(failDataList, &models.EdbdataImportFail{
  623. //Id: 0,
  624. ClassifyName: classifyName,
  625. CreateDate: createDate,
  626. SecName: indexName,
  627. Close: closeVal,
  628. Remark: "日期格式异常",
  629. SysUserId: strconv.Itoa(sysUserId),
  630. Frequency: frequency,
  631. Unit: unit,
  632. })
  633. continue
  634. }
  635. closeValFloat, tmpErr := cells[3].Float() //值
  636. if tmpErr != nil {
  637. failDataList = append(failDataList, &models.EdbdataImportFail{
  638. //Id: 0,
  639. ClassifyName: classifyName,
  640. CreateDate: currDate,
  641. SecName: indexName,
  642. Close: cells[3].Value,
  643. Remark: "值类型异常",
  644. SysUserId: strconv.Itoa(sysUserId),
  645. Frequency: frequency,
  646. Unit: unit,
  647. })
  648. continue
  649. }
  650. newDecimal := decimal.NewFromFloat(closeValFloat)
  651. newDecimal.Round(4)
  652. closeVal = newDecimal.String()
  653. if strings.Contains(closeVal, "#N/A") {
  654. continue
  655. }
  656. _, ok := indexDataListMap[indexName]
  657. if !ok {
  658. indexDataListMap[indexName] = ImportManualIndexData{
  659. IndexName: indexName,
  660. Unit: unit,
  661. ClassName: classifyName,
  662. Frequency: frequency,
  663. DataMap: make(map[string]string),
  664. }
  665. }
  666. indexDataListMap[indexName].DataMap[currDate] = closeVal
  667. }
  668. for _, v := range indexDataListMap {
  669. indexDataList = append(indexDataList, v)
  670. }
  671. return
  672. }
  673. // getDataByTemplate2
  674. // @Description: 根据模板2获取数据
  675. // @author: Roc
  676. // @datetime 2024-07-24 16:17:56
  677. // @param sheet *xlsx.Sheet
  678. // @return indexDataList []ImportManualIndexData
  679. // @return err error
  680. // @return errMsg string
  681. func getDataByTemplate2(sheet *xlsx.Sheet, sysUserId int) (indexDataList []ImportManualIndexData, failDataList []*models.EdbdataImportFail, err error, errMsg string) {
  682. fmt.Println("sheet name: ", sheet.Name)
  683. indexDataList = make([]ImportManualIndexData, 0)
  684. failDataList = make([]*models.EdbdataImportFail, 0)
  685. //遍历行读取
  686. maxRow := sheet.MaxRow
  687. fmt.Println("maxRow:", maxRow)
  688. varietyList := make([]string, 0)
  689. indexNameList := make([]string, 0)
  690. unitList := make([]string, 0)
  691. frequencyList := make([]string, 0)
  692. // make(map[指标下标]map[日期]值)
  693. indexDateValueMap := make(map[int]map[string]string)
  694. // 表头信息
  695. if maxRow <= 4 {
  696. errMsg = "模板异常1"
  697. err = errors.New(errMsg)
  698. return
  699. }
  700. // 表头处理
  701. for i := 1; i <= 4; i++ {
  702. row := sheet.Row(i)
  703. cells := row.Cells
  704. //if len(cells) < 1 {
  705. // errMsg = "模板异常2"
  706. // err = errors.New(errMsg)
  707. // return
  708. //}
  709. switch i {
  710. case 1:
  711. for k, v := range cells {
  712. if k == 0 {
  713. continue
  714. }
  715. varietyList = append(varietyList, strings.TrimSpace(v.String()))
  716. }
  717. case 2:
  718. for k, v := range cells {
  719. if k == 0 {
  720. continue
  721. }
  722. indexNameList = append(indexNameList, strings.TrimSpace(v.String()))
  723. }
  724. case 3:
  725. for k, v := range cells {
  726. if k == 0 {
  727. continue
  728. }
  729. unitList = append(unitList, strings.TrimSpace(v.String()))
  730. }
  731. case 4:
  732. for k, v := range cells {
  733. if k == 0 {
  734. continue
  735. }
  736. frequencyList = append(frequencyList, strings.TrimSpace(v.String()))
  737. }
  738. }
  739. }
  740. maxNameIndex := len(indexNameList) - 1
  741. maxUnitIndex := len(unitList) - 1
  742. maxClassNameIndex := len(varietyList) - 1
  743. maxFrequencyIndex := len(frequencyList) - 1
  744. // 数据处理
  745. for i := 5; i < maxRow; i++ {
  746. row := sheet.Row(i)
  747. cells := row.Cells
  748. //if len(cells) < 1 {
  749. // errMsg = "模板异常2"
  750. // err = errors.New(errMsg)
  751. // return
  752. //}
  753. // 当前日期
  754. var currDate string
  755. // 日期是否异常,异常的话,就跳过,进入下一行数据处理
  756. var dateErr bool
  757. // 数据处理
  758. for k, v := range cells {
  759. if k == 0 {
  760. tmpCurrDate, tmpErr := getExcelDate(v.Value)
  761. if tmpErr != nil {
  762. // TODO 错误数据记录
  763. //failItem := new(models.EdbdataImportFail)
  764. //failItem.SysUserId = strconv.Itoa(sysUser.AdminId)
  765. //failItem.ClassifyName = classifyName
  766. //failItem.CreateDate = createDate
  767. //failItem.SecName = secName
  768. //failItem.Close = closeVal
  769. //failItem.Remark = "日期格式异常"
  770. //failItem.Frequency = frequency
  771. //failItem.Unit = unit
  772. //failDataList = append(failDataList, failItem)
  773. //go utils.SendEmail(utils.APPNAME+"失败提醒", "导入数据 获取分类:Err:"+err.Error(), utils.EmailSendToUsers)
  774. dateErr = true
  775. }
  776. currDate = tmpCurrDate
  777. continue
  778. }
  779. key := k - 1
  780. // 日期异常,所以不处理该行了
  781. if dateErr {
  782. var indexName, unit, classifyName, frequency string
  783. if key <= maxNameIndex {
  784. indexName = indexNameList[key]
  785. }
  786. if key <= maxUnitIndex {
  787. unit = unitList[key]
  788. }
  789. if key <= maxClassNameIndex {
  790. classifyName = varietyList[key]
  791. }
  792. if key <= maxFrequencyIndex {
  793. frequency = frequencyList[key]
  794. }
  795. failDataList = append(failDataList, &models.EdbdataImportFail{
  796. //Id: 0,
  797. ClassifyName: classifyName,
  798. CreateDate: currDate,
  799. SecName: indexName,
  800. Close: cells[3].Value,
  801. Remark: "日期格式异常",
  802. SysUserId: strconv.Itoa(sysUserId),
  803. Frequency: frequency,
  804. Unit: unit,
  805. })
  806. continue
  807. }
  808. _, ok := indexDateValueMap[key]
  809. if !ok {
  810. indexDateValueMap[key] = make(map[string]string)
  811. }
  812. closeVal := v.Value //值
  813. // 没有数据,说明是空串
  814. if strings.Contains(closeVal, "#N/A") {
  815. indexDateValueMap[key][currDate] = ""
  816. continue
  817. }
  818. closeValFloat, tmpErr := v.Float() //值
  819. if tmpErr != nil {
  820. // TODO 错误数据记录
  821. //failItem := new(models.EdbdataImportFail)
  822. //failItem.SysUserId = strconv.Itoa(sysUser.AdminId)
  823. //failItem.ClassifyName = classifyName
  824. //failItem.CreateDate = createDate
  825. //failItem.SecName = secName
  826. //failItem.Close = cells[3].Value
  827. //failItem.Remark = "值类型异常"
  828. //failItem.Frequency = frequency
  829. //failItem.Unit = unit
  830. //failDataList = append(failDataList, failItem)
  831. indexDateValueMap[key][currDate] = ""
  832. continue
  833. }
  834. newDecimal := decimal.NewFromFloat(closeValFloat)
  835. newDecimal.Round(4)
  836. closeVal = newDecimal.String()
  837. if strings.Contains(closeVal, "#N/A") {
  838. indexDateValueMap[key][currDate] = ""
  839. continue
  840. }
  841. indexDateValueMap[key][currDate] = closeVal
  842. }
  843. }
  844. for i, indexName := range indexNameList {
  845. var unit, classifyName, frequency string
  846. if i <= maxUnitIndex {
  847. unit = unitList[i]
  848. }
  849. if i <= maxClassNameIndex {
  850. classifyName = varietyList[i]
  851. }
  852. if i <= maxFrequencyIndex {
  853. frequency = frequencyList[i]
  854. }
  855. indexData := ImportManualIndexData{
  856. IndexName: indexName,
  857. Unit: unit,
  858. ClassName: classifyName,
  859. Frequency: frequency,
  860. DataMap: indexDateValueMap[i],
  861. }
  862. indexDataList = append(indexDataList, indexData)
  863. }
  864. return
  865. }
  866. // getExcelDate
  867. // @Description: 获取excel的日期
  868. // @author: Roc
  869. // @datetime 2024-07-23 17:26:12
  870. // @param createDate string
  871. // @return newCreateDate string
  872. // @return err error
  873. func getExcelDate(createDate string) (newCreateDate string, err error) {
  874. if strings.Contains(createDate, "-") {
  875. //如果是带有 - 的普通日期格式文本
  876. _, err = time.Parse("2006-1-2", createDate)
  877. if err == nil {
  878. newCreateDate = createDate
  879. }
  880. } else if strings.Contains(createDate, "/") {
  881. //如果是带有 / 的普通日期格式文本
  882. createDateTime, timeErr := time.Parse("2006/1/2", createDate)
  883. if timeErr != nil {
  884. err = timeErr
  885. } else {
  886. newCreateDate = createDateTime.Format("2006-01-02")
  887. }
  888. } else {
  889. //可能是excel的日期格式
  890. _, tmpErr := strconv.Atoi(createDate)
  891. if tmpErr != nil {
  892. err = tmpErr
  893. } else {
  894. newCreateDate = utils.ConvertToFormatDay(createDate) //录入日期
  895. }
  896. }
  897. return
  898. }