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" ) // GetBotUserUsPreStockOrder // // @Description: 美股新股申购列表查询 // @receiver uo // @param ctx // @param id // @return []models.BotUserUsPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserUsPreStockOrder(ctx context.Context, id string) ([]models.BotUserUsPreStockOrder, error) { var botUs []models.BotUserUsPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserUsPreStockOrder). Where("pre_stock_id = ?", id). Where("status = 3"). Find(&botUs); err != nil { applogger.Error("%v GetBotUserUsPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botUs, nil } // GetBotUserUsPreStockOrderByOrderNo // // @Description: 美股新股申购列表查询--OrderNo // @receiver uo // @param ctx // @param code // @return []models.BotUserUsPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserUsPreStockOrderByOrderNo(ctx context.Context, code []string) ([]models.BotUserUsPreStockOrder, error) { var botUs []models.BotUserUsPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserUsPreStockOrder). In("order_no", code). Where("trade_status = 0"). Where("status = 3"). Find(&botUs); err != nil { applogger.Error("%v GetBotUserUsPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botUs, nil } // GetBotUserUsGiveStockOrder // // @Description: 美股赠送股票列表查询 // @receiver uo // @param ctx // @param id func (uo *userOrderRepo) GetBotUserUsGiveStockOrder(ctx context.Context, id string) ([]models.BotUserUsGiveStockOrder, error) { var botUs []models.BotUserUsGiveStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserUsGiveStockOrder). Where("order_no = ?", id). Find(&botUs); err != nil { applogger.Error("%v GetBotUserUsGiveStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botUs, nil } // GetBotStockList // // @Description: 美股股票代码列表 // @receiver uo // @param ctx // @return []models.BotStockList // @return error func (uo *userOrderRepo) GetBotStockList(ctx context.Context) ([]models.BotStockList, error) { var data []models.BotStockList if err := uo.data.mysqlDB.Table(flags.BotStockList).Find(&data); err != nil { applogger.Error("%v GetBotStockList 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.MarketShareUs, codeOld) Reds.HDel(context.Background(), setting.MarketShareBlkUs, codeOld) } } return data, nil } // CreateBotUserUsPreStockOrder // // @Description: 美股新股申购:开仓--持仓 // @receiver uo // @param ctx // @param code // @param order // @return error func (uo *userOrderRepo) CreateBotUserUsPreStockOrder(ctx context.Context, code string, order []structure.ShareOrder) error { // 同步写入股票列表缓存 SubscribeUsHSetCodeList(code, false) // 处理新股申购数据订单 session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v CreatBotUserUsPreStockOrder.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 强平阈值 flatRatio := GetCacheForcedClosure(flags.StockUsSystemSetUpKey, code) system := &structure.ShareSystem{ StrongFlatRatio: flatRatio.String(), } // 获取开盘时间 closingTime := SystemTimeGenerate(flags.StockMarketList, flags.UsMarket, time.Now()) // 初始化订单缓存 orderMap := make(map[string]share.ShareUsTallyCache) for _, value := range order { value.System = system // 生成订单ID var orderId string orderId, err = uo.VerifyBotStockTradeOrderId(session) if err != nil { applogger.Error("%v CreatBotUserUsPreStockOrder.VerifyBotStockTradeOrderId:%v", common.ErrSharePre, err) continue } // 查询账户是否存在,再存更新不存在新增 var check bool check, err = uo.CheckStockUsAmount(ctx, int64(value.UserId), value.StockId, value.OrderNumber, decimal.Zero.String()) if err != nil || !check { applogger.Error("%v CreatBotUserUsPreStockOrder.CheckStockUsAmount:%v", common.ErrSharePre, err) continue } // 写入订单数据表 trade := orders.BotStockTradePre(ctx, int64(value.UserId), orderId, value) if err = uo.CreateBotStockTrade(session, trade); err != nil { applogger.Error("%v CreatBotUserUsPreStockOrder.CreateBotStockTrade:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 resultMsg := share.ShareUsTallyCache{ 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 CreatBotUserUsPreStockOrder.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 CreatBotUserUsPreStockOrder.Marshal:%v", common.ErrSharePre, err) continue } // 写入持仓订单缓存 if err = Reds.HSet(context.Background(), setting.MarketShareUsPosition, resultMsg.OrderId, string(byteStr)).Err(); err != nil { continue } // 写入用户订单订阅缓存 orderIdKey := virtual.OrderIdListKey(setting.ShareUsSubscribe, resultMsg.UserId) if err = share.ShareUsHashUserOrder(Reds, orderIdKey, &resultMsg); err != nil { applogger.Error("%v CreatBotUserUsPreStockOrder.ShareUsSubscribe:%v", common.ErrSharePre, err) continue } // 写入管理员订单订阅缓存 if err = share.ShareUsHashUserOrder(Reds, setting.AdminShareUsSubscribe, &resultMsg); err != nil { applogger.Error("%v CreatBotUserUsPreStockOrder.AdminShareUsSubscribe:%v", common.ErrSharePre, err) continue } // 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("CreatBotUserUsPreStockOrder.Set err:%v", err) } } } return nil } // CheckStockUsAmount // // @Description: 美股新股申购【创建|更新】用戶资产 // @receiver uo // @param ctx // @param userId // @param stockId // @param usableNum // @param frozenNum // @return bool func (uo *userOrderRepo) CheckStockUsAmount(ctx context.Context, userId int64, stockId, usableNum, frozenNum string) (bool, error) { var botUserStock []models.BotUserStock if err := uo.data.mysqlDB.Table(flags.BotUserStock). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Find(&botUserStock); err != nil { applogger.Error("%v CheckStockUsAmount Find:%v", common.ErrSharePre, err) return false, err } var usableNumOld, frozenNumOld decimal.Decimal for _, value := range botUserStock { usableNumOld = decimal.RequireFromString(value.UsableNum) frozenNumOld = decimal.RequireFromString(value.FrozenNum) } if len(botUserStock) == 0 { userStock := orders.CreatBotUserStockPre(ctx, userId, stockId, usableNum, frozenNum) _, err := uo.data.mysqlDB.Table(flags.BotUserStock).Insert(&userStock) if err != nil { applogger.Error("%v CheckStockUsAmount Insert:%v", common.ErrSharePre, err) return false, err } } else { usableNumNew := usableNumOld.Add(decimal.RequireFromString(usableNum)) frozenNumNew := frozenNumOld.Add(decimal.RequireFromString(frozenNum)) userStock := orders.UpdateBotUserStockPre(ctx, usableNumNew.String(), frozenNumNew.String()) _, err := uo.data.mysqlDB.Table(flags.BotUserStock). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Update(&userStock) if err != nil { applogger.Error("%v CheckStockUsAmount Update:%v", common.ErrSharePre, err) return false, err } } return true, nil } // UpdateShareUsTradeStockId // // @Description: 更新-美股新股申购股票stockId // @receiver uo // @param ctx // @param code // @param codeOld // @param stock // @return error func (uo *userOrderRepo) UpdateShareUsTradeStockId(ctx context.Context, code, codeOld string) error { // 同步写入股票列表缓存 SubscribeUsHSetCodeList(code, false) // 新增到美股缓存列表 if err := Reds.HSet(context.Background(), setting.MarketShareUs, code, code).Err(); err != nil { applogger.Error("%v UpdateShareUsTradeStockId.HSet.MarketShareUs:%v", common.ErrSharePre, err) return flags.ErrCacheDB } session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateShareUsTradeStockId.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 1、更新订单股票代码 var userStockTrade []models.BotStockTrade if err = session.Table(flags.BotStockTrade).Where("stock_id =?", codeOld).Find(&userStockTrade); err != nil { return flags.ErrMySqlDB } for _, value := range userStockTrade { value.StockId = code // 1、更新订单信息 inCheck, err := session.Table(flags.BotStockTrade).Where("trade_id =?", value.TradeId).Update(&value) if err != nil || inCheck <= 0 { continue } // 2、处理缓存订单 orderMap, err := Reds.HGet(context.Background(), setting.AdminShareUsSubscribe, value.OrderId).Result() if err != nil { continue } var entrustJson share.ShareUsTallyCache if err = json.Unmarshal([]byte(orderMap), &entrustJson); err != nil { continue } var entrust *share.ShareUsTallyCache entrust = &entrustJson if entrust == nil { continue } entrust.Symbol = code entrust.Order.StockId = code // 3、处理挂单|持仓|用户订阅|管理员订阅缓存数据 switch value.Status { case 0: // 挂单 if err = share.ShareUsHashUserOrder(Reds, setting.MarketShareUsEntrust, entrust); err != nil { continue } case 1: // 持仓 if err = share.ShareUsHashUserOrder(Reds, setting.MarketShareUsPosition, entrust); err != nil { continue } default: continue } // 更新用户订阅缓存 if err = share.ShareUsHashUserOrder(Reds, fmt.Sprintf("%v-%v", setting.ShareUsSubscribe, entrust.UserId), entrust); err != nil { continue } // 更新管理员订阅缓存 if err = share.ShareUsHashUserOrder(Reds, setting.AdminShareUsSubscribe, entrust); err != nil { continue } } // 2、更新资产表股票代码 var userStock []models.BotUserStock if err = session.Table(flags.BotUserStock).Where("stock_id =?", codeOld).Find(&userStock); err != nil { return flags.ErrMySqlDB } for _, value := range userStock { value.StockId = code inCheck, err := session.Table(flags.BotUserStock).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareUsTradeStockId bot_user_stock err:%v", common.ErrSharePre, err) continue } } // 3、更新资产表日志股票代码 var userStockLog []models.BotUserStockLog if err = session.Table(flags.BotUserStockLog).Where("stock_id =?", codeOld).Find(&userStockLog); err != nil { return flags.ErrMySqlDB } for _, value := range userStockLog { value.StockId = code inCheck, err := session.Table(flags.BotUserStockLog).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareUsTradeStockId BotUserStockLog err:%v", common.ErrSharePre, err) continue } } if err = session.Commit(); err != nil { applogger.Error("%v UpdateShareUsTradeStockId.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } return nil }