package data

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/shopspring/decimal"
	"matchmaking-system/internal/biz/structure"
	"matchmaking-system/internal/data/memory"
	"matchmaking-system/internal/data/mq/consumer"
	"matchmaking-system/internal/data/socket/publicData"
	"matchmaking-system/internal/data/socket/shareData"
	"matchmaking-system/internal/data/tradedeal/share"
	"matchmaking-system/internal/data/tradedeal/virtual"
	"matchmaking-system/internal/pkg/flags"
	"matchmaking-system/internal/pkg/logging/applogger"
	"matchmaking-system/internal/pkg/logging/common"
	"matchmaking-system/internal/pkg/setting"
	"matchmaking-system/internal/pkg/utils"
	"sync"
	"time"
)

/*
马股行情订阅
1、接收用户马股下单的交易对马股行情
2、将订阅行情写入内存中用于并发计算
3、挂单(市价|限价)缓存队列处理-开仓
4、持仓缓存队列处理-平仓
*/

// ShareMysMessage
// @Description:
type ShareMysMessage struct {
	Symbol             string          `json:"symbol,omitempty"`               // 交易所:股票代码
	StockCode          string          `json:"stock_code,omitempty"`           // 股票代码
	StockName          string          `json:"stock_name,omitempty"`           // 股票名，可能为空
	Price              decimal.Decimal `json:"price,omitempty"`                // 实时价格
	UpDownRate         decimal.Decimal `json:"up_down_rate,omitempty"`         // 涨跌%
	UpDown             decimal.Decimal `json:"up_down,omitempty"`              // 涨跌额
	TradeV             decimal.Decimal `json:"trade_v,omitempty"`              // 技术评级指标值
	TradeK             string          `json:"trade_k,omitempty"`              // 技术评级
	Vol                int64           `json:"vol,omitempty"`                  // 成交量
	TurnoverPriceTotal decimal.Decimal `json:"turnover_price_total,omitempty"` // 成交量*价格
	PriceTotal         string          `json:"price_total,omitempty"`          // 总市值
	PE                 string          `json:"p_e,omitempty"`                  // pe
	Eps                string          `json:"eps,omitempty"`                  // EPS
	EmployeesNumber    string          `json:"employees_number,omitempty"`     // 雇员
	Plate              string          `json:"plate,omitempty"`                // 板块
	Desc               string          `json:"desc,omitempty"`                 // 描述
	PriceCode          string          `json:"price_code,omitempty"`           // 价格类型
	Country            string          `json:"country,omitempty"`              // 国家
	Ts                 int64           `json:"ts,omitempty"`                   // 时间
	Token              string          `json:"token,omitempty"`                // Token 令牌
}

// ShareMysSymbol
// @Description:
type ShareMysSymbol struct {
	MysMap          chan []byte // 订单交易对 -- 用户下单通知订阅
	MysSetMap       chan []byte // 订单交易对 -- 用户盘前|盘后行情订阅
	MysMapSymbol    sync.Map    // 订阅交易簿 -- 过滤重复
	MysSetMapSymbol sync.Map    // 订阅交易簿 -- 过滤重复
}

// InitCacheSymbolShareMys
//
//	@Description:
//	@param ctx
//	@param uo
func InitCacheSymbolShareMys(ctx context.Context, uo *Data) {
	for {
		mysList, err := GetBotStockMysListByStockId(ctx, uo)
		if err != nil {
			return
		}

		for _, value := range mysList {
			// Write to the redis list hot cache for initializing subscriptions when the program is pulled up
			checkBool, err := uo.redisDB.HExists(context.Background(), setting.MarketShareMys, value).Result()
			if err != nil {
				applogger.Error("%v SubscribeShareMys.MarketShareMys.HExists:%v", common.ErrShareMys, err)
				continue
			}
			if !checkBool {
				if err := uo.redisDB.HSet(context.Background(), setting.MarketShareMys, value, value).Err(); err != nil {
					applogger.Error("%v SubscribeShareMys.MarketShareMys.HSet:%v", common.ErrShareMys, err)
					continue
				}
			}
			if !flags.CheckSetting {
				applogger.Info("初始化马股股票代码:%v", value)
			}
		}

		time.Sleep(10 * time.Second)
	}
}

// InitSubscribeShareMys
//
//	@Description: 状态机加载订阅
//	@param ctx
//	@param uo
//	@param shareMys
func InitSubscribeShareMys(shareMys *ShareMysSymbol, check bool) {
	for {
		var key string
		if check {
			key = setting.MarketShareBlkMys
		} else {
			key = setting.MarketShareMys
		}

		listSpots, err := LoadLRangeList(key)
		if err != nil {
			applogger.Error("%v InitSubscribeShareMys.LoadLRangeList:%v", common.ErrShareMys, err)
			return
		}

		for _, value := range listSpots {
			// Handling duplicate subscriptions
			_, ok := shareMys.MysMapSymbol.Load(value)
			if ok {
				continue
			}

			shareMys.MysMap <- []byte(value)
		}

		time.Sleep(10 * time.Second)
	}
}

// InitShareMysCloseCachePrice
//
//	@Description:
//	@param shareMys
//	@param check
func InitShareMysCloseCachePrice(check bool) {
	for {
		var key string
		if check {
			key = setting.MarketShareBlkMys
		} else {
			key = setting.MarketShareMys
		}

		listSpots, err := LoadLRangeList(key)
		if err != nil {
			applogger.Error("%v InitSubscribeShareMys.LoadLRangeList:%v", common.ErrShareMys, err)
			return
		}

		for _, value := range listSpots {
			SubscribeMysClosePrice(value) // 处理闭盘价格取值
		}

		time.Sleep(10 * time.Second)
	}
}

// InitShareMysCloseNewPrice
//
//	@Description: 同步实时行情
//	@param check
func InitShareMysCloseNewPrice(check bool) {
	for {
		var key string
		if check {
			key = setting.MarketShareBlkMys
		} else {
			key = setting.MarketShareMys
		}
		listSpots, err := LoadLRangeList(key)
		if err != nil {
			applogger.Error("%v InitShareMysCloseNewPrice.LoadLRangeList:%v", common.ErrShareMys, err)
			time.Sleep(20 * time.Second)
			continue
		}
		for _, value := range listSpots {
			SubscribeMysCloseNewPrice(value)
		}

		time.Sleep(500 * time.Millisecond)
	}
}

