소스 검색

睿姿得数据爬取

gmy 5 달 전
부모
커밋
6dd54c39af
2개의 변경된 파일7개의 추가작업 그리고 4개의 파일을 삭제
  1. 6 3
      models/base_from_rzd_data.go
  2. 1 1
      models/base_from_rzd_index.go

+ 6 - 3
models/base_from_rzd_data.go

@@ -28,12 +28,15 @@ func AddRzdDataList(items []BaseFromRzdData) (err error) {
 }
 
 // GetRzdDataByIndexCodeAndDataTime 根据指标id和数据日期查询数据
-func GetRzdDataByIndexCodeAndDataTime(indexCode string, dataTime string) (items []BaseFromRzdData, err error) {
+func GetRzdDataByIndexCodeAndDataTime(indexCode string, dataTime string) (items BaseFromRzdData, err error) {
 	o := orm.NewOrm()
 	sql := `SELECT * FROM base_from_rzd_data WHERE index_code=? AND data_time=?`
-	_, err = o.Raw(sql, indexCode, dataTime).QueryRows(&items)
+	err = o.Raw(sql, indexCode, dataTime).QueryRow(&items) // 使用 QueryRow
 	if errors.Is(err, orm.ErrNoRows) {
-		return nil, nil
+		return items, nil
+	}
+	if err != nil {
+		return items, err
 	}
 	return
 }

+ 1 - 1
models/base_from_rzd_index.go

@@ -11,7 +11,7 @@ type BaseFromRzdIndex struct {
 	BaseFromRzdIndexId   int    `orm:"column(base_from_rzd_index_id);pk"`
 	CreateTime           string `orm:"column(create_time)"`
 	ModifyTime           string `orm:"column(modify_time)"`
-	BaseFromLyClassifyId int    `orm:"column(base_from_ly_classify_id)"`
+	BaseFromLyClassifyId int    `orm:"column(base_from_rzd_classify_id)"`
 	IndexCode            string `orm:"column(index_code)"`
 	IndexName            string `orm:"column(index_name)"`
 	Frequency            string `orm:"column(frequency)"`