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" ) // GetBotUserFurPreStockOrder // // @Description: 法股新股申购列表查詢 // @receiver uo // @param ctx // @param id // @return []models.BotUserEurPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserFurPreStockOrder(ctx context.Context, id string) ([]models.BotUserFurPreStockOrder, error) { var botSgd []models.BotUserFurPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserFurPreStockOrder). Where("pre_stock_id = ?", id). Where("status = 3"). Find(&botSgd); err != nil { applogger.Error("%v GetBotUserFurPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botSgd, nil } // GetBotUserFurPreStockOrderByOrderNo // // @Description: 法股新股申购列表查詢orderNo // @receiver uo // @param ctx // @param code // @return []models.BotUserFurPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserFurPreStockOrderByOrderNo(ctx context.Context, code []string) ([]models.BotUserFurPreStockOrder, error) { var botGbx []models.BotUserFurPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserFurPreStockOrder). In("order_no", code). Where("trade_status = 0"). Where("status = 3"). Find(&botGbx); err != nil { applogger.Error("%v GetBotUserFurPreStockOrderByOrderNo Find:%v", common.ErrSharePre, err) return nil, err } return botGbx, nil } // GetBotUserFurGiveStockOrders // // @Description: 法股赠股 // @receiver uo // @param ctx // @param id // @return []models.BotUserFurGiveStockOrder // @return error func (uo *userOrderRepo) GetBotUserFurGiveStockOrders(ctx context.Context, id string) ([]models.BotUserFurGiveStockOrder, error) { var botSgd []models.BotUserFurGiveStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserFurGiveStockOrder). Where("order_no = ?", id). Find(&botSgd); err != nil { applogger.Error("%v GetBotUserFurGiveStockOrders Find:%v", common.ErrSharePre, err) return nil, err } return botSgd, nil } // GetBotStockFurList // // @Description: 法股票代码列表 // @receiver uo // @param ctx // @return []models.BotStockFurList // @return error func (uo *userOrderRepo) GetBotStockFurList(ctx context.Context) ([]models.BotStockFurList, error) { var data []models.BotStockFurList if err := uo.data.mysqlDB.Table(flags.BotStockFurList).Find(&data); err != nil { applogger.Error("%v GetBotStockFurList 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.MarketShareFur, codeOld) Reds.HDel(context.Background(), setting.MarketShareBlkFur, codeOld) } } return data, nil } // CreateBotUserFurPreStockOrder // // @Description: 法股新股申购:开仓--持仓 // @receiver uo // @param ctx // @param code // @param order // @return error func (uo *userOrderRepo) CreateBotUserFurPreStockOrder(ctx context.Context, code string, order []structure.ShareOrder) error { // 同步写入股票列表缓存 SubscribeFurHSetCodeList(code, false) // 处理新股申购数据订单 session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v CreateBotUserFurPreStockOrder.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 强平阈值 flatRatio := GetCacheForcedClosure(flags.StockFURSystemSetUpKey, code) system := &structure.ShareSystem{ StrongFlatRatio: flatRatio.String(), } // 获取开盘时间 closingTime := SystemTimeGenerate(flags.StockMarketList, flags.FurMarket, time.Now()) // 初始化订单缓存map orderMap := make(map[string]share.ShareFurTallyCache) for _, value := range order { value.System = system // 生成订单ID var orderId string orderId, err = uo.VerifyBotStockFurTradeOrderId(session) if err != nil { applogger.Error("%v CreateBotUserFurPreStockOrder.VerifyBotStockFurTradeOrderId:%v", common.ErrSharePre, err) continue } // 查询账户是否存在,再存更新不存在新增 var check bool check, err = uo.CheckStockFurAmount(ctx, int64(value.UserId), value.StockId, value.OrderNumber, decimal.Zero.String()) if err != nil || !check { applogger.Error("%v CreateBotUserFurPreStockOrder.CheckStockFurAmount:%v", common.ErrSharePre, err) continue } // 写入订单数据表 trade := orders.BotStockFurTradePre(ctx, int64(value.UserId), orderId, value) if err = uo.CreateBotStockFurTrade(session, trade); err != nil { applogger.Error("%v CreateBotUserFurPreStockOrder.CreateBotStockFurTrade:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 resultMsg := share.ShareFurTallyCache{ 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 CreateBotUserFurPreStockOrder.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 CreateBotUserFurPreStockOrder.Marshal:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 if err = Reds.HSet(context.Background(), setting.MarketShareFurPosition, resultMsg.OrderId, string(byteStr)).Err(); err != nil { continue } // 写入用户订单订阅缓存 orderIdKey := virtual.OrderIdListKey(setting.ShareFurSubscribe, resultMsg.UserId) if err = share.ShareFurHashUserOrder(Reds, orderIdKey, &resultMsg); err != nil { applogger.Error("%v CreateBotUserFurPreStockOrder.ShareFurHashUserOrder:%v", common.ErrSharePre, err) return flags.ErrCacheDB } // 写入管理员订单订阅缓存 if err = share.ShareFurHashUserOrder(Reds, setting.AdminShareFurSubscribe, &resultMsg); err != nil { applogger.Error("%v CreateBotUserFurPreStockOrder.AdminShareFurSubscribe:%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("CreateBotUserFurPreStockOrder.Set err:%v", err) } } } return nil } // CheckStockFurAmount // // @Description: 法股新股申购【创建|更新】用戶资产 // @receiver uo // @param ctx // @param userId // @param stockId // @param usableNum // @param frozenNum // @return bool // @return error func (uo *userOrderRepo) CheckStockFurAmount(ctx context.Context, userId int64, stockId, usableNum, frozenNum string) (bool, error) { var botUserStockFur []models.BotUserStockFur if err := uo.data.mysqlDB.Table(flags.BotUserStockFur). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Find(&botUserStockFur); err != nil { applogger.Error("%v CheckStockFurAmount Find:%v", common.ErrSharePre, err) return false, err } var usableNumOld, frozenNumOld decimal.Decimal for _, value := range botUserStockFur { usableNumOld = decimal.RequireFromString(value.UsableNum) frozenNumOld = decimal.RequireFromString(value.FrozenNum) } if len(botUserStockFur) == 0 { userStock := orders.CreatBotUserStockFurPre(ctx, userId, stockId, usableNum, frozenNum) _, err := uo.data.mysqlDB.Table(flags.BotUserStockFur).Insert(&userStock) if err != nil { applogger.Error("%v CheckStockFurAmount Insert:%v", common.ErrSharePre, err) return false, err } } else { usableNumNew := usableNumOld.Add(decimal.RequireFromString(usableNum)) frozenNumNew := frozenNumOld.Add(decimal.RequireFromString(frozenNum)) userStock := orders.UpdateBotUserStockFurPre(ctx, usableNumNew.String(), frozenNumNew.String()) _, err := uo.data.mysqlDB.Table(flags.BotUserStockFur). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Update(&userStock) if err != nil { applogger.Error("%v CheckStockFurAmount Update:%v", common.ErrSharePre, err) return false, err } } return true, nil } // UpdateShareFurTradeStockId // // @Description: 更新-法股新股申购股票stockId // @receiver uo // @param ctx // @param code // @param codeOld // @return error func (uo *userOrderRepo) UpdateShareFurTradeStockId(ctx context.Context, code, codeOld string) error { // 同步写入股票列表缓存 SubscribeFurHSetCodeList(code, false) // 新增到法股缓存列表 if err := Reds.HSet(context.Background(), setting.MarketShareFur, code, code).Err(); err != nil { applogger.Error("%v UpdateShareFurTradeStockId.HSet.MarketShareFur:%v", common.ErrSharePre, err) return flags.ErrCacheDB } session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateShareFurTradeStockId.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 1、更新订单股票代码 var userStockTrade []models.BotStockFurTrade if err = session.Table(flags.BotStockFurTrade).Where("stock_id =?", codeOld).Find(&userStockTrade); err != nil { return flags.ErrMySqlDB } for _, value := range userStockTrade { value.StockId = code // 1、更新订单信息 inCheck, err := session.Table(flags.BotStockFurTrade).Where("trade_id =?", value.TradeId).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareFurTradeStockId BotStockFurTrade err:%v", common.ErrSharePre, err) continue } // 2、处理缓存订单 orderMap, err := Reds.HGet(context.Background(), setting.AdminShareFurSubscribe, value.OrderId).Result() if err != nil { continue } var entrustJson share.ShareFurTallyCache if err = json.Unmarshal([]byte(orderMap), &entrustJson); err != nil { continue } var entrust *share.ShareFurTallyCache entrust = &entrustJson if entrust == nil { continue } entrust.Symbol = code entrust.Order.StockId = code // 3、处理挂单|持仓|用户订阅|管理员订阅缓存数据 switch value.Status { case 0: // 挂单 if err = share.ShareFurHashUserOrder(Reds, setting.MarketShareFurEntrust, entrust); err != nil { continue } case 1: // 持仓 if err = share.ShareFurHashUserOrder(Reds, setting.MarketShareFurPosition, entrust); err != nil { continue } default: continue } // 更新用户订阅缓存 cacheKey := fmt.Sprintf("%v-%v", setting.ShareFurSubscribe, entrust.UserId) if err = share.ShareFurHashUserOrder(Reds, cacheKey, entrust); err != nil { continue } // 更新管理员订阅缓存 if err = share.ShareFurHashUserOrder(Reds, setting.AdminShareFurSubscribe, entrust); err != nil { continue } } // 2、更新资产表股票代码 var userStock []models.BotUserStockFur if err = session.Table(flags.BotUserStockFur).Where("stock_id =?", codeOld).Find(&userStock); err != nil { return flags.ErrMySqlDB } for _, value := range userStock { value.StockId = code inCheck, err := session.Table(flags.BotUserStockFur).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareFurTradeStockId BotUserStockFur err:%v", common.ErrSharePre, err) continue } } // 3、更新资产表日志股票代码 var userStockLog []models.BotUserStockFurLog if err = session.Table(flags.BotUserStockFurLog).Where("stock_id =?", codeOld).Find(&userStockLog); err != nil { return flags.ErrMySqlDB } for _, value := range userStockLog { value.StockId = code inCheck, err := session.Table(flags.BotUserStockFurLog).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareFurTradeStockId BotUserStockFurLog err:%v", common.ErrSharePre, err) continue } } if err = session.Commit(); err != nil { applogger.Error("%v UpdateShareFurTradeStockId.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } return nil }