// InitShareMysPriceSetUp
//
//	@Description: 同步插针行情
//	@param check
func InitShareMysPriceSetUp(check bool) {
	for {
		var key string
		if check {
			key = setting.MarketShareBlkMys
		} else {
			key = setting.MarketShareMys
		}
		listSpots, err := LoadLRangeList(key)
		if err != nil {
			applogger.Error("%v InitShareMysPriceSetUp.LoadLRangeList:%v", common.ErrShareMys, err)
			time.Sleep(20 * time.Second)
			continue
		}

		for _, value := range listSpots {
			mysKey := publicData.SymbolCache(flags.Mys, value, flags.TradeTypePrice)
			mysSetCache := fmt.Sprintf("%v%v:%v", flags.StockTradePrices, flags.ShareMysMarketType, value)
			vw, err := GetBeforeAndAfterSetPrice(mysSetCache)
			if err != nil || len(vw) == 0 {
				_ = memory.ShareMysPriceSetUp.Delete(mysKey)
				continue
			}
			if err = memory.ShareMysPriceSetUp.Set(mysKey, []byte(vw)); err != nil {
				applogger.Error("%v InitShareMysPriceSetUp.ShareMysPriceSetUp:%v", common.ErrShareMys, err)
			}
		}

		time.Sleep(500 * time.Millisecond)
	}
}

// SubscribeMysHSetCodeList
//
//	@Description: 同步列表缓存
//	@param symbolStr
//	@param check
func SubscribeMysHSetCodeList(symbolStr string, check bool) {
	var key string
	if check {
		key = setting.MarketShareBlkMys
	} else {
		key = setting.MarketShareMys
	}
	// Write to the redis list hot cache for initializing subscriptions when the program is pulled up
	checkBool, err := Reds.HExists(context.Background(), key, symbolStr).Result()
	if err != nil {
		return
	}
	if !checkBool {
		if err := Reds.HSet(context.Background(), key, symbolStr, symbolStr).Err(); err != nil {
			applogger.Error("%v SubscribeMysHSetCodeList.%v.HSet:%v", common.ErrShareMys, key, err)
			return
		}
	}
}

// SubscribeMysCloseNewPrice
//
//	@Description: 获取行情价
//	@param symbolStr
func SubscribeMysCloseNewPrice(symbolStr string) {
	// 获取市场真实开闭盘时间
	timeValue, checkWeekday := MarketCheckOpenAndCloseTime(flags.ShareMysMarketType)
	// 判定是否盘中
	if utils.CheckInPlateTime(timeValue) && checkWeekday {
		// 盘中实时价格
		realTimePrice, err := Reds.HGet(context.Background(), flags.ShareMalaysiaClosingNewPriceKey, symbolStr).Result()
		if err != nil {
			realTimePrice = decimal.Zero.String()
		}
		if realTimePrice != decimal.Zero.String() {
			keys := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypePrice)
			if err = memory.ShareMysCache.Set(keys, []byte(realTimePrice)); err != nil {
				applogger.Error("%v SubscribeMysCloseNewPrice.ShareMysCache:%v", common.ErrShareMys, err)
			}
		}
	}
}

// SubscribeMysClosePrice
//
//	@Description: 获取闭盘价
//	@param symbolStr
func SubscribeMysClosePrice(symbolStr string) {
	var err error
	var sub = decimal.Zero
	var closeNewPrice, closingPrice, price string

	// 闭盘价格
	closeNewPrice, err = Reds.HGet(context.Background(), flags.ShareMalaysiaClosingNewPriceKey, symbolStr).Result()
	if err != nil || closeNewPrice == decimal.Zero.String() {
		closeNewPrice, err = Reds.HGet(context.Background(), flags.ShareMalaysiaClosingPriceKey, symbolStr).Result()
		if err != nil || closeNewPrice == decimal.Zero.String() {
			closeNewPrice = decimal.Zero.String()
		}
	}

	// 市价(实时价格)
	closingPrice, err = Reds.HGet(context.Background(), flags.ShareMalaysiaBeforeClose, symbolStr).Result()
	if err != nil {
		closingPrice = decimal.Zero.String()
	}

	// 判定取值数据
	if closeNewPrice != decimal.Zero.String() {
		price = closeNewPrice
	} else {
		price = closingPrice
	}
	closeKey := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypeClosePrice)
	if err = memory.ShareMysClosePrice.Set(closeKey, []byte(price)); err != nil {
		applogger.Error("%v ShareMysClosePrice.Set.price:%v", common.ErrShareMys, err)
	}

	// 写入涨跌幅价格判定缓存
	cn := closeNewPrice != decimal.Zero.String()
	cp := closingPrice != decimal.Zero.String()
	lcn := utils.IsNumber(closeNewPrice)
	lcp := utils.IsNumber(closingPrice)
	if cn && cp && lcn && lcp {
		sub = decimal.RequireFromString(closingPrice).Sub(decimal.RequireFromString(closeNewPrice))
	}
	chgKey := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypeChg) // 涨跌幅标识Key
	if err = memory.ShareMysChgMark.Set(chgKey, []byte(sub.String())); err != nil {
		applogger.Error("%v ShareMysChgMark.Set.price:%v", common.ErrShareMys, err)
	}
}

