package data import ( "context" "encoding/json" "fmt" "github.com/shopspring/decimal" "matchmaking-system/internal/biz/structure" orders "matchmaking-system/internal/data/convert" "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" models "matchmaking-system/internal/pkg/model" "matchmaking-system/internal/pkg/setting" "strings" "time" ) // GetBotUserMysPreStockOrder // // @Description: 马股新股申购列表查询 // @receiver uo // @param ctx // @param id // @return []models.BotUserMysPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserMysPreStockOrder(ctx context.Context, id string) ([]models.BotUserMysPreStockOrder, error) { var botMys []models.BotUserMysPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserMysPreStockOrder). Where("pre_stock_id = ?", id). Where("status = 3"). Find(&botMys); err != nil { applogger.Error("%v GetBotUserMysPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botMys, nil } // GetBotUserMysPreStockOrderByOrderNo // // @Description: 马股新股申购列表查询orderNo // @receiver uo // @param ctx // @param code // @return []models.BotUserMysPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserMysPreStockOrderByOrderNo(ctx context.Context, code []string) ([]models.BotUserMysPreStockOrder, error) { var botMys []models.BotUserMysPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserMysPreStockOrder). In("order_no", code). Where("trade_status = 0"). Where("status = 3"). Find(&botMys); err != nil { applogger.Error("%v GetBotUserMysPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botMys, nil } // GetBotUserMysGiveStockOrders // // @Description: 马股赠送股票 // @receiver uo // @param ctx // @param id // @return []models.BotUserMysGiveStockOrder // @return error func (uo *userOrderRepo) GetBotUserMysGiveStockOrders(ctx context.Context, id string) ([]models.BotUserMysGiveStockOrder, error) { var botMys []models.BotUserMysGiveStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserMysGiveStockOrder). Where("order_no = ?", id). Find(&botMys); err != nil { applogger.Error("%v GetBotUserMysGiveStockOrders Find:%v", common.ErrSharePre, err) return nil, err } return botMys, nil } // GetBotStockMysList // // @Description: 马股股票代码列表 // @receiver uo // @param ctx // @return []models.BotStockMysList // @return error func (uo *userOrderRepo) GetBotStockMysList(ctx context.Context) ([]models.BotStockMysList, error) { var data []models.BotStockMysList if err := uo.data.mysqlDB.Table(flags.BotStockMysList).Find(&data); err != nil { applogger.Error("%v GetBotStockMysList Find:%v", common.ErrSharePre, err) return nil, err } // 清理股票缓存列表 for _, value := range data { strArray := strings.Split(value.StockCode, ":") if len(strArray) >= 2 { codeOld := strArray[1] Reds.HDel(context.Background(), setting.MarketShareMys, codeOld) Reds.HDel(context.Background(), setting.MarketShareBlkMys, codeOld) } } return data, nil } // CreateBotUserMysPreStockOrder // // @Description: 马股新股申购:开仓--持仓 // @receiver uo // @param ctx // @param code // @param order // @return error func (uo *userOrderRepo) CreateBotUserMysPreStockOrder(ctx context.Context, code string, order []structure.ShareOrder) error { // 同步写入股票列表缓存 SubscribeMysHSetCodeList(code, false) // 处理新股申购数据订单 session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 强平阈值 flatRatio := GetCacheForcedClosure(flags.StockMGSystemSetUpKey, code) system := &structure.ShareSystem{ StrongFlatRatio: flatRatio.String(), } // 获取开盘时间 closingTime := SystemTimeGenerate(flags.StockMarketList, flags.MysMarket, time.Now()) // 初始化订单缓存 orderMap := make(map[string]share.ShareMysTallyCache) for _, value := range order { value.System = system // 生成订单ID var orderId string orderId, err = uo.VerifyBotStockMysTradeOrderId(session) if err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.VerifyBotStockMysTradeOrderId:%v", common.ErrSharePre, err) continue } // 查询账户是否存在,再存更新不存在新增 var check bool check, err = uo.CheckStockMysAmount(ctx, int64(value.UserId), value.StockId, value.OrderNumber, decimal.Zero.String()) if err != nil || !check { applogger.Error("%v CreatBotUserMysPreStockOrder.CheckStockMysAmount:%v", common.ErrSharePre, err) continue } // 写入订单数据表 trade := orders.BotStockMysTradePre(ctx, int64(value.UserId), orderId, value) if err = uo.CreateBotStockMysTrade(session, trade); err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.CreateBotStockMysTrade:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 resultMsg := share.ShareMysTallyCache{ UserId: int64(value.UserId), // 用户Id OrderId: orderId, // 订单ID Symbol: value.StockId, // 下单交易对 OpenPrice: value.LimitPrice, // 开仓价格 Status: flags.Position, // 订单状态 Order: value, // 下单信息 ClosingTime: closingTime, // 平仓时间 } orderMap[orderId] = resultMsg } if err = session.Commit(); err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 写入持仓缓存、用户订单订阅缓存、管理员订单订阅缓存 for _, resultMsg := range orderMap { var byteStr []byte byteStr, err = json.Marshal(resultMsg) if err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.Marshal:%v", common.ErrSharePre, err) continue } // 写入持仓订单缓存 if err = Reds.HSet(context.Background(), setting.MarketShareMysPosition, resultMsg.OrderId, string(byteStr)).Err(); err != nil { continue } // 写入用户订单订阅缓存 orderIdKey := virtual.OrderIdListKey(setting.ShareMysSubscribe, resultMsg.UserId) if err = share.ShareMysHashUserOrder(Reds, orderIdKey, &resultMsg); err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.ShareMysSubscribe:%v", common.ErrSharePre, err) return flags.ErrCacheDB } // 写入管理员订单订阅缓存 if err = share.ShareMysHashUserOrder(Reds, setting.AdminShareMysSubscribe, &resultMsg); err != nil { applogger.Error("%v CreatBotUserMysPreStockOrder.AdminShareMysSubscribe:%v", common.ErrSharePre, err) return flags.ErrCacheDB } // TODO: 获取IPO未支付订单缓存,并更新缓存中的持仓订单ID keyCache := fmt.Sprintf("%v%v", flags.StockIpoList, resultMsg.Order.OrderNo) checkInt, err := Reds.Exists(ctx, keyCache).Result() if err != nil { continue } if checkInt > 0 { if err = Reds.Set(ctx, keyCache, resultMsg.OrderId, 0).Err(); err != nil { applogger.Error("CreatBotUserMysPreStockOrder.Set err:%v", err) } } } return nil } // CheckStockMysAmount // // @Description: 马股新股申购【创建|更新】用戶资产 // @receiver uo // @param ctx // @param userId // @param stockId // @param usableNum // @param frozenNum // @return bool // @return error func (uo *userOrderRepo) CheckStockMysAmount(ctx context.Context, userId int64, stockId, usableNum, frozenNum string) (bool, error) { var botUserStockMys []models.BotUserStockMys if err := uo.data.mysqlDB.Table(flags.BotUserStockMys). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Find(&botUserStockMys); err != nil { applogger.Error("%v CheckStockMysAmount Find:%v", common.ErrSharePre, err) return false, err } var usableNumOld, frozenNumOld decimal.Decimal for _, value := range botUserStockMys { usableNumOld = decimal.RequireFromString(value.UsableNum) frozenNumOld = decimal.RequireFromString(value.FrozenNum) } if len(botUserStockMys) == 0 { // 创建账户 userStock := orders.CreatBotUserStockMysPre(ctx, userId, stockId, usableNum, frozenNum) _, err := uo.data.mysqlDB.Table(flags.BotUserStockMys).Insert(&userStock) if err != nil { applogger.Error("%v CheckStockMysAmount Insert:%v", common.ErrSharePre, err) return false, err } } else { // 更新账户 usableNumNew := usableNumOld.Add(decimal.RequireFromString(usableNum)) frozenNumNew := frozenNumOld.Add(decimal.RequireFromString(frozenNum)) userStock := orders.UpdateBotUserStockMysPre(ctx, usableNumNew.String(), frozenNumNew.String()) _, err := uo.data.mysqlDB.Table(flags.BotUserStockMys). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Update(&userStock) if err != nil { applogger.Error("%v CheckStockMysAmount Update:%v", common.ErrSharePre, err) return false, err } } return true, nil } // UpdateShareMysTradeStockId // // @Description: 更新-马股新股申购股票stockId // @receiver uo // @param ctx // @param code // @param codeOld // @param stock // @return error func (uo *userOrderRepo) UpdateShareMysTradeStockId(ctx context.Context, code, codeOld string) error { // 同步写入股票列表缓存 SubscribeMysHSetCodeList(code, false) // 新增到马股缓存列表 if err := Reds.HSet(context.Background(), setting.MarketShareMys, code, code).Err(); err != nil { applogger.Error("%v UpdateShareMysTradeStockId.HSet.MarketShareMys:%v", common.ErrSharePre, err) return flags.ErrCacheDB } session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateShareMysTradeStockId.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 1、更新订单股票代码 var userStockTrade []models.BotStockMysTrade if err = session.Table(flags.BotStockMysTrade).Where("stock_id =?", codeOld).Find(&userStockTrade); err != nil { return flags.ErrMySqlDB } for _, value := range userStockTrade { value.StockId = code // 1、更新订单信息 inCheck, err := session.Table(flags.BotStockMysTrade).Where("trade_id =?", value.TradeId).Update(&value) if err != nil || inCheck <= 0 { continue } // 2、处理缓存订单 orderMap, err := Reds.HGet(context.Background(), setting.AdminShareMysSubscribe, value.OrderId).Result() if err != nil { continue } var entrustJson share.ShareMysTallyCache if err = json.Unmarshal([]byte(orderMap), &entrustJson); err != nil { continue } var entrust *share.ShareMysTallyCache entrust = &entrustJson if entrust == nil { continue } entrust.Symbol = code entrust.Order.StockId = code // 3、处理挂单|持仓|用户订阅|管理员订阅缓存数据 switch value.Status { case 0: // 挂单 if err = share.ShareMysHashUserOrder(Reds, setting.MarketShareMysEntrust, entrust); err != nil { continue } case 1: // 持仓 if err = share.ShareMysHashUserOrder(Reds, setting.MarketShareMysPosition, entrust); err != nil { continue } default: continue } // 更新用户订阅缓存 cacheKey := fmt.Sprintf("%v-%v", setting.ShareMysSubscribe, entrust.UserId) if err = share.ShareMysHashUserOrder(Reds, cacheKey, entrust); err != nil { continue } // 更新管理员订阅缓存 if err = share.ShareMysHashUserOrder(Reds, setting.AdminShareMysSubscribe, entrust); err != nil { continue } } // 2、更新资产表股票代码 var userStock []models.BotUserStockMys if err = session.Table(flags.BotUserStockMys).Where("stock_id =?", codeOld).Find(&userStock); err != nil { return flags.ErrMySqlDB } for _, value := range userStock { value.StockId = code inCheck, err := session.Table(flags.BotUserStockMys).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareMysTradeStockId BotUserStockMys err:%v", common.ErrSharePre, err) continue } } // 3、更新资产表日志股票代码 var userStockLog []models.BotUserStockMysLog if err = session.Table(flags.BotUserStockMysLog).Where("stock_id =?", codeOld).Find(&userStockLog); err != nil { return flags.ErrMySqlDB } for _, value := range userStockLog { value.StockId = code inCheck, err := session.Table(flags.BotUserStockMysLog).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareMysTradeStockId BotUserStockMysLog err:%v", common.ErrSharePre, err) continue } } if err = session.Commit(); err != nil { applogger.Error("%v UpdateShareMysTradeStockId.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } return nil }