|
@@ -1,1019 +0,0 @@
|
|
|
-package trade_analysis
|
|
|
-
|
|
|
-import (
|
|
|
- "eta/eta_api/models/data_manage"
|
|
|
- tradeAnalysisModel "eta/eta_api/models/data_manage/trade_analysis"
|
|
|
- "eta/eta_api/services/data"
|
|
|
- "eta/eta_api/utils"
|
|
|
- "fmt"
|
|
|
- "sort"
|
|
|
- "strings"
|
|
|
- "time"
|
|
|
-)
|
|
|
-
|
|
|
-// FormatCompanyTradeData2EdbMappings [公司-合约加总]转为指标数据
|
|
|
-func FormatCompanyTradeData2EdbMappings(companyTradeData []*tradeAnalysisModel.ContractCompanyTradeData, tradeType, dateType, dateTypeNum int, startDate, endDate string, chartEdbList []*data_manage.ChartSaveItem) (edbMappings []*data_manage.ChartEdbInfoMapping, chartName string, err error) {
|
|
|
- edbMappings = make([]*data_manage.ChartEdbInfoMapping, 0)
|
|
|
- if dateType <= 0 {
|
|
|
- dateType = utils.DateTypeOneMonth
|
|
|
- }
|
|
|
-
|
|
|
- // 期货公司名称作为标识进行匹配
|
|
|
- edbMap := make(map[string]*data_manage.ChartSaveItem)
|
|
|
- if len(chartEdbList) > 0 {
|
|
|
- for _, v := range chartEdbList {
|
|
|
- edbMap[v.UniqueFlag] = v
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- for k, v := range companyTradeData {
|
|
|
- mapping := new(data_manage.ChartEdbInfoMapping)
|
|
|
- mapping.EdbName = v.CompanyName
|
|
|
- mapping.EdbNameEn = v.CompanyName
|
|
|
- mapping.EdbAliasName = v.CompanyName
|
|
|
- mapping.EdbAliasNameEn = v.CompanyName
|
|
|
- mapping.Frequency = "日度"
|
|
|
- mapping.FrequencyEn = data.GetFrequencyEn(mapping.Frequency)
|
|
|
- mapping.SourceName = utils.SourceNameTradeAnalysis
|
|
|
- mapping.Source = utils.CHART_SOURCE_TRADE_ANALYSIS_PROCESS
|
|
|
- mapping.IsAxis = 1
|
|
|
- mapping.EdbInfoType = 1
|
|
|
- mapping.StartDate = v.StartDate.Format(utils.FormatDate)
|
|
|
- mapping.EndDate = v.EndDate.Format(utils.FormatDate)
|
|
|
- mapping.ConvertUnit = tradeAnalysisModel.WarehouseDefaultUnit // 固定单位
|
|
|
- mapping.UniqueFlag = v.CompanyName // 期货公司名称作为每条曲线的唯一标识
|
|
|
-
|
|
|
- // 有配置那么取配置中的图例名称和左右轴
|
|
|
- edbConf := edbMap[mapping.UniqueFlag]
|
|
|
- if edbConf != nil {
|
|
|
- mapping.EdbName = edbConf.EdbAliasName
|
|
|
- mapping.EdbNameEn = edbConf.EdbAliasName
|
|
|
- mapping.EdbAliasName = edbConf.EdbAliasName
|
|
|
- mapping.EdbAliasNameEn = edbConf.EdbAliasName
|
|
|
- mapping.IsAxis = edbConf.IsAxis
|
|
|
- }
|
|
|
-
|
|
|
- // 根据参数取日期范围
|
|
|
- var startTime, endTime time.Time
|
|
|
- if dateType > 0 {
|
|
|
- st, ed := utils.GetDateByDateTypeV2(dateType, startDate, endDate, dateTypeNum, time.Time{})
|
|
|
- if st != "" {
|
|
|
- startTime, _ = time.ParseInLocation(utils.FormatDate, st, time.Local)
|
|
|
- }
|
|
|
- if startTime.IsZero() {
|
|
|
- startTime = v.StartDate
|
|
|
- }
|
|
|
- if ed != "" {
|
|
|
- endTime, _ = time.ParseInLocation(utils.FormatDate, ed, time.Local)
|
|
|
- }
|
|
|
- if endTime.IsZero() {
|
|
|
- endTime = time.Now()
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 指标数据和最值
|
|
|
- edbData := make([]*data_manage.EdbDataList, 0)
|
|
|
- var minData, maxData float64
|
|
|
- var setMinMax bool
|
|
|
- for _, dv := range v.DataList {
|
|
|
- if dv.Date.Before(startTime) || dv.Date.After(endTime) {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- // 交易方向
|
|
|
- var (
|
|
|
- val float64
|
|
|
- hasVal bool
|
|
|
- )
|
|
|
- if tradeType == tradeAnalysisModel.WarehouseBuyChartType {
|
|
|
- if dv.BuyValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- hasVal = true
|
|
|
- val = float64(dv.BuyVal)
|
|
|
- }
|
|
|
- if tradeType == tradeAnalysisModel.WarehouseSoldChartType {
|
|
|
- if dv.SoldValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- hasVal = true
|
|
|
- val = float64(dv.SoldVal)
|
|
|
- }
|
|
|
- if tradeType == tradeAnalysisModel.WarehousePureBuyChartType {
|
|
|
- if dv.PureBuyValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- hasVal = true
|
|
|
- val = float64(dv.PureBuyVal)
|
|
|
- }
|
|
|
- if !hasVal {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- if !setMinMax {
|
|
|
- minData = val
|
|
|
- maxData = val
|
|
|
- setMinMax = true
|
|
|
- }
|
|
|
- if val < minData {
|
|
|
- minData = val
|
|
|
- }
|
|
|
- if val > maxData {
|
|
|
- maxData = val
|
|
|
- }
|
|
|
- edbData = append(edbData, &data_manage.EdbDataList{
|
|
|
- DataTime: dv.Date.Format(utils.FormatDate),
|
|
|
- DataTimestamp: dv.Date.UnixNano() / 1e6,
|
|
|
- Value: val,
|
|
|
- })
|
|
|
- }
|
|
|
- mapping.MinData = minData
|
|
|
- mapping.MaxData = maxData
|
|
|
- mapping.DataList = edbData
|
|
|
- edbMappings = append(edbMappings, mapping)
|
|
|
-
|
|
|
- // 图表默认名称
|
|
|
- if k == 0 {
|
|
|
- chartName += strings.ReplaceAll(v.ClassifyType, ",", "")
|
|
|
- }
|
|
|
- chartName += v.CompanyName
|
|
|
- }
|
|
|
-
|
|
|
- // 图表名称后缀
|
|
|
- chartName += tradeAnalysisModel.WarehouseTypeSuffixNames[tradeType]
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-func GetWarehouseTradeData(exchange, classifyName string, contracts, companies []string, predictRatio float64) (companyTradeData []*tradeAnalysisModel.ContractCompanyTradeData, err error) {
|
|
|
- // 获取合约持仓数据
|
|
|
- contractTradeData, lastBuyVal, lastSoldVal, e := GetContractCompanyTradeData(exchange, []string{classifyName}, contracts, companies, time.Time{}, time.Time{})
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取合约-持仓数据失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 填充[合约-公司]预估数据, 并根据[公司-多合约]分组, [公司]算作一个指标, 指标值为[多个合约]的计算加总
|
|
|
- companyContracts := make(map[string][]*tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- for _, v := range contractTradeData {
|
|
|
- td, fd, ed, e := PredictingTradeData(v.DataList, lastBuyVal[v.ClassifyType], lastSoldVal[v.ClassifyType], predictRatio)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("数据补全失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- v.DataList = td
|
|
|
- v.StartDate = fd
|
|
|
- v.EndDate = ed
|
|
|
-
|
|
|
- if companyContracts[v.CompanyName] == nil {
|
|
|
- companyContracts[v.CompanyName] = make([]*tradeAnalysisModel.ContractCompanyTradeData, 0)
|
|
|
- }
|
|
|
- companyContracts[v.CompanyName] = append(companyContracts[v.CompanyName], v)
|
|
|
- }
|
|
|
-
|
|
|
- // 以[公司]为组, 计算合约加总
|
|
|
- mussyTradeData := make(map[string]*tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- for k, v := range companyContracts {
|
|
|
- companyData := new(tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- companyData.CompanyName = k
|
|
|
- companyData.DataList = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- contractArr := make([]string, 0)
|
|
|
-
|
|
|
- // 合约加总
|
|
|
- sumDateData := make(map[time.Time]*tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- for _, vv := range v {
|
|
|
- contractArr = append(contractArr, vv.ClassifyType)
|
|
|
- for _, dv := range vv.DataList {
|
|
|
- if sumDateData[dv.Date] == nil {
|
|
|
- sumDateData[dv.Date] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- sumDateData[dv.Date].Date = dv.Date
|
|
|
- }
|
|
|
- // 数据类型以第一个非零值为准, 只处理多空和净多, 变化就不管了
|
|
|
- if sumDateData[dv.Date].BuyValType == tradeAnalysisModel.TradeDataTypeNull && dv.BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].BuyValType = dv.BuyValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].BuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.BuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].BuyValType = dv.BuyValType
|
|
|
- }
|
|
|
- if dv.BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].BuyVal += dv.BuyVal
|
|
|
- }
|
|
|
- // 空单
|
|
|
- if sumDateData[dv.Date].SoldValType == tradeAnalysisModel.TradeDataTypeNull && dv.SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].SoldValType = dv.SoldValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].SoldValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.SoldValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].SoldValType = dv.SoldValType
|
|
|
- }
|
|
|
- if dv.SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].SoldVal += dv.SoldVal
|
|
|
- }
|
|
|
- // 净多单
|
|
|
- if sumDateData[dv.Date].PureBuyValType == tradeAnalysisModel.TradeDataTypeNull && dv.PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].PureBuyValType = dv.PureBuyValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].PureBuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.PureBuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].PureBuyValType = dv.PureBuyValType
|
|
|
- }
|
|
|
- if dv.PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].PureBuyVal += dv.PureBuyVal
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 多个合约比对开始结束时间
|
|
|
- if companyData.StartDate.IsZero() {
|
|
|
- companyData.StartDate = vv.StartDate
|
|
|
- }
|
|
|
- if vv.StartDate.Before(companyData.StartDate) {
|
|
|
- companyData.StartDate = vv.StartDate
|
|
|
- }
|
|
|
- if companyData.EndDate.IsZero() {
|
|
|
- companyData.EndDate = vv.EndDate
|
|
|
- }
|
|
|
- if vv.EndDate.Before(companyData.EndDate) {
|
|
|
- companyData.EndDate = vv.EndDate
|
|
|
- }
|
|
|
- }
|
|
|
- for _, sv := range sumDateData {
|
|
|
- companyData.DataList = append(companyData.DataList, sv)
|
|
|
- }
|
|
|
- sort.Slice(companyData.DataList, func(i, j int) bool {
|
|
|
- return companyData.DataList[i].Date.Before(companyData.DataList[j].Date)
|
|
|
- })
|
|
|
- companyData.ClassifyType = strings.Join(contractArr, ",")
|
|
|
- mussyTradeData[k] = companyData
|
|
|
- }
|
|
|
-
|
|
|
- // 数据根据公司排序, 不然会随机乱
|
|
|
- companyTradeData = make([]*tradeAnalysisModel.ContractCompanyTradeData, 0)
|
|
|
- for _, v := range companies {
|
|
|
- // 没数据也需要加进去, 不然edbList会少
|
|
|
- if mussyTradeData[v] == nil {
|
|
|
- companyData := new(tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- companyData.CompanyName = v
|
|
|
- companyData.DataList = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- companyTradeData = append(companyTradeData, companyData)
|
|
|
- continue
|
|
|
- }
|
|
|
- companyTradeData = append(companyTradeData, mussyTradeData[v])
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// PredictingTradeData 根据数据库中的多空数据填充预估数据
|
|
|
-func PredictingTradeData(originData []*tradeAnalysisModel.ContractCompanyTradeDataList, lastBuyDateVal, lastSoldDateVal map[time.Time]int, predictRatio float64) (newData []*tradeAnalysisModel.ContractCompanyTradeDataList, firstDate, endDate time.Time, err error) {
|
|
|
- // 测试用的验证数据
|
|
|
- //lastBuyDateVal, lastSoldDateVal = make(map[time.Time]int), make(map[time.Time]int)
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 16, 0, 0, 0, 0, time.Local)] = 4602
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 17, 0, 0, 0, 0, time.Local)] = 5116
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 18, 0, 0, 0, 0, time.Local)] = 5130
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 19, 0, 0, 0, 0, time.Local)] = 5354
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 22, 0, 0, 0, 0, time.Local)] = 5916
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 23, 0, 0, 0, 0, time.Local)] = 6524
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 26, 0, 0, 0, 0, time.Local)] = 6575
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 29, 0, 0, 0, 0, time.Local)] = 7461
|
|
|
- //lastBuyDateVal[time.Date(2024, 7, 30, 0, 0, 0, 0, time.Local)] = 8488
|
|
|
- //
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 11, 0, 0, 0, 0, time.Local)] = 5467
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 12, 0, 0, 0, 0, time.Local)] = 5248
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 15, 0, 0, 0, 0, time.Local)] = 5102
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 16, 0, 0, 0, 0, time.Local)] = 4771
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 23, 0, 0, 0, 0, time.Local)] = 5989
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 26, 0, 0, 0, 0, time.Local)] = 6745
|
|
|
- //lastSoldDateVal[time.Date(2024, 7, 30, 0, 0, 0, 0, time.Local)] = 7272
|
|
|
- //
|
|
|
- //originData = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- //originData = append(originData, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 10, 0, 0, 0, 0, time.Local),
|
|
|
- // BuyVal: 14324,
|
|
|
- // BuyValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // BuyChange: -1107,
|
|
|
- // BuyChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // SoldVal: 0,
|
|
|
- // SoldValType: tradeAnalysisModel.TradeDataTypeNull,
|
|
|
- // SoldChange: 0,
|
|
|
- // SoldChangeType: tradeAnalysisModel.TradeDataTypeNull,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 11, 0, 0, 0, 0, time.Local),
|
|
|
- // BuyVal: 14280,
|
|
|
- // BuyValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // BuyChange: -44,
|
|
|
- // BuyChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 12, 0, 0, 0, 0, time.Local),
|
|
|
- // BuyVal: 14214,
|
|
|
- // BuyValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // BuyChange: -66,
|
|
|
- // BuyChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 15, 0, 0, 0, 0, time.Local),
|
|
|
- // BuyVal: 14269,
|
|
|
- // BuyValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // BuyChange: 55,
|
|
|
- // BuyChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 17, 0, 0, 0, 0, time.Local),
|
|
|
- // SoldVal: 5254,
|
|
|
- // SoldValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // SoldChange: 708,
|
|
|
- // SoldChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 18, 0, 0, 0, 0, time.Local),
|
|
|
- // SoldVal: 6595,
|
|
|
- // SoldValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // SoldChange: 1341,
|
|
|
- // SoldChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 19, 0, 0, 0, 0, time.Local),
|
|
|
- // SoldVal: 5938,
|
|
|
- // SoldValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // SoldChange: -657,
|
|
|
- // SoldChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 22, 0, 0, 0, 0, time.Local),
|
|
|
- // SoldVal: 6131,
|
|
|
- // SoldValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // SoldChange: 193,
|
|
|
- // SoldChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //}, &tradeAnalysisModel.ContractCompanyTradeDataList{
|
|
|
- // Date: time.Date(2024, 7, 29, 0, 0, 0, 0, time.Local),
|
|
|
- // SoldVal: 6679,
|
|
|
- // SoldValType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- // SoldChange: 312,
|
|
|
- // SoldChangeType: tradeAnalysisModel.TradeDataTypeOrigin,
|
|
|
- //})
|
|
|
-
|
|
|
- if len(originData) == 0 {
|
|
|
- return
|
|
|
- }
|
|
|
- if predictRatio < 0 || predictRatio > 1 {
|
|
|
- err = fmt.Errorf("估计参数不在0-1之间")
|
|
|
- return
|
|
|
- }
|
|
|
- sort.Slice(originData, func(i, j int) bool {
|
|
|
- return originData[i].Date.Before(originData[j].Date)
|
|
|
- })
|
|
|
- dateVal := make(map[time.Time]*tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- for _, v := range originData {
|
|
|
- dateVal[v.Date] = v
|
|
|
- }
|
|
|
-
|
|
|
- // 生成开始日期-1d(可能会往前面推算一天)至结束日期间的交易日, 以交易日为时间序列遍历
|
|
|
- tradeDays := utils.GetTradingDays(originData[0].Date.AddDate(0, 0, -1), originData[len(originData)-1].Date)
|
|
|
- for k, v := range tradeDays {
|
|
|
- // T日多空均无的情况
|
|
|
- //bothLast := false
|
|
|
- if dateVal[v] == nil {
|
|
|
- // T-1和T+1[原始数据]均无值, 那么T日无数据
|
|
|
- hasPrev, hasNext := false, false
|
|
|
- if k-1 >= 0 {
|
|
|
- hasPrev = true
|
|
|
- }
|
|
|
- if k+1 <= len(tradeDays)-1 {
|
|
|
- hasNext = true
|
|
|
- }
|
|
|
- if !hasPrev && !hasNext {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- // T+1有值, 优先从T+1推, 然后继续走下面计算净多单的逻辑
|
|
|
- if hasNext {
|
|
|
- nextDay := tradeDays[k+1]
|
|
|
- if dateVal[nextDay] != nil {
|
|
|
- // T+1有多/空及多空变化, 且是原始数据, 那么推出数据并在map中新加一日数据
|
|
|
- if dateVal[nextDay].BuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dateVal[nextDay].BuyChangeType == tradeAnalysisModel.TradeDataTypeOrigin {
|
|
|
- if _, ok := dateVal[v]; !ok {
|
|
|
- dateVal[v] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- dateVal[v].Date = v
|
|
|
- }
|
|
|
- dateVal[v].BuyVal = dateVal[nextDay].BuyVal - dateVal[nextDay].BuyChange
|
|
|
- dateVal[v].BuyValType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- }
|
|
|
- if dateVal[nextDay].SoldValType == tradeAnalysisModel.TradeDataTypeOrigin && dateVal[nextDay].SoldChangeType == tradeAnalysisModel.TradeDataTypeOrigin {
|
|
|
- if _, ok := dateVal[v]; !ok {
|
|
|
- dateVal[v] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- dateVal[v].Date = v
|
|
|
- }
|
|
|
- dateVal[v].SoldVal = dateVal[nextDay].SoldVal - dateVal[nextDay].SoldChange
|
|
|
- dateVal[v].SoldValType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // T+1没推出来而T-1有值, 那么T多空均取末位, 计算净多单
|
|
|
- _, has := dateVal[v]
|
|
|
- if hasPrev && !has {
|
|
|
- sv, sok := lastSoldDateVal[v]
|
|
|
- bv, bok := lastBuyDateVal[v]
|
|
|
- if !sok && !bok {
|
|
|
- continue
|
|
|
- }
|
|
|
- dateVal[v] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- dateVal[v].Date = v
|
|
|
- if sok {
|
|
|
- dateVal[v].SoldVal = int(predictRatio*float64(sv) + 0.5)
|
|
|
- dateVal[v].SoldValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- if bok {
|
|
|
- dateVal[v].BuyVal = int(predictRatio*float64(bv) + 0.5)
|
|
|
- dateVal[v].BuyValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- if dateVal[v].BuyValType > tradeAnalysisModel.TradeDataTypeNull && dateVal[v].SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- dateVal[v].PureBuyVal = dateVal[v].BuyVal - dateVal[v].SoldVal
|
|
|
- dateVal[v].PureBuyValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- continue
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 多空均有的情况下计算净多单
|
|
|
- if dateVal[v].BuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dateVal[v].SoldValType == tradeAnalysisModel.TradeDataTypeOrigin {
|
|
|
- dateVal[v].PureBuyVal = dateVal[v].BuyVal - dateVal[v].SoldVal
|
|
|
- dateVal[v].PureBuyValType = tradeAnalysisModel.TradeDataTypeOrigin // 原始值算出来的也作原始值
|
|
|
- }
|
|
|
-
|
|
|
- // 仅有多单, 空单取末位, 计算净多单
|
|
|
- if dateVal[v].BuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dateVal[v].SoldValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- if sv, ok := lastSoldDateVal[v]; ok {
|
|
|
- dateVal[v].SoldVal = int(predictRatio*float64(sv) + 0.5) // 估计参数*末位值, 向上取整
|
|
|
- dateVal[v].SoldValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- dateVal[v].PureBuyVal = dateVal[v].BuyVal - dateVal[v].SoldVal
|
|
|
- dateVal[v].PureBuyValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 仅有空单, 多单取末位, 计算净多单
|
|
|
- if dateVal[v].SoldValType == tradeAnalysisModel.TradeDataTypeOrigin && dateVal[v].BuyValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- if sv, ok := lastBuyDateVal[v]; ok {
|
|
|
- dateVal[v].BuyVal = int(predictRatio*float64(sv) + 0.5)
|
|
|
- dateVal[v].BuyValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- dateVal[v].PureBuyVal = dateVal[v].BuyVal - dateVal[v].SoldVal
|
|
|
- dateVal[v].PureBuyValType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 二次遍历, 计算与T-1的变化值
|
|
|
- for k, v := range tradeDays {
|
|
|
- // 无T/T-1数据, 忽略
|
|
|
- if dateVal[v] == nil {
|
|
|
- continue
|
|
|
- }
|
|
|
- if k-1 < 0 {
|
|
|
- continue
|
|
|
- }
|
|
|
- beforeDay := tradeDays[k-1]
|
|
|
- if dateVal[beforeDay] == nil {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- // 多单变化
|
|
|
- if dateVal[v].BuyChangeType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- if dateVal[v].BuyValType > tradeAnalysisModel.TradeDataTypeNull && dateVal[beforeDay].BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- dateVal[v].BuyChange = dateVal[v].BuyVal - dateVal[beforeDay].BuyVal
|
|
|
- // 如果当日多单或者前日多单是估计值, 那么多单变化也为估计值
|
|
|
- if dateVal[v].BuyValType == tradeAnalysisModel.TradeDataTypeCalculate || dateVal[beforeDay].BuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- dateVal[v].BuyChangeType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 空单变化
|
|
|
- if dateVal[v].SoldChangeType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- if dateVal[v].SoldValType > tradeAnalysisModel.TradeDataTypeNull && dateVal[beforeDay].SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- dateVal[v].SoldChange = dateVal[v].SoldVal - dateVal[beforeDay].SoldVal
|
|
|
- // 如果当日空单或者前日空单是估计值, 那么空单变化也为估计值
|
|
|
- if dateVal[v].SoldValType == tradeAnalysisModel.TradeDataTypeCalculate || dateVal[beforeDay].SoldValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- dateVal[v].SoldChangeType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 净多变化
|
|
|
- if dateVal[v].PureBuyChangeType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- if dateVal[v].PureBuyValType > tradeAnalysisModel.TradeDataTypeNull && dateVal[beforeDay].PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- dateVal[v].PureBuyChange = dateVal[v].PureBuyVal - dateVal[beforeDay].PureBuyVal
|
|
|
- dateVal[v].PureBuyChangeType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- // 如果当日净多单或者前日净多单是估计值, 那么净多单变化也为估计值
|
|
|
- if dateVal[v].PureBuyValType == tradeAnalysisModel.TradeDataTypeCalculate || dateVal[beforeDay].PureBuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- dateVal[v].PureBuyChangeType = tradeAnalysisModel.TradeDataTypeCalculate
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 重新遍历map, 生成数据序列并排序
|
|
|
- newData = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- for _, v := range dateVal {
|
|
|
- if v.BuyValType == tradeAnalysisModel.TradeDataTypeNull && v.SoldValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- newData = append(newData, v)
|
|
|
- }
|
|
|
- sort.Slice(newData, func(i, j int) bool {
|
|
|
- return newData[i].Date.Before(newData[j].Date)
|
|
|
- })
|
|
|
- if len(newData) > 0 {
|
|
|
- firstDate = newData[0].Date
|
|
|
- endDate = newData[len(newData)-1].Date
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// GetTopContractRank 获取TOP20根据成交量的合约排名
|
|
|
-func GetTopContractRank(exchange string, classifyNames []string, dataDate time.Time) (items []*tradeAnalysisModel.ContractTopRankData, err error) {
|
|
|
- // 郑商所/广期所查询方式不一样
|
|
|
- var tradeAnalysis TradeAnalysisInterface
|
|
|
- switch exchange {
|
|
|
- case tradeAnalysisModel.TradeExchangeZhengzhou:
|
|
|
- tradeAnalysis = &ZhengzhouTradeAnalysis{}
|
|
|
- case tradeAnalysisModel.TradeExchangeGuangzhou:
|
|
|
- tradeAnalysis = &GuangzhouTradeAnalysis{}
|
|
|
- default:
|
|
|
- tradeAnalysis = &BaseTradeAnalysis{}
|
|
|
- }
|
|
|
-
|
|
|
- // 郑商所-需要把所选品种转为实际合约进行后续的查询
|
|
|
- if exchange == tradeAnalysisModel.TradeExchangeZhengzhou {
|
|
|
- classifies, e := GetZhengzhouContractsByClassifyNames(classifyNames)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取郑商所实际合约失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- classifyNames = classifies
|
|
|
- }
|
|
|
-
|
|
|
- // 获取多单/空单原始数据
|
|
|
- rankData, e := tradeAnalysis.GetContractTopRankData(exchange, classifyNames, dataDate)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取多空单原始数据失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- items = make([]*tradeAnalysisModel.ContractTopRankData, 0)
|
|
|
- for _, v := range rankData {
|
|
|
- v.Exchange = exchange
|
|
|
- // 郑商所-这里注意把查出来的品种和合约赋值,不然后续是乱的
|
|
|
- if v.Exchange == tradeAnalysisModel.TradeExchangeZhengzhou {
|
|
|
- v.ClassifyType = v.ClassifyName
|
|
|
- v.ClassifyName = GetZhengzhouClassifyName(v.ClassifyName)
|
|
|
- }
|
|
|
- items = append(items, v)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// GetTableTradeData 获取多空分析表格持仓数据
|
|
|
-func GetTableTradeData(exchange string, classifyName string, contracts []string, companyName string, predictRatio float64, startDate, endDate time.Time, contractType int) (companyTradeData []*tradeAnalysisModel.ContractCompanyTradeData, err error) {
|
|
|
- companyTradeData = make([]*tradeAnalysisModel.ContractCompanyTradeData, 0)
|
|
|
-
|
|
|
- // 获取合约持仓数据
|
|
|
- contractTradeData, lastBuyVal, lastSoldVal, e := GetContractCompanyTradeData(exchange, []string{classifyName}, contracts, []string{companyName}, startDate, endDate)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取合约-持仓数据失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 填充[合约-公司]预估数据, 并根据[公司-多合约]分组, [公司]算作一个指标, 指标值为[多个合约]的计算加总
|
|
|
- companyContracts := make(map[string][]*tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- for _, v := range contractTradeData {
|
|
|
- td, fd, ed, e := PredictingTradeData(v.DataList, lastBuyVal[v.ClassifyType], lastSoldVal[v.ClassifyType], predictRatio)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("数据补全失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- v.DataList = td
|
|
|
- v.StartDate = fd
|
|
|
- v.EndDate = ed
|
|
|
-
|
|
|
- // 合约类型参数不为合约加总时, 每个合约算一行数据
|
|
|
- if contractType != tradeAnalysisModel.ContractQueryTypeTotal {
|
|
|
- companyTradeData = append(companyTradeData, v)
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- // 往下计算合约加总
|
|
|
- if companyContracts[v.CompanyName] == nil {
|
|
|
- companyContracts[v.CompanyName] = make([]*tradeAnalysisModel.ContractCompanyTradeData, 0)
|
|
|
- }
|
|
|
- companyContracts[v.CompanyName] = append(companyContracts[v.CompanyName], v)
|
|
|
- }
|
|
|
-
|
|
|
- // 类型为合约加总才往下合并
|
|
|
- if contractType != tradeAnalysisModel.ContractQueryTypeTotal {
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 以[公司]为组, 计算合约加总
|
|
|
- for k, v := range companyContracts {
|
|
|
- companyData := new(tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- companyData.CompanyName = k
|
|
|
- companyData.DataList = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- contractArr := make([]string, 0)
|
|
|
-
|
|
|
- // 合约加总
|
|
|
- sumDateData := make(map[time.Time]*tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- for _, vv := range v {
|
|
|
- contractArr = append(contractArr, vv.ClassifyType)
|
|
|
- for _, dv := range vv.DataList {
|
|
|
- if sumDateData[dv.Date] == nil {
|
|
|
- sumDateData[dv.Date] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- sumDateData[dv.Date].Date = dv.Date
|
|
|
- }
|
|
|
- // 数据类型以第一个非零值为准, 只处理多空和净多, 变化就不管了
|
|
|
- if sumDateData[dv.Date].BuyValType == tradeAnalysisModel.TradeDataTypeNull && dv.BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].BuyValType = dv.BuyValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].BuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.BuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].BuyValType = dv.BuyValType
|
|
|
- }
|
|
|
- if dv.BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].BuyVal += dv.BuyVal
|
|
|
- }
|
|
|
- // 空单
|
|
|
- if sumDateData[dv.Date].SoldValType == tradeAnalysisModel.TradeDataTypeNull && dv.SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].SoldValType = dv.SoldValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].SoldValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.SoldValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].SoldValType = dv.SoldValType
|
|
|
- }
|
|
|
- if dv.SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].SoldVal += dv.SoldVal
|
|
|
- }
|
|
|
- // 净多单
|
|
|
- if sumDateData[dv.Date].PureBuyValType == tradeAnalysisModel.TradeDataTypeNull && dv.PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].PureBuyValType = dv.PureBuyValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].PureBuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.PureBuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].PureBuyValType = dv.PureBuyValType
|
|
|
- }
|
|
|
- if dv.PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].PureBuyVal += dv.PureBuyVal
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 多个合约比对开始结束时间
|
|
|
- if companyData.StartDate.IsZero() {
|
|
|
- companyData.StartDate = vv.StartDate
|
|
|
- }
|
|
|
- if vv.StartDate.Before(companyData.StartDate) {
|
|
|
- companyData.StartDate = vv.StartDate
|
|
|
- }
|
|
|
- if companyData.EndDate.IsZero() {
|
|
|
- companyData.EndDate = vv.EndDate
|
|
|
- }
|
|
|
- if vv.EndDate.Before(companyData.EndDate) {
|
|
|
- companyData.EndDate = vv.EndDate
|
|
|
- }
|
|
|
- }
|
|
|
- for _, sv := range sumDateData {
|
|
|
- companyData.DataList = append(companyData.DataList, sv)
|
|
|
- }
|
|
|
- sort.Slice(companyData.DataList, func(i, j int) bool {
|
|
|
- return companyData.DataList[i].Date.Before(companyData.DataList[j].Date)
|
|
|
- })
|
|
|
- //companyData.ClassifyType = strings.Join(contractArr, ",")
|
|
|
- companyData.Exchange = exchange
|
|
|
- companyData.CompanyName = k
|
|
|
- companyData.ClassifyType = classifyName
|
|
|
- companyTradeData = append(companyTradeData, companyData)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// GetCorrelationTableTradeData 获取相关性表格持仓数据
|
|
|
-func GetCorrelationTableTradeData(exchange string, classifyName string, contracts, companies []string, predictRatio float64, contractType int) (companyTradeData []*tradeAnalysisModel.ContractCompanyTradeData, err error) {
|
|
|
- companyTradeData = make([]*tradeAnalysisModel.ContractCompanyTradeData, 0)
|
|
|
-
|
|
|
- // 获取合约持仓数据
|
|
|
- contractTradeData, lastBuyVal, lastSoldVal, e := GetContractCompanyTradeData(exchange, []string{classifyName}, contracts, companies, time.Time{}, time.Time{})
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取合约-持仓数据失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 填充[合约-公司]预估数据, 并根据[公司-多合约]分组, [公司]算作一个指标, 指标值为[多个合约]的计算加总
|
|
|
- companyContracts := make(map[string][]*tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- for _, v := range contractTradeData {
|
|
|
- td, fd, ed, e := PredictingTradeData(v.DataList, lastBuyVal[v.ClassifyType], lastSoldVal[v.ClassifyType], predictRatio)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("数据补全失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- v.DataList = td
|
|
|
- v.StartDate = fd
|
|
|
- v.EndDate = ed
|
|
|
-
|
|
|
- // 合约类型参数不为合约加总时, 每个合约算一行数据
|
|
|
- if contractType != tradeAnalysisModel.ContractQueryTypeTotal {
|
|
|
- companyTradeData = append(companyTradeData, v)
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- // 往下计算合约加总
|
|
|
- if companyContracts[v.CompanyName] == nil {
|
|
|
- companyContracts[v.CompanyName] = make([]*tradeAnalysisModel.ContractCompanyTradeData, 0)
|
|
|
- }
|
|
|
- companyContracts[v.CompanyName] = append(companyContracts[v.CompanyName], v)
|
|
|
- }
|
|
|
-
|
|
|
- // 类型为合约加总才往下合并
|
|
|
- if contractType != tradeAnalysisModel.ContractQueryTypeTotal {
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // 以[公司]为组, 计算合约加总
|
|
|
- for k, v := range companyContracts {
|
|
|
- companyData := new(tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- companyData.CompanyName = k
|
|
|
- companyData.DataList = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- contractArr := make([]string, 0)
|
|
|
-
|
|
|
- // 合约加总
|
|
|
- sumDateData := make(map[time.Time]*tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- for _, vv := range v {
|
|
|
- contractArr = append(contractArr, vv.ClassifyType)
|
|
|
- for _, dv := range vv.DataList {
|
|
|
- if sumDateData[dv.Date] == nil {
|
|
|
- sumDateData[dv.Date] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- sumDateData[dv.Date].Date = dv.Date
|
|
|
- }
|
|
|
- // 数据类型以第一个非零值为准, 只处理多空和净多, 变化就不管了
|
|
|
- if sumDateData[dv.Date].BuyValType == tradeAnalysisModel.TradeDataTypeNull && dv.BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].BuyValType = dv.BuyValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].BuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.BuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].BuyValType = dv.BuyValType
|
|
|
- }
|
|
|
- if dv.BuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].BuyVal += dv.BuyVal
|
|
|
- }
|
|
|
- // 空单
|
|
|
- if sumDateData[dv.Date].SoldValType == tradeAnalysisModel.TradeDataTypeNull && dv.SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].SoldValType = dv.SoldValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].SoldValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.SoldValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].SoldValType = dv.SoldValType
|
|
|
- }
|
|
|
- if dv.SoldValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].SoldVal += dv.SoldVal
|
|
|
- }
|
|
|
- // 净多单
|
|
|
- if sumDateData[dv.Date].PureBuyValType == tradeAnalysisModel.TradeDataTypeNull && dv.PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].PureBuyValType = dv.PureBuyValType
|
|
|
- }
|
|
|
- if sumDateData[dv.Date].PureBuyValType == tradeAnalysisModel.TradeDataTypeOrigin && dv.PureBuyValType == tradeAnalysisModel.TradeDataTypeCalculate {
|
|
|
- sumDateData[dv.Date].PureBuyValType = dv.PureBuyValType
|
|
|
- }
|
|
|
- if dv.PureBuyValType > tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- sumDateData[dv.Date].PureBuyVal += dv.PureBuyVal
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 多个合约比对开始结束时间
|
|
|
- if companyData.StartDate.IsZero() {
|
|
|
- companyData.StartDate = vv.StartDate
|
|
|
- }
|
|
|
- if vv.StartDate.Before(companyData.StartDate) {
|
|
|
- companyData.StartDate = vv.StartDate
|
|
|
- }
|
|
|
- if companyData.EndDate.IsZero() {
|
|
|
- companyData.EndDate = vv.EndDate
|
|
|
- }
|
|
|
- if vv.EndDate.Before(companyData.EndDate) {
|
|
|
- companyData.EndDate = vv.EndDate
|
|
|
- }
|
|
|
- }
|
|
|
- for _, sv := range sumDateData {
|
|
|
- companyData.DataList = append(companyData.DataList, sv)
|
|
|
- }
|
|
|
- sort.Slice(companyData.DataList, func(i, j int) bool {
|
|
|
- return companyData.DataList[i].Date.Before(companyData.DataList[j].Date)
|
|
|
- })
|
|
|
- //companyData.ClassifyType = strings.Join(contractArr, ",")
|
|
|
- companyData.IsTotal = true
|
|
|
- companyData.Exchange = exchange
|
|
|
- companyData.CompanyName = k
|
|
|
- companyData.ClassifyName = classifyName
|
|
|
- companyData.ClassifyType = classifyName
|
|
|
- companyTradeData = append(companyTradeData, companyData)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// TransTradeData2EdbData 持仓数据转为指标数据
|
|
|
-func TransTradeData2EdbData(tradeData []*tradeAnalysisModel.ContractCompanyTradeData, contractPosition int) (edbData []*tradeAnalysisModel.ContractCompanyTradeEdb, edbDataMap []map[time.Time]int, err error) {
|
|
|
- if len(tradeData) == 0 {
|
|
|
- return
|
|
|
- }
|
|
|
- edbData = make([]*tradeAnalysisModel.ContractCompanyTradeEdb, 0)
|
|
|
- edbDataMap = make([]map[time.Time]int, 0)
|
|
|
-
|
|
|
- for _, v := range tradeData {
|
|
|
- newEdb := new(tradeAnalysisModel.ContractCompanyTradeEdb)
|
|
|
- newEdb.Exchange = v.Exchange
|
|
|
- newEdb.ClassifyName = v.ClassifyName
|
|
|
- newEdb.ClassifyType = v.ClassifyType
|
|
|
- newEdb.CompanyName = v.CompanyName
|
|
|
- newEdb.IsTotal = v.IsTotal
|
|
|
- newEdb.ContractPosition = contractPosition
|
|
|
- newEdb.StartDate = v.StartDate
|
|
|
- newEdb.EndDate = v.EndDate
|
|
|
- newEdb.DataList = make([]*tradeAnalysisModel.ContractCompanyTradeEdbData, 0)
|
|
|
- dataMap := make(map[time.Time]int)
|
|
|
- for _, d := range v.DataList {
|
|
|
- var vd int
|
|
|
- switch contractPosition {
|
|
|
- case tradeAnalysisModel.ContractPositionBuy:
|
|
|
- if d.BuyValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- vd = d.BuyVal
|
|
|
- case tradeAnalysisModel.ContractPositionSold:
|
|
|
- if d.SoldValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- vd = d.SoldVal
|
|
|
- case tradeAnalysisModel.ContractPositionPureBuy:
|
|
|
- if d.PureBuyValType == tradeAnalysisModel.TradeDataTypeNull {
|
|
|
- continue
|
|
|
- }
|
|
|
- vd = d.PureBuyVal
|
|
|
- default:
|
|
|
- continue
|
|
|
- }
|
|
|
- newEdb.DataList = append(newEdb.DataList, &tradeAnalysisModel.ContractCompanyTradeEdbData{
|
|
|
- DataTime: d.Date,
|
|
|
- Val: vd,
|
|
|
- })
|
|
|
- dataMap[d.Date] = vd
|
|
|
- }
|
|
|
- edbData = append(edbData, newEdb)
|
|
|
- edbDataMap = append(edbDataMap, dataMap)
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// GetContractCompanyTradeData 获取合约持仓数据
|
|
|
-func GetContractCompanyTradeData(exchange string, classifyNames, contracts, companies []string, startDate, endDate time.Time) (contractTradeData map[string]*tradeAnalysisModel.ContractCompanyTradeData, lastBuyVal, lastSoldVal map[string]map[time.Time]int, err error) {
|
|
|
- // 各原始数据表期货公司名称不一致
|
|
|
- companyMap := make(map[string]string)
|
|
|
- {
|
|
|
- ob := new(tradeAnalysisModel.TradeFuturesCompany)
|
|
|
- list, e := ob.GetItemsByCondition(``, make([]interface{}, 0), []string{}, "")
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取期货公司名称失败: %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- switch exchange {
|
|
|
- case "zhengzhou":
|
|
|
- for _, v := range list {
|
|
|
- companyMap[v.CompanyName] = v.ZhengzhouName
|
|
|
- }
|
|
|
- case "dalian":
|
|
|
- for _, v := range list {
|
|
|
- companyMap[v.CompanyName] = v.DalianName
|
|
|
- }
|
|
|
- case "shanghai":
|
|
|
- for _, v := range list {
|
|
|
- companyMap[v.CompanyName] = v.ShanghaiName
|
|
|
- }
|
|
|
- case "cffex":
|
|
|
- for _, v := range list {
|
|
|
- companyMap[v.CompanyName] = v.CffexName
|
|
|
- }
|
|
|
- case "ine":
|
|
|
- for _, v := range list {
|
|
|
- companyMap[v.CompanyName] = v.IneName
|
|
|
- }
|
|
|
- case "guangzhou":
|
|
|
- for _, v := range list {
|
|
|
- companyMap[v.CompanyName] = v.GuangzhouName
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- var queryCompanies []string
|
|
|
- for _, v := range companies {
|
|
|
- if v == tradeAnalysisModel.TradeFuturesCompanyTop20 {
|
|
|
- queryCompanies = append(queryCompanies, tradeAnalysisModel.TradeFuturesCompanyTop20)
|
|
|
- continue
|
|
|
- }
|
|
|
- companyName, ok := companyMap[v]
|
|
|
- if !ok {
|
|
|
- utils.FileLog.Info(fmt.Sprintf("交易所%s公司名称映射不存在: %s", exchange, v))
|
|
|
- continue
|
|
|
- }
|
|
|
- queryCompanies = append(queryCompanies, companyName)
|
|
|
- }
|
|
|
-
|
|
|
- // 郑商所/广期所查询方式不一样
|
|
|
- var tradeAnalysis TradeAnalysisInterface
|
|
|
- switch exchange {
|
|
|
- case tradeAnalysisModel.TradeExchangeZhengzhou:
|
|
|
- tradeAnalysis = &ZhengzhouTradeAnalysis{}
|
|
|
- case tradeAnalysisModel.TradeExchangeGuangzhou:
|
|
|
- tradeAnalysis = &GuangzhouTradeAnalysis{}
|
|
|
- default:
|
|
|
- tradeAnalysis = &BaseTradeAnalysis{}
|
|
|
- }
|
|
|
-
|
|
|
- // 获取多单/空单原始数据
|
|
|
- originList, _, lastOriginList, e := tradeAnalysis.GetTradeDataByContracts(exchange, classifyNames, contracts, queryCompanies, startDate, endDate)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取多空单原始数据失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
-
|
|
|
- // [合约-期货公司]数据分组
|
|
|
- contractTradeData = make(map[string]*tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- {
|
|
|
- keyDateData := make(map[string]*tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- keyDateDataExist := make(map[string]bool)
|
|
|
- for _, v := range originList {
|
|
|
- companyName := v.CompanyName
|
|
|
-
|
|
|
- k := fmt.Sprintf("%s-%s", v.ClassifyType, companyName)
|
|
|
- if contractTradeData[k] == nil {
|
|
|
- contractTradeData[k] = new(tradeAnalysisModel.ContractCompanyTradeData)
|
|
|
- contractTradeData[k].Exchange = exchange
|
|
|
- contractTradeData[k].CompanyName = companyName
|
|
|
- contractTradeData[k].ClassifyName = v.ClassifyName
|
|
|
- contractTradeData[k].ClassifyType = v.ClassifyType
|
|
|
- contractTradeData[k].DataList = make([]*tradeAnalysisModel.ContractCompanyTradeDataList, 0)
|
|
|
- }
|
|
|
-
|
|
|
- kd := fmt.Sprintf("%s-%s", k, v.DataTime.Format(utils.FormatDate))
|
|
|
- if keyDateData[kd] == nil {
|
|
|
- keyDateData[kd] = new(tradeAnalysisModel.ContractCompanyTradeDataList)
|
|
|
- keyDateData[kd].Date = v.DataTime
|
|
|
- }
|
|
|
- if v.ValType == 1 {
|
|
|
- keyDateData[kd].BuyVal = v.Val
|
|
|
- keyDateData[kd].BuyValType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- keyDateData[kd].BuyChange = v.ValChange
|
|
|
- keyDateData[kd].BuyChangeType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- }
|
|
|
- if v.ValType == 2 {
|
|
|
- keyDateData[kd].SoldVal = v.Val
|
|
|
- keyDateData[kd].SoldValType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- keyDateData[kd].SoldChange = v.ValChange
|
|
|
- keyDateData[kd].SoldChangeType = tradeAnalysisModel.TradeDataTypeOrigin
|
|
|
- }
|
|
|
- if !keyDateDataExist[kd] {
|
|
|
- contractTradeData[k].DataList = append(contractTradeData[k].DataList, keyDateData[kd])
|
|
|
- keyDateDataExist[kd] = true
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- // 合约的[日期-末位值]
|
|
|
- lastBuyVal = make(map[string]map[time.Time]int)
|
|
|
- lastSoldVal = make(map[string]map[time.Time]int)
|
|
|
- {
|
|
|
- for _, v := range lastOriginList {
|
|
|
- if v.ValType == 1 {
|
|
|
- if lastBuyVal[v.ClassifyType] == nil {
|
|
|
- lastBuyVal[v.ClassifyType] = make(map[time.Time]int)
|
|
|
- }
|
|
|
- lastBuyVal[v.ClassifyType][v.DataTime] = v.Val
|
|
|
- continue
|
|
|
- }
|
|
|
- if lastSoldVal[v.ClassifyType] == nil {
|
|
|
- lastSoldVal[v.ClassifyType] = make(map[time.Time]int)
|
|
|
- }
|
|
|
- lastSoldVal[v.ClassifyType][v.DataTime] = v.Val
|
|
|
- }
|
|
|
- }
|
|
|
- return
|
|
|
-}
|
|
|
-
|
|
|
-// GetTradeClassifyNewestDataTime 获取数据最新日期
|
|
|
-func GetTradeClassifyNewestDataTime(exchange string, classifyNames []string) (dataTime time.Time, err error) {
|
|
|
- var tradeAnalysis TradeAnalysisInterface
|
|
|
- switch exchange {
|
|
|
- case tradeAnalysisModel.TradeExchangeZhengzhou:
|
|
|
- tradeAnalysis = &ZhengzhouTradeAnalysis{}
|
|
|
- case tradeAnalysisModel.TradeExchangeGuangzhou:
|
|
|
- tradeAnalysis = &GuangzhouTradeAnalysis{}
|
|
|
- default:
|
|
|
- tradeAnalysis = &BaseTradeAnalysis{}
|
|
|
- }
|
|
|
- if exchange == tradeAnalysisModel.TradeExchangeZhengzhou {
|
|
|
- classifies, e := GetZhengzhouContractsByClassifyNames(classifyNames)
|
|
|
- if e != nil {
|
|
|
- err = fmt.Errorf("获取郑商所实际合约失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- classifyNames = classifies
|
|
|
- }
|
|
|
-
|
|
|
- d, e := tradeAnalysis.GetClassifyNewestDataTime(exchange, classifyNames)
|
|
|
- if e != nil && e.Error() != utils.ErrNoRow() {
|
|
|
- err = fmt.Errorf("获取品种最新数据日期失败, %v", e)
|
|
|
- return
|
|
|
- }
|
|
|
- if !d.IsZero() {
|
|
|
- dataTime = d
|
|
|
- } else {
|
|
|
- dataTime = time.Now().Local()
|
|
|
- }
|
|
|
- return
|
|
|
-}
|