// OrderSubAdminShareMysSubscribeBySum
//
//	@Description: 持仓订单浮动盈亏计算
//	@param ctx
//	@param uo
func OrderSubAdminShareMysSubscribeBySum(uo *Data) {
	for {
		var topIc = setting.AdminShareMysSubscribe
		hashMap, err := uo.redisDB.HGetAll(context.Background(), topIc).Result()
		if err != nil {
			applogger.Error("OrderSubAdminShareMysSubscribeBySum.HGetAll:%v", err)
			time.Sleep(5 * time.Second)
			continue
		}
		var hashList []share.ShareMysTallyCache
		for _, value := range hashMap {
			var msg share.ShareMysTallyCache
			if err = json.Unmarshal([]byte(value), &msg); err != nil {
				time.Sleep(5 * time.Second)
				continue
			}
			switch msg.Status {
			case flags.Entrust: // 挂单
			case flags.Position: // 持仓
				hashList = append(hashList, msg)
			default: // 平仓|撤单
				err = uo.redisDB.HDel(context.Background(), topIc, msg.OrderId).Err()
				if err != nil {
					applogger.Error("AdminContractSubscribe.HDel:%v", err)
					continue
				}
			}
		}
		applogger.Info("马股数据量统计:%v", len(hashList))

		// 统计马股总持仓订单浮动盈亏
		priceSum := decimal.Zero
		for _, value := range hashList {
			orderModel := shareData.ShareMysOrderProcessing(topIc, value)
			if orderModel != nil {
				var subPrice decimal.Decimal
				newPrice, err := decimal.NewFromString(orderModel.Price)
				if err != nil {
					continue
				}
				openPrice, err := decimal.NewFromString(orderModel.OpenPrice)
				if err != nil {
					continue
				}
				switch value.Order.TradeType {
				case flags.TradeTypeBuy: // 买涨 = 新价格 - 开仓价
					subPrice = newPrice.Sub(openPrice)
				case flags.TradeTypeSell: // 卖跌 = 开仓价 - 新价格
					subPrice = openPrice.Sub(newPrice)
				default:
					subPrice = decimal.Zero
				}
				price := subPrice.Mul(decimal.RequireFromString(value.Order.OrderNumber))
				priceSum = priceSum.Add(price) // 累加盈亏统计
			}
		}

		// 写入缓存
		if err = memory.ShareMysFloating.Set(flags.FloatingMys, []byte(priceSum.String())); err != nil {
			applogger.Error("统计马股持仓订单浮动盈亏错误:%v", err)
			time.Sleep(5 * time.Second)
			continue
		}
		applogger.Info("统计马股持仓订单浮动盈亏:%v", priceSum)

		time.Sleep(1 * time.Second)
	}
}

// SubscribeShareMys
//
//	@Description: TODO: 马股订阅
//	@param ctx
//	@param uo
//	@param shareMys
func SubscribeShareMys(ctx context.Context, shareMys *ShareMysSymbol, check bool) {
	for {
		select {
		case symbol, _ := <-shareMys.MysMap:
			symbolStr := string(symbol)
			topIc := fmt.Sprintf("%v.%v", symbolStr, flags.CountryMys)
			// Handling duplicate subscriptions
			_, ok := shareMys.MysMapSymbol.Load(symbolStr)
			if ok {
				time.Sleep(5 * time.Second)
				continue
			}
			// Transaction determination
			var key string
			if check {
				key = setting.MarketShareBlkMys // 大宗股交易
			} else {
				key = setting.MarketShareMys // 马股交易
			}
			// Write to the redis list hot cache for initializing subscriptions when the program is pulled up
			checkBool, err := Reds.HExists(context.Background(), key, symbolStr).Result()
			if err != nil {
				time.Sleep(5 * time.Second)
				continue
			}
			if !checkBool {
				if err := Reds.HSet(context.Background(), key, symbolStr, symbolStr).Err(); err != nil {
					applogger.Error("%v SubscribeShareMys.%v.HSet:%v", common.ErrShareMys, key, err)
					time.Sleep(5 * time.Second)
					continue
				}
			}

			// 处理订阅实时行情数据
			go func() {
				symbolStrCache := symbolStr
				pubSub := Reds.Subscribe(ctx, topIc)
				defer pubSub.Close()
				if _, err := pubSub.Receive(ctx); err != nil {
					shareMys.MysMapSymbol.Delete(symbolStrCache)
					return
				}
				ch := pubSub.Channel()
				for {
					select {
					case msg, _ := <-ch:
						var subMsg ShareMysMessage
						if err := json.Unmarshal([]byte(msg.Payload), &subMsg); err != nil {
							applogger.Error("%v SubscribeShareMys.Unmarshal:%v", common.ErrShareMys, err)
							time.Sleep(5 * time.Second)
							continue
						}
						SubscribeMysPrice(subMsg.Price.String(), symbolStr, flags.RealTime)
					default:
						Reds.Ping(ctx) // 如果没有消息,发送 PING 命令以维护连接
						time.Sleep(10 * time.Second)
					}
				}
			}()

			// 盘前|盘后价格设置
			_, okc := shareMys.MysSetMapSymbol.Load(symbolStr)
			if !okc {
				go func() {
					for {
						keyCache := fmt.Sprintf("%v%v:%v", flags.StockTradePrices, flags.ShareMysMarketType, symbolStr)
						vw, err := GetBeforeAndAfterSetPrice(keyCache)
						if err != nil || len(vw) == 0 {
							priceCache := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypePrice)
							memory.ShareMysPriceSetUp.Delete(priceCache)
							time.Sleep(15 * time.Second)
							continue
						}
						SubscribeMysPrice(vw, symbolStr, flags.SetUp)
						time.Sleep(10 * time.Second)
					}
				}()
				// Write cache to determine whether to repeatedly open the protocol
				shareMys.MysSetMapSymbol.Store(symbolStr, symbolStr)
			}

			// Write in the map to determine whether to repeat subscription
			shareMys.MysMapSymbol.Store(symbolStr, symbolStr)
		}
	}
}

