package data import ( "context" "encoding/json" "github.com/go-xorm/xorm" "github.com/shopspring/decimal" "matchmaking-system/internal/biz/structure" "matchmaking-system/internal/data/socket/publicData" "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" "time" orders "matchmaking-system/internal/data/convert" models "matchmaking-system/internal/pkg/model" ) // GetBotStockHkdTradeList // // @Description: 港股订单列表查询 // @receiver uo // @param ctx // @param pageSize // @param pageCount // @param userId // @param status // @return []*models.BotStockHkdTrade // @return int64 // @return error func (uo *userOrderRepo) GetBotStockHkdTradeList(ctx context.Context, pageSize, pageCount, userId, status int64) ([]*models.BotStockHkdTrade, int64, error) { totalCount, err := uo.data.mysqlDB.Table(flags.BotStockHkdTrade). Where("user_id = ?", userId). Where("status = ?", status). Count() if err != nil { return nil, 0, flags.ErrMySqlDB } if totalCount == 0 { return nil, 0, nil } var stockList []*models.BotStockHkdTrade if err = uo.data.mysqlDB.Table(flags.BotStockHkdTrade). Where("user_id = ?", userId). Where("status = ?", status). Limit(int(pageSize), int(pageCount)). Desc(GetOrderByStatusSort(status)). Find(&stockList); err != nil { return nil, 0, flags.ErrMySqlDB } var stockUpdateList []*models.BotStockHkdTrade for _, value := range stockList { value.KeepDecimal = GetKeepDecimal(flags.StockHKDSystemSetUpKey, value.StockId) value.StockName = GetStockName(flags.StockHKDSystemSetUpKey, value.StockId) stockUpdateList = append(stockUpdateList, value) } return stockUpdateList, totalCount, nil } // PlacingStockHkdOrders // // @Description: 港股下单 // @receiver uo // @param ctx // @param userId // @param order // @return string // @return error func (uo *userOrderRepo) PlacingStockHkdOrders(ctx context.Context, userId int64, order structure.ShareOrder) (string, error) { // 1、通知下单订阅 symbol := order.StockId timeValue := QuerySystemCache(flags.ShareHkdMarketType) // 市场开闭盘时间 if !utils.CheckInPlateTime(timeValue) { return flags.SetNull, flags.ErrStopTread } SubscribeHkdHSetCodeList(symbol, false) // 同步写入股票列表缓存 SubscribeHkdCloseNewPrice(symbol) // 同步股票实时行情价 // 判定是否存在欠款订单 userIdMap, _ := GetBotUserArrears(ctx) _, isOk := userIdMap[userId] if isOk { return flags.ErrPublicServe.Error(), nil } order = PryNumInit(order) // 杠杆初始化 subKey := publicData.SymbolCache(flags.Hkd, symbol, flags.TradeTypePrice) // 港股市场系统设置 if CheckGlobalTread(flags.HkdMarket) { return flags.SetNull, flags.ErrShareSeven } // 全局设置停止交易操作 var blockTime time.Time var userKey, adminKey, priceNew string if !CheckTypeStatus(order.Type) { userKey = setting.ShareHkdSubscribe adminKey = setting.AdminShareHkdSubscribe } else { // 判定:1、是否开启股票交易 2、是否在交易时间内 3、是否达到最小交易量 check, openPrice, minNum, endTime := CheckBlockTread(flags.HkdBlk, symbol) if check { return flags.SetNull, flags.ErrShareSeven } if minNum > utils.StringToInt(order.OrderNumber) { return flags.SetNull, flags.ErrShareSix } priceNew = openPrice // 开仓价格 blockTime = endTime // 平仓时间 userKey = setting.ShareBlkSubscribe adminKey = setting.AdminShareBlkSubscribe } flatRatio := GetCacheForcedClosure(flags.StockHKDSystemSetUpKey, symbol) // 设置强平阈值 leverStatus := SystemShareMarketLeverStatus(flags.StockMarketList, flags.HkdMarket, userId) // 股票市场系统设置 system := &structure.ShareSystem{ StrongFlatRatio: flatRatio.String(), // 强平阈值 Status: leverStatus.Status, // 杠杆状态 StockMin: leverStatus.StockMin, // 开启杠杆的最小值 LevelMin: leverStatus.LevelMin, // 杠杆最小值 LevelMax: leverStatus.LevelMax, // 杠杆最大值 UserStatus: leverStatus.UserStatus, // 用户开启状态 } order.System = system // 2、股票下单判定 // 下单判定设置(false无设置|true止盈止损) checkBool, stopWinPrice, stopLossPrice, err := StockHkdVoteStopType(order) if err != nil { applogger.Error("%v PlacingStockHkdOrders.StockVoteStopType:%v", common.ErrShareHkd, err) return flags.SetNull, err } // 下单判定设置(限价|市价) limitOrMarketPrice, err := uo.StockHkdVoteDealType(order) if err != nil { applogger.Error("%v PlacingStockHkdOrders.StockVoteDealType:%v", common.ErrShareHkd, err) return flags.SetNull, err } // 下单判定设置(买涨|买跌) if err = uo.VoteTradeType(order.TradeType, stopWinPrice, stopLossPrice, limitOrMarketPrice, checkBool); err != nil { applogger.Error("%v PlacingStockHkdOrders.VoteTradeType:%v", common.ErrShareHkd, err) return flags.SetNull, err } // 3、订单持久化录入(并处理资产信息) pryNum := SystemDealWithLever(order) orderId, err := uo.StockOrderHkdWriteDB(ctx, userId, order, limitOrMarketPrice, pryNum) if err != nil || len(orderId) == 0 { applogger.Error("%v PlacingStockHkdOrders.StockOrderHkdWriteDB:%v", common.ErrShareHkd, err) return flags.SetNull, err } // 4、处理订单信息 chg := GetShareChgHkd(flags.Hkd, order.StockId) // 涨跌幅标识值 closingTime := SystemTimeGenerate(flags.StockMarketList, flags.HkdMarket, time.Now()) // 开盘时间 marketStatus, shareOrder, err := share.ShareHkdCacheDeal(ctx, userId, orderId, subKey, order, chg, priceNew, closingTime, blockTime) if err != nil || shareOrder == nil { applogger.Error("%v PlacingStockHkdOrders.ShareHkdCacheDeal:%v", common.ErrShareHkd, err) return flags.SetNull, err } // 计算保证金 = 订单价格 * 订单数量 / 杠杆 shareOrder.Order.MarketMoney = limitOrMarketPrice.Mul(decimal.RequireFromString(order.OrderNumber)).Div(pryNum).String() if !flags.CheckSetting { applogger.Debug("订单ID:%v", shareOrder.OrderId) applogger.Debug("订单交易对:%v", shareOrder.Symbol) applogger.Debug("订单状态:%v", shareOrder.Status) applogger.Debug("订单用户ID:%v", shareOrder.UserId) applogger.Debug("开盘价:%v", shareOrder.OpenPrice) applogger.Debug("收盘价:%v", shareOrder.ClosingPrice) applogger.Debug("下单信息:%v", shareOrder.Order) } // 6、订单(挂单|持仓)缓存列表,等待队列计算 if err = share.ShareHkdHashUserOrder(Reds, marketStatus, shareOrder); err != nil { applogger.Error("%v PlacingStockHkdOrders.ShareHkdHashUserOrder:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrCacheDB } // 7、用户订阅股票订单信息 orderIdKey := virtual.OrderIdListKey(userKey, userId) if err = share.ShareHkdHashUserOrder(Reds, orderIdKey, shareOrder); err != nil { applogger.Error("%v PlacingStockHkdOrders.%v:%v", common.ErrShareHkd, userKey, err) return flags.SetNull, flags.ErrCacheDB } // 8、管理员订阅订单信息 if err = share.ShareHkdHashUserOrder(Reds, adminKey, shareOrder); err != nil { applogger.Error("%v PlacingStockHkdOrders.%v:%v", common.ErrShareHkd, adminKey, err) return flags.SetNull, flags.ErrCacheDB } return orderId, nil } // StockOrderHkdWriteDB // // @Description: 港股处理下单操作 // @receiver uo // @param ctx // @param userId // @param order // @return string // @return error func (uo *userOrderRepo) StockOrderHkdWriteDB(ctx context.Context, userId int64, order structure.ShareOrder, limitOrMarketPrice, lever decimal.Decimal) (string, error) { session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v StockOrderHkdWriteDB.Begin:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } // 获取当前下单用户的资产 var usableOld, frozenOld decimal.Decimal stockUsd, err := uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit) if err != nil || stockUsd == nil { applogger.Error("%v StockOrderHkdWriteDB.Hkd:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } usableOld = decimal.RequireFromString(stockUsd.UsableNum) frozenOld = decimal.RequireFromString(stockUsd.FrozenNum) // 检查用户订单下单资产 if usableOld.IsNegative() || frozenOld.IsNegative() { return flags.SetNull, flags.ErrShareOne } // 获取当前系统下单用户的非资产 var usableFOld, frozenFOld decimal.Decimal stockFUsd, err := uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, order.StockId) if err != nil { applogger.Error("%v StockOrderHkdWriteDB.FeiHkd:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } if stockFUsd != nil { usableFOld = decimal.RequireFromString(stockFUsd.UsableNum) frozenFOld = decimal.RequireFromString(stockFUsd.FrozenNum) } if !flags.CheckSetting { applogger.Debug("下单可用资产:%v", usableOld) applogger.Debug("下单冻结资产:%v", frozenOld) applogger.Debug("下单查询非资产账户:%v", stockFUsd) applogger.Debug("下单可用非资产:%v", usableFOld) applogger.Debug("下单冻结非资产:%v", frozenFOld) } // 整理当前下单信息 serviceCost := decimal.RequireFromString(order.ServiceCost) // 订单手续费 orderNumber := decimal.RequireFromString(order.OrderNumber) // 订单数量 marketMoney := limitOrMarketPrice.Mul(orderNumber).Div(lever) // 订单金额(杠杆下单) = 订单数量 * 交易价格 / 杠杆 totalMoney := marketMoney.Add(serviceCost) // 订单总金额 order.MarketMoney = marketMoney.String() // 市值金额即为保证金 // 检查订单ID是否存在,新增下单信息 var orderId string if !CheckTypeStatus(order.Type) { orderId, err = uo.VerifyBotStockHkdTradeOrderId(session) if err != nil { applogger.Error("%v StockOrderHkdWriteDB.VerifyBotStockHkdTradeOrderId:%v", common.ErrShareHkd, err) return flags.SetNull, err } trade := orders.BotStockHkdTrade(ctx, userId, orderId, order) if err = uo.CreateBotStockHkdTrade(session, trade); err != nil { applogger.Error("%v StockOrderHkdWriteDB.CreateBotStockHkdTrade:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } } else { orderId, err = uo.VerifyBotStockBlockTradeOrderId(session) if err != nil { applogger.Error("%v StockOrderHkdWriteDB.VerifyBotStockBlockTradeOrderId:%v", common.ErrShareHkd, err) return flags.SetNull, err } trade := orders.BotStockBlockTrade(ctx, userId, orderId, order) if err = uo.CreateBotStockBlockTrade(session, trade); err != nil { applogger.Error("%v StockOrderHkdWriteDB.CreateBotStockBlockTrade:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } } // 下单判定 residue := usableOld.Sub(totalMoney).Div(usableOld) if usableOld.Cmp(totalMoney) <= 0 || residue.Cmp(utils.DecimalsStrInt()) <= 0 { return flags.SetNull, flags.ErrPublicTow } var usableNew, frozenNew, usableFNew, frozenFNew decimal.Decimal usableNew = usableOld.Sub(totalMoney) // 可用资产 frozenNew = frozenOld.Add(totalMoney) // 冻结资产 if order.StockId != flags.ShareHkdBasicUnit { usableFNew = usableFOld.Add(orderNumber) // 可用非资产 frozenFNew = frozenFOld // 冻结非资产 } // 检查用户订单下单资产 if usableNew.IsNegative() { return flags.SetNull, flags.ErrPublicTow } // 下单更新资产信息 stockUSDModel := orders.UpdateBotUserStockHkd(ctx, usableNew.String(), frozenNew.String()) err = uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit, stockUSDModel) if err != nil { applogger.Error("%v StockOrderHkdWriteDB.UpdateBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } // 下单(新增|更新)非资产信息 if stockFUsd == nil { stockHKDModel := orders.CreateBotUserStockHkd(ctx, userId, order.StockId, usableFNew.String(), frozenFNew.String()) err = uo.CreateBotUserStockHkd(session, stockHKDModel) if err != nil { applogger.Error("%v StockOrderHkdWriteDB.CreateBotUserStockHkd:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } } else { stockHKDModel := orders.UpdateBotUserStockHkd(ctx, usableFNew.String(), frozenFNew.String()) err = uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, order.StockId, stockHKDModel) if err != nil { applogger.Error("%v StockOrderHkdWriteDB.UpdateHKD:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } } if err = session.Commit(); err != nil { applogger.Error("%v StockOrderHkdWriteDB.Commit:%v", common.ErrShareHkd, err) return flags.SetNull, flags.ErrMySqlDB } return orderId, nil } // StockHkdOpenOrder // // @Description: 港股系统开仓 // @param ctx // @param db // @param userId // @param orderId // @param openPrice // @param order // @return error func StockHkdOpenOrder(ctx context.Context, db *xorm.EngineGroup, userId int64, orderId, openPrice string, order structure.ShareOrder) error { session := db.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v StockHkdOpenOrder.NewSession:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 查询当前下单用户可用账户金额和冻结金额 var usable, frozen, usableNo, frozenNo decimal.Decimal stockUsd, err := Uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit) if err != nil || stockUsd == nil { applogger.Error("%v StockHkdOpenOrder.GetBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } usable = decimal.RequireFromString(stockUsd.UsableNum) // 用户可用资产 frozen = decimal.RequireFromString(stockUsd.FrozenNum) // 用户冻结资产 // 检查用户订单开仓资产 if usable.IsNegative() || frozen.IsNegative() { return flags.ErrPublicThree } if !flags.CheckSetting { applogger.Debug("下单可用金额:%v", usable) applogger.Debug("下单冻结金额:%v", frozen) } if order.StockId != flags.ShareHkdBasicUnit { stockFUsd, err := Uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, order.StockId) if err != nil { applogger.Error("%v StockHkdOpenOrder.GetBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } usableNo = decimal.RequireFromString(stockFUsd.UsableNum) // 用户非可用资产 frozenNo = decimal.RequireFromString(stockFUsd.FrozenNum) // 用户非冻结资产 } if !flags.CheckSetting { applogger.Debug("下单非可用金额:%v", usableNo) applogger.Debug("下单非冻结金额:%v", frozenNo) } // 转换股票下单信息 orderNumber := decimal.RequireFromString(order.OrderNumber) // 下单股数 marketMoney := decimal.RequireFromString(order.MarketMoney) // 下单金额 serviceCost := decimal.RequireFromString(order.ServiceCost) // 下单手续费 totalMoney := marketMoney.Add(serviceCost) // 下单总金额 = (下单手续费 + 下单金额) openOrderPrice := decimal.RequireFromString(openPrice) // 开盘价格 if !flags.CheckSetting { applogger.Debug("下单订单数量:%v", orderNumber) applogger.Debug("下单订单金额:%v", marketMoney) applogger.Debug("下单订单手续费:%v", serviceCost) applogger.Debug("下单订单总金额:%v", totalMoney) } // 手续费记录 openPriceNew := openOrderPrice.Mul(orderNumber).String() // 开仓手续费 = 开仓价格 * 股数 * 开仓手续费 cost, err := Uo.CalculateHandlingFeesShare(ctx, session, int(userId), flags.ShareHkdMarketType, flags.OpenBrokType, orderId, openPriceNew) if err != nil { applogger.Error("%v StockHkdOpenOrder.CalculateHandlingFeesShare:%v", common.ErrShareHkd, err) return err } openServiceCost := decimal.RequireFromString(cost) // 开仓手续费 openMarketMoney := openOrderPrice.Mul(orderNumber).Div(decimal.RequireFromString(order.PryNum)) // 开仓订单金额 = 开仓价格 * 开仓数量 / 杠杆 openTotalMoney := openMarketMoney.Add(openServiceCost) // 开仓总金额 = (开仓订单金额 + 开仓手续费) floatPrice := totalMoney.Sub(openTotalMoney) // 计算可用金额容差 = 下单总额 - 开仓总额 if !flags.CheckSetting { applogger.Debug("开仓订单价格:%v", openOrderPrice) applogger.Debug("开仓手续费:%v", openServiceCost) applogger.Debug("开仓总金额:%v", openTotalMoney) applogger.Debug("下单->开仓总金额容差值:%v", floatPrice) } var usableNew, frozenNew, usableNoNew, frozenNoNew decimal.Decimal usableNew = usable.Add(floatPrice) // 可用资产(处理容差值) frozenNew = frozen.Sub(totalMoney).Add(openMarketMoney) // 冻结资产 if order.StockId != flags.ShareHkdBasicUnit { usableNoNew = usableNo // 非可用资产 frozenNoNew = frozenNo // 非冻结资产 } // 检查用户订单开仓资产 if frozenNew.IsNegative() || usableNew.IsNegative() { return flags.ErrPublicThree } if !flags.CheckSetting { applogger.Debug("下单后用户可用资产:%v", usableNew) applogger.Debug("下单后用户冻结资产:%v", frozenNew) } // 处理资产信息 userStockUsd := orders.UpdateBotUserStockHkd(ctx, usableNew.String(), frozenNew.String()) if err = Uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit, userStockUsd); err != nil { applogger.Error("%v StockHkdOpenOrder.HKD.UpdateBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 更新非资产信息 if order.StockId != flags.ShareHkdBasicUnit { userStockFUsd := orders.UpdateBotUserStockHkd(ctx, usableNoNew.String(), frozenNoNew.String()) if err = Uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, order.StockId, userStockFUsd); err != nil { applogger.Error("%v StockHkdOpenOrder.HKD.UpdateBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } } // 返佣记录|资金信息|资金详情信息 if err = Uo.ShareHkdRebateCalculation(ctx, session, int(userId), flags.ShareHkdMarketType, flags.OpenBrokType, flags.OpenMRebate, cost, orderId, order.Type); err != nil { applogger.Error("%v StockHkdOpenOrder.ShareHkdRebateCalculation:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 处理订单信息(成交价|仓位|开仓时间|订单总金额|手续费|持仓状态) if !CheckTypeStatus(order.Type) { trade := orders.UpdateOpenBotStockHkdTrade(ctx, openPrice, orderNumber.String(), openTotalMoney.String(), openServiceCost.String(), openMarketMoney.String()) if err = Uo.UpdateBotStockHkdTradeByOrderId(session, orderId, trade); err != nil { applogger.Error("%v StockHkdOpenOrder.UpdateBotStockHkdTradeByOrderId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } var list []models.BotUserStockHkdLog qData0 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.CostMoney, flags.ShareHkdBasicUnit, NegativeValue(openServiceCost.String()), orderId) // 资金变动明细表(开仓手续费) qData1 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.Freeze, flags.ShareHkdBasicUnit, NegativeValue(openMarketMoney.String()), orderId) // 资金变动明细表(开仓保证金) qData2 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.ChangeInto, order.StockId, orderNumber.String(), orderId) // 资金变动明细表(非资产) list = append(list, qData0, qData1, qData2) // 批量写入数据信息 if err = Uo.CreatBotUserStockHkdLogList(session, list); err != nil { applogger.Error("%v StockHkdOpenOrder.CreatBotUserStockHkdLogList:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } } else { trade := orders.UpdateOpenBotStockBlockTrade(ctx, openPrice, orderNumber.String(), openTotalMoney.String(), openServiceCost.String(), openMarketMoney.String()) if err = Uo.UpdateBotStockBlockTradeByOrderId(session, orderId, trade); err != nil { applogger.Error("%v StockHkdOpenOrder.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } var list []models.BotUserStockBlockLog qData0 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.CostMoney, flags.ShareHkdBasicUnit, NegativeValue(openServiceCost.String()), orderId, order.Type) // 资金变动明细表(开仓手续费) qData1 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.Freeze, flags.ShareHkdBasicUnit, NegativeValue(openMarketMoney.String()), orderId, order.Type) // 资金变动明细表(开仓保证金) qData2 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.ChangeInto, order.StockId, orderNumber.String(), orderId, order.Type) // 资金变动明细表(非资产) list = append(list, qData0, qData1, qData2) // 批量写入数据信息 if err = Uo.CreatBotUserStockBlockLogList(session, list); err != nil { applogger.Error("%v StockHkdOpenOrder.CreatBotUserStockBlockLogList:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } } if err = session.Commit(); err != nil { applogger.Error("%v StockHkdOpenOrder.Commit:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } return nil } // StockHkdClosingOrder // // @Description: 港股系统平仓 // @param ctx // @param db // @param orderId // @param price // @param order // @return error func StockHkdClosingOrder(ctx context.Context, db *xorm.EngineGroup, orderId string, price string, order structure.ShareOrder) error { session := db.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v StockHkdClosingOrder.NewSession:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 查询订单信息 var userId int64 var dealPrice, orderNumber, totalAmount, serviceCost, marketMoney decimal.Decimal if !CheckTypeStatus(order.Type) { tread, err := Uo.GetBotStockHkdTradeByOrderIdOrStatus(session, orderId, flags.PositionStatus) if err != nil || tread == nil { applogger.Error("%v StockHkdClosingOrder.GetBotStockHkdTradeByOrderIdOrStatus:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } userId = int64(tread.UserId) // 用户Id orderNumber = decimal.RequireFromString(tread.OrderNumber) // 下单数量 dealPrice = decimal.RequireFromString(tread.DealPrice) // 开仓价格 totalAmount = decimal.RequireFromString(tread.OrderMoney) // 开仓总金额 serviceCost = decimal.RequireFromString(tread.ServiceCost) // 开仓手续费 marketMoney = decimal.RequireFromString(tread.MarketMoney) // 开仓订单金额 } else { tread, err := Uo.GetBotStockBlockTradeByOrderIdOrStatus(session, orderId, flags.PositionStatus) if err != nil || tread == nil { applogger.Error("%v StockHkdClosingOrder.GetBotStockBlockTradeByOrderIdOrStatus:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } userId = int64(tread.UserId) // 用户Id orderNumber = decimal.RequireFromString(tread.OrderNumber) // 下单数量 dealPrice = decimal.RequireFromString(tread.DealPrice) // 开仓价格 totalAmount = decimal.RequireFromString(tread.OrderMoney) // 开仓总金额 serviceCost = decimal.RequireFromString(tread.ServiceCost) // 开仓手续费 marketMoney = decimal.RequireFromString(tread.MarketMoney) // 开仓订单金额 } if !flags.CheckSetting { applogger.Debug("下单开仓用户Id:%v", userId) applogger.Debug("下单开仓开仓价格:%v", dealPrice) applogger.Debug("下单开仓订单数量:%v", orderNumber) applogger.Debug("下单开仓总金额:%v", totalAmount) applogger.Debug("下单开仓手续费:%v", serviceCost) applogger.Debug("下单开仓订单金额:%v", marketMoney) } // 查询当前下单用户可用账户金额和冻结金额 var usable, frozen, usableNo, frozenNo decimal.Decimal userStockUsd, err := Uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit) if err != nil { applogger.Error("%v StockHkdClosingOrder.GetBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } usable = decimal.RequireFromString(userStockUsd.UsableNum) // 用户可用资金 frozen = decimal.RequireFromString(userStockUsd.FrozenNum) // 用户冻结资金 // 检查用户订单平仓资产 if usable.IsNegative() || frozen.IsNegative() { return flags.ErrShareTow } if !flags.CheckSetting { applogger.Debug("下单可用金额:%v", usable) applogger.Debug("下单冻结金额:%v", frozen) } // 查询当前下单用户可用非账户和冻结金额 if order.StockId != flags.ShareHkdBasicUnit { userStockFUsd, err := Uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, order.StockId) if err != nil { applogger.Error("%v StockHkdClosingOrder.GetBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } usableNo = decimal.RequireFromString(userStockFUsd.UsableNum) // 用户非可用资金 frozenNo = decimal.RequireFromString(userStockFUsd.FrozenNum) // 用户非冻结资金 if !flags.CheckSetting { applogger.Debug("下单非可用金额:%v", usableNo) applogger.Debug("下单非冻结金额:%v", frozenNo) } } // 手续费处理 closePrice := decimal.RequireFromString(price) // 平仓价格 cloePriceCost := closePrice.Mul(orderNumber).String() // 平仓手续费 = 平仓价格 * 股数 * 手续费比例 cost, err := Uo.CalculateHandlingFeesShare(ctx, session, int(userId), flags.ShareHkdMarketType, flags.ClosingBrokType, orderId, cloePriceCost) if err != nil { applogger.Error("%v StockHkdClosingOrder.CalculateHandlingFeesShare:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } closeServiceCost := decimal.RequireFromString(cost) if !flags.CheckSetting { applogger.Debug("平仓价格:%v", closePrice) applogger.Debug("平仓手续费手续费:%v", closeServiceCost) } //已经实现盈亏计算公式(已平仓结算方式) //买涨 已经实现盈亏 = (平仓成交价-开仓成交价)*股数 //买跌 已经实现盈亏 = (开仓成交价-平仓成交价)*股数 var usableNew, frozenNew, usableNewNo, frozenNewNo, subPrice, resultPrice decimal.Decimal switch order.TradeType { case flags.TradeTypeBuy: // 买涨 subPrice = closePrice.Sub(dealPrice) resultPrice = subPrice.Mul(orderNumber) case flags.TradeTypeSell: // 买跌 subPrice = dealPrice.Sub(closePrice) resultPrice = subPrice.Mul(orderNumber) default: } // TODO: 判定亏损不能超过 = 保证金 + 平仓手续费 if resultPrice.IsNegative() { amountLoss := marketMoney.Sub(closeServiceCost) if resultPrice.Abs().Cmp(amountLoss) >= 0 { resultPrice = amountLoss.Neg() } } // 用户资产账户 usableNew = usable.Add(resultPrice).Sub(closeServiceCost).Add(marketMoney) // 可用资产 = 原可用资产 + ((正负)盈亏) - 平仓手续费 + 保证金 if frozen.Cmp(marketMoney) == -1 { frozenNew = frozen.Sub(frozen) // 冻结资产 = 原冻结资产 - 保证金 } else { frozenNew = frozen.Sub(marketMoney) // 冻结资产 = 原冻结资产 - 保证金 } // 用户非资产账户 if order.StockId != flags.ShareHkdBasicUnit { usableNewNo = usableNo.Sub(orderNumber) // 非可用资产 = 原非可用资产 frozenNewNo = frozenNo // 非冻结资产 = 原非冻结资产 - 订单数量 } // 检查用户订单平仓资产 if frozenNew.IsNegative() || usableNewNo.IsNegative() { return flags.ErrShareTow } if !flags.CheckSetting { applogger.Debug("可用资产:%v", usableNew) applogger.Debug("冻结资产:%v", frozenNew) applogger.Debug("可用非资产:%v", usableNewNo) applogger.Debug("冻结非资产:%v", frozenNewNo) } // 平仓(处理资产表(资产)) userStockUSDT := orders.UpdateBotUserStockHkd(ctx, usableNew.String(), frozenNew.String()) if err = Uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit, userStockUSDT); err != nil { applogger.Error("%v StockHkdClosingOrder.UpdateBotUserStockHkd:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 平仓(处理资产表(非资产)) if order.StockId != flags.ShareHkdBasicUnit { userStockFHKD := orders.UpdateBotUserStockHkd(ctx, usableNewNo.String(), frozenNewNo.String()) if err = Uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, order.StockId, userStockFHKD); err != nil { applogger.Error("%v StockHkdClosingOrder.UpdateBotUserStockHkd:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } } // 平仓(返佣记录表|资产表|资金变动表) if err = Uo.ShareHkdRebateCalculation(ctx, session, int(userId), flags.ShareHkdMarketType, flags.ClosingBrokType, flags.CloseMRebate, cost, orderId, order.Type); err != nil { applogger.Error("%v StockHkdClosingOrder.ShareHkdRebateCalculation:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 平仓(处理订单信息(平仓价|保证金|平仓|订单总金额|平仓手续费|完成订单)) if !CheckTypeStatus(order.Type) { trade := orders.UpdateCloseBotStockHkdTrade(ctx, price, cost) if err = Uo.UpdateBotStockHkdTradeByOrderId(session, orderId, trade); err != nil { applogger.Error("%v StockHkdClosingOrder.UpdateBotStockHkdTradeByOrderId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } var list []models.BotUserStockHkdLog qData0 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.Thaw, flags.ShareHkdBasicUnit, marketMoney.String(), orderId) // 平仓(资金变动明细(平仓保证金)) qData1 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.CostMoney, flags.ShareHkdBasicUnit, NegativeValue(closeServiceCost.String()), orderId) // 平仓(资金变动明细表(平仓手续费)) qData2 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.TransferOut, order.StockId, NegativeValue(orderNumber.String()), orderId) // 平仓(资金变动明细表(非资产)) list = append(list, qData0, qData1, qData2) // 平仓(资金变动明细表(正负盈亏)) if resultPrice.IsNegative() { qData3 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.TransferOut, flags.ShareHkdBasicUnit, resultPrice.String(), orderId) list = append(list, qData3) } else { qData4 := orders.CreatBotUserStockHkdLog(ctx, userId, flags.ChangeInto, flags.ShareHkdBasicUnit, resultPrice.String(), orderId) list = append(list, qData4) } // 平仓(批量写入交易订单日志信息) if err = Uo.CreatBotUserStockHkdLogList(session, list); err != nil { applogger.Error("%v StockHkdClosingOrder.CreatBotUserStockHkdLogList:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } } else { trade := orders.UpdateCloseBotStockBlockTrade(ctx, price, cost) if err = Uo.UpdateBotStockBlockTradeByOrderId(session, orderId, trade); err != nil { applogger.Error("%v StockHkdClosingOrder.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } var list []models.BotUserStockBlockLog qData0 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.Thaw, flags.ShareHkdBasicUnit, marketMoney.String(), orderId, order.Type) // 平仓(资金变动明细(平仓保证金)) qData1 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.CostMoney, flags.ShareHkdBasicUnit, NegativeValue(closeServiceCost.String()), orderId, order.Type) // 平仓(资金变动明细表(平仓手续费)) qData2 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.TransferOut, order.StockId, NegativeValue(orderNumber.String()), orderId, order.Type) // 平仓(资金变动明细表(非资产)) list = append(list, qData0, qData1, qData2) // 平仓(资金变动明细表(正负盈亏)) if resultPrice.IsNegative() { qData3 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.TransferOut, flags.ShareHkdBasicUnit, resultPrice.String(), orderId, order.Type) list = append(list, qData3) } else { qData4 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.ChangeInto, flags.ShareHkdBasicUnit, resultPrice.String(), orderId, order.Type) list = append(list, qData4) } // 平仓(批量写入交易订单日志信息) if err = Uo.CreatBotUserStockBlockLogList(session, list); err != nil { applogger.Error("%v StockHkdClosingOrder.CreatBotUserStockBlockLogList:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } } if err = session.Commit(); err != nil { applogger.Error("%v StockHkdClosingOrder.Commit:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } return nil } // UpdateBotStockHkdCancelByOrderId // // @Description: 港股撤单 // @receiver uo // @param ctx // @param orderId // @return bool // @return error func (uo *userOrderRepo) UpdateBotStockHkdCancelByOrderId(ctx context.Context, orderId string, typeStatus int64) (bool, error) { session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.Begin:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } // 获取订单信息 var userId int64 var stockId string var marketMoneyOld, serviceCostOld, orderNumber decimal.Decimal if !CheckTypeStatus(typeStatus) { stockTrade, err := uo.GetBotStockHkdTradeByOrderId(session, orderId) if err != nil || stockTrade == nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.GetBotStockHkdTradeByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } marketMoneyOld = decimal.RequireFromString(stockTrade.MarketMoney) // 订单Id serviceCostOld = decimal.RequireFromString(stockTrade.ServiceCost) // 手续费 orderNumber = decimal.RequireFromString(stockTrade.OrderNumber) // 订单数量 userId = int64(stockTrade.UserId) // 用户Id stockId = stockTrade.StockId // 下单交易对 } else { stockTrade, err := uo.GetBotStockBlockTradeByOrderId(session, orderId) if err != nil || stockTrade == nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.GetBotStockBlockTradeByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } marketMoneyOld = decimal.RequireFromString(stockTrade.MarketMoney) // 订单Id serviceCostOld = decimal.RequireFromString(stockTrade.ServiceCost) // 手续费 orderNumber = decimal.RequireFromString(stockTrade.OrderNumber) // 订单数量 userId = int64(stockTrade.UserId) // 用户Id stockId = stockTrade.StockId // 下单交易对 } // 获取用户资产信息 var usable, frozen, usableNo, frozenNo decimal.Decimal stockUsd, err := uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit) if err != nil || stockUsd == nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.HKD.GetBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return false, err } usable = decimal.RequireFromString(stockUsd.UsableNum) // 用户账户可用资产 frozen = decimal.RequireFromString(stockUsd.FrozenNum) // 用户账户冻结资产 // 检查用户订单撤单资产 if usable.IsNegative() || frozen.IsNegative() { return false, flags.ErrShareNine } if stockId != flags.ShareHkdBasicUnit { stockFUsd, err := uo.GetBotUserStockHkdByUserIdAndStockId(session, userId, stockId) if err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.GetBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return false, flags.ErrPublicFour } usableNo = decimal.RequireFromString(stockFUsd.UsableNum) // 用户非账户可用资产 frozenNo = decimal.RequireFromString(stockFUsd.FrozenNum) // 用户非账户冻结资产 } if !flags.CheckSetting { applogger.Debug("用户原始可用资金:%v", usable) applogger.Debug("用户原始冻结资金:%v", frozen) applogger.Debug("用户原始可用非资金:%v", usableNo) applogger.Debug("用户原始冻结非资金:%v", frozenNo) } // 更新订单信息 var entrustKey, userKey, adminKey string if !CheckTypeStatus(typeStatus) { botStockTrade := orders.BotStockHkdCancelByOrderId(ctx) if err = uo.UpdateBotStockHkdTradeByOrderId(session, orderId, botStockTrade); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.HKD.UpdateBotStockHkdTradeByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } // 缓存Key entrustKey = setting.MarketShareHkdEntrust userKey = setting.ShareHkdSubscribe adminKey = setting.AdminShareHkdSubscribe } else { botStockTrade := orders.BotStockBlockCancelByOrderId(ctx) if err = uo.UpdateBotStockBlockTradeByOrderId(session, orderId, botStockTrade); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.HKD.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } // 缓存Key entrustKey = setting.MarketShareBlkEntrust userKey = setting.ShareBlkSubscribe adminKey = setting.AdminShareBlkSubscribe } // Operational Spot Asset Table totalMoney := marketMoneyOld.Add(serviceCostOld) // 订单总金额 // update total asset data usableNew := usable.Add(totalMoney) // 用户总资产 frozenNew := frozen.Sub(totalMoney) // 用户冻结资产 usableNoNew := usableNo.Sub(orderNumber) // 用户非可用资产 frozenNoNew := frozenNo // 用户非冻结资产 // 检查用户订单撤单资产 if frozenNew.IsNegative() || usableNoNew.IsNegative() { return false, flags.ErrShareNine } if !flags.CheckSetting { applogger.Debug("用户撤单可用总资产:%v", usableNew) applogger.Debug("用户撤单冻结总资产:%v", frozenNew) applogger.Debug("用户非撤单可用总资产:%v", usableNoNew) applogger.Debug("用户非撤单冻结总资产:%v", frozenNoNew) } // 更新资产信息和非资产信息 userStockUSD := orders.UpdateBotUserStockHkd(ctx, usableNew.String(), frozenNew.String()) if err = uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, flags.ShareHkdBasicUnit, userStockUSD); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.UpdateBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } if stockId != flags.ShareHkdBasicUnit { userStockFHKD := orders.UpdateBotUserStockHkd(ctx, usableNoNew.String(), frozenNoNew.String()) if err = uo.UpdateBotUserStockHkdByUserIdAndStockId(session, userId, stockId, userStockFHKD); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.UpdateBotUserStockHkdByUserIdAndStockId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } } // 删除订单缓存列表 if err = Reds.HDel(context.Background(), entrustKey, orderId).Err(); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.HDel:%v", common.ErrShareHkd, err) return false, flags.ErrCacheDB } // 更新用户订阅订单缓存列表 userSubKey := virtual.OrderIdListKey(userKey, userId) if err = share.UpdateShareHkdHashByOrderId(Reds, orderId, userSubKey, flags.Cancel); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.%v:%v", common.ErrShareHkd, userKey, err) return false, flags.ErrCacheDB } // 更新管理员订阅订单缓存列表 if err = UpdateShareHkdSubscribeHashStatusByOrderId(orderId, adminKey, flags.Cancel, flags.SetNull); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.%v:%v", common.ErrShareHkd, adminKey, err) return false, flags.ErrCacheDB } if err = session.Commit(); err != nil { applogger.Error("%v UpdateBotStockHkdCancelByOrderId.Commit:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } return true, nil } // UpdateBotStockHkdStopByOrderId // // @Description: 港股设置止盈止损 // @receiver uo // @param ctx // @param order // @return bool // @return error func (uo *userOrderRepo) UpdateBotStockHkdStopByOrderId(ctx context.Context, order structure.StopOrder) (bool, error) { session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.NewSession:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } // 设置止盈止损 var userId int64 if order.StopType == 1 { // 股票下单信息 checkBool, stopWinPrice, stopLossPrice, err := uo.UpdateVoteStopType(order) if err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.UpdateVoteStopType:%v", common.ErrShareHkd, err) return false, err } if !flags.CheckSetting { applogger.Debug("当前止盈止损数据:%v,%v,%v", checkBool, stopWinPrice, stopLossPrice) } // 设置止盈止损判定 var tradeType int var limitOrMarketPrice decimal.Decimal if !CheckTypeStatus(order.Type) { trade, err := uo.GetBotStockHkdTradeByOrderIdOrStatus(session, order.OrderId, flags.PositionStatus) if err != nil || trade == nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.GetBotStockHkdTradeByOrderIdOrStatus:%v", common.ErrShareHkd, err) return false, flags.ErrShareFour } tradeType = trade.TradeType userId = int64(trade.UserId) // 下单判定设置(限价|市价|开仓价) limitOrMarketPrice, err = uo.UpdateShareHkdVoteDealType(trade) if err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.UpdateShareHkdVoteDealType:%v", common.ErrShareHkd, err) return false, err } } else { trade, err := uo.GetBotStockBlockTradeByOrderIdOrStatus(session, order.OrderId, flags.PositionStatus) if err != nil || trade == nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.GetBotStockBlockTradeByOrderIdOrStatus:%v", common.ErrShareHkd, err) return false, flags.ErrShareFour } tradeType = trade.TradeType userId = int64(trade.UserId) // 下单判定设置(限价|市价|开仓价) limitOrMarketPrice, err = uo.UpdateShareBlockVoteDealType(trade) if err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.UpdateShareBlockVoteDealType:%v", common.ErrShareHkd, err) return false, err } } if !flags.CheckSetting { applogger.Debug("当前开仓价格:%v", limitOrMarketPrice) } // 下单判定设置(买涨|买跌) if err = uo.VoteTradeType(int64(tradeType), stopWinPrice, stopLossPrice, limitOrMarketPrice, checkBool); err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.VoteTradeType:%v", common.ErrShareHkd, err) return false, err } } else { order.StopLossPrice = decimal.Zero.String() order.StopWinPrice = decimal.Zero.String() } // TODO: 获取存在IPO未支付订单的用户 userIdMap, _ := GetBotUserArrears(ctx) _, ok := userIdMap[userId] if ok { return false, flags.ErrPublicServe } // 更新订单表 var positionKey string if !CheckTypeStatus(order.Type) { botStockTrade := orders.BotStockStopHkdByOrderId(ctx, order) if err = uo.UpdateBotStockHkdTradeByOrderId(session, order.OrderId, botStockTrade); err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.UpdateBotStockHkdTradeByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } positionKey = setting.MarketShareHkdPosition } else { botStockTrade := orders.BotStockBlockStopByOrderId(ctx, order) if err = uo.UpdateBotStockBlockTradeByOrderId(session, order.OrderId, botStockTrade); err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } positionKey = setting.MarketShareBlkPosition } // 修改止損止盈持仓缓存队列数据 if err = share.UpdateShareHkdStopByOrderId(Reds, order, positionKey, flags.Position); err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.UpdateShareHkdStopByOrderId:%v", common.ErrShareHkd, err) return false, flags.ErrCacheDB } if err = session.Commit(); err != nil { applogger.Error("%v UpdateBotStockHkdStopByOrderId.Commit:%v", common.ErrShareHkd, err) return false, flags.ErrMySqlDB } return true, nil } // UpdateBotStockHkdClosingByOrderId // // @Description: 港股用户平仓 // @receiver uo // @param ctx // @param orderId // @return bool // @return error func (uo *userOrderRepo) UpdateBotStockHkdClosingByOrderId(ctx context.Context, orderId string, typeStatus int64) (bool, error) { // 下单判定-冷静期停止所有交易 if CheckGlobalTread(flags.HkdMarket) { return false, flags.ErrStopTread } // 1、查询持仓缓存列表数据并清理对应缓存 var positionKey, userKey, adminKey string if !CheckTypeStatus(typeStatus) { positionKey = setting.MarketShareHkdPosition userKey = setting.ShareHkdSubscribe adminKey = setting.AdminShareHkdSubscribe } else { positionKey = setting.MarketShareBlkPosition userKey = setting.ShareBlkSubscribe adminKey = setting.AdminShareBlkSubscribe } var entrustCache *share.ShareHkdTallyCache entrust, err := Reds.HGet(context.Background(), positionKey, orderId).Result() if err != nil { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.HGet:%v", common.ErrShareHkd, err) return false, flags.ErrShareFive } var entrustJson share.ShareHkdTallyCache if err = json.Unmarshal([]byte(entrust), &entrustJson); err != nil { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.Unmarshal:%v", common.ErrShareHkd, err) return false, flags.ErrCacheDB } if orderId == entrustJson.OrderId { entrustCache = &entrustJson } // 3、判定是否存在平仓订单 if entrustCache == nil { return false, flags.ErrShareFive } // TODO: 获取存在IPO未支付订单的用户 userIdMap, _ := GetBotUserArrears(ctx) _, ok := userIdMap[entrustJson.UserId] if ok { return false, flags.ErrPublicServe } // 市场设置交易时间(T+0,T+1,T+2s......) if !CheckShareSystemTime(entrustCache.ClosingTime) { return false, flags.ErrStopTread } if !flags.CheckSetting { applogger.Debug("手动平仓:%v", entrustCache) } // 2、获取当前最新价格 var closingPrice decimal.Decimal subKey := publicData.SymbolCache(flags.Hkd, entrustCache.Symbol, flags.TradeTypePrice) closingPrice, err = uo.GetShareHkdTheLatestPrice(subKey, entrustCache.Order.TradeType) if err != nil { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.GetShareHkdTheLatestPrice:%v", common.ErrShareHkd, err) return false, flags.ErrStopTread } // 4、平仓处理 if err = StockHkdClosingOrder(ctx, Msql, orderId, closingPrice.String(), entrustCache.Order); err != nil { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.StockHkdClosingOrder:%v", common.ErrShareHkd, err) return false, err } // 5、清理持仓订单信息 var removedCount int64 removedCount, err = Reds.HDel(context.Background(), positionKey, orderId).Result() if err != nil || removedCount == 0 { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.HDel:%v", common.ErrShareHkd, err) return false, flags.ErrCacheDB } // 6、平仓更新订阅缓存订单状态 userSubKey := virtual.OrderIdListKey(userKey, entrustCache.UserId) if err = UpdateShareHkdSubscribeHashStatusByOrderId(orderId, userSubKey, flags.Close, flags.SetNull); err != nil { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.ShareHkdSubscribe:%v", common.ErrShareHkd, err) return false, flags.ErrCacheDB } // 7、平仓更新管理员订阅缓存订单状态 if err = UpdateShareHkdSubscribeHashStatusByOrderId(orderId, adminKey, flags.Close, flags.SetNull); err != nil { applogger.Error("%v UpdateBotStockHkdClosingByOrderId.%v:%v", common.ErrShareHkd, adminKey, err) return false, flags.ErrCacheDB } return true, nil } // UpdateBotStockHkdAllClosingByOrderId // // @Description: 港股用户一键平仓 // @receiver uo // @param ctx // @param userId // @return error func (uo *userOrderRepo) UpdateBotStockHkdAllClosingByOrderId(ctx context.Context, userId int64) error { // 下单判定-冷静期停止所有交易 if CheckGlobalTread(flags.HkdMarket) { return flags.ErrStopTread } // TODO: 获取存在IPO未支付订单的用户 userIdMap, _ := GetBotUserArrears(ctx) _, isOk := userIdMap[userId] if isOk { return flags.ErrPublicServe } // 1、查询当前用户所有订单 orderMap, err := uo.GetBotStockHkdTradeByUserId(userId) if err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.GetBotStockHkdTradeByUserId:%v", common.ErrShareHkd, err) return flags.ErrMySqlDB } // 2、查询持仓缓存列表数据-1、清理持仓列表缓存 2、处理平仓数据 3、清理订单ID缓存列表 entrust, err := Reds.HGetAll(context.Background(), setting.MarketShareHkdPosition).Result() if err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.HGetAll:%v", common.ErrShareHkd, err) return flags.ErrShareFive } // 3、循环检查股票交易对最新价格(有一个查询不到既不能平仓) var entrustList []share.ShareHkdTallyCache closePriceMap := make(map[string]decimal.Decimal) for key, value := range entrust { var entrustJson share.ShareHkdTallyCache if err = json.Unmarshal([]byte(value), &entrustJson); err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.Unmarshal:%v", common.ErrShareHkd, err) return flags.ErrStopTread } _, ok := orderMap[entrustJson.OrderId] if ok { var closingPrice decimal.Decimal subKey := publicData.SymbolCache(flags.Hkd, entrustJson.Symbol, flags.TradeTypePrice) closingPrice, err = uo.GetShareHkdTheLatestPrice(subKey, entrustJson.Order.TradeType) if err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.GetShareHkdTheLatestPrice:%v", common.ErrShareHkd, err) return flags.ErrStopTread } if !flags.CheckSetting { applogger.Debug("订阅Key:%v,最新价格:%v,订单ID:%v", subKey, closingPrice, entrustJson.OrderId) } closePriceMap[key] = closingPrice entrustList = append(entrustList, entrustJson) } } // 4、股票一键平仓 for _, value := range entrustList { closePrice, ok := closePriceMap[value.OrderId] if ok { // 平仓 if err = StockHkdClosingOrder(ctx, uo.data.mysqlDB, value.OrderId, closePrice.String(), value.Order); err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.StockHkdClosingOrder:%v", common.ErrShareHkd, err) return err } // 清理持仓缓存列表 var removedCount int64 removedCount, err = Reds.HDel(context.Background(), setting.MarketShareHkdPosition, value.OrderId).Result() if err != nil || removedCount == 0 { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.HDel:%v", common.ErrShareHkd, err) return flags.ErrCacheDB } // 平仓更新用户订阅订单状态 userSubKey := virtual.OrderIdListKey(setting.ShareHkdSubscribe, value.UserId) if err = UpdateShareHkdSubscribeHashStatusByOrderId(value.OrderId, userSubKey, flags.Close, flags.SetNull); err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.ShareHkdSubscribe:%v", common.ErrShareHkd, err) return flags.ErrCacheDB } // 平仓更新管理员订阅订单状态 if err = UpdateShareHkdSubscribeHashStatusByOrderId(value.OrderId, setting.AdminShareHkdSubscribe, flags.Close, flags.SetNull); err != nil { applogger.Error("%v UpdateBotStockHkdAllClosingByOrderId.AdminShareHkdSubscribe:%v", common.ErrShareHkd, err) return flags.ErrCacheDB } } } return nil } // GetBotStockHkdTradeByUserId // // @Description: // @receiver uo // @param ctx // @param userId // @return map[string]string // @return error func (uo *userOrderRepo) GetBotStockHkdTradeByUserId(userId int64) (map[string]string, error) { var botStockHkdTrade []models.BotStockHkdTrade if err := uo.data.mysqlDB.Table(flags.BotStockHkdTrade). Where("user_id = ?", userId). Where("status = 1"). Find(&botStockHkdTrade); err != nil { return nil, err } botMap := make(map[string]string) for _, value := range botStockHkdTrade { botMap[value.OrderId] = value.OrderId } return botMap, nil } // GetShareHkdTheLatestPrice // // @Description: // @receiver uo // @param ctx // @param subKey // @param tradeType // @return decimal.Decimal // @return error func (uo *userOrderRepo) GetShareHkdTheLatestPrice(subKey string, tradeType int64) (decimal.Decimal, error) { priceNew, err := share.ShareHkdSubMarketPrice(subKey) if err != nil { applogger.Error("%v GetShareHkdTheLatestPrice.ShareHkdSubMarketPrice:%v", common.ErrShareHkd, err) return decimal.Decimal{}, err } priceS := decimal.RequireFromString(priceNew) var openPrice decimal.Decimal difference := priceS.Mul(utils.Difference()) // 设置价差 switch tradeType { case flags.TradeTypeBuy: // 买涨 openPrice = priceS.Sub(difference) // 开仓价格 case flags.TradeTypeSell: // 买跌 openPrice = priceS.Add(difference) // 开仓价格 default: } return openPrice, nil } // GetBotStockHkdTradeByOrderId // // @Description: // @receiver uo // @param ctx // @param session // @param orderId // @return *models.BotStockHkdTrade // @return error func (uo *userOrderRepo) GetBotStockHkdTradeByOrderId(session *xorm.Session, orderId string) (*models.BotStockHkdTrade, error) { var botStockHkdTrade []models.BotStockHkdTrade if err := session.Table(flags.BotStockHkdTrade). Where("order_id = ?", orderId). Find(&botStockHkdTrade); err != nil { return nil, err } for _, value := range botStockHkdTrade { return &value, nil } return nil, nil } // GetBotUserStockHkdByUserIdAndStockId // // @Description: // @receiver uo // @param ctx // @param session // @param userId // @param stockId // @return *models.BotUserStock // @return error func (uo *userOrderRepo) GetBotUserStockHkdByUserIdAndStockId(session *xorm.Session, userId int64, stockId string) (*models.BotUserStockHkd, error) { var stockUSD []models.BotUserStockHkd if err := session.Table(flags.BotUserStockHkd). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Find(&stockUSD); err != nil || len(stockUSD) < 0 { return nil, err } for _, value := range stockUSD { return &value, nil } return nil, nil } // UpdateBotStockHkdTradeByOrderId // // @Description: // @receiver uo // @param ctx // @param session // @param orderId // @param stock // @return error func (uo *userOrderRepo) UpdateBotStockHkdTradeByOrderId(session *xorm.Session, orderId string, stock models.BotStockHkdTrade) error { checkInt, err := session.Table(flags.BotStockHkdTrade). Where("order_id = ?", orderId). Update(&stock) if err != nil || checkInt < 0 { return err } return nil } // CreateBotStockHkdTrade // // @Description: // @receiver uo // @param ctx // @param session // @param trade // @return error func (uo *userOrderRepo) CreateBotStockHkdTrade(session *xorm.Session, trade models.BotStockHkdTrade) error { _, err := session.Table(flags.BotStockHkdTrade).Insert(&trade) if err != nil { return err } return nil } // CreateBotUserStockHkd // // @Description: // @receiver uo // @param ctx // @param session // @param stock // @return error func (uo *userOrderRepo) CreateBotUserStockHkd(session *xorm.Session, stock models.BotUserStockHkd) error { _, err := session.Table(flags.BotUserStockHkd).Insert(&stock) if err != nil { return err } return nil } // UpdateBotUserStockHkdByUserIdAndStockId // // @Description: // @receiver uo // @param ctx // @param session // @param userId // @param stockId // @param stock // @return error func (uo *userOrderRepo) UpdateBotUserStockHkdByUserIdAndStockId(session *xorm.Session, userId int64, stockId string, stock models.BotUserStockHkd) error { checkNum, err := session.Table(flags.BotUserStockHkd). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Update(&stock) if err != nil || checkNum < 0 { return err } return nil } // StockHkdVoteDealType // // @Description: // @receiver uo // @param ctx // @param order // @return decimal.Decimal // @return error func (uo *userOrderRepo) StockHkdVoteDealType(order structure.ShareOrder) (decimal.Decimal, error) { var limitOrMarketPrice decimal.Decimal switch order.DealType { case flags.DealTypeLimited: // 限价 if len(order.LimitPrice) != 0 { limitOrMarketPrice = decimal.RequireFromString(order.LimitPrice) } case flags.DealTypeMarket: // 市价 if len(order.MarketPrice) != 0 { limitOrMarketPrice = decimal.RequireFromString(order.MarketPrice) } default: return decimal.Decimal{}, flags.ErrPublicFive } return limitOrMarketPrice, nil } // UpdateShareHkdVoteDealType // // @Description: // @receiver uo // @param ctx // @param order // @return decimal.Decimal // @return error func (uo *userOrderRepo) UpdateShareHkdVoteDealType(order *models.BotStockHkdTrade) (decimal.Decimal, error) { var limitOrMarketPrice decimal.Decimal switch order.DealType { case flags.DealTypeLimited: // 限价 if len(order.LimitPrice) != 0 { limitOrMarketPrice = decimal.RequireFromString(order.LimitPrice) } case flags.DealTypeMarket: // 市价 if len(order.MarketPrice) != 0 { limitOrMarketPrice = decimal.RequireFromString(order.MarketPrice) } default: return decimal.Decimal{}, flags.ErrPublicFive } if len(order.DealPrice) != 0 { dealPrice := decimal.RequireFromString(order.DealPrice) if !dealPrice.IsZero() { limitOrMarketPrice = dealPrice } } return limitOrMarketPrice, nil } // VerifyBotStockHkdTradeOrderId // // @Description: // @receiver uo // @param ctx // @param session // @return string // @return error func (uo *userOrderRepo) VerifyBotStockHkdTradeOrderId(session *xorm.Session) (string, error) { var orderId string for { // 生成订单ID orderId = orders.CreateRandCodeOrder(10) var orderList []models.BotStockHkdTrade err := session.Table(flags.BotStockHkdTrade). Where("order_id = ?", orderId). Find(&orderList) if err != nil || len(orderList) > 0 { applogger.Error("%v VerifyBotStockHkdTradeOrderId.Find:%v", common.ErrShareHkd, err) continue } break } return orderId, nil } // GetBotStockHkdTradeByOrderIdOrStatus // // @Description: // @receiver uo // @param ctx // @param session // @param orderId // @param status // @return *models.BotStockTrade // @return error func (uo *userOrderRepo) GetBotStockHkdTradeByOrderIdOrStatus(session *xorm.Session, orderId string, status int) (*models.BotStockHkdTrade, error) { var botStockHkdTrade []models.BotStockHkdTrade if err := session.Table(flags.BotStockHkdTrade). Where("order_id = ?", orderId). Where("`status` = ?", status). Find(&botStockHkdTrade); err != nil { return nil, err } for _, value := range botStockHkdTrade { return &value, nil } return nil, nil }