base_from_rzd_data.go 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. // @Author gmy 2024/10/21 9:50:00
  2. package models
  3. import (
  4. "errors"
  5. "github.com/beego/beego/v2/client/orm"
  6. )
  7. type BaseFromRzdData struct {
  8. BaseFromRzdDataId int `orm:"column(base_from_rzd_data_id);pk"`
  9. BaseFromRzdIndexId int `orm:"column(base_from_rzd_index_id)"`
  10. CreateTime string `orm:"column(create_time)"`
  11. DataTime string `orm:"column(data_time)"`
  12. IndexCode string `orm:"column(index_code)"`
  13. ModifyTime string `orm:"column(modify_time)"`
  14. Value float64 `orm:"column(value)"`
  15. }
  16. func init() {
  17. orm.RegisterModel(new(BaseFromRzdData))
  18. }
  19. // AddRzdDataList 批量插入数据记录列表
  20. func AddRzdDataList(items []BaseFromRzdData) (err error) {
  21. o := orm.NewOrm()
  22. _, err = o.InsertMulti(len(items), items)
  23. return
  24. }
  25. // GetRzdDataByIndexCodeAndDataTime 根据指标id和数据日期查询数据
  26. func GetRzdDataByIndexCodeAndDataTime(indexCode string, dataTime string) (items []BaseFromRzdData, err error) {
  27. o := orm.NewOrm()
  28. sql := `SELECT * FROM base_from_rzd_data WHERE index_code=? AND data_time=?`
  29. _, err = o.Raw(sql, indexCode, dataTime).QueryRows(&items)
  30. if errors.Is(err, orm.ErrNoRows) {
  31. return nil, nil
  32. }
  33. return
  34. }
  35. // UpdateRzdDataById 根据主键id更新数据
  36. func UpdateRzdDataById(dataId int, value float64) (err error) {
  37. o := orm.NewOrm()
  38. sql := `UPDATE base_from_rzd_data SET value=? WHERE base_from_rzd_data_id=?`
  39. _, err = o.Raw(sql, value, dataId).Exec()
  40. return
  41. }
  42. // GetBaseFromRzdDataByCondition 添加查询
  43. func GetBaseFromRzdDataByCondition(condition string, pars []interface{}) (items []BaseFromRzdData, err error) {
  44. sql := `SELECT * FROM base_from_rzd_data WHERE 1=1 `
  45. o := orm.NewOrm()
  46. if condition != "" {
  47. sql += condition
  48. }
  49. _, err = o.Raw(sql, pars...).QueryRows(&items)
  50. if errors.Is(err, orm.ErrNoRows) {
  51. return nil, nil
  52. }
  53. return
  54. }