// SubscribeMysPrice
//
//	@Description: TODO: 订阅获取马股股实时价格或者[盘前|盘后]设置价格
//	@param setPrice
//	@param symbolStr
//	@param check
func SubscribeMysPrice(setPrice, symbolStr string, check int) {
	var err error
	var sub = decimal.Zero

	// 计算涨跌幅
	yClosingPrice, err := Reds.HGet(context.Background(), flags.ShareMalaysiaClosingPriceKey, symbolStr).Result()
	if err != nil || yClosingPrice == decimal.Zero.String() {
		yClosingPrice, err = Reds.HGet(context.Background(), flags.ShareMalaysiaClosingNewPriceKey, symbolStr).Result()
		if err != nil || yClosingPrice == decimal.Zero.String() {
			yClosingPrice = setPrice
		}
	}

	// 写入涨跌幅缓存数据
	yc := yClosingPrice != decimal.Zero.String()
	sp := setPrice != decimal.Zero.String()
	lyc := utils.IsNumber(yClosingPrice)
	lsp := utils.IsNumber(setPrice)
	if yc && sp && lyc && lsp {
		sub = decimal.RequireFromString(setPrice).Sub(decimal.RequireFromString(yClosingPrice))
	}
	chgKey := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypeChg) // 涨跌幅标识Key
	if err = memory.ShareMysChgMark.Set(chgKey, []byte(sub.String())); err != nil {
		applogger.Error("%v SubscribeShareMys.ShareMysChgMark.Set.price:%v", common.ErrShareMys, err)
	}
	if !flags.CheckSetting {
		applogger.Debug("涨跌幅交易对内存key:%v,涨跌幅数值:%v", symbolStr, sub)
	}

	// 交易类型:0最新价,1买入,2卖出
	price := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypePrice)
	switch check {
	case flags.RealTime: // 实时价格（优先级高）
		if err = memory.ShareMysCache.Set(price, []byte(setPrice)); err != nil {
			applogger.Error("%v SubscribeShareMys.ShareMysCache.Set.price:%v", common.ErrShareMys, err)
		}
	case flags.SetUp: // 设置[盘前|盘后]价格
		if err = memory.ShareMysPriceSetUp.Set(price, []byte(setPrice)); err != nil {
			applogger.Error("%v SubscribeShareMys.ShareMysPriceSetUp:%v", common.ErrShareMys, err)
		}
	}
	if !flags.CheckSetting {
		applogger.Debug("股票交易对内存Key:%v,最新价格:%v", price, setPrice)
	}
}

// SubscribeMysSetForcedClosure
//
//	@Description: TODO: 处理马股强平阈值
//	@param symbolStr
func SubscribeMysSetForcedClosure(symbolStr string) {
	flatRatio := GetCacheForcedClosure(flags.StockMGSystemSetUpKey, symbolStr)
	chgKey := publicData.SymbolCache(flags.Mys, symbolStr, flags.TradeTypeForcedClosure)
	if err := memory.ShareMysForcedClosure.Set(chgKey, []byte(flatRatio.String())); err != nil {
		applogger.Error("%v SubscribeMysSetForcedClosure.StockMGSystemSetUpKey err:%v", common.ErrShareMys, err)
		return
	}
}

// ShareMysTransactionEntrust
//
//	@Description: 马股委托订单
//	@param ctx
func ShareMysTransactionEntrust(ctx context.Context) {
	for {
		ShareMysTransactionCalculationEntrust(ctx)
		time.Sleep(600 * time.Millisecond)
	}
}

// ShareMysTransactionCalculationEntrust
//
//	@Description: 马股挂单缓存队列计算
//	@param ctx
func ShareMysTransactionCalculationEntrust(ctx context.Context) {
	var wg sync.WaitGroup

	// 全局变量处理
	if CheckGlobalTread(flags.MysMarket) {
		time.Sleep(5 * time.Second)
		return
	}
	// 获取挂单缓存列表
	orderMap, err := Reds.HGetAll(context.Background(), setting.MarketShareMysEntrust).Result()
	if err != nil {
		applogger.Error("%v ShareMysTransactionCalculationEntrust.HGetAll:%v", common.ErrShareMys, err)
		return
	}

	for _, value := range orderMap {
		var msg = make(chan share.ShareMysTallyCache, 1)
		var entrust share.ShareMysTallyCache
		if err = json.Unmarshal([]byte(value), &entrust); err != nil {
			applogger.Error("%v ShareMysTransactionCalculationEntrust.Unmarshal:%v", common.ErrShareMys, err)
			continue
		}

		// 股票实时价格
		var newPrice decimal.Decimal
		newPrice, err = GetShareMysPrice(entrust.Symbol)
		if err != nil {
			applogger.Warn("%v ShareMysTransactionCalculationEntrust.GetShareMysPrice:%v---%v", common.ErrShareMys, entrust.Symbol, err)
			continue
		}

		// 判定涨跌幅
		chg := GetShareChgMys(flags.MysMarket, entrust.Symbol) // 涨跌幅标识
		if share.CheckOrderMysByChg(setting.MarketShareMysEntrust, entrust, chg) {
			//applogger.Warn("当前股票%v订单在挂单状态已达到涨跌幅上限.", entrust.Symbol)
			continue
		}

		wg.Add(1)
		go func(value string) {
			var resultMsg share.ShareMysTallyCache
			switch entrust.Order.DealType {
			case flags.DealTypeLimited: // 限价挂单
				resultMsg = ShareMysConcurrentComputingEntrustLimited(&entrust, newPrice, &wg)
			case flags.DealTypeMarket: // 市价挂单(开盘即成交)
				resultMsg = ShareMysConcurrentComputingEntrustMarket(&entrust, newPrice, &wg)
			}
			msg <- resultMsg // 通知管道异步处理订单操作
		}(value)

		// 处理并发结果(需要改成通过信道通知)
		select {
		case resultMsg, _ := <-msg:
			switch resultMsg.Status {
			case flags.Entrust: // 挂单中
				if !flags.CheckSetting {
					applogger.Debug("从信道接收马股挂单信息:%v", resultMsg)
				}
			case flags.Position: // 持仓中
				if !flags.CheckSetting {
					applogger.Debug("从信道接收到马股持仓信息:%v", resultMsg)
				}

				// 开盘逻辑
				if err = ShareMysLiquidationEntrust(ctx, &resultMsg); err != nil {
					applogger.Error("%v ShareMysTransactionCalculationEntrust.ShareMysLiquidationEntrust:%v", common.ErrShareMys, err)
					continue
				}

				// 写入持仓缓存列表
				var byteStr []byte
				byteStr, err = json.Marshal(resultMsg)
				if err != nil {
					applogger.Error("%v ShareMysTransactionCalculationEntrust.Marshal:%v", common.ErrShareMys, err)
					continue
				}

				// TODO: 写入持仓消息队列
				//if err = uo.mqProducer.Entrust.PushNotice(byteStr); err != nil {
				//	applogger.Error("%v ShareMysTransactionCalculationEntrust.PushNotice:%v", common.ShareMysError, err)
				//	continue
				//}

				// 写入持仓队列缓存
				if err = Reds.HSet(context.Background(), setting.MarketShareMysPosition, resultMsg.OrderId, string(byteStr)).Err(); err != nil {
					continue
				}
			}
		}
	}

	wg.Wait()
	applogger.Info("马股挂单并发计算执行完毕......")
}

