package models

import (
	"fmt"
	"gitee.com/quant1x/engine/market"
	"gitee.com/quant1x/gox/concurrent"
	"gitee.com/quant1x/gox/progressbar"
	"gitee.com/quant1x/gox/tags"
	"gitee.com/quant1x/pkg/tablewriter"
	"os"
)

// ExecuteStrategy 执行策略
func ExecuteStrategy(model Strategy, barIndex *int) {
	fmt.Printf("策略模块: %s\n", model.Name())
	// 加载即时行情
	SyncAllSnapshots(barIndex)
	fmt.Println()
	// 执行策略
	allCodes := market.GetCodeList()
	count := len(allCodes)
	bar := progressbar.NewBar(*barIndex, "执行["+model.Name()+"]", count)
	results := concurrent.NewTreeMap[string, ResultInfo]()
	for _, securityCode := range allCodes {
		// 此处可以增加过滤规则
		model.Evaluate(securityCode, results)
		bar.Add(1)
	}
	// 输出一个换行符, 结束上一个进度条
	fmt.Println()
	table := tablewriter.NewWriter(os.Stdout)
	table.SetHeader(tags.GetHeadersByTags(ResultInfo{}))
	results.Each(func(key string, value ResultInfo) {
		table.Append(tags.GetValuesByTags(value))
	})
	table.Render()
}