eta_index_lib/services/base_from_ths.go

package services

import (
	"encoding/json"
	"errors"
	"eta_gn/eta_index_lib/models"
	"eta_gn/eta_index_lib/models/future_good"
	"eta_gn/eta_index_lib/utils"
	"fmt"
	"github.com/rdlucklib/rdluck_tools/http"
	"github.com/shopspring/decimal"
	"reflect"
)

// EdbDataFromThsInterface 数据类型转为interface
type EdbDataFromThsInterface struct {
	DataVol   int64       `json:"dataVol"`
	Errmsg    string      `json:"errmsg"`
	Errorcode int64       `json:"errorcode"`
	Perf      interface{} `json:"perf"`
	Tables    []struct {
		ID    []string      `json:"id"`
		Time  []string      `json:"time"`
		Value []interface{} `json:"value"`
	} `json:"tables"`
}

func GetEdbDataFromThs(edbCode, startDate, endDate, edbTerminalCode string) (item models.EdbDataFromThs, err error) {
	terminal, err := GetTerminal(utils.DATA_SOURCE_THS, edbTerminalCode)
	if err != nil {
		err = fmt.Errorf("获取同花顺接口配置出错 Err: %s", err)
		return
	}

	if edbTerminalCode == "" {
		// 设置指标与终端关系的缓存
		terminalCodeCacheKey := utils.CACHE_EDB_TERMINAL_CODE_URL + edbCode
		_ = utils.Rc.Put(terminalCodeCacheKey, terminal.TerminalCode, utils.GetTodayLastSecond())
	}

	// 如果没有配置，获取配置的方式是api，那么就走官方接口
	if utils.ThsDataMethod == "" || utils.ThsDataMethod == "api" {
		if terminal.Value == "" {
			err = fmt.Errorf("同花顺接口未配置")
			return
		}
		var token string
		token, err = GetAccessToken(false, terminal.Value)
		if err != nil {
			return
		}
		return getEdbDataFromThsHttp(edbCode, startDate, endDate, terminal.Value, token)
	}

	return getEdbDataFromThsApp(edbCode, startDate, endDate, 0, terminal.ServerUrl)
}

// getEdbDataFromThs 获取同花顺接口数据
func getEdbDataFromThsApp(edbCode, startDate, endDate string, num int, serverUrl string) (item models.EdbDataFromThs, err error) {
	if serverUrl == `` {
		err = errors.New("同花顺接口未配置")
		return
	}
	thsUrl := serverUrl + `edbInfo/ths?EdbCode=%s&StartDate=%s&EndDate=%s`
	thsUrl = fmt.Sprintf(thsUrl, edbCode, startDate, endDate)
	utils.FileLog.Info("thsUrl:" + thsUrl)
	body, err := http.Get(thsUrl)
	utils.FileLog.Info("ths result:" + string(body))
	if err != nil {
		err = errors.New(" Err:" + err.Error() + ";result:" + string(body))
		return
	}
	if string(body) == "null" {
		err = errors.New("同花顺数据获取异常:" + err.Error() + ";result:" + string(body))
		return
	}

	tmpItems := new(EdbDataFromThsInterface)
	err = json.Unmarshal(body, &tmpItems)
	if err != nil {
		err = errors.New("GetEdbDataFromThs json.Unmarshal Err:" + err.Error())
		return
	}
	if tmpItems.Errorcode != 0 {

		//session has expired,please re-login after using the system
		//如果是同花顺登录session失效了，那么就重新请求获取数据
		if tmpItems.Errorcode == -1020 && num == 0 {
			return getEdbDataFromThsApp(edbCode, startDate, endDate, 1, serverUrl)
		}

		err = errors.New(string(body))
		return
	}

	// 因为table里面的value有的时候返回的是string,有的是float64，所以需要用interface来反射取值
	tablesList := make([]models.Tables, 0)
	for _, table := range tmpItems.Tables {
		tableIdList := make([]string, 0)
		tableTimeList := make([]string, 0)
		tableValueList := make([]float64, 0)

		for _, tableId := range table.ID {
			tableIdList = append(tableIdList, tableId)
		}
		for _, tableTime := range table.Time {
			tableTimeList = append(tableTimeList, tableTime)
		}

		//指标数据
		for _, tmpValue := range table.Value {
			var tableValue float64
			if reflect.TypeOf(tmpValue).Kind() == reflect.Float64 {
				tableValue = reflect.ValueOf(tmpValue).Float()
			} else if reflect.TypeOf(tmpValue).Kind() == reflect.String {
				tmpTableValue, tmpErr := decimal.NewFromString(reflect.ValueOf(tmpValue).String())
				if tmpErr != nil {
					err = tmpErr
					return
				}
				tableValue, _ = tmpTableValue.Truncate(4).Float64()
			} else {
				err = errors.New("错误的数据类型" + reflect.TypeOf(tmpValue).String())
				return
			}
			tableValueList = append(tableValueList, tableValue)
		}
		tmpTable := models.Tables{
			ID:    tableIdList,
			Time:  tableTimeList,
			Value: tableValueList,
		}
		tablesList = append(tablesList, tmpTable)
	}
	item = models.EdbDataFromThs{
		DataVol:   tmpItems.DataVol,
		Errmsg:    tmpItems.Errmsg,
		Errorcode: tmpItems.Errorcode,
		Perf:      tmpItems.Perf,
		Tables:    tablesList,
	}
	return item, nil
}