// ShareMysConcurrentComputingEntrustLimited
//
//	@Description:
//	@param order
//	@param newPrice
//	@param bidPrice
//	@param askPrice
//	@param wg
//	@return tradedeal.ShareMysTallyCache
func ShareMysConcurrentComputingEntrustLimited(order *share.ShareMysTallyCache, newPrice decimal.Decimal, wg *sync.WaitGroup) share.ShareMysTallyCache {
	var deleteCache bool
	var dealPrice string
	var limitPrice = decimal.RequireFromString(order.Order.LimitPrice) // 限价-价格
	var status = order.Status                                          // 原始价格

	switch order.Order.TradeType {
	case flags.TradeTypeBuy: // 买涨
		if !flags.CheckSetting {
			applogger.Debug("用户ID:%v,限价买入下单价格:%v,最新市价:%v,判定结果:%v,原始状态:%v", order.UserId, limitPrice, newPrice, limitPrice.Cmp(newPrice), order.Status)
		}

		// 限价买入逻辑判定 -> 挂单金额 > 当前价格
		if limitPrice.Cmp(newPrice) > 0 {
			deleteCache = true
			difference := newPrice.Mul(utils.Difference()) // 设置价差
			dealPrice = newPrice.Add(difference).String()  // 开仓价格
		}
	case flags.TradeTypeSell: // 买跌
		if !flags.CheckSetting {
			applogger.Debug("用户ID:%v,限价卖出下单价格:%v,最新市价:%v,判定结果:%v,原始状态:%v", order.UserId, limitPrice, newPrice, limitPrice.Cmp(newPrice), order.Status)
		}

		// 限价卖入逻辑判定 -> 挂单金额 < 当前价格
		if limitPrice.Cmp(newPrice) < 0 {
			deleteCache = true
			difference := newPrice.Mul(utils.Difference()) // 设置价差
			dealPrice = newPrice.Sub(difference).String()  // 开仓价格
		}
	}

	if deleteCache {
		order.Status = flags.Position
		if err := Reds.HDel(context.Background(), setting.MarketShareMysEntrust, order.OrderId).Err(); err != nil {
			applogger.Error("%v ShareMysConcurrentComputingEntrustLimited.MarketShareMysEntrust.HDel:%v", common.ErrShareMys, err)
			order.Status = status
		}
	}

	wg.Done()
	return share.ShareMysTallyCache{
		UserId:      order.UserId,      // 用户Id
		OrderId:     order.OrderId,     // 订单Id
		Symbol:      order.Symbol,      // 下单交易对
		OpenPrice:   dealPrice,         // 开仓价格
		Status:      order.Status,      // 订单状态
		Order:       order.Order,       // 下单信息
		ClosingTime: order.ClosingTime, // 平仓时间
	}
}

// ShareMysConcurrentComputingEntrustMarket
//
//	@Description: 马股市价下单进入挂单队列(休市)
//	@param order
//	@param newPrice
//	@param bidPrice
//	@param askPrice
//	@param wg
//	@return tradedeal.ShareMysTallyCache
func ShareMysConcurrentComputingEntrustMarket(order *share.ShareMysTallyCache, newPrice decimal.Decimal, wg *sync.WaitGroup) share.ShareMysTallyCache {
	var deleteCache bool
	var dealPrice string
	var marketPrice = decimal.RequireFromString(order.Order.MarketPrice) // 市价-价格
	var status = order.Status                                            // 原始价格

	switch order.Order.TradeType {
	case flags.TradeTypeBuy: // 买涨
		if !flags.CheckSetting {
			applogger.Debug("用户ID:%v,限价买入下单价格:%v,最新市价:%v,原始状态:%v", order.UserId, marketPrice, newPrice, order.Status)
		}

		deleteCache = true
		difference := newPrice.Mul(utils.Difference()) // 设置价差
		dealPrice = newPrice.Add(difference).String()  // 开仓价格
	case flags.TradeTypeSell: // 买跌
		if !flags.CheckSetting {
			applogger.Debug("用户ID:%v,限价卖出下单价格:%v,最新市价:%v,原始状态:%v", order.UserId, marketPrice, newPrice, order.Status)
		}

		deleteCache = true
		difference := newPrice.Mul(utils.Difference()) // 设置价差
		dealPrice = newPrice.Sub(difference).String()  // 开仓价格
	}

	if deleteCache {
		order.Status = flags.Position
		if err := Reds.HDel(context.Background(), setting.MarketShareMysEntrust, order.OrderId).Err(); err != nil {
			applogger.Error("%v ShareUsConcurrentComputingEntrustMarket.MarketShareMysEntrust.HDel:%v", common.ErrShareMys, err)
			order.Status = status
		}
	}

	wg.Done()
	return share.ShareMysTallyCache{
		UserId:      order.UserId,      // 用户Id
		OrderId:     order.OrderId,     // 订单Id
		Symbol:      order.Symbol,      // 下单交易对
		OpenPrice:   dealPrice,         // 开仓价格
		Status:      order.Status,      // 订单状态
		Order:       order.Order,       // 下单信息
		ClosingTime: order.ClosingTime, // 平仓时间
	}
}

// ShareBlockMysConcurrentComputingEntrustMarket
//
//	@Description: 大宗(马股)交易
//	@param order
//	@param newPrice
//	@param wg
//	@return share.ShareMysTallyCache
func ShareBlockMysConcurrentComputingEntrustMarket(order *share.ShareMysTallyCache, newPrice decimal.Decimal, wg *sync.WaitGroup) share.ShareMysTallyCache {
	var status = order.Status // 原始状态

	order.Status = flags.Position
	if err := Reds.HDel(context.Background(), setting.MarketShareBlkEntrust, order.OrderId).Err(); err != nil {
		applogger.Error("%v ShareBlockMysConcurrentComputingEntrustMarket.MarketShareBlkEntrust.HDel:%v", common.ErrShareBlk, err)
		order.Status = status
	}

	wg.Done()
	return share.ShareMysTallyCache{
		UserId:      order.UserId,      // 用户Id
		OrderId:     order.OrderId,     // 订单Id
		Symbol:      order.Symbol,      // 下单交易对
		OpenPrice:   newPrice.String(), // 开仓价格
		Status:      order.Status,      // 订单状态
		Order:       order.Order,       // 下单信息
		ClosingTime: order.ClosingTime, // 平仓时间
	}
}

