This commit is contained in:
2025-02-18 09:59:51 +08:00
parent eb2455f967
commit 08a7da607f
9 changed files with 420 additions and 548 deletions

View File

@ -12,14 +12,11 @@ import (
"go-admin/common/helper"
models2 "go-admin/models"
"go-admin/pkg/utility"
"go-admin/pkg/utility/snowflakehelper"
"strconv"
"strings"
"time"
"github.com/bytedance/sonic"
"github.com/go-admin-team/go-admin-core/logger"
"github.com/jinzhu/copier"
"github.com/shopspring/decimal"
"gorm.io/gorm"
)
@ -115,12 +112,6 @@ func handleFutOrderByType(db *gorm.DB, preOrder *DbModels.LinePreOrder, orderSta
// 减仓回调
func handleReduceFilled(db *gorm.DB, preOrder *DbModels.LinePreOrder) {
apiUserInfo, _ := GetApiInfo(preOrder.ApiId)
mainId := preOrder.Id
if preOrder.MainId > 0 {
mainId = preOrder.MainId
}
if apiUserInfo.Id == 0 {
logger.Errorf("handleMainReduceFilled 获取api信息失败,订单号:%s", preOrder.OrderSn)
return
@ -161,74 +152,12 @@ func handleReduceFilled(db *gorm.DB, preOrder *DbModels.LinePreOrder) {
}
orders := make([]models.LinePreOrder, 0)
rate := utility.StringAsFloat(preOrder.Rate)
ext := models.LinePreOrderExt{}
//获取订单配置
db.Model(&ext).Where("order_id =?", preOrder.Pid).First(&ext)
// 不是100%减仓 就需要挂止盈止损
if rate >= 100 {
removeFutLossAndAddPosition(preOrder)
ids := []int{preOrder.MainId, preOrder.Pid}
if err := db.Model(&DbModels.LinePreOrder{}).Where("id IN ? AND status =6", ids).Update("status", 9).Error; err != nil {
logger.Info("100%减仓完毕,终结流程")
}
if err := db.Model(&models.LinePreOrder{}).Where("pid =? AND order_type IN (1,2) AND status = 0", preOrder.Id).Find(&orders).Error; err != nil {
logger.Errorf("handleMainReduceFilled 获取待触发订单失败,订单号:%s", preOrder.OrderSn)
return
}
totalLossAmountU, _ := GetTotalLossAmount(db, mainId)
totalNum := getFuturesPositionAvailableQuantity(db, apiUserInfo, preOrder, tradeSet)
takeProfitOrder := models.LinePreOrder{}
copier.Copy(&takeProfitOrder, &preOrder)
takeProfitOrder.Id = 0
takeProfitOrder.Pid = preOrder.Id
takeProfitOrder.OrderSn = utility.Int64ToString(snowflakehelper.GetOrderId())
takeProfitOrder.Status = 0
takeProfitOrder.Price = price.Mul(decimal.NewFromInt(1).Add(ext.TakeProfitRatio)).Truncate(int32(tradeSet.PriceDigit)).String()
takeProfitOrder.OrderType = 1
takeProfitOrder.Rate = ext.ReduceTakeProfitRatio.String()
takeProfitOrder.SignPrice = preOrder.Price
takeProfitOrder.CreatedAt = time.Now()
takeProfitOrder.BuyPrice = "0"
takeProfitOrder.MainOrderType = "LIMIT"
takeProfitOrder.Num = totalNum.String()
takeProfitOrder.Rate = ext.ReduceTakeProfitRatio.String()
//止盈需要累加之前的亏损
if totalLossAmountU.Cmp(decimal.Zero) > 0 {
percent := totalLossAmountU.Div(totalNum).Div(price).Abs()
takeProfitOrder.Rate = percent.Mul(decimal.NewFromInt(100)).Add(ext.ReduceTakeProfitRatio).Truncate(2).String()
}
setPrice(&takeProfitOrder, preOrder, tradeSet)
orders = append(orders, takeProfitOrder)
//有止损单
if ext.ReduceStopLossRatio.Cmp(decimal.Zero) > 0 {
var stoploss models.LinePreOrder
copier.Copy(&stoploss, &preOrder)
stoploss.Id = 0
stoploss.Pid = preOrder.Id
stoploss.OrderSn = utility.Int64ToString(snowflakehelper.GetOrderId())
stoploss.Status = 0
stoploss.CreatedAt = time.Now()
stoploss.OrderType = 2
stoploss.SignPrice = preOrder.Price
stoploss.BuyPrice = "0"
stoploss.Rate = ext.ReduceStopLossRatio.String()
stoploss.MainOrderType = "LIMIT"
stoploss.Num = totalNum.String()
setPrice(&stoploss, preOrder, tradeSet)
orders = append(orders, stoploss)
}
if err := db.Create(&orders).Error; err != nil {
logger.Errorf("handleMainReduceFilled 创建止盈止损单失败:%v", err)
return
}
futApi := FutRestApi{}
for _, v := range orders {
if v.OrderType == 1 {
@ -435,37 +364,27 @@ func handleFutMainOrderFilled(db *gorm.DB, preOrder *models.LinePreOrder) {
return
}
//预生成加仓单
shouldReturn := createFutPreAddPosition(preOrder, db, tradeSet)
if shouldReturn {
return
}
if preOrder.OrderCategory == 3 {
if err := cancelSymbolTakeAndStop(db, preOrder.MainId, preOrder.SymbolType); err != nil {
logger.Errorf("取消止盈止损订单失败 orderSn:%s err:%v", preOrder.OrderSn, err)
}
}
apiInfo, err := GetApiInfo(preOrder.ApiId)
if apiInfo.Id == 0 {
logger.Error("订单回调查询apiuserinfo失败 err:", err)
return
}
if err := db.Model(&DbModels.LinePreOrder{}).
Where("pid = ? AND order_type > 0 AND status = '0' ", preOrder.Id).
Find(&orders).Error; err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
logger.Error("订单回调查询止盈止损单失败:", err)
return
} else if len(orders) == 0 && preOrder.OrderCategory == 3 {
orders, err = makeFuturesTakeAndReduce(preOrder, db, tradeSet, orders)
if err != nil {
return
}
}
futApi := FutRestApi{}
num, _ := decimal.NewFromString(preOrder.Num)
num := getFuturesPositionAvailableQuantity(db, apiInfo, preOrder, tradeSet)
for i, order := range orders {
if i >= 2 { // 最多处理 2 个订单
break
}
for _, order := range orders {
price := utility.StrToDecimal(order.Price).Truncate(int32(tradeSet.PriceDigit))
num = num.Truncate(int32(tradeSet.AmountDigit))
order.Price = price.String()
@ -495,184 +414,6 @@ func handleFutMainOrderFilled(db *gorm.DB, preOrder *models.LinePreOrder) {
}
}
// 生成加仓单
func createFutPreAddPosition(preOrder *DbModels.LinePreOrder, db *gorm.DB, tradeSet models2.TradeSet) bool {
//主单类型
if preOrder.OrderCategory == 1 {
orderExts, err := GetOrderExts(db, preOrder.Id)
if err != nil {
logger.Errorf("预生成加仓单失败, 回调订单号:%s 获取主单拓展配置失败:%v", preOrder.OrderSn, err)
return true
}
price := utility.StrToDecimal(preOrder.Price)
for _, v := range orderExts {
if v.OrderId == 0 {
var data DbModels.LinePreOrder
copier.Copy(&data, &preOrder)
data.Id = 0
data.Pid = preOrder.Id
data.OrderSn = utility.Int64ToString(snowflakehelper.GetOrderId())
data.MainId = preOrder.Id
data.CreatedAt = time.Now()
data.MainOrderType = v.AddPositionOrderType
data.Status = 0
data.OrderCategory = 3
data.Rate = v.AddPositionPriceRatio.String()
var percentage decimal.Decimal
if data.Site == "BUY" {
percentage = decimal.NewFromInt(1).Sub(v.AddPositionPriceRatio.Div(decimal.NewFromInt(100)))
} else {
percentage = decimal.NewFromInt(1).Add(v.AddPositionPriceRatio.Div(decimal.NewFromInt(100)))
}
dataPrice := price.Mul(percentage).Truncate(int32(tradeSet.PriceDigit))
data.Price = dataPrice.String()
if v.AddPositionType == 1 {
data.Num = utility.StrToDecimal(preOrder.Num).Truncate(int32(tradeSet.AmountDigit)).String()
data.BuyPrice = "0"
} else {
data.BuyPrice = v.AddPositionVal.String()
data.Num = v.AddPositionVal.Div(dataPrice).Truncate(int32(tradeSet.AmountDigit)).String()
}
err := db.Transaction(func(tx *gorm.DB) error {
if err2 := tx.Create(&data).Error; err2 != nil {
return err2
}
v.OrderId = data.Id
if err2 := tx.Model(&v).Update("order_id", data.Id).Error; err2 != nil {
return err2
}
return nil
})
if err != nil {
logger.Errorf("预生成加仓单失败, 回调订单号:%s 预生成加仓单失败:%v", preOrder.OrderSn, err)
return true
}
}
}
}
return false
}
// 构建合约止盈、减仓单
func makeFuturesTakeAndReduce(preOrder *DbModels.LinePreOrder, db *gorm.DB, tradeSet models2.TradeSet, orders []DbModels.LinePreOrder) ([]DbModels.LinePreOrder, error) {
ext := models.LinePreOrderExt{}
apiInfo, err := GetApiInfo(preOrder.ApiId)
price := utility.StrToDecimal(preOrder.Price)
if apiInfo.Id == 0 {
logger.Error("订单回调查询apiuserinfo失败 err:", err)
return nil, errors.New("订单回调查询apiuserinfo失败")
}
if err := db.Model(&ext).Where("order_id = ?", preOrder.Id).First(&ext).Error; err != nil {
logger.Error("订单回调查询止盈止损单扩展表失败:", err)
return nil, errors.New("订单回调查询止盈止损单扩展表失败")
}
totalLossAmountU, _ := GetTotalLossAmount(db, preOrder.MainId)
num := getFuturesPositionAvailableQuantity(db, apiInfo, preOrder, tradeSet)
//止盈单
if ext.TakeProfitRatio.Cmp(decimal.Zero) > 0 {
profitOrder := models.LinePreOrder{}
copier.Copy(&profitOrder, preOrder)
profitOrder.Id = 0
profitOrder.OrderSn = strconv.FormatInt(snowflakehelper.GetOrderId(), 10)
profitOrder.Pid = preOrder.Id
profitOrder.OrderType = 1
profitOrder.Status = 0
profitOrder.MainId = ext.MainOrderId
profitOrder.Num = num.String()
profitOrder.Rate = ext.TakeProfitRatio.String()
//止盈需要累加之前的亏损
if totalLossAmountU.Cmp(decimal.Zero) > 0 {
percent := totalLossAmountU.Div(num).Div(price).Abs()
profitOrder.Rate = percent.Mul(decimal.NewFromInt(100)).Add(ext.TakeProfitRatio).Truncate(2).String()
}
if strings.ToUpper(preOrder.Site) == "BUY" {
profitOrder.Site = "SELL"
} else {
profitOrder.Site = "BUY"
}
setPrice(&profitOrder, preOrder, tradeSet)
orders = append(orders, profitOrder)
}
//减仓单
if ext.ReducePriceRatio.Cmp(decimal.Zero) > 0 {
stopOrder := models.LinePreOrder{}
copier.Copy(&stopOrder, preOrder)
stopOrder.Id = 0
stopOrder.OrderSn = strconv.FormatInt(snowflakehelper.GetOrderId(), 10)
stopOrder.Pid = preOrder.Id
stopOrder.MainId = ext.MainOrderId
stopOrder.OrderType = 4
stopOrder.Status = 0
stopOrder.Rate = ext.ReducePriceRatio.String()
stopOrder.Num = num.String()
if ext.ReduceNumRatio.Cmp(decimal.Zero) > 0 {
stopOrder.Num = num.Mul(ext.ReduceNumRatio.Div(decimal.NewFromInt(100))).Truncate(int32(tradeSet.AmountDigit)).String()
} else {
stopOrder.Num = num.String()
}
if strings.ToUpper(preOrder.Site) == "BUY" {
stopOrder.Site = "SELL"
} else {
stopOrder.Site = "BUY"
}
setPrice(&stopOrder, preOrder, tradeSet)
orders = append(orders, stopOrder)
}
for index := range orders {
orderRate := utility.StrToDecimal(orders[index].Rate)
orderType := orders[index].OrderType
if strings.ToUpper(preOrder.Site) == "BUY" {
orders[index].Site = "SELL"
} else {
orders[index].Site = "BUY"
}
switch {
//做多止盈、做空止损或减仓
case (orderType == 1 && preOrder.Site == "BUY"), ((orderType == 2 || orderType == 4) && preOrder.Site == "SELL"):
orders[index].Price = utility.StrToDecimal(preOrder.Price).Mul(decimal.NewFromInt(1).Add(orderRate.Div(decimal.NewFromInt(100)))).Truncate(int32(tradeSet.PriceDigit)).String()
//做多止损或减仓、做空止盈
case ((orderType == 2 || orderType == 4) && preOrder.Site == "BUY"), (orderType == 1 && preOrder.Site == "SELL"):
orders[index].Price = utility.StrToDecimal(preOrder.Price).Mul(decimal.NewFromInt(1).Sub(orderRate.Div(decimal.NewFromInt(100)))).Truncate(int32(tradeSet.PriceDigit)).String()
}
}
if len(orders) > 0 {
if err := db.Create(&orders).Error; err != nil {
logger.Error("主单回调,创建止盈、减仓单失败")
return orders, errors.New("主单回调,创建止盈、减仓单失败")
}
}
return orders, nil
}
// 减仓单
func processFutReduceOrder(order DbModels.LinePreOrder, price, num decimal.Decimal) {
key := fmt.Sprintf(rediskey.FuturesReduceList, global.EXCHANGE_BINANCE)