base_from_mysteel_chemical.go 24 KB

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