// ShareMysMqOpenBusiness
//
//	@Description: TODO: 马股持仓消息队列
//	@param ctx
//	@param uo
func ShareMysMqOpenBusiness(ctx context.Context, uo *Data) {
	go func() {
		uo.mqConsumer.Entrust.ConsumerNotice()
	}()
	for {
		select {
		case message, ok := <-consumer.ConsumerEntrustMq.ShareMysMq:
			if !ok {
				time.Sleep(5 * time.Second)
				applogger.Error("%v ShareMysMqOpenBusiness.ShareMysMq err....", common.ErrShareMys)
				continue
			}

			// 持仓订单缓存数据序列化解析
			var resultMsg share.ShareMysTallyCache
			if err := json.Unmarshal(message, &resultMsg); err != nil {
				applogger.Error("%v ShareMysMqOpenBusiness.Unmarshal:%v", common.ErrShareMys, err)
				time.Sleep(5 * time.Second)
				continue
			}

			// 持仓平仓
			if err := ShareMysLiquidationEntrust(ctx, &resultMsg); err != nil {
				applogger.Error("%v ShareMysMqOpenBusiness.ShareMysLiquidationEntrust:%v", common.ErrShareMys, err)
				continue
			}
		default:
			// TODO: 是否处理空队列缓存的情况
			applogger.Info("这里没有监控到持仓mq队列消息.........")
			time.Sleep(2 * time.Second)
		}
	}
}

// ShareMysLiquidationEntrust
//
//	@Description: 马股挂单-->开仓业务处理
//	@param ctx
//	@param order
//	@return error
func ShareMysLiquidationEntrust(ctx context.Context, order *share.ShareMysTallyCache) error {
	// 1、开盘
	err := StockMysOpenOrder(ctx, Msql, order.UserId, order.OrderId, order.OpenPrice, order.Order)
	if err != nil {
		applogger.Error("%v ShareMysLiquidationEntrust.StockMysOpenOrder:%v", common.ErrShareMys, err)
		return err
	}

	// 大宗(马股)交易判定
	var entrustKey, adminKey string
	if !CheckTypeStatus(order.Order.Type) {
		entrustKey = setting.ShareMysSubscribe
		adminKey = setting.AdminShareMysSubscribe
	} else {
		entrustKey = setting.ShareBlkSubscribe
		adminKey = setting.AdminShareBlkSubscribe
	}

	// 2、更新用户订阅缓存列表订单状态
	updateKey := virtual.OrderIdListKey(entrustKey, order.UserId)
	if err = UpdateShareMysSubscribeHashStatusByOrderId(order.OrderId, updateKey, flags.Position, order.OpenPrice); err != nil {
		applogger.Error("%v ShareMysLiquidationEntrust.%v:%v", common.ErrShareMys, entrustKey, err)
		return err
	}

	// 3、更新管理员订阅缓存列表订单状态
	if err = UpdateShareMysSubscribeHashStatusByOrderId(order.OrderId, adminKey, flags.Position, order.OpenPrice); err != nil {
		applogger.Error("%v ShareMysLiquidationEntrust.Entrust.%v:%v", common.ErrShareMys, adminKey, err)
		return err
	}

	return nil
}

// ShareMysTransactionPosition
//
//	@Description: 马股持仓订单
//	@param ctx
func ShareMysTransactionPosition(ctx context.Context) {
	for {
		ShareMysTransactionCalculationPosition(ctx)
		time.Sleep(600 * time.Millisecond)
	}
}

// ShareMysTransactionCalculationPosition
//
//	@Description: 马股持仓缓存队列计算
//	@param ctx
func ShareMysTransactionCalculationPosition(ctx context.Context) {
	var wg sync.WaitGroup

	// 判定是否交易
	if CheckGlobalTread(flags.MysMarket) {
		time.Sleep(5 * time.Second)
		return
	}

	// TODO: 获取IPO未支付订单缓存
	userIdMap, _ := GetBotUserArrears(ctx)

	// 获取持仓缓存列表
	orderMap, err := Reds.HGetAll(context.Background(), setting.MarketShareMysPosition).Result()
	if err != nil {
		applogger.Error("%v ShareMysTransactionCalculationPosition.HGetAll:%v", common.ErrShareMys, err)
		return
	}

	for _, value := range orderMap {
		var msg = make(chan share.ShareMysTallyCache, 1)
		var entrust share.ShareMysTallyCache
		if err = json.Unmarshal([]byte(value), &entrust); err != nil {
			applogger.Error("%v ShareMysTransactionCalculationPosition.Unmarshal:%v", common.ErrShareMys, err)
			continue
		}

		// TODO: 判定是否未IPO未支付订单-不能平仓
		_, isOk := userIdMap[entrust.UserId]
		if isOk {
			continue
		}

		// 处理时间
		if !CheckShareSystemTime(entrust.ClosingTime) {
			applogger.Warn("马股:%v,平仓时间:%v暂未到达...", entrust.Symbol, entrust.ClosingTime)
			continue
		}

		// TODO: 实时获取强平阈值
		//entrust.Order.System.StrongFlatRatio = GetForcedClosureValue(flags.Mys, entrust.Symbol)

		// 股票实时价格
		var newPrice decimal.Decimal
		newPrice, err = GetShareMysPrice(entrust.Symbol)
		if err != nil {
			applogger.Warn("%v ShareMysTransactionCalculationPosition.Get:%v--%v", common.ErrShareMys, entrust.Symbol, err)
			continue
		}

		// 判定涨跌幅
		chg := GetShareChgMys(flags.Mys, entrust.Symbol) // 涨跌幅标识
		if share.CheckOrderMysByChg(setting.MarketShareMysEntrust, entrust, chg) {
			//applogger.Warn("当前股票%v订单在挂单状态已达到涨跌幅上限.", entrust.Symbol)
			continue
		}

		// 判定配额强平
		checkFlattening := GetStrongFlatteningThreshold(entrust.UserId)

		wg.Add(1)
		go func(value string) {
			resultMsg := ShareMysConcurrentComputingPosition(&entrust, newPrice, &wg, checkFlattening)
			msg <- resultMsg
		}(value)

		// 处理并发结果
		select {
		case resultMsg, _ := <-msg:
			switch resultMsg.Status {
			case flags.Position: // 持仓
				if !flags.CheckSetting {
					applogger.Debug("从信道接收到马股持仓信息:%v", resultMsg)
				}
			case flags.Close: // 平仓
				if !flags.CheckSetting {
					applogger.Debug("从信道接收到马股平仓信息:%v", resultMsg)
				}

				// TODO: 平仓操作(优化写入队列中等待平仓)
				//var byteStr []byte
				//byteStr, err = json.Marshal(resultMsg)
				//if err != nil {
				//	applogger.Error("%v ShareMysTransactionCalculationPosition.Marshal:%v", common.ShareMysError, err)
				//	continue
				//}
				//if err = uo.mqProducer.Position.PushNotice(byteStr); err != nil {
				//	applogger.Error("%v ShareMysTransactionCalculationPosition.PushNotice:%v", common.ShareMysError, err)
				//	continue
				//}

				if err = ShareMysLiquidationPosition(ctx, &resultMsg); err != nil {
					applogger.Error("%v ShareMysTransactionCalculationPosition.ShareMysLiquidationPosition:%v", common.ErrShareMys, err)
					continue
				}
			}
		}
	}

	wg.Wait()
	applogger.Info("马股持仓并发计算执行完毕......")
}

