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" ) // GetBotUserThaPreStockOrder // // @Description: 泰股新股申购列表查詢 // @receiver uo // @param ctx // @param id // @return []models.BotUserThaPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserThaPreStockOrder(ctx context.Context, id string) ([]models.BotUserThaPreStockOrder, error) { var botTha []models.BotUserThaPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserThaPreStockOrder). Where("pre_stock_id = ?", id). Where("status = 3"). Find(&botTha); err != nil { applogger.Error("%v GetBotUserThaPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botTha, nil } // GetBotUserThaPreStockOrderByOrderNo // // @Description: 泰股新股申购列表查詢OrderNo // @receiver uo // @param ctx // @param code // @return []models.BotUserThaPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserThaPreStockOrderByOrderNo(ctx context.Context, code []string) ([]models.BotUserThaPreStockOrder, error) { var botTha []models.BotUserThaPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserThaPreStockOrder). In("order_no", code). Where("trade_status = 0"). Where("status = 3"). Find(&botTha); err != nil { applogger.Error("%v GetBotUserThaPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botTha, nil } // GetBotUserThaGiveStockOrders // // @Description: 泰股赠股票 // @receiver uo // @param ctx // @param id // @return []models.BotUserThaGiveStockOrder // @return error func (uo *userOrderRepo) GetBotUserThaGiveStockOrders(ctx context.Context, id string) ([]models.BotUserThaGiveStockOrder, error) { var botTha []models.BotUserThaGiveStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserThaGiveStockOrder). Where("order_no = ?", id). Find(&botTha); err != nil { applogger.Error("%v GetBotUserThaGiveStockOrders Find:%v", common.ErrSharePre, err) return nil, err } return botTha, nil } // GetBotStockThaList // // @Description: 泰股股票代码列表 // @receiver uo // @param ctx // @return []models.BotStockThaList // @return error func (uo *userOrderRepo) GetBotStockThaList(ctx context.Context) ([]models.BotStockThaList, error) { var data []models.BotStockThaList if err := uo.data.mysqlDB.Table(flags.BotStockThaList).Find(&data); err != nil { applogger.Error("%v GetBotStockThaList 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.MarketShareTha, codeOld) Reds.HDel(context.Background(), setting.MarketShareBlkTha, codeOld) } } return data, nil } // CreateBotUserThaPreStockOrder // // @Description: 泰股新股申购:开仓--持仓 // @receiver uo // @param ctx // @param code // @param order // @return error func (uo *userOrderRepo) CreateBotUserThaPreStockOrder(ctx context.Context, code string, order []structure.ShareOrder) error { // 同步写入股票列表缓存 SubscribeThaHSetCodeList(code, false) // 处理新股申购数据订单 session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v CreatBotUserThaPreStockOrder.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 强平阈值 flatRatio := GetCacheForcedClosure(flags.StockTGSystemSetUpKey, code) system := &structure.ShareSystem{ StrongFlatRatio: flatRatio.String(), } // 获取开盘时间 closingTime := SystemTimeGenerate(flags.StockMarketList, flags.ThaMarket, time.Now()) // 初始化订单缓存 orderMap := make(map[string]share.ShareThaTallyCache) for _, value := range order { value.System = system // 生成订单ID orderId, err := uo.VerifyBotStockThaTradeOrderId(session) if err != nil { applogger.Error("%v CreatBotUserThaPreStockOrder.VerifyBotStockThaTradeOrderId:%v", common.ErrSharePre, err) continue } // 查询账户是否存在,再存更新不存在新增 check, err := uo.CheckStockThaAmount(ctx, int64(value.UserId), value.StockId, value.OrderNumber, decimal.Zero.String()) if err != nil || !check { applogger.Error("%v CreatBotUserThaPreStockOrder.CheckStockThaAmount:%v", common.ErrSharePre, err) continue } // 写入订单数据表 trade := orders.BotStockThaTradePre(ctx, int64(value.UserId), orderId, value) if err := uo.CreateBotStockThaTrade(session, trade); err != nil { applogger.Error("%v CreatBotUserThaPreStockOrder.CreateBotStockThaTrade:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 resultMsg := share.ShareThaTallyCache{ 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 CreatBotUserThaPreStockOrder.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 写入持仓缓存、用户订单订阅缓存、管理员订单订阅缓存 for _, resultMsg := range orderMap { byteStr, err := json.Marshal(resultMsg) if err != nil { applogger.Error("%v CreatBotUserThaPreStockOrder.Marshal:%v", common.ErrSharePre, err) continue } // 写入持仓订单缓存 if err = Reds.HSet(context.Background(), setting.MarketShareThaPosition, resultMsg.OrderId, string(byteStr)).Err(); err != nil { continue } // 写入用户订单订阅缓存 orderIdKey := virtual.OrderIdListKey(setting.ShareThaSubscribe, resultMsg.UserId) if err = share.ShareThaHashUserOrder(Reds, orderIdKey, &resultMsg); err != nil { applogger.Error("%v CreatBotUserThaPreStockOrder.ShareThaSubscribe:%v", common.ErrSharePre, err) return flags.ErrCacheDB } // 写入管理员订单订阅缓存 if err = share.ShareThaHashUserOrder(Reds, setting.AdminShareThaSubscribe, &resultMsg); err != nil { applogger.Error("%v CreatBotUserThaPreStockOrder.AdminShareThaSubscribe:%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("CreatBotUserThaPreStockOrder.Set err:%v", err) } } } return nil } // CheckStockThaAmount // // @Description: 泰股新股申购【创建|更新】用戶资产 // @receiver uo // @param ctx // @param userId // @param stockId // @param usableNum // @param frozenNum // @return bool // @return error func (uo *userOrderRepo) CheckStockThaAmount(ctx context.Context, userId int64, stockId, usableNum, frozenNum string) (bool, error) { var botUserStockTha []models.BotUserStockTha if err := uo.data.mysqlDB.Table(flags.BotUserStockTha). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Find(&botUserStockTha); err != nil { applogger.Error("%v CheckStockThaAmount Find:%v", common.ErrSharePre, err) return false, err } var usableNumOld, frozenNumOld decimal.Decimal for _, value := range botUserStockTha { usableNumOld = decimal.RequireFromString(value.UsableNum) frozenNumOld = decimal.RequireFromString(value.FrozenNum) } if len(botUserStockTha) == 0 { // 创建账户 userStock := orders.CreatBotUserStockThaPre(ctx, userId, stockId, usableNum, frozenNum) _, err := uo.data.mysqlDB.Table(flags.BotUserStockTha).Insert(&userStock) if err != nil { applogger.Error("%v CheckStockThaAmount Insert:%v", common.ErrSharePre, err) return false, err } } else { // 更新账户 usableNumNew := usableNumOld.Add(decimal.RequireFromString(usableNum)) frozenNumNew := frozenNumOld.Add(decimal.RequireFromString(frozenNum)) userStock := orders.UpdateBotUserStockThaPre(ctx, usableNumNew.String(), frozenNumNew.String()) _, err := uo.data.mysqlDB.Table(flags.BotUserStockTha). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Update(&userStock) if err != nil { applogger.Error("%v CheckStockThaAmount Update:%v", common.ErrSharePre, err) return false, err } } return true, nil } // UpdateShareThaTradeStockId // // @Description: 更新-泰股新股申购股票stockId // @receiver uo // @param ctx // @param code // @param codeOld // @param stock // @return error func (uo *userOrderRepo) UpdateShareThaTradeStockId(ctx context.Context, code, codeOld string) error { // 同步写入股票列表缓存 SubscribeThaHSetCodeList(code, false) // 新增到泰股缓存列表 if err := Reds.HSet(context.Background(), setting.MarketShareTha, code, code).Err(); err != nil { applogger.Error("%v UpdateShareThaTradeStockId.HSet.MarketShareTha:%v", common.ErrSharePre, err) return flags.ErrCacheDB } session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateShareThaTradeStockId.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 1、更新订单股票代码 var userStockTrade []models.BotStockThaTrade if err = session.Table(flags.BotStockThaTrade).Where("stock_id =?", codeOld).Find(&userStockTrade); err != nil { return flags.ErrMySqlDB } for _, value := range userStockTrade { value.StockId = code // 1、更新订单信息 inCheck, err := session.Table(flags.BotStockThaTrade).Where("trade_id =?", value.TradeId).Update(&value) if err != nil || inCheck <= 0 { continue } // 2、处理缓存订单 orderMap, err := Reds.HGet(context.Background(), setting.AdminShareThaSubscribe, value.OrderId).Result() if err != nil { continue } var entrustJson share.ShareThaTallyCache if err = json.Unmarshal([]byte(orderMap), &entrustJson); err != nil { continue } var entrust *share.ShareThaTallyCache entrust = &entrustJson if entrust == nil { continue } entrust.Symbol = code entrust.Order.StockId = code // 3、处理挂单|持仓|用户订阅|管理员订阅缓存数据 switch value.Status { case 0: // 挂单 if err = share.ShareThaHashUserOrder(Reds, setting.MarketShareThaEntrust, entrust); err != nil { continue } case 1: // 持仓 if err = share.ShareThaHashUserOrder(Reds, setting.MarketShareThaPosition, entrust); err != nil { continue } default: continue } // 更新用户订阅缓存 cacheKey := fmt.Sprintf("%v-%v", setting.ShareThaSubscribe, entrust.UserId) if err = share.ShareThaHashUserOrder(Reds, cacheKey, entrust); err != nil { continue } // 更新管理员订阅缓存 if err = share.ShareThaHashUserOrder(Reds, setting.AdminShareThaSubscribe, entrust); err != nil { continue } } // 2、更新资产表股票代码 var userStock []models.BotUserStockTha if err = session.Table(flags.BotUserStockTha).Where("stock_id =?", codeOld).Find(&userStock); err != nil { return flags.ErrMySqlDB } for _, value := range userStock { value.StockId = code inCheck, err := session.Table(flags.BotUserStockTha).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareThaTradeStockId BotUserStockTha err:%v", common.ErrSharePre, err) continue } } // 3、更新资产表日志股票代码 var userStockLog []models.BotUserStockThaLog if err = session.Table(flags.BotUserStockThaLog).Where("stock_id =?", codeOld).Find(&userStockLog); err != nil { return flags.ErrMySqlDB } for _, value := range userStockLog { value.StockId = code inCheck, err := session.Table(flags.BotUserStockThaLog).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareThaTradeStockId BotUserStockThaLog err:%v", common.ErrSharePre, err) continue } } if err = session.Commit(); err != nil { applogger.Error("%v UpdateShareThaTradeStockId.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } return nil }