You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1534 lines
59 KiB
1534 lines
59 KiB
package data
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"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"
|
|
)
|
|
|
|
// GetBotStockMysTradeList
|
|
//
|
|
// @Description: 马股订单列表查询
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param pageSize
|
|
// @param pageCount
|
|
// @param userId
|
|
// @param status
|
|
// @return []*models.BotStockMysTrade
|
|
// @return int64
|
|
// @return error
|
|
func (uo *userOrderRepo) GetBotStockMysTradeList(ctx context.Context, pageSize, pageCount, userId, status int64) ([]*models.BotStockMysTrade, int64, error) {
|
|
totalCount, err := uo.data.mysqlDB.Table(flags.BotStockMysTrade).
|
|
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.BotStockMysTrade
|
|
if err = uo.data.mysqlDB.Table(flags.BotStockMysTrade).
|
|
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.BotStockMysTrade
|
|
for _, value := range stockList {
|
|
value.NumericCode = GetCacheNumericCode(fmt.Sprintf("%v%v", flags.StockMGSystemSetUpKey, value.StockId))
|
|
value.KeepDecimal = GetKeepDecimal(flags.StockMGSystemSetUpKey, value.StockId)
|
|
value.StockName = GetStockName(flags.StockMGSystemSetUpKey, value.StockId)
|
|
stockUpdateList = append(stockUpdateList, value)
|
|
}
|
|
|
|
return stockUpdateList, totalCount, nil
|
|
}
|
|
|
|
// PlacingStockMysOrders
|
|
//
|
|
// @Description: 马股下单
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param userId
|
|
// @param order
|
|
// @return string
|
|
// @return error
|
|
func (uo *userOrderRepo) PlacingStockMysOrders(ctx context.Context, userId int64, order structure.ShareOrder) (string, error) {
|
|
// 1、通知下单订阅
|
|
symbol := order.StockId
|
|
timeValue := QuerySystemCache(flags.ShareMysMarketType) // 市场开闭盘时间
|
|
if !utils.CheckInPlateTime(timeValue) {
|
|
return flags.SetNull, flags.ErrStopTread
|
|
}
|
|
SubscribeMysHSetCodeList(symbol, false) // 同步写入股票列表缓存
|
|
SubscribeMysCloseNewPrice(symbol) // 同步股票实时行情价
|
|
|
|
// 判定是否存在欠款订单
|
|
userIdMap, _ := GetBotUserArrears(ctx)
|
|
_, isOk := userIdMap[userId]
|
|
if isOk {
|
|
return flags.ErrPublicServe.Error(), nil
|
|
}
|
|
|
|
order = PryNumInit(order) // 杠杆初始化
|
|
subKey := publicData.SymbolCache(flags.Mys, symbol, flags.TradeTypePrice)
|
|
|
|
// 马股市场系统设置
|
|
if CheckGlobalTread(flags.MysMarket) {
|
|
return flags.SetNull, flags.ErrShareSeven
|
|
}
|
|
|
|
// 股票交易系统设置
|
|
var blockTime time.Time
|
|
var priceNew, userKey, adminKey string
|
|
if !CheckTypeStatus(order.Type) {
|
|
userKey = setting.ShareMysSubscribe
|
|
adminKey = setting.AdminShareMysSubscribe
|
|
} else {
|
|
// 判定:1、是否开启股票交易 2、是否在交易时间内 3、是否达到最小交易量
|
|
check, openPrice, minNum, endTime := CheckBlockTread(flags.MysBlk, 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.StockMGSystemSetUpKey, symbol) // 强平阈值
|
|
leverStatus := SystemShareMarketLeverStatus(flags.StockMarketList, flags.MysMarket, 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、股票下单判定(1>下单判定设置(false无设置|true止盈止损) 2、下单判定设置(限价|市价) 3、下单判定设置(买涨|买跌))
|
|
checkBool, stopWinPrice, stopLossPrice, err := StockMysVoteStopType(order)
|
|
if err != nil {
|
|
applogger.Error("%v PlacingStockMysOrders.StockMysVoteStopType:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, err
|
|
}
|
|
limitOrMarketPrice, err := uo.StockMysVoteDealType(order)
|
|
if err != nil {
|
|
applogger.Error("%v PlacingStockMysOrders.StockMysVoteDealType:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, err
|
|
}
|
|
if err = uo.VoteTradeType(order.TradeType, stopWinPrice, stopLossPrice, limitOrMarketPrice, checkBool); err != nil {
|
|
applogger.Error("%v PlacingStockMysOrders.VoteTradeType:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, err
|
|
}
|
|
|
|
// 3、订单持久化录入(并处理资产信息)
|
|
pryNum := SystemDealWithLever(order) // 杠杆值处理
|
|
orderId, err := uo.StockMysOrderWriteDB(ctx, userId, order, limitOrMarketPrice, pryNum)
|
|
if err != nil || len(orderId) == 0 {
|
|
applogger.Error("%v PlacingStockMysOrders.StockMysOrderWriteDB:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, err
|
|
}
|
|
|
|
// 4、处理订单信息
|
|
chg := GetShareChgMys(flags.Mys, order.StockId) // 涨跌幅标识值
|
|
closingTime := SystemTimeGenerate(flags.StockMarketList, flags.MysMarket, time.Now()) // 开盘时间
|
|
marketStatus, shareOrder, err := share.ShareMysCacheDeal(ctx, userId, orderId, subKey, order, chg, priceNew, closingTime, blockTime)
|
|
if err != nil || shareOrder == nil {
|
|
applogger.Error("%v PlacingStockMysOrders.ShareMysCacheDeal:%v", common.ErrShareMys, 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.ShareMysHashUserOrder(Reds, marketStatus, shareOrder); err != nil {
|
|
applogger.Error("%v PlacingStockMysOrders.ShareMysHashUserOrder:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrCacheDB
|
|
}
|
|
|
|
// 7、用户订阅订单信息
|
|
orderIdKey := virtual.OrderIdListKey(userKey, userId)
|
|
if err = share.ShareMysHashUserOrder(Reds, orderIdKey, shareOrder); err != nil {
|
|
applogger.Error("%v PlacingStockMysOrders.%v:%v", common.ErrShareMys, userKey, err)
|
|
return flags.SetNull, flags.ErrCacheDB
|
|
}
|
|
|
|
// 8、管理员订阅订单信息
|
|
if err = share.ShareMysHashUserOrder(Reds, adminKey, shareOrder); err != nil {
|
|
applogger.Error("%v PlacingStockMysOrders.%v:%v", common.ErrShareMys, adminKey, err)
|
|
return flags.SetNull, flags.ErrCacheDB
|
|
}
|
|
|
|
return orderId, nil
|
|
}
|
|
|
|
// StockMysOrderWriteDB
|
|
//
|
|
// @Description: 马股下单操作
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param userId
|
|
// @param order
|
|
// @return string
|
|
// @return error
|
|
func (uo *userOrderRepo) StockMysOrderWriteDB(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 StockMysOrderWriteDB.Begin:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
|
|
// 获取当前下单用户的资产
|
|
var usableOld, frozenOld decimal.Decimal
|
|
stockMyr, err := uo.GetBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit)
|
|
if err != nil || stockMyr == nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.MYR:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
usableOld = decimal.RequireFromString(stockMyr.UsableNum)
|
|
frozenOld = decimal.RequireFromString(stockMyr.FrozenNum)
|
|
|
|
// 检查用户订单资产
|
|
if usableOld.IsZero() || usableOld.IsNegative() || frozenOld.IsNegative() {
|
|
return flags.SetNull, flags.ErrShareOne
|
|
}
|
|
|
|
// 获取当前系统下单用户的非资产
|
|
var usableFOld, frozenFOld decimal.Decimal
|
|
stockFMyr, err := uo.GetBotUserStockMysByUserIdAndStockId(session, userId, order.StockId)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.FMyr:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
if stockFMyr != nil {
|
|
usableFOld = decimal.RequireFromString(stockFMyr.UsableNum)
|
|
frozenFOld = decimal.RequireFromString(stockFMyr.FrozenNum)
|
|
}
|
|
if !flags.CheckSetting {
|
|
applogger.Debug("下单可用资产:%v", usableOld)
|
|
applogger.Debug("下单冻结资产:%v", frozenOld)
|
|
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.VerifyBotStockMysTradeOrderId(session)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.VerifyBotStockMysTradeOrderId:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, err
|
|
}
|
|
trade := orders.BotStockMysTrade(ctx, userId, orderId, order)
|
|
if err = uo.CreateBotStockMysTrade(session, trade); err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.CreateBotStockMysTrade:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
} else {
|
|
orderId, err = uo.VerifyBotStockBlockTradeOrderId(session)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.VerifyBotStockBlockTradeOrderId:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, err
|
|
}
|
|
trade := orders.BotStockBlockTrade(ctx, userId, orderId, order)
|
|
if err = uo.CreateBotStockBlockTrade(session, trade); err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.CreateBotStockBlockTrade:%v", common.ErrShareMys, 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.ShareMysBasicUnit {
|
|
usableFNew = usableFOld.Add(orderNumber) // 可用非资产
|
|
frozenFNew = frozenFOld // 冻结非资产
|
|
}
|
|
|
|
// 检查用户订单资产
|
|
if usableNew.IsNegative() {
|
|
return flags.SetNull, flags.ErrPublicTow
|
|
}
|
|
|
|
// 下单更新资产信息
|
|
stockMYRModel := orders.UpdateBotUserStockMys(ctx, usableNew.String(), frozenNew.String())
|
|
err = uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit, stockMYRModel)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
|
|
// 下单(新增|更新)非资产信息
|
|
if stockFMyr == nil {
|
|
stockFMYRModel := orders.CreateBotUserStockMys(ctx, userId, order.StockId, usableFNew.String(), frozenFNew.String())
|
|
err = uo.CreateBotUserStockMysFMYR(session, stockFMYRModel)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.CreateBotUserStockFUSD:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
} else {
|
|
stockFMYRModel := orders.UpdateBotUserStockMys(ctx, usableFNew.String(), frozenFNew.String())
|
|
err = uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, order.StockId, stockFMYRModel)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.UpdateFUSDL:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
}
|
|
|
|
if err = session.Commit(); err != nil {
|
|
applogger.Error("%v StockMysOrderWriteDB.Commit:%v", common.ErrShareMys, err)
|
|
return flags.SetNull, flags.ErrMySqlDB
|
|
}
|
|
|
|
return orderId, nil
|
|
}
|
|
|
|
// StockMysOpenOrder
|
|
//
|
|
// @Description: 马股系统开仓
|
|
// @param ctx
|
|
// @param db
|
|
// @param userId
|
|
// @param orderId
|
|
// @param openPrice
|
|
// @param order
|
|
// @return error
|
|
func StockMysOpenOrder(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 StockMysOpenOrder.NewSession:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 查询当前下单用户可用账户金额和冻结金额
|
|
var usable, frozen, usableNo, frozenNo decimal.Decimal
|
|
stockMyr, err := Uo.GetBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit)
|
|
if err != nil || stockMyr == nil {
|
|
applogger.Error("%v StockMysOpenOrder.GetBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
usable = decimal.RequireFromString(stockMyr.UsableNum) // 用户可用资产
|
|
frozen = decimal.RequireFromString(stockMyr.FrozenNum) // 用户冻结资产
|
|
|
|
// 检查用户订单开仓资产
|
|
if usable.IsNegative() || frozen.IsNegative() {
|
|
return flags.ErrPublicThree
|
|
}
|
|
|
|
if order.StockId != flags.ShareMysBasicUnit {
|
|
stockFMyr, err := Uo.GetBotUserStockMysByUserIdAndStockId(session, userId, order.StockId)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.GetBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
usableNo = decimal.RequireFromString(stockFMyr.UsableNum) // 用户非可用资产
|
|
frozenNo = decimal.RequireFromString(stockFMyr.FrozenNum) // 用户非冻结资产
|
|
}
|
|
if !flags.CheckSetting {
|
|
applogger.Debug("下单可用金额:%v", usable)
|
|
applogger.Debug("下单冻结金额:%v", frozen)
|
|
applogger.Debug("下单非可用金额:%v", usableNo)
|
|
applogger.Debug("下单非冻结金额:%v", frozenNo)
|
|
}
|
|
|
|
// 转换下单信息
|
|
orderNumber := decimal.RequireFromString(order.OrderNumber) // 下单股数
|
|
markerMoney := decimal.RequireFromString(order.MarketMoney) // 下单金额
|
|
serviceCost := decimal.RequireFromString(order.ServiceCost) // 下单手续费
|
|
totalMoney := markerMoney.Add(serviceCost) // 下单总金额 = 下单手续费 + 下单金额
|
|
openOrderPrice := decimal.RequireFromString(openPrice) // 开盘价格
|
|
if !flags.CheckSetting {
|
|
applogger.Debug("下单订单数量:%v", orderNumber)
|
|
applogger.Debug("下单订单金额:%v", markerMoney)
|
|
applogger.Debug("下单订单手续费:%v", serviceCost)
|
|
applogger.Debug("下单订单总金额:%v", totalMoney)
|
|
}
|
|
|
|
// 手续费记录
|
|
openPriceNew := openOrderPrice.Mul(orderNumber).String() // 开仓手续费 = 开仓价格 * 股数 * 开仓手续费
|
|
cost, err := Uo.CalculateHandlingFeesShare(ctx, session, int(userId), flags.ShareMysMarketType, flags.OpenBrokType, orderId, openPriceNew)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.CalculateHandlingFeesShare:%v", common.ErrShareMys, err)
|
|
return err
|
|
}
|
|
openServiceCost := decimal.RequireFromString(cost) // 开仓手续费
|
|
openMarkerMoney := openOrderPrice.Mul(orderNumber).Div(decimal.RequireFromString(order.PryNum)) // 开仓金额 = 开仓价格 * 股数 / 杠杆
|
|
openTotalMoney := openMarkerMoney.Add(openServiceCost) // 开仓总金额 = 开仓金额 + 开仓手续费
|
|
floatPrice := totalMoney.Sub(openTotalMoney) // 计算容差 = 下单总额 - 开仓总额
|
|
if !flags.CheckSetting {
|
|
applogger.Debug("开仓订单价格:%v", openOrderPrice)
|
|
applogger.Debug("开仓手续费:%v", openServiceCost)
|
|
applogger.Debug("开仓总金额:%v", openTotalMoney)
|
|
applogger.Debug("开仓订单金额:%v", openMarkerMoney)
|
|
applogger.Debug("下单->开仓总金额容差值:%v", floatPrice)
|
|
}
|
|
|
|
var usableNew, frozenNew, usableNoNew, frozenNoNew decimal.Decimal
|
|
usableNew = usable.Add(floatPrice) // 可用资产(处理容差值)
|
|
frozenNew = frozen.Sub(totalMoney).Add(openMarkerMoney) // 冻结资产(处理容差值)---> 当前冻结金额 = 原冻结金额 - 下单总额
|
|
if order.StockId != flags.ShareMysBasicUnit {
|
|
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.UpdateBotUserStockMys(ctx, usableNew.String(), frozenNew.String())
|
|
if err = Uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit, userStockUsd); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.MYR.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 更新非资产信息
|
|
if order.StockId != flags.ShareMysBasicUnit {
|
|
userStockFMyr := orders.UpdateBotUserStockMys(ctx, usableNoNew.String(), frozenNoNew.String())
|
|
if err = Uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, order.StockId, userStockFMyr); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
}
|
|
|
|
// 返佣记录|资金信息|资金详情信息
|
|
if err = Uo.ShareMysRebateCalculation(ctx, session, int(userId), flags.ShareMysMarketType, flags.OpenBrokType, flags.OpenMRebate, cost, orderId, order.Type); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.ShareUsRebateCalculation:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 处理订单信息(成交价|仓位|开仓时间|订单总金额|手续费|持仓状态)
|
|
if !CheckTypeStatus(order.Type) {
|
|
trade := orders.UpdateOpenBotStockMysTrade(ctx, openPrice, orderNumber.String(), openTotalMoney.String(), openServiceCost.String(), openMarkerMoney.String())
|
|
if err = Uo.UpdateBotStockMysTradeByOrderId(session, orderId, trade); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.UpdateBotStockMysTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
var list []models.BotUserStockMysLog
|
|
qData0 := orders.CreatBotUserStockMysLog(ctx, userId, flags.CostMoney, flags.ShareMysBasicUnit, NegativeValue(openServiceCost.String()), orderId) // 资金变动明细表(开仓手续费)
|
|
qData1 := orders.CreatBotUserStockMysLog(ctx, userId, flags.Freeze, flags.ShareMysBasicUnit, NegativeValue(openMarkerMoney.String()), orderId) // 资金变动明细表(开仓保证金)
|
|
qData2 := orders.CreatBotUserStockMysLog(ctx, userId, flags.ChangeInto, order.StockId, orderNumber.String(), orderId) // 资金变动明细表(非资产)
|
|
list = append(list, qData0, qData1, qData2)
|
|
|
|
// 批量写入数据信息
|
|
if err = Uo.CreatBotUserStockMysLogList(session, list); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.CreatBotUserStockMysLogList:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
} else {
|
|
trade := orders.UpdateOpenBotStockBlockTrade(ctx, openPrice, orderNumber.String(), openTotalMoney.String(), openServiceCost.String(), openMarkerMoney.String())
|
|
if err = Uo.UpdateBotStockBlockTradeByOrderId(session, orderId, trade); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
var list []models.BotUserStockBlockLog
|
|
qData0 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.CostMoney, flags.ShareMysBasicUnit, NegativeValue(openServiceCost.String()), orderId, order.Type) // 资金变动明细表(开仓手续费)
|
|
qData1 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.Freeze, flags.ShareMysBasicUnit, NegativeValue(openMarkerMoney.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 StockMysOpenOrder.CreatBotUserStockBlockLogList:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
}
|
|
|
|
if err = session.Commit(); err != nil {
|
|
applogger.Error("%v StockMysOpenOrder.Commit:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// StockMysClosingOrder
|
|
//
|
|
// @Description: 马股系统平仓
|
|
// @param ctx
|
|
// @param db
|
|
// @param orderId
|
|
// @param price
|
|
// @param order
|
|
// @return error
|
|
func StockMysClosingOrder(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 StockMysClosingOrder.NewSession:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 查询订单信息
|
|
var userId int64
|
|
var dealPrice, orderNumber, totalAmount, serviceCost, marketMoney decimal.Decimal
|
|
if !CheckTypeStatus(order.Type) {
|
|
tread, err := Uo.GetBotStockMysTradeByOrderIdOrStatus(session, orderId, flags.PositionStatus)
|
|
if err != nil || tread == nil {
|
|
applogger.Error("%v StockMysClosingOrder.GetBotStockMysTradeByOrderIdOrStatus:%v", common.ErrShareMys, 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 StockMysClosingOrder.GetBotStockBlockTradeByOrderIdOrStatus:%v", common.ErrShareMys, 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
|
|
userStockMyr, err := Uo.GetBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.GetBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
usable = decimal.RequireFromString(userStockMyr.UsableNum) // 用户可用资金
|
|
frozen = decimal.RequireFromString(userStockMyr.FrozenNum) // 用户非冻结资金
|
|
|
|
// 检查用户订单平仓资产
|
|
if usable.IsNegative() || frozen.IsNegative() {
|
|
return flags.ErrShareTow
|
|
}
|
|
// 查询当前下单用户可用非账户和冻结金额
|
|
if order.StockId != flags.ShareMysBasicUnit {
|
|
userStockFUsd, err := Uo.GetBotUserStockMysByUserIdAndStockId(session, userId, order.StockId)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.GetBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
usableNo = decimal.RequireFromString(userStockFUsd.UsableNum) // 用户非可用资金
|
|
frozenNo = decimal.RequireFromString(userStockFUsd.FrozenNum) // 用户非冻结资金
|
|
}
|
|
if !flags.CheckSetting {
|
|
applogger.Debug("下单可用金额:%v", usable)
|
|
applogger.Debug("下单冻结金额:%v", frozen)
|
|
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.ShareMysMarketType, flags.ClosingBrokType, orderId, cloePriceCost)
|
|
if err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.CalculateHandlingFeesShare:%v", common.ErrShareMys, err)
|
|
return err
|
|
}
|
|
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.ShareMysBasicUnit {
|
|
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)
|
|
}
|
|
|
|
// 平仓(处理资产表(资产))
|
|
userStockMYR := orders.UpdateBotUserStockMys(ctx, usableNew.String(), frozenNew.String())
|
|
if err = Uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit, userStockMYR); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.USD.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 平仓(处理资产表(非资产))
|
|
if order.StockId != flags.ShareMysBasicUnit {
|
|
userStockFMYR := orders.UpdateBotUserStockMys(ctx, usableNewNo.String(), frozenNewNo.String())
|
|
if err = Uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, order.StockId, userStockFMYR); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
}
|
|
|
|
// 平仓(返佣记录表|资产表|资金变动表)
|
|
if err = Uo.ShareMysRebateCalculation(ctx, session, int(userId), flags.ShareMysMarketType, flags.ClosingBrokType, flags.CloseMRebate, cost, orderId, order.Type); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.ShareMysRebateCalculation:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 平仓(处理订单信息(平仓价|保证金|平仓|订单总金额|平仓手续费|完成订单))
|
|
if !CheckTypeStatus(order.Type) {
|
|
trade := orders.UpdateCloseBotStockMysTrade(ctx, price, cost)
|
|
if err = Uo.UpdateBotStockMysTradeByOrderId(session, orderId, trade); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.UpdateBotStockMysTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
var list []models.BotUserStockMysLog
|
|
qData0 := orders.CreatBotUserStockMysLog(ctx, userId, flags.Thaw, flags.ShareMysBasicUnit, marketMoney.String(), orderId) // 平仓(资金变动明细表(开仓保证金))
|
|
qData1 := orders.CreatBotUserStockMysLog(ctx, userId, flags.CostMoney, flags.ShareMysBasicUnit, NegativeValue(closeServiceCost.String()), orderId) // 平仓(资金变动明细表(-平仓手续费))
|
|
qData2 := orders.CreatBotUserStockMysLog(ctx, userId, flags.TransferOut, order.StockId, NegativeValue(orderNumber.String()), orderId) // 平仓(资金变动明细表(-非资产))
|
|
list = append(list, qData0, qData1, qData2)
|
|
// 平仓(资金变动明细表(正负盈亏))
|
|
if resultPrice.IsNegative() {
|
|
qData3 := orders.CreatBotUserStockMysLog(ctx, userId, flags.TransferOut, flags.ShareMysBasicUnit, resultPrice.String(), orderId)
|
|
list = append(list, qData3)
|
|
} else {
|
|
qData4 := orders.CreatBotUserStockMysLog(ctx, userId, flags.ChangeInto, flags.ShareMysBasicUnit, resultPrice.String(), orderId)
|
|
list = append(list, qData4)
|
|
}
|
|
// 平仓(批量写入交易订单日志信息)
|
|
if err = Uo.CreatBotUserStockMysLogList(session, list); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.CreatBotUserStockMysLogList:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
} else {
|
|
trade := orders.UpdateCloseBotStockBlockTrade(ctx, price, cost)
|
|
if err = Uo.UpdateBotStockBlockTradeByOrderId(session, orderId, trade); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
var list []models.BotUserStockBlockLog
|
|
qData0 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.Thaw, flags.ShareMysBasicUnit, marketMoney.String(), orderId, order.Type) // 平仓(资金变动明细表(开仓保证金))
|
|
qData1 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.CostMoney, flags.ShareMysBasicUnit, 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.ShareMysBasicUnit, resultPrice.String(), orderId, order.Type)
|
|
list = append(list, qData3)
|
|
} else {
|
|
qData4 := orders.CreatBotUserStockBlockLog(ctx, userId, flags.ChangeInto, flags.ShareMysBasicUnit, resultPrice.String(), orderId, order.Type)
|
|
list = append(list, qData4)
|
|
}
|
|
// 平仓(批量写入交易订单日志信息)
|
|
if err = Uo.CreatBotUserStockBlockLogList(session, list); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.CreatBotUserStockBlockLogList:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
}
|
|
|
|
if err = session.Commit(); err != nil {
|
|
applogger.Error("%v StockMysClosingOrder.Commit:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UpdateBotStockMysCancelByOrderId
|
|
//
|
|
// @Description: 马股撤单
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param orderId
|
|
// @return bool
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateBotStockMysCancelByOrderId(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 UpdateBotStockMysCancelByOrderId.Begin:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
|
|
// 获取订单信息
|
|
var userId int64
|
|
var stockId string
|
|
var marketMoneyOld, serviceCostOld, orderNumber decimal.Decimal
|
|
if !CheckTypeStatus(typeStatus) {
|
|
stockTrade, err := uo.GetBotStockMysTradeByOrderId(session, orderId)
|
|
if err != nil || stockTrade == nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.GetBotStockMysTradeByOrderId:%v", common.ErrShareMys, 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 UpdateBotStockMysCancelByOrderId.GetBotStockBlockTradeByOrderId:%v", common.ErrShareMys, 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.GetBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit)
|
|
if err != nil || stockUsd == nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.USD.GetBotUserStockByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
usable = decimal.RequireFromString(stockUsd.UsableNum) // 用户账户可用资产
|
|
frozen = decimal.RequireFromString(stockUsd.FrozenNum) // 用户账户冻结资产
|
|
|
|
// 检查用户订单撤单资产
|
|
if frozen.IsNegative() || usable.IsNegative() {
|
|
return false, flags.ErrShareNine
|
|
}
|
|
if stockId != flags.ShareMysBasicUnit {
|
|
stockFMyr, err := uo.GetBotUserStockMysByUserIdAndStockId(session, userId, stockId)
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.GetBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrPublicFour
|
|
}
|
|
usableNo = decimal.RequireFromString(stockFMyr.UsableNum) // 用户非账户可用资产
|
|
frozenNo = decimal.RequireFromString(stockFMyr.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.BotStockMysCancelByOrderId(ctx)
|
|
if err = uo.UpdateBotStockMysTradeByOrderId(session, orderId, botStockTrade); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.MYS.UpdateBotStockMysTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
// 缓存Key
|
|
entrustKey = setting.MarketShareMysEntrust
|
|
userKey = setting.ShareMysSubscribe
|
|
adminKey = setting.AdminShareMysSubscribe
|
|
} else {
|
|
botStockTrade := orders.BotStockBlockCancelByOrderId(ctx)
|
|
if err = uo.UpdateBotStockBlockTradeByOrderId(session, orderId, botStockTrade); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.MYS.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareMys, 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
|
|
var usableNew, frozenNew, usableNoNew, frozenNoNew decimal.Decimal
|
|
usableNew = usable.Add(totalMoney) // 用户总资产
|
|
frozenNew = frozen.Sub(totalMoney) // 用户冻结资产
|
|
if stockId != flags.ShareMysBasicUnit {
|
|
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)
|
|
}
|
|
|
|
// 更新资产信息和非资产信息
|
|
userStockMYR := orders.UpdateBotUserStockMys(ctx, usableNew.String(), frozenNew.String())
|
|
if err = uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, flags.ShareMysBasicUnit, userStockMYR); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
if stockId != flags.ShareMysBasicUnit {
|
|
userStockFMYT := orders.UpdateBotUserStockMys(ctx, usableNoNew.String(), frozenNoNew.String())
|
|
if err = uo.UpdateBotUserStockMysByUserIdAndStockId(session, userId, stockId, userStockFMYT); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.UpdateBotUserStockMysByUserIdAndStockId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
}
|
|
|
|
// 删除订单缓存列表
|
|
if err = Reds.HDel(context.Background(), entrustKey, orderId).Err(); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.HDel:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
// 更新用户订阅订单缓存列表
|
|
userSubKey := virtual.OrderIdListKey(userKey, userId)
|
|
if err = share.UpdateShareMysHashByOrderId(Reds, orderId, userSubKey, flags.Cancel); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.%v:%v", common.ErrShareMys, userKey, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
// 更新管理员订阅订单缓存列表
|
|
if err = UpdateShareMysSubscribeHashStatusByOrderId(orderId, adminKey, flags.Cancel, flags.SetNull); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.%v:%v", common.ErrShareMys, adminKey, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
if err = session.Commit(); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysCancelByOrderId.Commit:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// UpdateBotStockMysStopByOrderId
|
|
//
|
|
// @Description: 马股设置止盈止损
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param order
|
|
// @return bool
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateBotStockMysStopByOrderId(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 UpdateBotStockMysStopByOrderId.NewSession:%v", common.ErrShareMys, 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 UpdateBotStockMysStopByOrderId.UpdateVoteStopType:%v", common.ErrShareMys, 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.GetBotStockMysTradeByOrderIdOrStatus(session, order.OrderId, flags.PositionStatus)
|
|
if err != nil || trade == nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.GetBotStockMysTradeByOrderIdOrStatus:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrShareFour
|
|
}
|
|
tradeType = trade.TradeType
|
|
userId = int64(trade.UserId)
|
|
// 下单判定设置(限价|市价|开仓价)
|
|
limitOrMarketPrice, err = uo.UpdateShareMysVoteDealType(trade)
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.UpdateShareMysVoteDealType:%v", common.ErrShareMys, err)
|
|
return false, err
|
|
}
|
|
} else {
|
|
trade, err := uo.GetBotStockBlockTradeByOrderIdOrStatus(session, order.OrderId, flags.PositionStatus)
|
|
if err != nil || trade == nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.GetBotStockBlockTradeByOrderIdOrStatus:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrShareFour
|
|
}
|
|
tradeType = trade.TradeType
|
|
userId = int64(trade.UserId)
|
|
// 下单判定设置(限价|市价|开仓价)
|
|
limitOrMarketPrice, err = uo.UpdateShareBlockVoteDealType(trade)
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.UpdateShareBlockVoteDealType:%v", common.ErrShareMys, 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 UpdateBotStockMysStopByOrderId.VoteTradeType:%v", common.ErrShareMys, 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.BotStockMysStopByOrderId(ctx, order)
|
|
if err = uo.UpdateBotStockMysTradeByOrderId(session, order.OrderId, botStockTrade); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.UpdateBotStockMysTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
positionKey = setting.MarketShareMysPosition
|
|
} else {
|
|
botStockTrade := orders.BotStockBlockStopByOrderId(ctx, order)
|
|
if err = uo.UpdateBotStockBlockTradeByOrderId(session, order.OrderId, botStockTrade); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.UpdateBotStockBlockTradeByOrderId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
positionKey = setting.MarketShareBlkPosition
|
|
}
|
|
|
|
// 修改持倉缓存队列
|
|
if err = share.UpdateShareMysStopByOrderId(Reds, order, positionKey, flags.Position); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.UpdateShareMysStopByOrderId:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
if err = session.Commit(); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysStopByOrderId.Commit:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrMySqlDB
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// UpdateBotStockMysClosingByOrderId
|
|
//
|
|
// @Description: 马股用户平仓
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param orderId
|
|
// @return bool
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateBotStockMysClosingByOrderId(ctx context.Context, orderId string, typeStatus int64) (bool, error) {
|
|
// 下单判定-冷静期停止所有交易
|
|
if CheckGlobalTread(flags.MysMarket) {
|
|
return false, flags.ErrStopTread
|
|
}
|
|
|
|
// 1、查询持仓缓存列表数据并清理对应缓存
|
|
var positionKey, userKey, adminKey string
|
|
if !CheckTypeStatus(typeStatus) {
|
|
positionKey = setting.MarketShareMysPosition
|
|
userKey = setting.ShareMysSubscribe
|
|
adminKey = setting.AdminShareMysSubscribe
|
|
} else {
|
|
positionKey = setting.MarketShareBlkPosition
|
|
userKey = setting.ShareBlkSubscribe
|
|
adminKey = setting.AdminShareBlkSubscribe
|
|
}
|
|
var entrustCache *share.ShareMysTallyCache
|
|
entrust, err := Reds.HGet(context.Background(), positionKey, orderId).Result()
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysClosingByOrderId.HGet:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrShareFive
|
|
}
|
|
var entrustJson share.ShareMysTallyCache
|
|
if err = json.Unmarshal([]byte(entrust), &entrustJson); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysClosingByOrderId.Unmarshal:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
if orderId == entrustJson.OrderId {
|
|
entrustCache = &entrustJson
|
|
}
|
|
// 2、判定是否存在平仓订单
|
|
if entrustCache == nil {
|
|
return false, err
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
|
|
// 3、获取当前最新价格
|
|
var closingPrice decimal.Decimal
|
|
subKey := publicData.SymbolCache(flags.Mys, entrustCache.Symbol, flags.TradeTypePrice)
|
|
closingPrice, err = uo.GetShareMysTheLatestPrice(subKey, entrustCache.Order.TradeType)
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysClosingByOrderId.GetShareMysTheLatestPrice:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrStopTread
|
|
}
|
|
|
|
// 4、平仓处理
|
|
if err = StockMysClosingOrder(ctx, Msql, orderId, closingPrice.String(), entrustCache.Order); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysClosingByOrderId.StockMysClosingOrder:%v", common.ErrShareMys, 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 UpdateBotStockMysClosingByOrderId.HDel:%v", common.ErrShareMys, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
// 6、平仓更新订阅缓存订单状态
|
|
userSubKey := virtual.OrderIdListKey(userKey, entrustCache.UserId)
|
|
if err = UpdateShareMysSubscribeHashStatusByOrderId(orderId, userSubKey, flags.Close, flags.SetNull); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysClosingByOrderId.%v:%v", common.ErrShareMys, userKey, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
// 7、平仓更新管理员订阅缓存订单状态
|
|
if err = UpdateShareMysSubscribeHashStatusByOrderId(orderId, adminKey, flags.Close, flags.SetNull); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysClosingByOrderId.%v:%v", common.ErrShareMys, adminKey, err)
|
|
return false, flags.ErrCacheDB
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// UpdateBotStockMysAllClosingByOrderId
|
|
//
|
|
// @Description: 马股用户一键平仓
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param userId
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateBotStockMysAllClosingByOrderId(ctx context.Context, userId int64) error {
|
|
// 下单判定-冷静期停止所有交易
|
|
if CheckGlobalTread(flags.MysMarket) {
|
|
return flags.ErrStopTread
|
|
}
|
|
|
|
// TODO: 获取存在IPO未支付订单的用户
|
|
userIdMap, _ := GetBotUserArrears(ctx)
|
|
_, isOk := userIdMap[userId]
|
|
if isOk {
|
|
return flags.ErrPublicServe
|
|
}
|
|
|
|
// 1、查询当前用户所有订单
|
|
orderMap, err := uo.GetBotStockMysTradeByUserId(userId)
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.GetBotStockMysTradeByUserId:%v", common.ErrShareMys, err)
|
|
return flags.ErrMySqlDB
|
|
}
|
|
|
|
// 2、查询持仓缓存列表数据-1、清理持仓列表缓存 2、处理平仓数据 3、清理订单ID缓存列表
|
|
entrust, err := Reds.HGetAll(context.Background(), setting.MarketShareMysPosition).Result()
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.HGetAll:%v", common.ErrShareMys, err)
|
|
return flags.ErrShareFive
|
|
}
|
|
|
|
// 3、循环检查股票交易对最新价格(有一个查询不到既不能平仓)
|
|
var entrustList []share.ShareMysTallyCache
|
|
closePriceMap := make(map[string]decimal.Decimal)
|
|
for key, value := range entrust {
|
|
var entrustJson share.ShareMysTallyCache
|
|
if err = json.Unmarshal([]byte(value), &entrustJson); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.Unmarshal:%v", common.ErrShareMys, err)
|
|
return flags.ErrPosition
|
|
}
|
|
_, ok := orderMap[entrustJson.OrderId]
|
|
if ok {
|
|
var closingPrice decimal.Decimal
|
|
subKey := publicData.SymbolCache(flags.Mys, entrustJson.Symbol, flags.TradeTypePrice)
|
|
closingPrice, err = uo.GetShareMysTheLatestPrice(subKey, entrustJson.Order.TradeType)
|
|
if err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.GetShareMysTheLatestPrice:%v", common.ErrShareMys, err)
|
|
return flags.ErrPosition
|
|
}
|
|
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 = StockMysClosingOrder(ctx, uo.data.mysqlDB, value.OrderId, closePrice.String(), value.Order); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.StockMysClosingOrder:%v", common.ErrShareMys, err)
|
|
return err
|
|
}
|
|
|
|
// 清理持仓缓存列表
|
|
var removedCount int64
|
|
removedCount, err = Reds.HDel(context.Background(), setting.MarketShareMysPosition, value.OrderId).Result()
|
|
if err != nil || removedCount == 0 {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.HDel:%v", common.ErrShareMys, err)
|
|
return flags.ErrCacheDB
|
|
}
|
|
|
|
// 平仓更新用户订阅订单状态
|
|
userSubKey := virtual.OrderIdListKey(setting.ShareMysSubscribe, value.UserId)
|
|
if err = UpdateShareMysSubscribeHashStatusByOrderId(value.OrderId, userSubKey, flags.Close, flags.SetNull); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.ShareMysSubscribe:%v", common.ErrShareMys, err)
|
|
return flags.ErrCacheDB
|
|
}
|
|
|
|
// 平仓更新管理员订阅订单状态
|
|
if err = UpdateShareMysSubscribeHashStatusByOrderId(value.OrderId, setting.AdminShareMysSubscribe, flags.Close, flags.SetNull); err != nil {
|
|
applogger.Error("%v UpdateBotStockMysAllClosingByOrderId.AdminShareMysSubscribe:%v", common.ErrShareMys, err)
|
|
return flags.ErrCacheDB
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetBotStockMysTradeByUserId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param userId
|
|
// @return map[string]string
|
|
// @return error
|
|
func (uo *userOrderRepo) GetBotStockMysTradeByUserId(userId int64) (map[string]string, error) {
|
|
var botStockMysTrade []models.BotStockMysTrade
|
|
if err := uo.data.mysqlDB.Table(flags.BotStockMysTrade).
|
|
Where("user_id = ?", userId).
|
|
Where("status = 1").
|
|
Find(&botStockMysTrade); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
botMap := make(map[string]string)
|
|
for _, value := range botStockMysTrade {
|
|
botMap[value.OrderId] = value.OrderId
|
|
}
|
|
|
|
return botMap, nil
|
|
}
|
|
|
|
// GetShareMysTheLatestPrice
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param subKey
|
|
// @param tradeType
|
|
// @return decimal.Decimal
|
|
// @return error
|
|
func (uo *userOrderRepo) GetShareMysTheLatestPrice(subKey string, tradeType int64) (decimal.Decimal, error) {
|
|
priceNew, err := share.ShareMysSubMarketPrice(subKey)
|
|
if err != nil {
|
|
applogger.Error("%v GetShareMysTheLatestPrice.ShareMysSubMarketPrice:%v", common.ErrShareMys, 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
|
|
}
|
|
|
|
// GetBotStockMysTradeByOrderId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param orderId
|
|
// @return *models.BotStockMysTrade
|
|
// @return error
|
|
func (uo *userOrderRepo) GetBotStockMysTradeByOrderId(session *xorm.Session, orderId string) (*models.BotStockMysTrade, error) {
|
|
var botStockMysTrade []models.BotStockMysTrade
|
|
if err := session.Table(flags.BotStockMysTrade).
|
|
Where("order_id = ?", orderId).
|
|
Find(&botStockMysTrade); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, value := range botStockMysTrade {
|
|
return &value, nil
|
|
}
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
// GetBotUserStockMysByUserIdAndStockId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param userId
|
|
// @param stockId
|
|
// @return *models.BotUserStockMys
|
|
// @return error
|
|
func (uo *userOrderRepo) GetBotUserStockMysByUserIdAndStockId(session *xorm.Session, userId int64, stockId string) (*models.BotUserStockMys, error) {
|
|
var stockMYR []models.BotUserStockMys
|
|
if err := session.Table(flags.BotUserStockMys).
|
|
Where("user_id = ?", userId).
|
|
Where("stock_id = ?", stockId).
|
|
Find(&stockMYR); err != nil || len(stockMYR) <= 0 {
|
|
return nil, err
|
|
}
|
|
|
|
for _, value := range stockMYR {
|
|
return &value, nil
|
|
}
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
// UpdateBotStockMysTradeByOrderId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param orderId
|
|
// @param stock
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateBotStockMysTradeByOrderId(session *xorm.Session, orderId string, stock models.BotStockMysTrade) error {
|
|
checkInt, err := session.Table(flags.BotStockMysTrade).
|
|
Where("order_id = ?", orderId).
|
|
Update(&stock)
|
|
if err != nil || checkInt < 0 {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CreateBotStockMysTrade
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param trade
|
|
// @return error
|
|
func (uo *userOrderRepo) CreateBotStockMysTrade(session *xorm.Session, trade models.BotStockMysTrade) error {
|
|
_, err := session.Table(flags.BotStockMysTrade).Insert(&trade)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CreateBotUserStockMysFMYR
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param stock
|
|
// @return error
|
|
func (uo *userOrderRepo) CreateBotUserStockMysFMYR(session *xorm.Session, stock models.BotUserStockMys) error {
|
|
_, err := session.Table(flags.BotUserStockMys).Insert(&stock)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UpdateBotUserStockMysByUserIdAndStockId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param userId
|
|
// @param stockId
|
|
// @param stock
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateBotUserStockMysByUserIdAndStockId(session *xorm.Session, userId int64, stockId string, stock models.BotUserStockMys) error {
|
|
checkNum, err := session.Table(flags.BotUserStockMys).
|
|
Where("user_id = ?", userId).
|
|
Where("stock_id = ?", stockId).
|
|
Update(&stock)
|
|
|
|
if err != nil || checkNum < 0 {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// StockMysVoteDealType
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param order
|
|
// @return decimal.Decimal
|
|
// @return error
|
|
func (uo *userOrderRepo) StockMysVoteDealType(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
|
|
}
|
|
|
|
// UpdateShareMysVoteDealType
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param order
|
|
// @return decimal.Decimal
|
|
// @return error
|
|
func (uo *userOrderRepo) UpdateShareMysVoteDealType(order *models.BotStockMysTrade) (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
|
|
}
|
|
|
|
// VerifyBotStockMysTradeOrderId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @return string
|
|
// @return error
|
|
func (uo *userOrderRepo) VerifyBotStockMysTradeOrderId(session *xorm.Session) (string, error) {
|
|
var orderId string
|
|
for {
|
|
// 生成订单ID
|
|
orderId = orders.CreateRandCodeOrder(10)
|
|
|
|
var orderList []models.BotStockMysTrade
|
|
err := session.Table(flags.BotStockMysTrade).
|
|
Where("order_id = ?", orderId).
|
|
Find(&orderList)
|
|
|
|
if err != nil || len(orderList) > 0 {
|
|
applogger.Error("%v VerifyBotStockMysTradeOrderId.Find:%v", common.ErrShareMys, err)
|
|
continue
|
|
}
|
|
|
|
break
|
|
}
|
|
|
|
return orderId, nil
|
|
}
|
|
|
|
// VerifyBotStockBlockTradeOrderId
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param session
|
|
// @return string
|
|
// @return error
|
|
func (uo *userOrderRepo) VerifyBotStockBlockTradeOrderId(session *xorm.Session) (string, error) {
|
|
var orderId string
|
|
for {
|
|
// 生成订单ID
|
|
orderId = orders.CreateRandCodeOrder(10)
|
|
|
|
var orderList []models.BotStockBlockTrade
|
|
err := session.Table(flags.BotStockBlockTrade).
|
|
Where("order_id = ?", orderId).
|
|
Find(&orderList)
|
|
|
|
if err != nil || len(orderList) > 0 {
|
|
applogger.Error("%v VerifyBotStockMysTradeOrderId.Find:%v", common.ErrShareMys, err)
|
|
continue
|
|
}
|
|
|
|
break
|
|
}
|
|
|
|
return orderId, nil
|
|
}
|
|
|
|
// GetBotStockMysTradeByOrderIdOrStatus
|
|
//
|
|
// @Description:
|
|
// @receiver uo
|
|
// @param ctx
|
|
// @param session
|
|
// @param orderId
|
|
// @param status
|
|
// @return *models.BotStockMysTrade
|
|
// @return error
|
|
func (uo *userOrderRepo) GetBotStockMysTradeByOrderIdOrStatus(session *xorm.Session, orderId string, status int) (*models.BotStockMysTrade, error) {
|
|
var botStockMysTrade []models.BotStockMysTrade
|
|
if err := session.Table(flags.BotStockMysTrade).
|
|
Where("order_id = ?", orderId).
|
|
Where("`status` = ?", status).
|
|
Find(&botStockMysTrade); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, value := range botStockMysTrade {
|
|
return &value, nil
|
|
}
|
|
|
|
return nil, nil
|
|
}
|
|
|