// ShareMysConcurrentComputingPosition
//
//	@Description: 马股持仓缓存列表业务处理
//	@param order
//	@param newPrice
//	@param bidPrice
//	@param askPrice
//	@param wg
//	@return tradedeal.ShareMysTallyCache
func ShareMysConcurrentComputingPosition(order *share.ShareMysTallyCache, newPrice decimal.Decimal, wg *sync.WaitGroup, checkFlattening bool) share.ShareMysTallyCache {
	var deleteCache, checkBool bool
	var dealPrice string
	var sellShort decimal.Decimal
	var openPrice = decimal.RequireFromString(order.OpenPrice)           // 限价
	var orderNumber = decimal.RequireFromString(order.Order.OrderNumber) // 仓位

	var pryNum decimal.Decimal
	lenPryNum := len(utils.ReplaceStr(order.Order.PryNum)) == 0
	if lenPryNum || order.Order.PryNum == flags.SetZero || order.Order.System.UserStatus != flags.SetThree || !utils.IsNumberInt(order.Order.PryNum) {
		pryNum = decimal.RequireFromString(flags.SetOne)
	} else {
		pryNum = decimal.RequireFromString(order.Order.PryNum)
	}

	// 下单判定设置(false无设置|true止盈止损)
	_, stopWinPrice, stopLossPrice, _ := StockMysVoteStopType(order.Order)

	switch order.Order.TradeType {
	case flags.TradeTypeBuy: // 买涨(强平)
		if !flags.CheckSetting {
			applogger.Debug("用户ID:%v,持仓开仓价格:%v,最新市价:%v,止盈:%v,止损:%v,原始状态:%v", order.UserId, openPrice, newPrice, stopWinPrice, stopLossPrice, order.Status)
		}
		// 买涨：挂止损止盈，止盈价必须<当前价，止损价必须>当前价;
		if !stopWinPrice.IsZero() {
			if stopWinPrice.Cmp(newPrice) < 0 {
				checkBool = true
				difference := newPrice.Mul(utils.Difference()) // 设置价差
				dealPrice = newPrice.Sub(difference).String()  // 平仓价格
			}
		}
		if !stopLossPrice.IsZero() {
			if stopLossPrice.Cmp(newPrice) > 0 {
				checkBool = true
				difference := newPrice.Mul(utils.Difference()) // 设置价差
				dealPrice = newPrice.Add(difference).String()  // 平仓价格
			}
		}
		// 强行平仓(做多)
		if !checkBool {
			sellShort = newPrice.Sub(openPrice).Mul(orderNumber)
		}
	case flags.TradeTypeSell: // 买跌(存在强行平仓)
		if !flags.CheckSetting {
			applogger.Debug("用户ID:%v,限价持仓买跌下单价格:%v,最新市价:%v,止盈:%v,止损:%v,原始状态:%v", order.UserId, openPrice, newPrice, stopWinPrice, stopLossPrice, order.Status)
		}
		// 买跌：挂止损止盈，止盈价必须>当前价，止损价必须<当前价;
		if !stopWinPrice.IsZero() {
			if stopWinPrice.Cmp(newPrice) > 0 {
				checkBool = true
				difference := newPrice.Mul(utils.Difference()) // 设置价差
				dealPrice = newPrice.Add(difference).String()  // 平仓价格
			}
		}
		if !stopLossPrice.IsZero() {
			if stopLossPrice.Cmp(newPrice) < 0 {
				checkBool = true
				difference := newPrice.Mul(utils.Difference()) // 设置价差
				dealPrice = newPrice.Sub(difference).String()  // 平仓价格
			}
		}
		// 强行平仓(做空)
		if !checkBool {
			sellShort = openPrice.Sub(newPrice).Mul(orderNumber)
		}
	}
	/* 强行平仓(做多|做空)
	1、做多:(最新成交价-开仓成交价)*仓位
	2、做空:(开仓成交价-最新成交价)*仓位
	3、当浮动盈亏 >= 保证金的70%订单强行平仓
	*/
	if sellShort.IsNegative() && checkFlattening {
		orderAmount := newPrice.Mul(decimal.RequireFromString(order.Order.OrderNumber)).Div(pryNum) // 保证金 = 当前价格 * 订单数量 / 杠杆
		floatPrice := orderAmount.Mul(decimal.RequireFromString(order.Order.System.StrongFlatRatio))
		if !flags.CheckSetting {
			applogger.Debug("强平保证金:%v,强平浮动70:%v,强行平仓判定:%v", orderAmount, floatPrice, sellShort.Abs().Cmp(floatPrice))
		}

		if sellShort.Abs().Cmp(floatPrice) >= 0 {
			checkBool = true
			difference := newPrice.Mul(utils.Difference()) // 设置价差
			switch order.Order.TradeType {
			case flags.TradeTypeBuy: //买涨
				dealPrice = newPrice.Sub(difference).String() // 平仓价格
			case flags.TradeTypeSell: // 买跌
				dealPrice = newPrice.Add(difference).String() // 平仓价格
			}
		}
	}
	// 判定交易状态
	if checkBool {
		deleteCache = true
		order.Status = flags.Close
	}
	if deleteCache {
		// 根据订单ID-删除持仓列表缓存订单
		if err := Reds.HDel(context.Background(), setting.MarketShareMysPosition, order.OrderId).Err(); err != nil {
			applogger.Error("%v ShareMysConcurrentComputingPosition.MarketShareMysPosition.HDel:%v", common.ErrShareMys, err)
		}
	}

	wg.Done()
	return share.ShareMysTallyCache{
		UserId:       order.UserId,      // 用户Id
		OrderId:      order.OrderId,     // 订单Id
		Symbol:       order.Symbol,      // 下单交易对
		ClosingPrice: dealPrice,         // 平仓价格
		Status:       order.Status,      // 订单状态
		Order:        order.Order,       // 下单信息
		ClosingTime:  order.ClosingTime, // 平仓时间
	}
}

