base_from_mysteel_chemical.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671
  1. package models
  2. import (
  3. sql2 "database/sql"
  4. "eta/eta_index_lib/global"
  5. "eta/eta_index_lib/utils"
  6. "fmt"
  7. "gorm.io/gorm"
  8. "strconv"
  9. "strings"
  10. "time"
  11. )
  12. //钢联
  13. type MysteelChemicalData struct {
  14. InputValue float64 `gorm:"column:value" description:"值"`
  15. DataTime string `gorm:"column:data_time" description:"日期"`
  16. }
  17. // AfterFind 在该模型上设置钩子函数,把日期转成正确的string,所以查询函数只能用Find函数,First或者Scan是不会触发该函数的来获取数据
  18. func (m *MysteelChemicalData) AfterFind(db *gorm.DB) (err error) {
  19. m.DataTime = utils.GormDateStrToDateStr(m.DataTime)
  20. return
  21. }
  22. func GetMysteelChemicalDataByCondition(condition string, pars []interface{}) (item []*MysteelChemicalData, err error) {
  23. sql1 := ` SELECT * FROM base_from_mysteel_chemical_data WHERE 1=1 `
  24. //o := orm.NewOrm()
  25. if condition != "" {
  26. sql1 += condition
  27. }
  28. sql := `select * from (` + sql1 + ` having 1 order by modify_time DESC ) tmp GROUP BY data_time ORDER BY data_time DESC `
  29. //_, err = o.Raw(sql, pars).QueryRows(&item)
  30. err = global.DEFAULT_DB.Raw(sql, pars...).Find(&item).Error
  31. return
  32. }
  33. // AddEdbDataFromMysteelChemical 新增钢联指标数据
  34. func AddEdbDataFromMysteelChemical(edbCode string) (err error) {
  35. //o := orm.NewOrm()
  36. var condition string
  37. var pars []interface{}
  38. if edbCode != "" {
  39. condition += " AND index_code = ? "
  40. pars = append(pars, edbCode)
  41. }
  42. mysteelChemicalDataList, err := GetMysteelChemicalDataByCondition(condition, pars)
  43. if err != nil {
  44. return
  45. }
  46. dataLen := len(mysteelChemicalDataList)
  47. existMap := make(map[string]string)
  48. if dataLen > 0 {
  49. var isAdd bool
  50. addSql := ` INSERT INTO edb_data_mysteel_chemical (edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  51. addSql = utils.ReplaceDriverKeywords("", addSql)
  52. for i := 0; i < dataLen; i++ {
  53. item := mysteelChemicalDataList[i]
  54. eDate := item.DataTime
  55. sValue := utils.SubFloatToString(item.InputValue, 30)
  56. if sValue != "" {
  57. if _, ok := existMap[eDate]; !ok {
  58. dataTime, err := time.ParseInLocation(utils.FormatDate, eDate, time.Local)
  59. if err != nil {
  60. return err
  61. }
  62. timestamp := dataTime.UnixNano() / 1e6
  63. timeStr := fmt.Sprintf("%d", timestamp)
  64. addSql += GetAddSql("0", edbCode, eDate, timeStr, sValue)
  65. isAdd = true
  66. }
  67. }
  68. existMap[eDate] = eDate
  69. }
  70. if isAdd {
  71. addSql = strings.TrimRight(addSql, ",")
  72. utils.FileLog.Info("addSql:" + addSql)
  73. //_, err = o.Raw(addSql).Exec()
  74. err = global.DEFAULT_DB.Exec(addSql).Error
  75. if err != nil {
  76. return err
  77. }
  78. }
  79. }
  80. return
  81. }
  82. // RefreshEdbDataFromMysteelChemical 刷新钢联指标数据
  83. func RefreshEdbDataFromMysteelChemical(edbInfoId int, edbCode, startDate string) (err error) {
  84. source := utils.DATA_SOURCE_MYSTEEL_CHEMICAL
  85. subSource := utils.DATA_SUB_SOURCE_EDB
  86. //o := orm.NewOrm()
  87. if err != nil {
  88. return
  89. }
  90. edbInfoIdStr := strconv.Itoa(edbInfoId)
  91. //计算数据
  92. var condition string
  93. var pars []interface{}
  94. if edbCode != "" {
  95. condition += " AND index_code=? "
  96. pars = append(pars, edbCode)
  97. }
  98. if startDate != "" {
  99. condition += " AND data_time>=? "
  100. pars = append(pars, startDate)
  101. }
  102. mysteelChemicalDataList, err := GetMysteelChemicalDataByCondition(condition, pars)
  103. if err != nil {
  104. return
  105. }
  106. // 真实数据的最大日期 , 插入规则配置的日期
  107. var realDataMaxDate, edbDataInsertConfigDate time.Time
  108. var edbDataInsertConfig *EdbDataInsertConfig
  109. var isFindConfigDateRealData bool //是否找到配置日期的实际数据的值
  110. {
  111. edbDataInsertConfig, err = GetEdbDataInsertConfigByEdbId(edbInfoId)
  112. if err != nil && !utils.IsErrNoRow(err) {
  113. return
  114. }
  115. if edbDataInsertConfig != nil {
  116. edbDataInsertConfigDate = edbDataInsertConfig.Date
  117. }
  118. }
  119. var existCondition string
  120. var existPars []interface{}
  121. existCondition += " AND edb_info_id=? "
  122. existPars = append(existPars, edbInfoId)
  123. if startDate != "" {
  124. existCondition += " AND data_time>=? "
  125. existPars = append(existPars, startDate)
  126. }
  127. //获取指标所有数据
  128. existList, err := GetEdbDataByCondition(source, subSource, existCondition, existPars)
  129. if err != nil {
  130. return err
  131. }
  132. existMap := make(map[string]*EdbInfoSearchData)
  133. for _, v := range existList {
  134. existMap[v.DataTime] = v
  135. }
  136. addSql := ` INSERT INTO edb_data_mysteel_chemical(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  137. addSql = utils.ReplaceDriverKeywords("", addSql)
  138. var isAdd bool
  139. addMap := make(map[string]string)
  140. for _, v := range mysteelChemicalDataList {
  141. item := v
  142. eDate := item.DataTime
  143. sValue := utils.SubFloatToString(item.InputValue, 30)
  144. dataTime, err := time.ParseInLocation(utils.FormatDate, eDate, time.Local)
  145. if err != nil {
  146. return err
  147. }
  148. if findItem, ok := existMap[v.DataTime]; !ok {
  149. if sValue != "" {
  150. timestamp := dataTime.UnixNano() / 1e6
  151. timeStr := fmt.Sprintf("%d", timestamp)
  152. saveValue := sValue
  153. if _, addOk := addMap[eDate]; !addOk {
  154. addSql += GetAddSql(edbInfoIdStr, edbCode, eDate, timeStr, saveValue)
  155. isAdd = true
  156. }
  157. }
  158. } else {
  159. if findItem != nil && utils.SubFloatToString(findItem.Value, 30) != sValue {
  160. err = ModifyEdbDataById(source, subSource, findItem.EdbDataId, sValue)
  161. if err != nil {
  162. return err
  163. }
  164. }
  165. }
  166. addMap[v.DataTime] = v.DataTime
  167. // 下面代码主要目的是处理掉手动插入的数据判断
  168. {
  169. if realDataMaxDate.IsZero() || dataTime.After(realDataMaxDate) {
  170. realDataMaxDate = dataTime
  171. }
  172. if edbDataInsertConfigDate.IsZero() || dataTime.Equal(edbDataInsertConfigDate) {
  173. isFindConfigDateRealData = true
  174. }
  175. }
  176. }
  177. // 处理手工数据补充的配置
  178. HandleConfigInsertEdbData(realDataMaxDate, edbDataInsertConfig, edbInfoId, source, subSource, existMap, isFindConfigDateRealData)
  179. if isAdd {
  180. addSql = strings.TrimRight(addSql, ",")
  181. //_, err = o.Raw(addSql).Exec()
  182. err = global.DEFAULT_DB.Exec(addSql).Error
  183. if err != nil {
  184. return err
  185. }
  186. }
  187. return
  188. }
  189. // 钢联化工指标数据
  190. type BaseFromMysteelChemicalIndex struct {
  191. BaseFromMysteelChemicalIndexId int64 `gorm:"column:base_from_mysteel_chemical_index_id;primaryKey"`
  192. //BaseFromMysteelChemicalIndexId int64 `orm:"column(base_from_mysteel_chemical_index_id);pk"`
  193. BaseFromMysteelChemicalClassifyId int `description:"分类id"`
  194. IndexCode string `description:"指标编码"`
  195. IndexName string `description:"指标名称"`
  196. Unit string `description:"单位"`
  197. Source string `description:"数据来源"`
  198. Frequency string `description:"频度"`
  199. StartDate time.Time `description:"开始日期"`
  200. EndDate time.Time `description:"结束日期"`
  201. Describe string `description:"指标描述"`
  202. UpdateWeek string `description:"更新周期"`
  203. UpdateTime string `description:"更新时间,多个时间点用英文,隔开"`
  204. UpdateTime2 string `description:"更新时间2"`
  205. SysUserId int `description:"创建人id"`
  206. SysUserRealName string `description:"创建人姓名"`
  207. FilePath string `description:"文件存储路径"`
  208. MergeFilePath string `description:"更新文件"`
  209. FileIndex int `description:"文件索引"`
  210. MergeUpdateWeek string `description:"合并文件的更新周"`
  211. UpdateDate string `description:"更新日期"`
  212. CreateTime time.Time `description:"创建时间"`
  213. ModifyTime time.Time `description:"修改时间"`
  214. IsStop int `description:"是否停更:1:停更,0:未停更"`
  215. TerminalCode string `description:"终端编码"`
  216. IsSupplierStop int `description:"是否供应商停更:1:停更,0:未停更"`
  217. }
  218. func (e *BaseFromMysteelChemicalIndex) AfterFind(db *gorm.DB) (err error) {
  219. e.UpdateTime = utils.GormDateStrToDateStr(e.UpdateTime)
  220. e.UpdateTime2 = utils.GormDateStrToDateStr(e.UpdateTime2)
  221. return
  222. }
  223. type MysteelChemicalAPiCheck struct {
  224. IsEnable bool
  225. ErrMsg string
  226. }
  227. // GetIndexRefreshAllByMergeFile 根据合并文件去分组查询需要刷新的文件
  228. func (m *BaseFromMysteelChemicalIndex) GetIndexRefreshAllByMergeFile() (items []*BaseFromMysteelChemicalIndex, err error) {
  229. //o := orm.NewOrm()
  230. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE merge_file_path <>'' AND base_from_mysteel_chemical_classify_id NOT IN(54,55,56,57) GROUP BY merge_file_path`
  231. //_, err = o.Raw(sql).QueryRows(&items)
  232. err = global.DEFAULT_DB.Raw(sql).Find(&items).Error
  233. return
  234. }
  235. func (m *BaseFromMysteelChemicalIndex) GetIndexItem(indexCode string) (item *BaseFromMysteelChemicalIndex, err error) {
  236. //o := orm.NewOrm()
  237. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_code = ? `
  238. //err = o.Raw(sql, indexCode).QueryRow(&item)
  239. err = global.DEFAULT_DB.Raw(sql, indexCode).First(&item).Error
  240. return
  241. }
  242. func (m *BaseFromMysteelChemicalIndex) GetBatchIndexItem(indexCodes []string) (items []*BaseFromMysteelChemicalIndex, err error) {
  243. if len(indexCodes) <= 0 {
  244. return
  245. }
  246. //o := orm.NewOrm()
  247. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_code IN (%s) `
  248. holder := make([]string, 0, len(indexCodes))
  249. for range indexCodes {
  250. holder = append(holder, "?")
  251. }
  252. sql = fmt.Sprintf(sql, strings.Join(holder, ","))
  253. err = global.DEFAULT_DB.Raw(sql, indexCodes).Find(&items).Error
  254. //_, err = o.Raw(sql, indexCodes).QueryRows(&items)
  255. return
  256. }
  257. func (m *BaseFromMysteelChemicalIndex) GetIndexCreate(terminalCode string) (items []*BaseFromMysteelChemicalIndex, err error) {
  258. //o := orm.NewOrm()
  259. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_name = '' AND terminal_code = ? `
  260. //_, err = o.Raw(sql, terminalCode).QueryRows(&items)
  261. err = global.DEFAULT_DB.Raw(sql, terminalCode).Find(&items).Error
  262. return
  263. }
  264. // Add 新增
  265. func (m *BaseFromMysteelChemicalIndex) Add() (err error) {
  266. //o := orm.NewOrm()
  267. //_, err = o.Insert(m)
  268. err = global.DEFAULT_DB.Create(&m).Error
  269. return
  270. }
  271. func (m *BaseFromMysteelChemicalIndex) Update(cols []string) (err error) {
  272. //o := orm.NewOrm()
  273. //_, err = o.Update(m, cols...)
  274. err = global.DEFAULT_DB.Model(&m).Select(cols).Updates(&m).Error
  275. return
  276. }
  277. // GetNoMergeIndexByFrequencyCount 获取未合并的指标总数量
  278. func (m *BaseFromMysteelChemicalIndex) GetNoMergeIndexByFrequencyCount(frequency string) (total int64, err error) {
  279. //o := orm.NewOrm()
  280. sql := `SELECT count(1) AS total FROM base_from_mysteel_chemical_index WHERE frequency = ? AND merge_file_path = '' AND index_name NOT LIKE "%停%" `
  281. //err = o.Raw(sql, frequency).QueryRow(&total)
  282. err = global.DEFAULT_DB.Raw(sql, frequency).Scan(&total).Error
  283. return
  284. }
  285. // GetMaxFileIndexReq 获取最大文件数请求
  286. type GetMaxFileIndexReq struct {
  287. Frequency string
  288. EndDate string
  289. TerminalCode string `description:"终端编码"`
  290. }
  291. // GetMaxFileIndex 获取最大文件的
  292. func (m *BaseFromMysteelChemicalIndex) GetMaxFileIndex(frequency string) (item *BaseFromMysteelChemicalIndex, err error) {
  293. //o := orm.NewOrm()
  294. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE frequency=? AND index_name NOT LIKE "%停%" ORDER BY file_index DESC `
  295. ///err = o.Raw(sql, frequency).QueryRow(&item)
  296. err = global.DEFAULT_DB.Raw(sql, frequency).First(&item).Error
  297. return
  298. }
  299. // GetNoMergeIndexByFrequencyPageReq 获取未合并的指标列表
  300. type GetNoMergeIndexByFrequencyPageReq struct {
  301. Frequency string
  302. Limit int
  303. }
  304. // GetNoMergeIndexByFrequencyPage 获取未合并的指标列表
  305. func (m *BaseFromMysteelChemicalIndex) GetNoMergeIndexByFrequencyPage(frequency string, limit int) (items []*BaseFromMysteelChemicalIndex, err error) {
  306. //o := orm.NewOrm()
  307. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE frequency=? AND merge_file_path = '' limit ? `
  308. //_, err = o.Raw(sql, frequency, limit).QueryRows(&items)
  309. err = global.DEFAULT_DB.Raw(sql, frequency, limit).Find(&items).Error
  310. return
  311. }
  312. type GetIndexCountByMergeFilePathReq struct {
  313. MergeFilePath string
  314. }
  315. func (m *BaseFromMysteelChemicalIndex) GetIndexCountByMergeFilePath(mergeFilePath string) (total int64, err error) {
  316. //o := orm.NewOrm()
  317. sql := `SELECT count(1) AS total FROM base_from_mysteel_chemical_index WHERE merge_file_path = ?`
  318. //err = o.Raw(sql, mergeFilePath).QueryRow(&total)
  319. err = global.DEFAULT_DB.Raw(sql, mergeFilePath).Scan(&total).Error
  320. return
  321. }
  322. // GetIndexRefreshMethanolByMergeFile 根据合并文件去分组查询需要刷新的甲醇文件
  323. func (m *BaseFromMysteelChemicalIndex) GetIndexRefreshMethanolByMergeFile() (items []*BaseFromMysteelChemicalIndex, err error) {
  324. //o := orm.NewOrm()
  325. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE merge_file_path <> '' AND base_from_mysteel_chemical_classify_id IN(54,55,56,57) GROUP BY merge_file_path`
  326. //_, err = o.Raw(sql).QueryRows(&items)
  327. err = global.DEFAULT_DB.Raw(sql).Find(&items).Error
  328. return
  329. }
  330. // GetIndexRefreshMethanolByTimely 获取需要及时刷新的文件
  331. func (m *BaseFromMysteelChemicalIndex) GetIndexRefreshMethanolByTimely() (items []*BaseFromMysteelChemicalIndex, err error) {
  332. //o := orm.NewOrm()
  333. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE merge_file_path <> '' AND is_refresh = 1 GROUP BY merge_file_path`
  334. //_, err = o.Raw(sql).QueryRows(&items)
  335. err = global.DEFAULT_DB.Raw(sql).Find(&items).Error
  336. return
  337. }
  338. type GetIndexByFrequencyReq struct {
  339. Frequency string
  340. }
  341. // GetIndexByFrequency 根据频率获取指标
  342. func (m *BaseFromMysteelChemicalIndex) GetIndexByFrequency(frequency string) (items []*BaseFromMysteelChemicalIndex, err error) {
  343. //o := orm.NewOrm()
  344. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE frequency = ? AND merge_file_path = '' `
  345. //_, err = o.Raw(sql, frequency).QueryRows(&items)
  346. err = global.DEFAULT_DB.Raw(sql, frequency).Find(&items).Error
  347. return
  348. }
  349. type GetIndexRefreshWeekReq struct {
  350. NowWeekZn string
  351. StartTime string
  352. EndTime string
  353. }
  354. func (m *BaseFromMysteelChemicalIndex) GetIndexRefreshWeek(nowWeekZn, startTime, endTime string) (items []*BaseFromMysteelChemicalIndex, err error) {
  355. where := `update_week = ? AND ((update_time >= ? AND update_time <= ?) OR (update_time2 >= ? AND update_time2 <= ?)) `
  356. //o := orm.NewOrm()
  357. sql := fmt.Sprintf(`SELECT * FROM base_from_mysteel_chemical_index WHERE %s `, where)
  358. //_, err = o.Raw(sql, nowWeekZn, startTime, endTime, startTime, endTime).QueryRows(&items)
  359. err = global.DEFAULT_DB.Raw(sql, nowWeekZn, startTime, endTime, startTime, endTime).Find(&items).Error
  360. return
  361. }
  362. // UpdateIndexReq 根据编码更新指标请求体
  363. type UpdateIndexReq struct {
  364. Item *BaseFromMysteelChemicalIndex
  365. UpdateCols []string
  366. }
  367. func (m *BaseFromMysteelChemicalIndex) UpdateIndex(item *BaseFromMysteelChemicalIndex, updateCols []string) (err error) {
  368. if item == nil {
  369. return
  370. }
  371. if len(updateCols) == 0 {
  372. return
  373. }
  374. //o := orm.NewOrm()
  375. //_, err = o.Update(item, updateCols...)
  376. err = global.DEFAULT_DB.Model(&item).Select(updateCols).Updates(&item).Error
  377. return
  378. }
  379. // MultiUpdateIndex 批量更新指标
  380. func (m *BaseFromMysteelChemicalIndex) MultiUpdateIndex(items []UpdateIndexReq) (err error) {
  381. if len(items) == 0 {
  382. return
  383. }
  384. //o := orm.NewOrm()
  385. for _, v := range items {
  386. if len(v.UpdateCols) == 0 {
  387. continue
  388. }
  389. if v.Item == nil {
  390. continue
  391. }
  392. //if _, e := o.Update(v.Item, v.UpdateCols...); e != nil {
  393. if e := global.DEFAULT_DB.Model(&v.Item).Select(v.UpdateCols).Updates(&v.Item).Error; e != nil {
  394. err = fmt.Errorf("update err: %s", e.Error())
  395. return
  396. }
  397. }
  398. return
  399. }
  400. type HandleMysteelIndex struct {
  401. IndexName string `description:"指标名称"`
  402. IndexCode string `description:"指标编码"`
  403. Unit string `description:"单位"`
  404. Source string `description:"数据来源"`
  405. Frequency string `description:"频度"`
  406. StartDate string `description:"开始日期"`
  407. EndDate string `description:"结束日期"`
  408. Describe string `description:"指标描述"`
  409. UpdateDate string `description:"更新日期"`
  410. ExcelDataMap map[string]string
  411. }
  412. type HandleMysteelIndexResp struct {
  413. List []*HandleMysteelIndex
  414. }
  415. // 钢联化工指标数据
  416. type BaseFromMysteelChemicalData struct {
  417. BaseFromMysteelChemicalDataId int64 `gorm:"column:base_from_mysteel_chemical_data_id;primaryKey"`
  418. //BaseFromMysteelChemicalDataId int64 `orm:"column(base_from_mysteel_chemical_data_id);pk"`
  419. BaseFromMysteelChemicalIndexId int64
  420. IndexCode string
  421. DataTime time.Time
  422. Value string
  423. UpdateDate string
  424. CreateTime time.Time `description:"创建时间"`
  425. ModifyTime time.Time `description:"修改时间"`
  426. }
  427. // GetIndexDataList 根据指标编码获取数据
  428. func (d *BaseFromMysteelChemicalData) GetIndexDataList(indexCode string) (item []*BaseFromMysteelChemicalData, err error) {
  429. //o := orm.NewOrm()
  430. sql := `SELECT * FROM base_from_mysteel_chemical_data WHERE index_code = ? `
  431. //_, err = o.Raw(sql, indexCode).QueryRows(&item)
  432. err = global.DEFAULT_DB.Raw(sql, indexCode).Find(&item).Error
  433. return
  434. }
  435. // Update 修改
  436. func (r *BaseFromMysteelChemicalData) Update(updateCols []string) (err error) {
  437. //o := orm.NewOrm()
  438. //_, err = o.Update(r, updateCols...)
  439. err = global.DEFAULT_DB.Model(&r).Select(updateCols).Updates(&r).Error
  440. return
  441. }
  442. // Add 新增
  443. func (r *BaseFromMysteelChemicalData) Add(list []BaseFromMysteelChemicalData) (err error) {
  444. //o := orm.NewOrm()
  445. //_, err = o.InsertMulti(500, list)
  446. err = global.DEFAULT_DB.CreateInBatches(list, utils.MultiAddNum).Error
  447. return
  448. }
  449. // AddV2 新增
  450. func (r *BaseFromMysteelChemicalData) AddV2(list []*BaseFromMysteelChemicalData) (err error) {
  451. if len(list) == 0 {
  452. return
  453. }
  454. //o := orm.NewOrm()
  455. //_, err = o.InsertMulti(500, list)
  456. err = global.DEFAULT_DB.CreateInBatches(list, utils.MultiAddNum).Error
  457. return
  458. }
  459. type AddMysteelIndexResp struct {
  460. EdbCode string `description:"指标编码"`
  461. TerminalCode string `description:"指标终端编码"`
  462. BaseFromMysteelChemicalClassifyId int `description:"指标分类"`
  463. SysUserId int `description:"操作人id"`
  464. SysUserRealName string `description:"操作人真实名称"`
  465. }
  466. type GetIndexDetailReq struct {
  467. IndexCode string `description:"指标编码"`
  468. }
  469. // GetIndexCountByFrequency 获取未合并的指标总数量
  470. func (m *BaseFromMysteelChemicalIndex) GetIndexCountByFrequency(condition string, pars []interface{}) (total int64, err error) {
  471. //o := orm.NewOrm()
  472. sql := `SELECT count(1) AS total FROM base_from_mysteel_chemical_index WHERE 1=1 `
  473. if condition != "" {
  474. sql += condition
  475. }
  476. //err = o.Raw(sql, pars).QueryRow(&total)
  477. err = global.DEFAULT_DB.Raw(sql, pars...).Scan(&total).Error
  478. return
  479. }
  480. // GetNoMergeIndexByFrequencyPage 获取未合并的指标列表
  481. func (m *BaseFromMysteelChemicalIndex) GetIndexLimitByFrequency(frequency string, limit int) (items []*BaseFromMysteelChemicalIndex, err error) {
  482. //o := orm.NewOrm()
  483. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE frequency=? limit ? `
  484. //_, err = o.Raw(sql, frequency, limit).QueryRows(&items)
  485. err = global.DEFAULT_DB.Raw(sql, frequency, limit).Find(&items).Error
  486. return
  487. }
  488. type GetIndexPageByFrequencyPageReq struct {
  489. Frequency string
  490. EndDate string
  491. StartSize int
  492. PageSize int
  493. TerminalCode string `description:"终端编码"`
  494. }
  495. // GetNoMergeIndexByFrequencyPage 获取未合并的指标列表
  496. func (m *BaseFromMysteelChemicalIndex) GetIndexPageByFrequency(condition string, pars []interface{}, startSize, pageSize int) (items []*BaseFromMysteelChemicalIndex, err error) {
  497. //o := orm.NewOrm()
  498. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 `
  499. if condition != "" {
  500. sql += condition
  501. }
  502. sql += ` LIMIT ?,? `
  503. //_, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
  504. pars = append(pars, startSize, pageSize)
  505. err = global.DEFAULT_DB.Raw(sql, pars...).Find(&items).Error
  506. return
  507. }
  508. func (d *BaseFromMysteelChemicalData) GetMysteelIndexInfoMaxAndMinInfo(indexCode string) (item *EdbInfoMaxAndMinInfo, err error) {
  509. //o := orm.NewOrm()
  510. sql := ` SELECT MIN(data_time) AS min_date,MAX(data_time) AS max_date,MIN(value) AS min_value,MAX(value) AS max_value FROM base_from_mysteel_chemical_data WHERE index_code=? `
  511. //err = o.Raw(sql, indexCode).QueryRow(&item)
  512. err = global.DEFAULT_DB.Raw(sql, indexCode).First(&item).Error
  513. if err != nil {
  514. return
  515. }
  516. // 获取最新值
  517. sql = ` SELECT value AS latest_value FROM base_from_mysteel_chemical_data WHERE index_code=? ORDER BY data_time DESC LIMIT 1 `
  518. //err = o.Raw(sql, indexCode).QueryRow(&latest_value)
  519. var latestValueNull sql2.NullFloat64
  520. err = global.DEFAULT_DB.Raw(sql, indexCode).Scan(&latestValueNull).Error
  521. if err == nil && latestValueNull.Valid {
  522. item.LatestValue = latestValueNull.Float64
  523. }
  524. return
  525. }
  526. func (d *BaseFromMysteelChemicalData) ModifyMysteelIndexMaxAndMinInfo(indexCode string, item *EdbInfoMaxAndMinInfo) (err error) {
  527. //o := orm.NewOrm()
  528. sql := ` UPDATE base_from_mysteel_chemical_index SET start_date=?,end_date=?,end_value=?,modify_time=NOW() WHERE index_code=? `
  529. //_, err = o.Raw(sql, item.MinDate, item.MaxDate, item.LatestValue, indexCode).Exec()
  530. err = global.DEFAULT_DB.Exec(sql, item.MinDate, item.MaxDate, item.LatestValue, indexCode).Error
  531. return
  532. }
  533. // GetIndexByCondition 获取指标
  534. func (m *BaseFromMysteelChemicalIndex) GetIndexByCondition(condition string, pars []interface{}) (items []*BaseFromMysteelChemicalIndex, err error) {
  535. //o := orm.NewOrm()
  536. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 `
  537. if condition != "" {
  538. sql += condition
  539. }
  540. //_, err = o.Raw(sql, pars).QueryRows(&items)
  541. err = global.DEFAULT_DB.Raw(sql, pars...).Find(&items).Error
  542. return
  543. }
  544. type BaseFromMysteelChemicalRecord struct {
  545. BaseFromMysteelChemicalRecordId int64 `gorm:"column:base_from_mysteel_chemical_record_id;primaryKey"`
  546. //BaseFromMysteelChemicalRecordId int64 `orm:"column(base_from_mysteel_chemical_record_id);pk"`
  547. BaseFromMysteelChemicalIndexId int64
  548. OldIndexName string `description:"原始名称"`
  549. NewIndexName string `description:"新的名称"`
  550. CreateTime time.Time `description:"记录创建时间"`
  551. Timestamp int64 `description:"记录创建时间戳"`
  552. }
  553. func (m *BaseFromMysteelChemicalRecord) AddBaseFromMysteelChemicalRecord() (err error) {
  554. //o := orm.NewOrm()
  555. //_, err = o.Insert(m)
  556. err = global.DEFAULT_DB.Create(&m).Error
  557. return
  558. }
  559. type MySteelChemicalApiResp struct {
  560. Code string `json:"code" description:"200成功,其他失败"`
  561. Success bool `json:"success" description:"true 成功,false 失败"`
  562. Timestamp int64 `json:"timestamp" description:"时间戳"`
  563. Message string `json:"message" description:"显示执行信息"`
  564. Data []*MySteelChemicalApiData `json:"data" description:"数据"`
  565. }
  566. type MySteelChemicalApiInfoResp struct {
  567. Code string `json:"code" description:"200成功,其他失败"`
  568. Success bool `json:"success" description:"true 成功,false 失败"`
  569. Timestamp int64 `json:"timestamp" description:"时间戳"`
  570. Message string `json:"message" description:"显示执行信息"`
  571. Data *MySteelChemicalApiInfo `json:"data" description:"数据"`
  572. }
  573. type MySteelChemicalApiInfo struct {
  574. Total int `json:"total" description:"总条数"`
  575. Pages int `json:"pages" description:"总页数"`
  576. List []*MySteelChemicalApiInfoItem `json:"list" description:"数据列表"`
  577. }
  578. type MySteelChemicalApiInfoItem struct {
  579. IndexCode string `json:"INDEX_CODE"`
  580. IndexName string `json:"INDEX_NAME"`
  581. FrequencyName string `json:"FREQUENCY_NAME"`
  582. UnitName string `json:"UNIT_NAME"`
  583. }
  584. type MySteelChemicalApiData struct {
  585. IndexCode string `json:"INDEX_CODE"`
  586. DataList []*MySteelChemicalApiDataList `json:"dataList"`
  587. }
  588. type MySteelChemicalApiDataList struct {
  589. PublishTime int64 `json:"PUBLISH_TIME"`
  590. IndexCode string `json:"INDEX_CODE"`
  591. DataDate string `json:"DATA_DATE"`
  592. DataValue float64 `json:"DATA_VALUE"`
  593. }