chart_series.go 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251
  1. package data_manage
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "github.com/beego/beego/v2/client/orm"
  6. "time"
  7. )
  8. type ChartSeries struct {
  9. ChartSeriesId int `orm:"column(chart_series_id);pk"`
  10. SeriesName string `description:"系列名称"`
  11. SeriesNameEn string `description:"系列英文名称"`
  12. ChartInfoId int `description:"图表ID"`
  13. ChartStyle string `description:"图表类型"`
  14. ChartColor string `description:"颜色"`
  15. ChartWidth int `description:"线条大小"`
  16. IsPoint int `description:"是否用数据点展示(0 否,1是)"`
  17. IsNumber int `description:"是否用数值展示(0 否,1是)"`
  18. IsAxis int `description:"1:左轴,0:右轴"`
  19. MaxData float64 `description:"上限"`
  20. MinData float64 `description:"下限"`
  21. IsOrder bool `description:"true:正序,false:逆序"`
  22. CreateTime time.Time `description:"创建时间"`
  23. ModifyTime time.Time `description:"修改时间"`
  24. }
  25. func (c *ChartSeries) TableName() string {
  26. return "chart_series"
  27. }
  28. func GetChartSeriesByChartInfoId(chartInfoId int) (items []*ChartSeries, err error) {
  29. o := orm.NewOrmUsingDB("data")
  30. sql := "SELECT * FROM chart_series WHERE chart_info_id = ?"
  31. _, err = o.Raw(sql, chartInfoId).QueryRows(&items)
  32. return
  33. }
  34. // EditChartSeriesAndEdbMapping
  35. func EditChartSeriesAndEdbMapping(extraConfigStr string, chartInfoId int) (err error) {
  36. o := orm.NewOrmUsingDB("data")
  37. to, err := o.Begin()
  38. if err != nil {
  39. return
  40. }
  41. defer func() {
  42. if err != nil {
  43. _ = to.Rollback()
  44. } else {
  45. _ = to.Commit()
  46. }
  47. }()
  48. // 获取已经存在的系列
  49. series, err := GetChartSeriesByChartInfoId(chartInfoId)
  50. if err != nil {
  51. return
  52. }
  53. //整理成系列map
  54. seriesMap := make(map[int]struct{})
  55. seriesDeleteMap := make(map[int]struct{})
  56. for _, v := range series {
  57. seriesMap[v.ChartSeriesId] = struct{}{}
  58. seriesDeleteMap[v.ChartSeriesId] = struct{}{}
  59. }
  60. var sectionExtraConfig ChartSectionAllExtraConf
  61. err = json.Unmarshal([]byte(extraConfigStr), &sectionExtraConfig)
  62. if err != nil {
  63. err = fmt.Errorf("截面组合图配置异常")
  64. return
  65. }
  66. // 删除所有的指标映射
  67. _, err = o.Raw("DELETE FROM chart_series_edb_mapping WHERE chart_info_id = ?", chartInfoId).Exec()
  68. if err != nil {
  69. return
  70. }
  71. seriesList := sectionExtraConfig.SeriesList
  72. for _, v := range seriesList {
  73. seriesId := v.ChartSeriesId
  74. tmp := &ChartSeries{
  75. ChartSeriesId: v.ChartSeriesId,
  76. SeriesName: v.SeriesName,
  77. SeriesNameEn: v.SeriesNameEn,
  78. ChartInfoId: chartInfoId,
  79. ChartStyle: v.ChartStyle,
  80. ChartColor: v.ChartColor,
  81. ChartWidth: v.ChartWidth,
  82. IsPoint: v.IsPoint,
  83. IsNumber: v.IsNumber,
  84. IsAxis: v.IsAxis,
  85. MaxData: v.MaxData,
  86. MinData: v.MinData,
  87. ModifyTime: time.Now(),
  88. }
  89. if _, ok := seriesMap[v.ChartSeriesId]; !ok {
  90. if seriesId > 0 {
  91. err = fmt.Errorf("系列ID错误")
  92. return
  93. }
  94. //新增
  95. delete(seriesDeleteMap, v.ChartSeriesId)
  96. tmp.CreateTime = time.Now()
  97. seriesIdTmp, e := to.Insert(tmp)
  98. if e != nil {
  99. err = fmt.Errorf("AddChartSeries Err:" + e.Error())
  100. return
  101. }
  102. seriesId = int(seriesIdTmp)
  103. } else {
  104. _, e := to.Update(tmp)
  105. if e != nil {
  106. err = fmt.Errorf("UpdateChartSeries Err:" + e.Error())
  107. return
  108. }
  109. }
  110. addSeriesEdbList := make([]*ChartSeriesEdbMapping, 0)
  111. for _, edbItem := range v.EdbInfoList {
  112. dateConfStrByte, e := json.Marshal(edbItem.DateConf)
  113. if e != nil {
  114. err = e
  115. return
  116. }
  117. dateConfStr := string(dateConfStrByte)
  118. edbTmp := &ChartSeriesEdbMapping{
  119. ChartSeriesId: seriesId, //todo 系列ID
  120. ChartInfoId: chartInfoId, //todo 表图ID
  121. EdbInfoId: edbItem.EdbInfoId,
  122. //EdbAliasName: "",
  123. //EdbAliasNameEn: "",
  124. DateConfName: edbItem.DateConfName,
  125. DateConf: dateConfStr,
  126. DateConfType: edbItem.DateConfType,
  127. CreateTime: time.Now(),
  128. ModifyTime: time.Now(),
  129. }
  130. addSeriesEdbList = append(addSeriesEdbList, edbTmp)
  131. }
  132. if len(addSeriesEdbList) > 0 {
  133. _, err = to.InsertMulti(len(addSeriesEdbList), addSeriesEdbList)
  134. if err != nil {
  135. err = fmt.Errorf("AddChartSeries Err:" + err.Error())
  136. return
  137. }
  138. }
  139. }
  140. return
  141. }
  142. func AddChartSeriesAndEdbMapping(extraConfigStr string, chartInfoId int) (err error) {
  143. o := orm.NewOrmUsingDB("data")
  144. to, err := o.Begin()
  145. if err != nil {
  146. return
  147. }
  148. defer func() {
  149. if err != nil {
  150. _ = to.Rollback()
  151. } else {
  152. _ = to.Commit()
  153. }
  154. }()
  155. var sectionExtraConfig ChartSectionAllExtraConf
  156. err = json.Unmarshal([]byte(extraConfigStr), &sectionExtraConfig)
  157. if err != nil {
  158. err = fmt.Errorf("截面组合图配置异常")
  159. return
  160. }
  161. seriesList := sectionExtraConfig.SeriesList
  162. for _, v := range seriesList {
  163. tmp := &ChartSeries{
  164. SeriesName: v.SeriesName,
  165. SeriesNameEn: v.SeriesNameEn,
  166. ChartInfoId: chartInfoId,
  167. ChartStyle: v.ChartStyle,
  168. ChartColor: v.ChartColor,
  169. ChartWidth: v.ChartWidth,
  170. IsPoint: v.IsPoint,
  171. IsNumber: v.IsNumber,
  172. IsAxis: v.IsAxis,
  173. MaxData: v.MaxData,
  174. MinData: v.MinData,
  175. IsOrder: false, //todo 是否排序
  176. CreateTime: time.Now(),
  177. ModifyTime: time.Now(),
  178. }
  179. seriesIdTmp, e := to.Insert(tmp)
  180. if e != nil {
  181. err = fmt.Errorf("AddChartSeries Err:" + e.Error())
  182. return
  183. }
  184. seriesId := int(seriesIdTmp)
  185. addSeriesEdbList := make([]*ChartSeriesEdbMapping, 0)
  186. for _, edbItem := range v.EdbInfoList {
  187. dateConfStrByte, e := json.Marshal(edbItem.DateConf)
  188. if e != nil {
  189. err = e
  190. return
  191. }
  192. dateConfStr := string(dateConfStrByte)
  193. edbTmp := &ChartSeriesEdbMapping{
  194. ChartSeriesId: seriesId, //todo 系列ID
  195. ChartInfoId: chartInfoId, //todo 表图ID
  196. EdbInfoId: edbItem.EdbInfoId,
  197. //EdbAliasName: "",
  198. //EdbAliasNameEn: "",
  199. DateConfName: edbItem.DateConfName,
  200. DateConfType: edbItem.DateConfType,
  201. DateConf: dateConfStr,
  202. CreateTime: time.Now(),
  203. ModifyTime: time.Now(),
  204. }
  205. addSeriesEdbList = append(addSeriesEdbList, edbTmp)
  206. }
  207. if len(addSeriesEdbList) > 0 {
  208. _, e = to.InsertMulti(len(addSeriesEdbList), addSeriesEdbList)
  209. if e != nil {
  210. err = fmt.Errorf("AddChartSeries Err:" + e.Error())
  211. return
  212. }
  213. }
  214. }
  215. return
  216. }
  217. func DeleteChartSeriesAndEdbMapping(chartInfoId int) (err error) {
  218. o := orm.NewOrmUsingDB("data")
  219. to, err := o.Begin()
  220. if err != nil {
  221. return
  222. }
  223. defer func() {
  224. if err != nil {
  225. _ = to.Rollback()
  226. } else {
  227. _ = to.Commit()
  228. }
  229. }()
  230. sql := ` DELETE FROM chart_series WHERE chart_info_id=? `
  231. _, err = to.Raw(sql, chartInfoId).Exec()
  232. if err != nil {
  233. return
  234. }
  235. sql = ` DELETE FROM chart_series_edb_mapping WHERE chart_info_id=? `
  236. _, err = to.Raw(sql, chartInfoId).Exec()
  237. return
  238. }