You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
212 lines
5.3 KiB
212 lines
5.3 KiB
2 months ago
|
package biz
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
|
||
|
"github.com/go-kratos/kratos/v2/log"
|
||
|
|
||
|
"matchmaking-system/internal/biz/structure"
|
||
|
"matchmaking-system/internal/pkg/flags"
|
||
|
|
||
|
models "matchmaking-system/internal/pkg/model"
|
||
|
)
|
||
|
|
||
|
// UserShareBrlOrder
|
||
|
// @Description:
|
||
|
type UserShareBrlOrder struct {
|
||
|
ud UserShareBrlOrderRepo
|
||
|
|
||
|
log *log.Helper
|
||
|
}
|
||
|
|
||
|
// UserShareBrlOrderRepo
|
||
|
// @Description: 订单操作(订单列表|下单|设置止盈止损|撤单|平仓|一键平仓)
|
||
|
type UserShareBrlOrderRepo interface {
|
||
|
CheckToken(ctx context.Context) (int, error)
|
||
|
GetBotUserIsRealByUserId(ctx context.Context, userId int) bool
|
||
|
GetBotStockBrlTradeList(ctx context.Context, pageSize, pageCount, userId, status int64) ([]*models.BotStockBrlTrade, int64, error)
|
||
|
PlacingStockBrlOrders(ctx context.Context, userId int64, order structure.ShareOrder) (string, error)
|
||
|
UpdateBotStockBrlStopByOrderId(ctx context.Context, order structure.StopOrder) (bool, error)
|
||
|
UpdateBotStockBrlCancelByOrderId(ctx context.Context, orderId string, typeStatus int64) (bool, error)
|
||
|
UpdateBotStockBrlClosingByOrderId(ctx context.Context, orderId string, typeStatus int64) (bool, error)
|
||
|
UpdateBotStockBrlAllClosingByOrderId(ctx context.Context, userId int64) error
|
||
|
}
|
||
|
|
||
|
// NewUserShareBrlOrderRepo
|
||
|
//
|
||
|
// @Description:
|
||
|
// @param uo
|
||
|
// @param logger
|
||
|
// @return *UserShareBrlOrder
|
||
|
func NewUserShareBrlOrderRepo(uo UserShareBrlOrderRepo, logger log.Logger) *UserShareBrlOrder {
|
||
|
return &UserShareBrlOrder{ud: uo, log: log.NewHelper(logger)}
|
||
|
}
|
||
|
|
||
|
// GetUserIdByToken
|
||
|
//
|
||
|
// @Description: 通过token获取用户Id
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @return int64
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) GetUserIdByToken(ctx context.Context) (int64, bool, error) {
|
||
|
userId, err := uo.ud.CheckToken(ctx)
|
||
|
if err != nil {
|
||
|
return 0, false, flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
return int64(userId), true, nil
|
||
|
}
|
||
|
|
||
|
// BotStockBrlTradeList
|
||
|
//
|
||
|
// @Description: 订单列表
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @param pageSize
|
||
|
// @param pageCount
|
||
|
// @param status
|
||
|
// @return []*models.BotStockBrlTrade
|
||
|
// @return int64
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) BotStockBrlTradeList(ctx context.Context, pageSize, pageCount, status int64) ([]*models.BotStockBrlTrade, int64, error) {
|
||
|
userId, token, err := uo.GetUserIdByToken(ctx)
|
||
|
if !token || err != nil || userId == 0 {
|
||
|
return nil, 0, flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
stockList, totalCount, err := uo.ud.GetBotStockBrlTradeList(ctx, pageSize, pageCount-1, userId, status)
|
||
|
if err != nil {
|
||
|
return nil, 0, err
|
||
|
}
|
||
|
|
||
|
return stockList, totalCount, nil
|
||
|
}
|
||
|
|
||
|
// ShareBrlPlaceOrder
|
||
|
//
|
||
|
// @Description: 下单
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @param order
|
||
|
// @return string
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) ShareBrlPlaceOrder(ctx context.Context, order structure.ShareOrder) (string, error) {
|
||
|
userId, token, err := uo.GetUserIdByToken(ctx)
|
||
|
if !token || err != nil || userId == 0 {
|
||
|
return flags.SetNull, flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
if !uo.ud.GetBotUserIsRealByUserId(ctx, int(userId)) {
|
||
|
return flags.SetNull, flags.ErrIsReal
|
||
|
}
|
||
|
|
||
|
orderId, err := uo.ud.PlacingStockBrlOrders(ctx, userId, order)
|
||
|
if err != nil {
|
||
|
return flags.SetNull, err
|
||
|
}
|
||
|
|
||
|
return orderId, nil
|
||
|
}
|
||
|
|
||
|
// ShareBrlUpdateOrder
|
||
|
//
|
||
|
// @Description: 设置止盈止损
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @param order
|
||
|
// @return bool
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) ShareBrlUpdateOrder(ctx context.Context, order structure.StopOrder) (bool, error) {
|
||
|
userId, token, err := uo.GetUserIdByToken(ctx)
|
||
|
if !token || err != nil || userId == 0 {
|
||
|
return false, flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
if !uo.ud.GetBotUserIsRealByUserId(ctx, int(userId)) {
|
||
|
return false, flags.ErrIsReal
|
||
|
}
|
||
|
|
||
|
check, err := uo.ud.UpdateBotStockBrlStopByOrderId(ctx, order)
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
|
||
|
return check, nil
|
||
|
}
|
||
|
|
||
|
// ShareBrlCancel
|
||
|
//
|
||
|
// @Description: 撤单
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @param orderId
|
||
|
// @return bool
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) ShareBrlCancel(ctx context.Context, orderId string) (bool, error) {
|
||
|
userId, token, err := uo.GetUserIdByToken(ctx)
|
||
|
if !token || err != nil || userId == 0 {
|
||
|
return false, flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
if !uo.ud.GetBotUserIsRealByUserId(ctx, int(userId)) {
|
||
|
return false, flags.ErrIsReal
|
||
|
}
|
||
|
|
||
|
check, err := uo.ud.UpdateBotStockBrlCancelByOrderId(ctx, orderId, 0)
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
|
||
|
return check, nil
|
||
|
}
|
||
|
|
||
|
// ShareBrlPosition
|
||
|
//
|
||
|
// @Description: 平仓
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @param orderId
|
||
|
// @return bool
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) ShareBrlPosition(ctx context.Context, orderId string) (bool, error) {
|
||
|
userId, token, err := uo.GetUserIdByToken(ctx)
|
||
|
if !token || err != nil || userId == 0 {
|
||
|
return false, flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
if !uo.ud.GetBotUserIsRealByUserId(ctx, int(userId)) {
|
||
|
return false, flags.ErrIsReal
|
||
|
}
|
||
|
|
||
|
check, err := uo.ud.UpdateBotStockBrlClosingByOrderId(ctx, orderId, 0)
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
|
||
|
return check, nil
|
||
|
}
|
||
|
|
||
|
// ShareBrlAllPosition
|
||
|
//
|
||
|
// @Description: 一键平仓
|
||
|
// @receiver uo
|
||
|
// @param ctx
|
||
|
// @return error
|
||
|
func (uo *UserShareBrlOrder) ShareBrlAllPosition(ctx context.Context) error {
|
||
|
userId, token, err := uo.GetUserIdByToken(ctx)
|
||
|
if !token || err != nil || userId == 0 {
|
||
|
return flags.ErrTokenMessage
|
||
|
}
|
||
|
|
||
|
if !uo.ud.GetBotUserIsRealByUserId(ctx, int(userId)) {
|
||
|
return flags.ErrIsReal
|
||
|
}
|
||
|
|
||
|
if err = uo.ud.UpdateBotStockBrlAllClosingByOrderId(ctx, userId); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|