package services

import (
	"gitee.com/quant1x/engine/cache"
	"gitee.com/quant1x/engine/factors"
	"gitee.com/quant1x/engine/market"
	"gitee.com/quant1x/engine/models"
	"gitee.com/quant1x/exchange"
	"gitee.com/quant1x/gotdx/quotes"
	"gitee.com/quant1x/gox/api"
	"gitee.com/quant1x/gox/coroutine"
	"gitee.com/quant1x/gox/logger"
	"gitee.com/quant1x/gox/progressbar"
	"gitee.com/quant1x/gox/runtime"
	"time"
)

// 更新快照
func jobUpdateMiscAndSnapshot() {
	now := time.Now()
	updateInRealTime, _ := exchange.CanUpdateInRealtime()
	// 集合竞价时段更新数据
	if updateInRealTime && exchange.CheckCallAuctionTime(now) {
		realtimeUpdateMiscAndSnapshot()
	} else {
		if runtime.Debug() {
			realtimeUpdateMiscAndSnapshot()
		}
	}
}

var (
	snapshotDate = cache.DefaultCanReadDate()
	mapSnapshot  = map[string][]quotes.Snapshot{}
	onceSnapshot coroutine.RollingOnce
)

func resetSnapshotCache() {
	date := cache.DefaultCanReadDate()
	if date > snapshotDate {
		clear(mapSnapshot)
		snapshotDate = date
		factors.SwitchDate(snapshotDate)
	}
}

// realtimeUpdateMiscAndSnapshot 更新快照缓存
func realtimeUpdateMiscAndSnapshot() {
	onceSnapshot.Do(resetSnapshotCache)
	moduleName := "执行[misc]"
	logger.Infof("%s: begin", moduleName)
	allCodes := market.GetCodeList()
	count := len(allCodes)
	currentDate := cache.DefaultCanReadDate()
	bar := progressbar.NewBar(barIndexUpdateExchangeAndSnapshot, moduleName, count)
	for _, securityCode := range allCodes {
		bar.Add(1)
		if api.StartsWith(securityCode, []string{"88"}) {
			securityCode = "sh" + securityCode
		}
		v := models.GetTickFromMemory(securityCode)
		if v == nil || v.Date != currentDate {
			// 如果snapshot缓存无效, 或者日期不是当前日期, 跳过
			continue
		}
		timestamp := time.Now()
		// 1. 修订日期
		v.Date = currentDate
		securityCode := v.SecurityCode
		misc := factors.GetL5Misc(securityCode)
		if misc == nil {
			logger.Errorf("加载misc[%s-%s]失败", currentDate, securityCode)
			misc = &factors.Misc{
				Date: currentDate,
				Code: securityCode,
			}
		} else {
			misc.Date = currentDate
		}
		// 2. 计算开盘和收盘的成交量
		misc.OpenVolume = int64(v.OpenVolume)
		misc.CloseVolume = int64(v.CloseVolume)
		// 计算开盘换手z和收盘换手z
		f10 := factors.GetL5F10(securityCode)
		if f10 != nil {
			misc.OpenTurnZ = f10.TurnZ(misc.OpenVolume)
			misc.CloseTurnZ = f10.TurnZ(misc.CloseVolume)
		}
		// 3. 计算快照扩展数据
		if exchange.CheckCallAuctionOpen(timestamp) {
			// 3.1 早盘情绪有时效性
			// 计算早盘竞价方向
			misc.OpenBiddingDirection, misc.OpenVolumeDirection = v.CheckDirection()
			// 3.2 计算早盘情绪
			misc.OpenSentiment, misc.OpenConsistent = market.SnapshotSentiment(*v)
		} else {
			// 3.3盘 中及盘后的数据的计算都没有问题
			// 计算收盘竞价方向
			misc.CloseBiddingDirection, misc.CloseVolumeDirection = v.CheckDirection()
			// 3.4 计算收盘情绪
			misc.CloseSentiment, misc.CloseConsistent = market.SnapshotSentiment(*v)
		}
		// 4. 竞价上午竞价观察
		if exchange.CheckCallAuctionOpen(timestamp) {
			// 4.1 竞价开盘
			if misc.BidOpen == 0 {
				misc.BidOpen = v.Ask1
			}
			// 4.2 竞价结束
			misc.BidClose = v.Price
			// 4.3 竞价最高
			if misc.BidHigh == 0 || misc.BidHigh < v.Ask1 {
				misc.BidHigh = v.Ask1
			}
			// 4.4 竞价最低
			if misc.BidLow == 0 || misc.BidLow > v.Ask1 {
				misc.BidLow = v.Ask1
			}
			// 4.4 竞价匹配量
			misc.BidMatched = float64(v.BidVol1)
			// 4.5 竞价未匹配量
			if v.BidVol2 == 0 {
				misc.BidUnmatched = float64(v.AskVol2)
				misc.BidDirection = -1
			}
			if v.AskVol2 == 0 {
				misc.BidUnmatched = float64(v.BidVol2)
				misc.BidDirection = 1
			}
		}
		// 5. 缓存数据
		//cacheSnapshots = append(cacheSnapshots, *exchange)

		// 6. 更新内存中的数据
		factors.UpdateL5Misc(misc)
		// 7. 刷新缓存
		cacheList, ok := mapSnapshot[securityCode]
		if !ok {
			cacheList = []quotes.Snapshot{}
		}
		if len(cacheList) > 0 {
			lastDay := cacheList[len(cacheList)-1].Date
			lastServerTime := cacheList[len(cacheList)-1].ServerTime
			if currentDate == lastDay && v.ServerTime <= lastServerTime {
				// 时间戳在缓存之前, 忽略
				continue
			}
		}
		cacheList = append(cacheList, *v)
		if len(cacheList) > 0 {
			mapSnapshot[securityCode] = cacheList
		}
	}
	bar.Wait()
	// 刷新Misc快照本地cache
	factors.RefreshL5Misc()
	timestamp := time.Now()
	if exchange.CheckCallAuctionOpenFinished(timestamp) || exchange.CheckCallAuctionCloseFinished(timestamp) {
		// 早盘和尾盘集合竞价结束后刷新缓存文件
		for _, listSnapshot := range mapSnapshot {
			if len(listSnapshot) == 0 {
				continue
			}
			// 获取第一条记录
			first := listSnapshot[0]
			securityCode := first.SecurityCode
			filename := cache.SnapshotFilename(securityCode)
			cacheList := []quotes.Snapshot{}
			err := api.CsvToSlices(filename, &cacheList)
			if err == nil && len(cacheList) > 0 {
				// 缓存中最后一条记录
				rows := len(cacheList)
				last := cacheList[rows-1]
				// 日期
				lastDay := last.Date
				// 时间戳
				lastServerTime := last.ServerTime
				for _, v := range listSnapshot {
					if currentDate == lastDay && v.ServerTime <= lastServerTime {
						// 时间戳在缓存之前, 忽略
						continue
					}
					cacheList = append(cacheList, v)
				}
			} else {
				// 如果缓存文件不存在, 用缓存数据
				cacheList = listSnapshot
			}
			if len(cacheList) > 0 {
				_ = api.SlicesToCsv(filename, cacheList)
			}
		}
	}
	logger.Infof("%s: end", moduleName)
}