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" ) // GetBotUserJpyPreStockOrder // // @Description: 日股新股申购列表查詢 // @receiver uo // @param ctx // @param id // @return []models.BotUserEurPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserJpyPreStockOrder(ctx context.Context, id string) ([]models.BotUserJpPreStockOrder, error) { var botSgd []models.BotUserJpPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserJpyPreStockOrder). Where("pre_stock_id = ?", id). Where("status = 3"). Find(&botSgd); err != nil { applogger.Error("%v GetBotUserJpyPreStockOrder Find:%v", common.ErrSharePre, err) return nil, err } return botSgd, nil } // GetBotUserJpyPreStockOrderByOrderNo // // @Description: 日股新股申购列表查詢orderNo // @receiver uo // @param ctx // @param code // @return []models.BotUserJpyPreStockOrder // @return error func (uo *userOrderRepo) GetBotUserJpyPreStockOrderByOrderNo(ctx context.Context, code []string) ([]models.BotUserJpPreStockOrder, error) { var botGbx []models.BotUserJpPreStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserJpyPreStockOrder). In("order_no", code). Where("trade_status = 0"). Where("status = 3"). Find(&botGbx); err != nil { applogger.Error("%v GetBotUserJpyPreStockOrderByOrderNo Find:%v", common.ErrSharePre, err) return nil, err } return botGbx, nil } // GetBotUserJpyGiveStockOrders // // @Description: 日股赠股 // @receiver uo // @param ctx // @param id // @return []models.BotUserJpyGiveStockOrder // @return error func (uo *userOrderRepo) GetBotUserJpyGiveStockOrders(ctx context.Context, id string) ([]models.BotUserJpGiveStockOrder, error) { var botSgd []models.BotUserJpGiveStockOrder if err := uo.data.mysqlDB.Table(flags.BotUserJpyGiveStockOrder). Where("order_no = ?", id). Find(&botSgd); err != nil { applogger.Error("%v GetBotUserJpyGiveStockOrders Find:%v", common.ErrSharePre, err) return nil, err } return botSgd, nil } // GetBotStockJpyList // // @Description: 日股票代码列表 // @receiver uo // @param ctx // @return []models.BotStockJpyList // @return error func (uo *userOrderRepo) GetBotStockJpyList(ctx context.Context) ([]models.BotStockJpList, error) { var data []models.BotStockJpList if err := uo.data.mysqlDB.Table(flags.BotStockJpyList).Find(&data); err != nil { applogger.Error("%v GetBotStockJpyList 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.MarketShareJpy, codeOld) Reds.HDel(context.Background(), setting.MarketShareBlkJpy, codeOld) } } return data, nil } // CreateBotUserJpyPreStockOrder // // @Description: 日股新股申购:开仓--持仓 // @receiver uo // @param ctx // @param code // @param order // @return error func (uo *userOrderRepo) CreateBotUserJpyPreStockOrder(ctx context.Context, code string, order []structure.ShareOrder) error { // 同步写入股票列表缓存 SubscribeJpyHSetCodeList(code, false) // 处理新股申购数据订单 session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v CreateBotUserJpyPreStockOrder.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 强平阈值 flatRatio := GetCacheForcedClosure(flags.StockFURSystemSetUpKey, code) system := &structure.ShareSystem{ StrongFlatRatio: flatRatio.String(), } // 获取开盘时间 closingTime := SystemTimeGenerate(flags.StockMarketList, flags.JpyMarket, time.Now()) // 初始化订单缓存map orderMap := make(map[string]share.ShareJpyTallyCache) for _, value := range order { value.System = system // 生成订单ID var orderId string orderId, err = uo.VerifyBotStockJpyTradeOrderId(session) if err != nil { applogger.Error("%v CreateBotUserJpyPreStockOrder.VerifyBotStockJpyTradeOrderId:%v", common.ErrSharePre, err) continue } // 查询账户是否存在,再存更新不存在新增 var check bool check, err = uo.CheckStockJpyAmount(ctx, int64(value.UserId), value.StockId, value.OrderNumber, decimal.Zero.String()) if err != nil || !check { applogger.Error("%v CreateBotUserJpyPreStockOrder.CheckStockJpyAmount:%v", common.ErrSharePre, err) continue } // 写入订单数据表 trade := orders.BotStockJpyTradePre(ctx, int64(value.UserId), orderId, value) if err = uo.CreateBotStockJpyTrade(session, trade); err != nil { applogger.Error("%v CreateBotUserJpyPreStockOrder.CreateBotStockJpyTrade:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 resultMsg := share.ShareJpyTallyCache{ 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 CreateBotUserJpyPreStockOrder.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 CreateBotUserJpyPreStockOrder.Marshal:%v", common.ErrSharePre, err) continue } // 写入持仓缓存 if err = Reds.HSet(context.Background(), setting.MarketShareJpyPosition, resultMsg.OrderId, string(byteStr)).Err(); err != nil { continue } // 写入用户订单订阅缓存 orderIdKey := virtual.OrderIdListKey(setting.ShareJpySubscribe, resultMsg.UserId) if err = share.ShareJpyHashUserOrder(Reds, orderIdKey, &resultMsg); err != nil { applogger.Error("%v CreateBotUserJpyPreStockOrder.ShareJpyHashUserOrder:%v", common.ErrSharePre, err) return flags.ErrCacheDB } // 写入管理员订单订阅缓存 if err = share.ShareJpyHashUserOrder(Reds, setting.AdminShareJpySubscribe, &resultMsg); err != nil { applogger.Error("%v CreateBotUserJpyPreStockOrder.AdminShareJpySubscribe:%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("CreateBotUserJpyPreStockOrder.Set err:%v", err) } } } return nil } // CheckStockJpyAmount // // @Description: 日股新股申购【创建|更新】用戶资产 // @receiver uo // @param ctx // @param userId // @param stockId // @param usableNum // @param frozenNum // @return bool // @return error func (uo *userOrderRepo) CheckStockJpyAmount(ctx context.Context, userId int64, stockId, usableNum, frozenNum string) (bool, error) { var botUserStockJpy []models.BotUserStockJp if err := uo.data.mysqlDB.Table(flags.BotUserStockJpy). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Find(&botUserStockJpy); err != nil { applogger.Error("%v CheckStockJpyAmount Find:%v", common.ErrSharePre, err) return false, err } var usableNumOld, frozenNumOld decimal.Decimal for _, value := range botUserStockJpy { usableNumOld = decimal.RequireFromString(value.UsableNum) frozenNumOld = decimal.RequireFromString(value.FrozenNum) } if len(botUserStockJpy) == 0 { userStock := orders.CreatBotUserStockJpyPre(ctx, userId, stockId, usableNum, frozenNum) _, err := uo.data.mysqlDB.Table(flags.BotUserStockJpy).Insert(&userStock) if err != nil { applogger.Error("%v CheckStockJpyAmount Insert:%v", common.ErrSharePre, err) return false, err } } else { usableNumNew := usableNumOld.Add(decimal.RequireFromString(usableNum)) frozenNumNew := frozenNumOld.Add(decimal.RequireFromString(frozenNum)) userStock := orders.UpdateBotUserStockJpyPre(ctx, usableNumNew.String(), frozenNumNew.String()) _, err := uo.data.mysqlDB.Table(flags.BotUserStockJpy). Where("user_id = ?", userId). Where("stock_id = ?", stockId). Update(&userStock) if err != nil { applogger.Error("%v CheckStockJpyAmount Update:%v", common.ErrSharePre, err) return false, err } } return true, nil } // UpdateShareJpyTradeStockId // // @Description: 更新-日股新股申购股票stockId // @receiver uo // @param ctx // @param code // @param codeOld // @return error func (uo *userOrderRepo) UpdateShareJpyTradeStockId(ctx context.Context, code, codeOld string) error { // 同步写入股票列表缓存 SubscribeJpyHSetCodeList(code, false) // 新增到日股缓存列表 if err := Reds.HSet(context.Background(), setting.MarketShareJpy, code, code).Err(); err != nil { applogger.Error("%v UpdateShareJpyTradeStockId.HSet.MarketShareJpy:%v", common.ErrSharePre, err) return flags.ErrCacheDB } session := uo.data.mysqlDB.NewSession() defer session.Close() err := session.Begin() if err != nil { applogger.Error("%v UpdateShareJpyTradeStockId.Begin:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } // 1、更新订单股票代码 var userStockTrade []models.BotStockJpTrade if err = session.Table(flags.BotStockJpyTrade).Where("stock_id =?", codeOld).Find(&userStockTrade); err != nil { return flags.ErrMySqlDB } for _, value := range userStockTrade { value.StockId = code // 1、更新订单信息 inCheck, err := session.Table(flags.BotStockJpyTrade).Where("trade_id =?", value.TradeId).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareJpyTradeStockId BotStockJpyTrade err:%v", common.ErrSharePre, err) continue } // 2、处理缓存订单 orderMap, err := Reds.HGet(context.Background(), setting.AdminShareJpySubscribe, value.OrderId).Result() if err != nil { continue } var entrustJson share.ShareJpyTallyCache if err = json.Unmarshal([]byte(orderMap), &entrustJson); err != nil { continue } var entrust *share.ShareJpyTallyCache entrust = &entrustJson if entrust == nil { continue } entrust.Symbol = code entrust.Order.StockId = code // 3、处理挂单|持仓|用户订阅|管理员订阅缓存数据 switch value.Status { case 0: // 挂单 if err = share.ShareJpyHashUserOrder(Reds, setting.MarketShareJpyEntrust, entrust); err != nil { continue } case 1: // 持仓 if err = share.ShareJpyHashUserOrder(Reds, setting.MarketShareJpyPosition, entrust); err != nil { continue } default: continue } // 更新用户订阅缓存 cacheKey := fmt.Sprintf("%v-%v", setting.ShareJpySubscribe, entrust.UserId) if err = share.ShareJpyHashUserOrder(Reds, cacheKey, entrust); err != nil { continue } // 更新管理员订阅缓存 if err = share.ShareJpyHashUserOrder(Reds, setting.AdminShareJpySubscribe, entrust); err != nil { continue } } // 2、更新资产表股票代码 var userStock []models.BotUserStockJp if err = session.Table(flags.BotUserStockJpy).Where("stock_id =?", codeOld).Find(&userStock); err != nil { return flags.ErrMySqlDB } for _, value := range userStock { value.StockId = code inCheck, err := session.Table(flags.BotUserStockJpy).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareJpyTradeStockId BotUserStockJpy err:%v", common.ErrSharePre, err) continue } } // 3、更新资产表日志股票代码 var userStockLog []models.BotUserStockJpLog if err = session.Table(flags.BotUserStockJpyLog).Where("stock_id =?", codeOld).Find(&userStockLog); err != nil { return flags.ErrMySqlDB } for _, value := range userStockLog { value.StockId = code inCheck, err := session.Table(flags.BotUserStockJpyLog).Where("id =?", value.Id).Update(&value) if err != nil || inCheck <= 0 { applogger.Error("%v UpdateShareJpyTradeStockId BotUserStockJpyLog err:%v", common.ErrSharePre, err) continue } } if err = session.Commit(); err != nil { applogger.Error("%v UpdateShareJpyTradeStockId.Commit:%v", common.ErrSharePre, err) return flags.ErrMySqlDB } return nil }