// ShareMysMqClosingBusiness
//
//	@Description: TODO: 处理马股平仓消息队列
//	@param ctx
//	@param uo
func ShareMysMqClosingBusiness(ctx context.Context, uo *Data) {
	go func() {
		uo.mqConsumer.Position.ConsumerNotice()
	}()
	for {
		select {
		case message, ok := <-consumer.ConsumerPositionMq.ShareMysMq:
			if !ok {
				time.Sleep(5 * time.Second)
				applogger.Error("%v ShareMysMqClosingBusiness.ShareMysMq err....", common.ErrShareMys)
				continue
			}
			applogger.Info("接收到平仓信息:%v", string(message))

			// 持仓订单缓存数据序列化解析
			var resultMsg share.ShareMysTallyCache
			if err := json.Unmarshal(message, &resultMsg); err != nil {
				applogger.Error("%v ShareMysMqClosingBusiness.Unmarshal:%v", common.ErrShareMys, err)
				time.Sleep(5 * time.Second)
				continue
			}

			// 持仓平仓
			if err := ShareMysLiquidationPosition(ctx, &resultMsg); err != nil {
				applogger.Error("%v ShareMysMqClosingBusiness.ShareMysLiquidationPosition:%v", common.ErrShareMys, err)
				continue
			}
		default:
			// TODO: 是否处理空队列缓存的情况
			time.Sleep(2 * time.Second)
		}
	}
}

// ShareMysLiquidationPosition
//
//	@Description: 马股持仓-->平仓业务处理
//	@param ctx
//	@param order
//	@return error
func ShareMysLiquidationPosition(ctx context.Context, order *share.ShareMysTallyCache) error {
	// 1、平仓操作
	err := StockMysClosingOrder(ctx, Msql, order.OrderId, order.ClosingPrice, order.Order)
	if err != nil {
		applogger.Error("%v ShareMysLiquidationPosition.StockMysClosingOrder:%v", common.ErrShareMys, err)
		return err
	}

	// 2、平仓更新用户订阅订单状态
	userSubKey := virtual.OrderIdListKey(setting.ShareMysSubscribe, order.UserId)
	if err = UpdateShareMysSubscribeHashStatusByOrderId(order.OrderId, userSubKey, flags.Close, flags.SetNull); err != nil {
		applogger.Error("%v ShareMysLiquidationPosition.Position.ShareMysSubscribe:%v", common.ErrShareMys, err)
		return err
	}

	// 3、平仓更新管理员订阅订单状态
	if err = UpdateShareMysSubscribeHashStatusByOrderId(order.OrderId, setting.AdminShareMysSubscribe, flags.Close, flags.SetNull); err != nil {
		applogger.Error("%v ShareMysLiquidationPosition.Position.AdminShareMysSubscribe:%v", common.ErrShareMys, err)
		return err
	}

	return nil
}

// StockMysVoteStopType
//
//	@Description: 马股判定取值止损止盈
//	@param order
//	@return bool
//	@return decimal.Decimal
//	@return decimal.Decimal
//	@return error
func StockMysVoteStopType(order structure.ShareOrder) (bool, decimal.Decimal, decimal.Decimal, error) {
	var checkBool bool
	var stopWinPrice, stopLossPrice decimal.Decimal
	switch order.StopType {
	case flags.StopTypeNone: // 暂无设置止盈止损
		checkBool = false
	case flags.StopTypeSet: // 设置止盈止损
		checkBool = true
		stopWinPrice, _ = decimal.NewFromString(order.StopWinPrice)
		stopLossPrice, _ = decimal.NewFromString(order.StopLossPrice)
	default:
		return checkBool, stopWinPrice, stopLossPrice, flags.ErrContractThirteen
	}

	return checkBool, stopWinPrice, stopLossPrice, nil
}

// GetShareMysPrice
//
//	@Description: 查询马股股票价格
//	@param symbol
//	@return decimal.Decimal
//	@return decimal.Decimal
//	@return decimal.Decimal
//	@return error
func GetShareMysPrice(symbol string) (decimal.Decimal, error) {
	var err error
	var priceByte []byte
	var newPrice decimal.Decimal

	key := publicData.SymbolCache(flags.Mys, symbol, flags.TradeTypePrice)
	priceByte, err = memory.GetShareMysCache(key)
	if err != nil {
		return decimal.Decimal{}, err
	}

	newPrice, err = decimal.NewFromString(string(priceByte))
	if err != nil {
		return decimal.Decimal{}, err
	}

	applogger.Info("shareMysCode:%v,topIc:%v,shareMysPrice:%v", symbol, key, newPrice)

	return newPrice, nil
}