mysteel_chemical_index.go 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806
  1. package data_manage
  2. import (
  3. "eta/eta_api/global"
  4. "eta/eta_api/utils"
  5. "fmt"
  6. "gorm.io/gorm"
  7. "strings"
  8. "time"
  9. "github.com/rdlucklib/rdluck_tools/paging"
  10. )
  11. // BaseFromMysteelChemicalIndex 上海钢联指标表
  12. type BaseFromMysteelChemicalIndex struct {
  13. BaseFromMysteelChemicalIndexId int `orm:"column(base_from_mysteel_chemical_index_id);pk" gorm:"primaryKey"`
  14. BaseFromMysteelChemicalClassifyId int `orm:"column(base_from_mysteel_chemical_classify_id)" description:"上海钢联指标分类id"`
  15. IndexCode string `description:"指标编码"`
  16. IndexName string `description:"指标名称"`
  17. Unit string `description:"单位"`
  18. Source string `description:"数据来源"`
  19. Frequency string `description:"频度"`
  20. StartDate string `description:"开始日期"`
  21. EndDate string `description:"结束日期"`
  22. Describe string `description:"指标描述"`
  23. UpdateWeek string `description:"更新周期"`
  24. UpdateTime string `description:"更新时间1"`
  25. UpdateTime2 string `description:"更新时间2"`
  26. SysUserId int `description:"创建人id"`
  27. SysUserRealName string `description:"创建人姓名"`
  28. ModifyTime time.Time `description:"修改时间"`
  29. CreateTime time.Time `description:"创建时间"`
  30. Sort int `description:"排序字段"`
  31. FilePath string `description:"文件"`
  32. MergeFilePath string `description:"合并文件"`
  33. TerminalCode string `description:"终端编码"`
  34. IsStop int `description:"是否停更:1:停更,0:未停更"`
  35. EndValue float64 `description:"指标的最新值"`
  36. IsSupplierStop int `description:"是否供应商停更:1:停更,0:未停更"`
  37. }
  38. type BaseFromMysteelChemicalIndexItem struct {
  39. BaseFromMysteelChemicalIndexId int64 `gorm:"column:base_from_mysteel_chemical_index_id);primaryKey"`
  40. IndexCode string `description:"指标编码"`
  41. IndexName string `description:"指标名称"`
  42. Unit string `description:"单位"`
  43. Frequency string `description:"频度"`
  44. Source string `description:"数据来源"`
  45. StartDate string `description:"开始日期"`
  46. EndDate string `description:"结束日期"`
  47. CreateTime string `description:"创建时间"`
  48. ModifyTime string `description:"修改时间"`
  49. EdbInfoId int `description:"eta指标库的id"`
  50. EdbUniqueCode string `description:"指标库唯一编码"`
  51. EdbClassifyId int `description:"指标库分类ID"`
  52. IsStop int `description:"是否停更:1:停更,0:未停更"`
  53. EdbExist int `description:"指标库是否已添加:0-否;1-是"`
  54. }
  55. func (m *BaseFromMysteelChemicalIndexItem) AfterFind(tx *gorm.DB) (err error) {
  56. m.StartDate = utils.GormDateStrToDateStr(m.StartDate)
  57. m.EndDate = utils.GormDateStrToDateStr(m.EndDate)
  58. m.CreateTime = utils.GormDateStrToDateTimeStr(m.CreateTime)
  59. m.ModifyTime = utils.GormDateStrToDateTimeStr(m.ModifyTime)
  60. return
  61. }
  62. var BaseFromMysteelChemicalIndexCols = struct {
  63. BaseFromMysteelChemicalIndexId string
  64. IndexCode string
  65. IndexName string
  66. Unit string
  67. Frequency string
  68. Source string
  69. StartDate string
  70. EndDate string
  71. CreateTime string
  72. ModifyTime string
  73. }{
  74. BaseFromMysteelChemicalIndexId: "base_from_mysteel_chemical_index_id",
  75. IndexCode: "index_code",
  76. IndexName: "index_name",
  77. Unit: "unit",
  78. Frequency: "frequency",
  79. Source: "source",
  80. StartDate: "start_date",
  81. EndDate: "end_date",
  82. CreateTime: "create_time",
  83. ModifyTime: "modify_time",
  84. }
  85. // Update 更新上海钢联指标基础信息
  86. func (item *BaseFromMysteelChemicalIndex) Update(cols []string) (err error) {
  87. o := global.DbMap[utils.DbNameIndex]
  88. err = o.Select(cols).Updates(item).Error
  89. return
  90. }
  91. func (m *BaseFromMysteelChemicalIndex) GeItemsByCondition(condition string, pars []interface{}, limitSize int) (items []*BaseFromMysteelChemicalIndexItem, err error) {
  92. if pars == nil {
  93. pars = make([]interface{}, 0)
  94. }
  95. //pars = append(pars, utils.DATA_SOURCE_MYSTEEL_CHEMICAL)
  96. pars = utils.ForwardPars(pars, utils.DATA_SOURCE_MYSTEEL_CHEMICAL)
  97. sql := `select a.*,b.edb_info_id,b.unique_code as edb_unique_code,b.classify_id as edb_classify_id FROM base_from_mysteel_chemical_index AS a LEFT JOIN edb_info b on a.index_code=b.edb_code AND b.source=? WHERE 1=1 `
  98. if condition != "" {
  99. sql += condition
  100. }
  101. sql += ` order by a.base_from_mysteel_chemical_index_id desc `
  102. if limitSize > 0 {
  103. sql += " limit ? "
  104. pars = append(pars, limitSize)
  105. }
  106. o := global.DbMap[utils.DbNameIndex]
  107. err = o.Raw(sql, pars...).Find(&items).Error
  108. return
  109. }
  110. // AddBaseFromMysteelChemicalIndex 添加上海钢联指标
  111. func AddBaseFromMysteelChemicalIndex(item *BaseFromMysteelChemicalIndex) (lastId int64, err error) {
  112. o := global.DbMap[utils.DbNameIndex]
  113. err = o.Create(item).Error
  114. if err != nil {
  115. return
  116. }
  117. lastId = int64(item.BaseFromMysteelChemicalIndexId)
  118. return
  119. }
  120. // AddBaseFromMysteelChemicalIndex 添加上海钢联指标
  121. func BatchAddBaseFromMysteelChemicalIndex(items []*BaseFromMysteelChemicalIndex) (lastId int64, err error) {
  122. o := global.DbMap[utils.DbNameIndex]
  123. err = o.CreateInBatches(items, utils.MultiAddNum).Error
  124. return
  125. }
  126. // BaseFromMysteelChemicalData 上海钢联指标数据表
  127. type BaseFromMysteelChemicalData struct {
  128. BaseFromMysteelChemicalDataId int `orm:"column(base_from_mysteel_chemical_data_id);pk" gorm:"primaryKey"`
  129. BaseFromMysteelChemicalIndexId int `orm:"column(base_from_mysteel_chemical_index_id)" description:"上海钢联指标id"`
  130. IndexCode string `description:"指标编码"`
  131. DataTime time.Time `description:"数据日期"`
  132. Value float64 `description:"数据值"`
  133. ModifyTime time.Time `description:"修改时间"`
  134. CreateTime time.Time `description:"创建时间"`
  135. }
  136. // MysteelChemicalFrequency 上海钢联频度
  137. type MysteelChemicalFrequency struct {
  138. Frequency string `description:"频度:1-日度 2-周度 3-月度 4-季度 5-年度 99-无固定频率"`
  139. }
  140. // GetMysteelChemicalIndexByClassifyId 用于分类展示
  141. func GetMysteelChemicalIndexByClassifyId(classifyId int) (items []*BaseFromMysteelChemicalClassifyItems, err error) {
  142. o := global.DbMap[utils.DbNameIndex]
  143. sql := ` SELECT base_from_mysteel_chemical_index_id,base_from_mysteel_chemical_classify_id,index_name AS classify_name,
  144. sys_user_id,sys_user_real_name,sort,index_code
  145. FROM base_from_mysteel_chemical_index where base_from_mysteel_chemical_classify_id=? ORDER BY sort asc, create_time ASC `
  146. err = o.Raw(sql, classifyId).Find(&items).Error
  147. return
  148. }
  149. // MysteelChemicalFrequencyByClassifyId 根据分类id获取上海钢联频度数据列表
  150. func MysteelChemicalFrequencyByClassifyId(classifyId int) (items []*MysteelChemicalFrequency, err error) {
  151. o := global.DbMap[utils.DbNameIndex]
  152. if classifyId == 0 {
  153. sql := ` SELECT frequency FROM base_from_mysteel_chemical_index WHERE frequency != '' GROUP BY frequency ORDER BY frequency ASC `
  154. err = o.Raw(sql).Find(&items).Error
  155. return
  156. } else {
  157. sql := ` SELECT frequency FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id = ? AND frequency != '' GROUP BY frequency ORDER BY frequency ASC `
  158. err = o.Raw(sql, classifyId).Find(&items).Error
  159. return
  160. }
  161. }
  162. // GetMysteelChemicalFrequency 获取上海钢联频度数据列表
  163. func GetMysteelChemicalFrequency(condition string, pars []interface{}) (items []*MysteelChemicalFrequency, err error) {
  164. o := global.DbMap[utils.DbNameIndex]
  165. sql := ` SELECT frequency FROM base_from_mysteel_chemical_index WHERE 1=1 AND frequency != '' `
  166. if condition != "" {
  167. sql += condition
  168. }
  169. sql += ` GROUP BY frequency ORDER BY frequency ASC `
  170. err = o.Raw(sql, pars...).Find(&items).Error
  171. return
  172. }
  173. // MysteelChemicalList 上海钢联指标列表
  174. type MysteelChemicalList struct {
  175. Id int `gorm:"column:base_from_mysteel_chemical_index_id"`
  176. BaseFromMysteelChemicalClassifyId int `gorm:"column:base_from_mysteel_chemical_classify_id" description:"上海钢联指标分类id"`
  177. ParentClassifyId int `description:"上海钢联指标父级分类id"`
  178. IndexCode string `description:"指标编码"`
  179. IndexName string `description:"指标名称"`
  180. UnitName string `gorm:"column:unit"`
  181. UniqueCode string `description:"唯一编码"`
  182. FrequencyName string `gorm:"column:frequency"`
  183. EdbInfoId int `description:"指标库的id"`
  184. UpdateTime string `grom:"column:modify_time"`
  185. IsStop int `description:"是否停更:1:停更,0:未停更"`
  186. IsSupplierStop int `description:"是否供应商停更:1:停更,0:未停更"`
  187. Paging *paging.PagingItem `description:"分页数据" gorm:"-"`
  188. EndDate string `description:"结束日期"`
  189. EndValue float64 `description:"最新值"`
  190. DataList []*MysteelChemicalData `gorm:"-"`
  191. }
  192. func (m *MysteelChemicalList) AfterFind(tx *gorm.DB) (err error) {
  193. m.UpdateTime = utils.GormDateStrToDateTimeStr(m.UpdateTime)
  194. m.EndDate = utils.GormDateStrToDateStr(m.EndDate)
  195. return
  196. }
  197. // MysteelChemicalData 上海钢联数据列表
  198. type MysteelChemicalData struct {
  199. InputValue string `gorm:"column:value" description:"值"`
  200. DataTime string `gorm:"column:data_time" description:"日期"`
  201. }
  202. func (m *MysteelChemicalData) AfterFind(tx *gorm.DB) (err error) {
  203. m.DataTime = utils.GormDateStrToDateStr(m.DataTime)
  204. return
  205. }
  206. // GetMysteelChemicalIndex 根据分类id获取上海钢联频度数据列表
  207. func GetMysteelChemicalIndex(condition string, pars []interface{}) (items []*MysteelChemicalList, err error) {
  208. o := global.DbMap[utils.DbNameIndex]
  209. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 `
  210. if condition != "" {
  211. sql += condition
  212. }
  213. sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC `
  214. err = o.Raw(sql, pars...).Find(&items).Error
  215. return
  216. }
  217. // MysteelChemicalData 上海钢联数据列表
  218. type MysteelChemicalDataItem struct {
  219. InputValue float64 `gorm:"column:value" description:"值"`
  220. DataTime string `gorm:"column:data_time" description:"日期"`
  221. }
  222. func (m *MysteelChemicalDataItem) AfterFind(tx *gorm.DB) (err error) {
  223. m.DataTime = utils.GormDateStrToDateStr(m.DataTime)
  224. return
  225. }
  226. // GetMysteelChemicalIndexData 根据指标code获取上海钢联数据列表
  227. func GetMysteelChemicalIndexData(indexCode string, startSize, pageSize int) (items []*MysteelChemicalDataItem, err error) {
  228. sql := ` SELECT * FROM (
  229. SELECT DISTINCT a.index_code,a.value,a.data_time FROM base_from_mysteel_chemical_data AS a WHERE index_code=?
  230. ORDER BY data_time DESC
  231. )AS t
  232. GROUP BY t.data_time
  233. ORDER BY t.data_time DESC LIMIT ?,? `
  234. sql = utils.ReplaceDriverKeywords(utils.DbDriverName, sql)
  235. o := global.DbMap[utils.DbNameIndex]
  236. err = o.Raw(sql, indexCode, startSize, pageSize).Find(&items).Error
  237. return
  238. }
  239. // GetMysteelChemicalIndexDataCount 根据指标code获取上海钢联数据列表 获取钢联数据总数
  240. func GetMysteelChemicalIndexDataCount(indexCode string) (count int, err error) {
  241. sql := `SELECT COUNT(1) AS count FROM (
  242. SELECT * FROM (
  243. SELECT DISTINCT a.index_code,a.value,a.data_time FROM base_from_mysteel_chemical_data AS a WHERE index_code=?
  244. ORDER BY data_time DESC
  245. )AS t
  246. GROUP BY t.data_time
  247. ORDER BY t.data_time DESC
  248. )AS n `
  249. o := global.DbMap[utils.DbNameIndex]
  250. err = o.Raw(sql, indexCode).Scan(&count).Error
  251. return
  252. }
  253. // GetBaseFromMysteelChemicalIndexByIndexId 根据指标id获取指标信息
  254. func GetBaseFromMysteelChemicalIndexByIndexId(indexId int) (item *BaseFromMysteelChemicalIndex, err error) {
  255. o := global.DbMap[utils.DbNameIndex]
  256. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id=? `
  257. err = o.Raw(sql, indexId).Find(&item).Error
  258. return
  259. }
  260. // GetBaseFromMysteelChemicalIndexByCode 根据指标code获取指标信息
  261. func GetBaseFromMysteelChemicalIndexByCode(indexCode string) (item *BaseFromMysteelChemicalIndex, err error) {
  262. o := global.DbMap[utils.DbNameIndex]
  263. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_code=? `
  264. err = o.Raw(sql, indexCode).Find(&item).Error
  265. return
  266. }
  267. // GetBaseFromMysteelChemicalIndexByCodeList 根据指标code获取指标信息
  268. func GetBaseFromMysteelChemicalIndexByCodeList(indexCodeList []string) (items []*BaseFromMysteelChemicalIndex, err error) {
  269. if len(indexCodeList) <= 0 {
  270. return
  271. }
  272. o := global.DbMap[utils.DbNameIndex]
  273. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE index_code IN (%s) `
  274. holder := make([]string, 0)
  275. for range indexCodeList {
  276. holder = append(holder, "?")
  277. }
  278. sql = fmt.Sprintf(sql, strings.Join(holder, ","))
  279. err = o.Raw(sql, indexCodeList).Find(&items).Error
  280. return
  281. }
  282. // GetBaseFromMysteelChemicalIndexByClassifyIdAndName 根据分类id和指标名名获取指标信息
  283. func GetBaseFromMysteelChemicalIndexByClassifyIdAndName(classifyId int, chartName string) (item *BaseFromMysteelChemicalIndex, err error) {
  284. o := global.DbMap[utils.DbNameIndex]
  285. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id = ? and index_name = ? `
  286. err = o.Raw(sql, classifyId, chartName).Find(&item).Error
  287. return
  288. }
  289. // GetBaseFromMysteelChemicalIndexListByClassifyId 根据指标分类id获取指标列表信息
  290. func GetBaseFromMysteelChemicalIndexListByClassifyId(classifyId int) (items []*BaseFromMysteelChemicalIndex, err error) {
  291. o := global.DbMap[utils.DbNameIndex]
  292. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id=? `
  293. err = o.Raw(sql, classifyId).Find(&items).Error
  294. return
  295. }
  296. // GetBaseFromMysteelChemicalIndexListByClassifyIdList 根据指标分类id集合获取指标列表信息
  297. func GetBaseFromMysteelChemicalIndexListByClassifyIdList(classifyIdList []int) (items []*BaseFromMysteelChemicalIndex, err error) {
  298. num := len(classifyIdList)
  299. if num <= 0 {
  300. return
  301. }
  302. o := global.DbMap[utils.DbNameIndex]
  303. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id IN (` + utils.GetOrmInReplace(num) + `) `
  304. err = o.Raw(sql, classifyIdList).Find(&items).Error
  305. return
  306. }
  307. // GetBaseFromMysteelChemicalDataMaxCount 获取分类下指标的最大数量
  308. func GetBaseFromMysteelChemicalDataMaxCount(classifyId int) (count int, err error) {
  309. o := global.DbMap[utils.DbNameIndex]
  310. sql := `SELECT COALESCE(MAX(t.num), 0) AS count FROM (
  311. SELECT COUNT(1) AS num FROM base_from_mysteel_chemical_index AS a
  312. INNER JOIN base_from_mysteel_chemical_data AS b ON a.base_from_mysteel_chemical_index_id=b.base_from_mysteel_chemical_index_id
  313. WHERE a.base_from_mysteel_chemical_classify_id=?
  314. GROUP BY a.base_from_mysteel_chemical_index_id
  315. )AS t `
  316. err = o.Raw(sql, classifyId).Scan(&count).Error
  317. return
  318. }
  319. // GetBaseFromMysteelChemicalDataTimeByIndexId 根据指标id获取指标数据的日期列表
  320. func GetBaseFromMysteelChemicalDataTimeByIndexId(indexIdList []int) (items []string, err error) {
  321. if len(indexIdList) == 0 {
  322. return
  323. }
  324. o := global.DbMap[utils.DbNameIndex]
  325. sql := ` SELECT DISTINCT data_time FROM base_from_mysteel_chemical_data WHERE base_from_mysteel_chemical_index_id IN (` + utils.GetOrmInReplace(len(indexIdList)) + `) ORDER BY data_time DESC`
  326. err = o.Raw(sql, indexIdList).Find(&items).Error
  327. if err != nil {
  328. return
  329. }
  330. for i := range items {
  331. items[i] = utils.GormDateStrToDateStr(items[i])
  332. }
  333. return
  334. }
  335. // GetMysteelChemicalIndexDataByCode 通过上海钢联指标code获取所有数据列表
  336. func GetMysteelChemicalIndexDataByCode(indexCode string) (items []*MysteelChemicalData, err error) {
  337. sql := ` SELECT * FROM (
  338. SELECT DISTINCT a.index_code,a.value,a.data_time FROM base_from_mysteel_chemical_data AS a WHERE index_code=?
  339. ORDER BY data_time DESC
  340. )AS t
  341. GROUP BY t.data_time
  342. ORDER BY t.data_time DESC `
  343. o := global.DbMap[utils.DbNameIndex]
  344. err = o.Raw(sql, indexCode).Find(&items).Error
  345. return
  346. }
  347. // MoveBaseFromMysteelChemicalIndex 移动上海钢联指标分类
  348. func MoveBaseFromMysteelChemicalIndex(chartInfoId, classifyId int) (err error) {
  349. o := global.DbMap[utils.DbNameIndex]
  350. sql := ` UPDATE base_from_mysteel_chemical_index
  351. SET
  352. base_from_mysteel_chemical_classify_id = ?
  353. WHERE base_from_mysteel_chemical_index_id = ?`
  354. err = o.Exec(sql, classifyId, chartInfoId).Error
  355. return
  356. }
  357. // UpdateBaseFromMysteelChemicalIndexByClassifyId 根据指标id更新排序
  358. func UpdateBaseFromMysteelChemicalIndexByClassifyId(classifyId, nowSort, prevIndexInfoId int, updateSort string) (err error) {
  359. o := global.DbMap[utils.DbNameIndex]
  360. sql := ` update base_from_mysteel_chemical_index set sort = ` + updateSort + ` WHERE base_from_mysteel_chemical_classify_id=? AND `
  361. if prevIndexInfoId > 0 {
  362. sql += ` ( sort > ? or (base_from_mysteel_chemical_index_id > ` + fmt.Sprint(prevIndexInfoId) + ` and sort = ` + fmt.Sprint(nowSort) + `))`
  363. }
  364. err = o.Exec(sql, classifyId, nowSort).Error
  365. return
  366. }
  367. // GetFirstBaseFromMysteelChemicalIndexByClassifyId 获取当前分类下,且排序数相同 的排序第一条的数据
  368. func GetFirstBaseFromMysteelChemicalIndexByClassifyId(classifyId int) (item *BaseFromMysteelChemicalIndex, err error) {
  369. o := global.DbMap[utils.DbNameIndex]
  370. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_classify_id=? order by sort asc,base_from_mysteel_chemical_index_id asc limit 1`
  371. err = o.Raw(sql, classifyId).Find(&item).Error
  372. return
  373. }
  374. // GetMysteelChemicalIndexCount 根据条件获取上海钢联数据
  375. func GetMysteelChemicalIndexCount(condition string, pars []interface{}) (count int, err error) {
  376. o := global.DbMap[utils.DbNameIndex]
  377. sql := ` SELECT COUNT(1) AS count FROM base_from_mysteel_chemical_index WHERE 1=1 `
  378. if condition != "" {
  379. sql += condition
  380. }
  381. sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC `
  382. err = o.Raw(sql, pars...).Scan(&count).Error
  383. return
  384. }
  385. // GetMysteelChemicalIndexList 根据分类id获取上海钢联频度数据列表
  386. func GetMysteelChemicalIndexList(condition string, pars []interface{}, startSize, pageSize int, orderDesc string) (items []*BaseFromMysteelChemicalIndex, err error) {
  387. o := global.DbMap[utils.DbNameIndex]
  388. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 `
  389. if condition != "" {
  390. sql += condition
  391. }
  392. if orderDesc == `` {
  393. orderDesc = ` ASC `
  394. }
  395. sql += ` ORDER BY base_from_mysteel_chemical_index_id `
  396. sql += orderDesc
  397. sql += ` LIMIT ?,? `
  398. pars = append(pars, startSize, pageSize)
  399. err = o.Raw(sql, pars...).Find(&items).Error
  400. return
  401. }
  402. // GetMysteelChemicalIndexListGroupByUserId 根据指标id列表、用户分组获取指标信息
  403. func GetMysteelChemicalIndexListGroupByUserId(edbIdList []string) (items []*BaseFromMysteelChemicalIndex, err error) {
  404. num := len(edbIdList)
  405. if num <= 0 {
  406. return
  407. }
  408. o := global.DbMap[utils.DbNameIndex]
  409. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id in (` + utils.GetOrmInReplace(num) + `) GROUP BY sys_user_id `
  410. err = o.Raw(sql, edbIdList).Find(&items).Error
  411. return
  412. }
  413. // GetMysteelChemicalIndexListByIndexId
  414. // @Description: 根据指标id列表获取列表信息
  415. // @param edbIdList
  416. // @return items
  417. // @return err
  418. func GetMysteelChemicalIndexListByIndexId(edbIdList []string) (items []*BaseFromMysteelChemicalIndex, err error) {
  419. num := len(edbIdList)
  420. if num <= 0 {
  421. return
  422. }
  423. o := global.DbMap[utils.DbNameIndex]
  424. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id in (` + utils.GetOrmInReplace(num) + `) `
  425. err = o.Raw(sql, edbIdList).Find(&items).Error
  426. return
  427. }
  428. // GetMysteelChemicalIndexListByUserId
  429. // @Description: 根据用户id列表获取列表信息
  430. // @param userIdList
  431. // @return items
  432. // @return err
  433. func GetMysteelChemicalIndexListByUserId(userIdList []int) (items []*BaseFromMysteelChemicalIndex, err error) {
  434. num := len(userIdList)
  435. if num <= 0 {
  436. return
  437. }
  438. o := global.DbMap[utils.DbNameIndex]
  439. sql := ` SELECT * FROM base_from_mysteel_chemical_index WHERE sys_user_id in (` + utils.GetOrmInReplace(num) + `) `
  440. err = o.Raw(sql, userIdList).Find(&items).Error
  441. return
  442. }
  443. // ModifyMysteelChemicalIndexUserIdByCodeList 根据指标code列表修改创建人
  444. func ModifyMysteelChemicalIndexUserIdByCodeList(edbIdList []string, userId int, userName string) (err error) {
  445. num := len(edbIdList)
  446. if num <= 0 {
  447. return
  448. }
  449. o := global.DbMap[utils.DbNameIndex]
  450. sql := `UPDATE base_from_mysteel_chemical_index SET sys_user_id=?,sys_user_real_name=? WHERE base_from_mysteel_chemical_index_id in (` + utils.GetOrmInReplace(num) + `) `
  451. err = o.Exec(sql, userId, userName, edbIdList).Error
  452. return
  453. }
  454. // ModifyMysteelChemicalIndexUserIdByOldUserId
  455. // @Description: 根据旧用户id修改新用户id
  456. // @author: Roc
  457. // @datetime 2024-03-25 17:59:08
  458. // @param oldUserId int
  459. // @param userId int
  460. // @param userName string
  461. // @return err error
  462. func ModifyMysteelChemicalIndexUserIdByOldUserId(oldUserIdList []int, userId int, userName string) (err error) {
  463. num := len(oldUserIdList)
  464. if num <= 0 {
  465. return
  466. }
  467. o := global.DbMap[utils.DbNameIndex]
  468. sql := `UPDATE base_from_mysteel_chemical_index SET sys_user_id=?,sys_user_real_name=? WHERE sys_user_id in (` + utils.GetOrmInReplace(num) + `) `
  469. err = o.Exec(sql, userId, userName, oldUserIdList).Error
  470. return
  471. }
  472. // GetMysteelChemicalIndexAdminList 获取所有指标创建人
  473. func GetMysteelChemicalIndexAdminList() (list []int, err error) {
  474. o := global.DbMap[utils.DbNameIndex]
  475. sql := ` SELECT sys_user_id FROM base_from_mysteel_chemical_index GROUP BY sys_user_id `
  476. err = o.Raw(sql).Find(&list).Error
  477. return
  478. }
  479. type TerminalNum struct {
  480. TerminalCode string `description:"terminal_code"`
  481. Num int `description:"num"`
  482. }
  483. // GetMysteelChemicalGroupTerminalNum 获取上海钢联指标的终端分布
  484. func GetMysteelChemicalGroupTerminalNum() (items []*TerminalNum, err error) {
  485. o := global.DbMap[utils.DbNameIndex]
  486. sql := ` SELECT terminal_code,count(1) num FROM base_from_mysteel_chemical_index GROUP BY terminal_code ORDER BY num ASC `
  487. err = o.Raw(sql).Find(&items).Error
  488. return
  489. }
  490. // BaseRefreshEdbInfo
  491. // @Description: 刷新配置的基础指标信息结构体
  492. type BaseRefreshEdbInfo struct {
  493. EdbInfoId int
  494. ClassifyId int `description:"上海钢联指标分类id"`
  495. IndexCode string `description:"指标编码"`
  496. IndexName string `description:"指标名称"`
  497. EndDate string `description:"最新日期"`
  498. EndValue string `description:"最新值"`
  499. SysUserId int `description:"创建人id"`
  500. SysUserRealName string `description:"创建人姓名"`
  501. Frequency string `description:"频度"`
  502. IsStop int `description:"是否停更:1:停更,0:未停更"`
  503. TerminalCode string `description:"终端编码"`
  504. RefreshTime string `description:"刷新时间"`
  505. }
  506. func (obj *BaseRefreshEdbInfo) AfterFind(tx *gorm.DB) (err error) {
  507. obj.EndDate = utils.GormDateStrToDateStr(obj.EndDate)
  508. return
  509. }
  510. // RefreshBaseEdbInfoResp
  511. // @Description: 刷新数据源的数据返回
  512. type RefreshBaseEdbInfoResp struct {
  513. Paging *paging.PagingItem
  514. List []*BaseRefreshEdbInfo
  515. }
  516. // GetMysteelChemicalBaseInfoList
  517. // @Description: 获取上海钢联数据列表
  518. // @author: Roc
  519. // @datetime 2024-01-10 14:28:35
  520. // @param condition string
  521. // @param pars []interface{}
  522. // @param orderBy string
  523. // @param startSize int
  524. // @param pageSize int
  525. // @return total int
  526. // @return items []*BaseRefreshEdbInfo
  527. // @return err error
  528. func GetMysteelChemicalBaseInfoList(condition string, pars []interface{}, orderBy string, startSize, pageSize int) (total int, items []*BaseRefreshEdbInfo, err error) {
  529. o := global.DbMap[utils.DbNameIndex]
  530. // 数量汇总
  531. totalSql := ` SELECT count(1) FROM base_from_mysteel_chemical_index WHERE 1=1 `
  532. if condition != "" {
  533. totalSql += condition
  534. }
  535. err = o.Raw(totalSql, pars...).Scan(&total).Error
  536. if err != nil {
  537. return
  538. }
  539. // 列表数据
  540. sql := ` SELECT base_from_mysteel_chemical_index_id as edb_info_id, base_from_mysteel_chemical_classify_id as classify_id,index_code,index_name,end_date,end_value,sys_user_id,sys_user_real_name,frequency,is_stop,terminal_code FROM base_from_mysteel_chemical_index WHERE 1=1 `
  541. if condition != "" {
  542. sql += condition
  543. }
  544. if orderBy != "" {
  545. sql += ` ORDER BY ` + orderBy
  546. } else {
  547. sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC `
  548. }
  549. sql += ` LIMIT ?,? `
  550. pars = append(pars, startSize, pageSize)
  551. err = o.Raw(sql, pars...).Find(&items).Error
  552. return
  553. }
  554. // ModifyMysteelChemicalUpdateStatus
  555. // @Description: 修改上海钢联数据停更状态
  556. // @author: Roc
  557. // @datetime 2024-01-08 16:23:31
  558. // @param edbIdList []int
  559. // @param indexCodeList []string
  560. // @param isStop int
  561. // @return err error
  562. func ModifyMysteelChemicalUpdateStatus(edbIdList []int, indexCodeList []string, isStop int) (err error) {
  563. idNum := len(edbIdList)
  564. if idNum <= 0 {
  565. return
  566. }
  567. o := global.DbMap[utils.DbNameIndex].Begin()
  568. defer func() {
  569. if err != nil {
  570. _ = o.Rollback()
  571. return
  572. }
  573. _ = o.Commit()
  574. }()
  575. // 更改数据源的更新状态
  576. sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = ? WHERE base_from_mysteel_chemical_index_id IN (` + utils.GetOrmInReplace(idNum) + `) `
  577. err = o.Exec(sql, isStop, edbIdList).Error
  578. if err != nil {
  579. return
  580. }
  581. codeNum := len(indexCodeList)
  582. if codeNum <= 0 {
  583. // 需要通过指标id列表查找code列表
  584. sql := ` SELECT index_code FROM base_from_mysteel_chemical_index WHERE base_from_mysteel_chemical_index_id IN (` + utils.GetOrmInReplace(idNum) + `) `
  585. err = o.Raw(sql, edbIdList).Find(&indexCodeList).Error
  586. if err != nil {
  587. return
  588. }
  589. }
  590. codeNum = len(indexCodeList)
  591. // 查出来的编码是空的话,那么就直接返回了
  592. if codeNum <= 0 {
  593. return
  594. }
  595. // 更改指标的更新状态
  596. sql = ` UPDATE edb_info SET no_update = ?, set_update_time=? WHERE source = ? AND sub_source= ? AND edb_code IN (` + utils.GetOrmInReplace(codeNum) + `) `
  597. err = o.Exec(sql, isStop, time.Now(), utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 0, indexCodeList).Error
  598. if err != nil {
  599. return
  600. }
  601. return
  602. }
  603. // ModifyMysteelChemicalUpdateStatusByEdbInfoId
  604. // @Description: 修改单个上海钢联指标停更状态,同时停更依赖于该指标的计算指标
  605. // @author: Roc
  606. // @datetime 2024-01-08 16:23:31
  607. // @param edbIdList []int
  608. // @param indexCodeList []string
  609. // @param isStop int
  610. // @return err error
  611. func ModifyMysteelChemicalUpdateStatusByEdbInfoId(edbInfoId, isStop int, edbCode string, calculateEdbInfoIds []int) (err error) {
  612. o := global.DbMap[utils.DbNameIndex].Begin()
  613. defer func() {
  614. if err != nil {
  615. _ = o.Rollback()
  616. return
  617. }
  618. _ = o.Commit()
  619. }()
  620. // 更改数据源的更新状态
  621. sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = ? WHERE index_code = ? `
  622. err = o.Exec(sql, isStop, edbCode).Error
  623. if err != nil {
  624. return
  625. }
  626. // 更改指标的更新状态
  627. sql = ` UPDATE edb_info SET no_update = ?, set_update_time=? WHERE source = ? AND sub_source= ? AND edb_info_id=? `
  628. err = o.Exec(sql, isStop, time.Now(), utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 0, edbInfoId).Error
  629. if err != nil {
  630. return
  631. }
  632. if len(calculateEdbInfoIds) > 0 {
  633. // 批量更新相关联的指标ID
  634. sql = ` UPDATE edb_info SET no_update = ?, set_update_time=? WHERE edb_info_id IN (` + utils.GetOrmInReplace(len(calculateEdbInfoIds)) + `) `
  635. err = o.Exec(sql, isStop, time.Now(), calculateEdbInfoIds).Error
  636. if err != nil {
  637. return
  638. }
  639. }
  640. return
  641. }
  642. func ModifyMysteelChemicalUpdateStatusByEdbInfoIds(edbInfoIds []int, isStop int, edbCodes []string, calculateEdbInfoIds []int) (err error) {
  643. o := global.DbMap[utils.DbNameIndex].Begin()
  644. defer func() {
  645. if err != nil {
  646. _ = o.Rollback()
  647. return
  648. }
  649. _ = o.Commit()
  650. }()
  651. // 更改数据源的更新状态
  652. sql := ` UPDATE base_from_mysteel_chemical_index SET is_stop = ? WHERE index_code IN (` + utils.GetOrmInReplace(len(edbCodes)) + `) `
  653. err = o.Exec(sql, isStop, edbCodes).Error
  654. if err != nil {
  655. return
  656. }
  657. // 更改指标的更新状态
  658. sql = ` UPDATE edb_info SET no_update = ?, set_update_time=? WHERE source = ? AND edb_info_id IN (` + utils.GetOrmInReplace(len(edbInfoIds)) + `) `
  659. err = o.Exec(sql, isStop, time.Now(), utils.DATA_SOURCE_MYSTEEL_CHEMICAL, edbInfoIds).Error
  660. if err != nil {
  661. return
  662. }
  663. if len(calculateEdbInfoIds) > 0 {
  664. // 批量更新相关联的指标ID
  665. sql = ` UPDATE edb_info SET no_update = ?, set_update_time=? WHERE edb_info_id IN (` + utils.GetOrmInReplace(len(calculateEdbInfoIds)) + `) `
  666. err = o.Exec(sql, isStop, time.Now(), calculateEdbInfoIds).Error
  667. if err != nil {
  668. return
  669. }
  670. }
  671. return
  672. }
  673. // GetNotIsSupplierStopIndexByCodeList
  674. // @Description: 获取未被供应商停更的指标
  675. // @author: Roc
  676. // @datetime 2024-08-28 18:15:03
  677. // @param codeList []string
  678. // @param isStop int
  679. // @return items []*BaseFromMysteelChemicalIndex
  680. // @return err error
  681. func GetNotIsSupplierStopIndexByCodeList(codeList []string, isStop int) (items []*BaseFromMysteelChemicalIndex, err error) {
  682. num := len(codeList)
  683. if num <= 0 {
  684. return
  685. }
  686. o := global.DbMap[utils.DbNameIndex]
  687. sql := `SELECT * FROM base_from_mysteel_chemical_index WHERE is_supplier_stop = ? AND index_code in (` + utils.GetOrmInReplace(num) + `) `
  688. err = o.Raw(sql, isStop, codeList).Find(&items).Error
  689. return
  690. }
  691. type MysteelChemicalPageListResp struct {
  692. List []*MysteelChemicalList
  693. Paging *paging.PagingItem `description:"分页数据"`
  694. }
  695. // GetNoEdbMysteelChemicalIndexCount 获取未加入指标库的钢联原始指标-分页
  696. func GetNoEdbMysteelChemicalIndexCount(condition string, pars []interface{}) (count int, err error) {
  697. o := global.DbMap[utils.DbNameIndex]
  698. sql := fmt.Sprintf(`SELECT COUNT(1) FROM base_from_mysteel_chemical_index WHERE 1=1 AND index_code NOT IN (
  699. SELECT DISTINCT edb_code FROM edb_info WHERE source = %d
  700. ) `, utils.DATA_SOURCE_MYSTEEL_CHEMICAL)
  701. if condition != "" {
  702. sql += condition
  703. }
  704. err = o.Raw(sql, pars...).Scan(&count).Error
  705. return
  706. }
  707. // GetNoEdbMysteelChemicalIndexPageList 获取未加入指标库的钢联原始指标-分页
  708. func GetNoEdbMysteelChemicalIndexPageList(condition string, pars []interface{}, startSize, pageSize int) (items []*MysteelChemicalList, err error) {
  709. o := global.DbMap[utils.DbNameIndex]
  710. sql := fmt.Sprintf(`SELECT * FROM base_from_mysteel_chemical_index WHERE 1=1 AND index_code NOT IN (
  711. SELECT DISTINCT edb_code FROM edb_info WHERE source = %d
  712. ) `, utils.DATA_SOURCE_MYSTEEL_CHEMICAL)
  713. if condition != "" {
  714. sql += condition
  715. }
  716. sql += ` ORDER BY base_from_mysteel_chemical_index_id ASC LIMIT ?,?`
  717. pars = append(pars, startSize, pageSize)
  718. err = o.Raw(sql, pars...).Find(&items).Error
  719. return
  720. }