// FutureGoodDataFromThsInterface 同花顺商品数据类型转为interface
type FutureGoodDataFromThsInterface struct {
	Errmsg    string      `json:"errmsg"`
	Errorcode int64       `json:"errorcode"`
	DataVol   int64       `json:"dataVol"`
	Perf      interface{} `json:"perf"`
	Tables    []struct {
		ThsCode string   `json:"thscode"`
		Time    []string `json:"time"`
		Table   struct {
			LastClose      []float64 `json:"lastclose"`
			Open           []float64 `json:"open"`
			High           []float64 `json:"high"`
			Low            []float64 `json:"low"`
			Close          []float64 `json:"close"`
			AvgPrice       []float64 `json:"avgprice"`
			Change         []float64 `json:"change"`
			ChangePer      []float64 `json:"changeper"`
			Volume         []float64 `json:"volume"`
			Amount         []float64 `json:"amount"`
			Hsl            []float64 `json:"hsl"`
			LastSettlement []float64 `json:"lastsettlement"`
			Settlement     []float64 `json:"settlement"`
			ZdSettlement   []float64 `json:"zdsettlement"`
			ZdfSettlement  []float64 `json:"zdfsettlement"`
			Ccl            []float64 `json:"ccl"`
			Ccbd           []float64 `json:"ccbd"`
			Zf             []float64 `json:"zf"`
			Zjlx           []float64 `json:"zjlx"`
			Zjcd           []float64 `json:"zjcd"`
		} `json:"table"`
	} `json:"tables"`
}

func GetFutureGoodDataFromThs(edbCode, startDate, endDate, edbTerminalCode string) (item future_good.FutureGoodDataFromThs, err error) {
	terminal, err := GetFirstTerminal(utils.DATA_SOURCE_THS, edbTerminalCode)
	if err != nil {
		err = fmt.Errorf("获取同花顺接口配置出错 Err: %s", err)
		return
	}
	if terminal.ServerUrl == "" {
		err = fmt.Errorf("同花顺接口未配置")
		return
	}

	if edbTerminalCode == "" {
		terminalCodeCacheKey := utils.CACHE_EDB_TERMINAL_CODE_GOOD_URL + edbCode
		_ = utils.Rc.Put(terminalCodeCacheKey, terminal.TerminalCode, utils.GetTodayLastSecond())
	}

	if utils.ThsDataMethod == "" || utils.ThsDataMethod == "api" { // 生产环境走官方http请求，测试环境走终端
		var token string
		token, err = GetAccessToken(false, terminal.Value)
		if err != nil {
			return
		}
		return getFutureGoodDataFromThsHttp(edbCode, startDate, endDate, terminal.Value, token)
	} else {
		return getFutureGoodDataFromThsApp(edbCode, startDate, endDate, 0, terminal.ServerUrl)
	}
}

// getFutureGoodDataFromThsApp 通过终端获取wind的商品数据
func getFutureGoodDataFromThsApp(edbCode, startDate, endDate string, num int, serverUrl string) (item future_good.FutureGoodDataFromThs, err error) {
	/*if utils.Hz_Ths_Data_Url == `` {
		err = errors.New("同花顺接口未配置")
		return
	}*/
	if serverUrl == "" {
		err = errors.New("同花顺接口未配置")
		return
	}
	thsUrl := serverUrl + `edbInfo/ths/future_good?EdbCode=%s&StartDate=%s&EndDate=%s`
	thsUrl = fmt.Sprintf(thsUrl, edbCode, startDate, endDate)
	utils.FileLog.Info("thsUrl:" + thsUrl)
	body, err := http.Get(thsUrl)
	utils.FileLog.Info("ths result:" + string(body))
	if err != nil {
		err = errors.New(" Err:" + err.Error() + ";result:" + string(body))
		return
	}
	if string(body) == "null" {
		err = errors.New("同花顺数据获取异常:" + err.Error() + ";result:" + string(body))
		return
	}

	tmpItems := new(FutureGoodDataFromThsInterface)
	err = json.Unmarshal(body, &tmpItems)
	if err != nil {
		err = errors.New("GetEdbDataFromThs json.Unmarshal Err:" + err.Error())
		return
	}
	if tmpItems.Errorcode != 0 {

		//session has expired,please re-login after using the system
		//如果是同花顺登录session失效了，那么就重新请求获取数据
		if tmpItems.Errorcode == -1020 && num == 0 {
			return getFutureGoodDataFromThsApp(edbCode, startDate, endDate, 1, serverUrl)
		}

		err = errors.New(string(body))
		return
	}

	if len(tmpItems.Tables) <= 0 {
		return
	}
	table := tmpItems.Tables[0]
	item = future_good.FutureGoodDataFromThs{
		DataVol:   tmpItems.DataVol,
		Errmsg:    tmpItems.Errmsg,
		Errorcode: tmpItems.Errorcode,
		Perf:      tmpItems.Perf,
		Tables: future_good.FutureGoodDataTables{
			Time:       table.Time,
			Open:       table.Table.Open,
			High:       table.Table.High,
			Low:        table.Table.Low,
			Close:      table.Table.Close,
			Volume:     table.Table.Volume,
			Amount:     table.Table.Amount,
			Ccl:        table.Table.Ccl,
			Settlement: table.Table.Settlement,
		},
	}
	return
}

type StockDatas struct {
	Time    string `json:"time"`
	ThsCode string `json:"thscode"`
	//OpenPriceStock   *float64 `json:"ths_open_price_stock"`
	//HighPriceStock   *float64 `json:"ths_high_price_stock"`
	//LowStock         *float64 `json:"ths_low_stock,omitempty"`
	Value *float64
}

type TerminalResponse struct {
	ErrorCode int                      `json:"errorcode"`
	ErrMsg    string                   `json:"errmsg"`
	Data      []map[string]interface{} `json:"data"`
}