Browse Source

修复谷歌数据

longyu 2 năm trước cách đây
mục cha
commit
1f0c06ca67
2 tập tin đã thay đổi với 8 bổ sung0 xóa
  1. 2 0
      models/data_manage/edb_data_base.go
  2. 6 0
      models/data_manage/edb_info.go

+ 2 - 0
models/data_manage/edb_data_base.go

@@ -63,6 +63,8 @@ func GetEdbDataTableName(source int) (tableName string) {
 		tableName = "edb_data_lt"
 	case utils.DATA_SOURCE_COAL:
 		tableName = "edb_data_coal"
+	case utils.DATA_SOURCE_GOOGLE_TRAVEL:
+		tableName = "edb_data_google_travel"
 	default:
 		tableName = ""
 		errors.New("无效的渠道:" + strconv.Itoa(source))

+ 6 - 0
models/data_manage/edb_info.go

@@ -1,8 +1,10 @@
 package data_manage
 
 import (
+	"errors"
 	"fmt"
 	"github.com/beego/beego/v2/client/orm"
+	"strconv"
 	"time"
 )
 
@@ -91,6 +93,10 @@ func GetEdbInfoMaxAndMinInfo(source int, edbCode string) (item *EdbInfoMaxAndMin
 	o := orm.NewOrmUsingDB("data")
 	sql := ``
 	tableName := GetEdbDataTableName(source)
+	if tableName == "" {
+		err = errors.New("无效的表名称:source:" + strconv.Itoa(source))
+		return nil, err
+	}
 	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 %s WHERE edb_code=? `
 	sql = fmt.Sprintf(sql, tableName)
 	err = o.Raw(sql, edbCode).QueryRow(&item)