package flash

import (
	"gitee.com/quant1x/engine/cache"
	"gitee.com/quant1x/engine/cachel5"
	"gitee.com/quant1x/engine/datasets"
	"gitee.com/quant1x/engine/features"
	"sync"
)

var (
	__l5Once sync.Once
	// 历史数据
	__l5History *cachel5.Cache1D[*features.History] = nil
	// 基本面F10
	__l5F10 *cachel5.Cache1D[*features.F10] = nil
)

func init() {
	__l5Once.Do(lazyInitFeatures)
}

func lazyInitFeatures() {
	// 历史数据
	__l5History = cachel5.NewCache1D[*features.History](features.CacheL5KeyHistory, features.NewHistory)
	err := cache.Register(__l5History)
	if err != nil {
		panic(err)
	}
	// 基本面F10
	__l5F10 = cachel5.NewCache1D[*features.F10](features.CacheL5KeyF10, features.NewF10)
	err = cache.Register(__l5F10)
	if err != nil {
		panic(err)
	}
}

// CacheList 缓存列表
func CacheList() []cachel5.CacheAdapter {
	__l5Once.Do(lazyInitFeatures)
	list := []cachel5.CacheAdapter{
		__l5F10,
		__l5History,
	}
	return list
}

func GetHistory() *cachel5.Cache1D[*features.History] {
	__l5Once.Do(lazyInitFeatures)
	return __l5History
}

func GetL5History(securityCode string, date ...string) *features.History {
	__l5Once.Do(lazyInitFeatures)
	data := __l5History.Get(securityCode, date...)
	if data == nil {
		return nil
	}
	return *data
}

func GetL5F10(securityCode string, date ...string) *features.F10 {
	__l5Once.Do(lazyInitFeatures)
	data := __l5F10.Get(securityCode, date...)
	if data == nil {
		return nil
	}
	return *data
}

var (
	__dataSetOnce sync.Once
	__dataSets    []datasets.DataSet = nil
)

func lazyInitDataSets() {
	__dataSets = []datasets.DataSet{
		new(datasets.DataQuarterlyReport), // 季报
		new(datasets.DataXdxr),            // 除权除息
		new(datasets.DataKLine),           // 基础K线
		//new(datasets.DataSafetyScore),     // 安全分
	}
}

// DataSetList 数据集 列表
func DataSetList() []datasets.DataSet {
	__dataSetOnce.Do(lazyInitDataSets)
	return __dataSets
}