843 lines
25 KiB
Go
843 lines
25 KiB
Go
package binanceservice
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"go-admin/app/admin/models"
|
|
DbModels "go-admin/app/admin/models"
|
|
"go-admin/app/admin/service/dto"
|
|
"go-admin/common/const/rediskey"
|
|
"go-admin/common/global"
|
|
"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"
|
|
)
|
|
|
|
/*
|
|
订单回调
|
|
*/
|
|
func ChangeSpotOrder(mapData map[string]interface{}) {
|
|
// 检查订单号是否存在
|
|
orderSn, ok := mapData["c"]
|
|
originOrderSn := mapData["C"] //取消操作 代表原始订单号
|
|
|
|
if !ok {
|
|
logger.Error("订单回调失败, 没有订单号", mapData)
|
|
return
|
|
}
|
|
|
|
if originOrderSn != "" {
|
|
orderSn = originOrderSn
|
|
}
|
|
|
|
// 获取数据库连接
|
|
db := GetDBConnection()
|
|
if db == nil {
|
|
logger.Error("订单回调失败, 无法获取数据库连接")
|
|
return
|
|
}
|
|
|
|
// 获取 Redis 锁
|
|
lockKey := fmt.Sprintf(rediskey.SpotCallBack, orderSn)
|
|
lock := helper.NewRedisLock(lockKey, 200, 5, 100*time.Millisecond)
|
|
if err := acquireLock(lock, orderSn); err != nil {
|
|
return
|
|
}
|
|
defer lock.Release()
|
|
|
|
// 查询订单
|
|
preOrder, err := getPreOrder(db, orderSn)
|
|
if err != nil {
|
|
logger.Error("订单回调失败, 查询订单失败:", orderSn, " err:", err)
|
|
return
|
|
}
|
|
|
|
// 解析订单状态
|
|
status, ok := mapData["X"]
|
|
if !ok {
|
|
logMapData(mapData)
|
|
return
|
|
}
|
|
|
|
// 更新订单状态
|
|
orderStatus, reason := parseOrderStatus(status, mapData)
|
|
|
|
if orderStatus == 0 {
|
|
logger.Error("订单回调失败,状态错误:", orderSn, " status:", status, " reason:", reason)
|
|
return
|
|
}
|
|
|
|
if err := updateOrderStatus(db, preOrder, orderStatus, reason, true, mapData); err != nil {
|
|
logger.Error("修改订单状态失败:", orderSn, " err:", err)
|
|
return
|
|
}
|
|
|
|
// 根据订单类型和状态处理逻辑
|
|
handleOrderByType(db, preOrder, orderStatus)
|
|
}
|
|
|
|
// 获取 Redis 锁
|
|
func acquireLock(lock *helper.RedisLock, orderSn interface{}) error {
|
|
acquired, err := lock.AcquireWait(context.Background())
|
|
if err != nil {
|
|
logger.Error("订单回调失败, 获取锁失败:", orderSn, " err:", err)
|
|
return err
|
|
}
|
|
if !acquired {
|
|
logger.Error("订单回调失败, 获取锁失败:", orderSn)
|
|
return fmt.Errorf("failed to acquire lock")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 记录 mapData 数据
|
|
func logMapData(mapData map[string]interface{}) {
|
|
mapStr, _ := sonic.Marshal(&mapData)
|
|
logger.Error("订单回调失败, 没有状态:", string(mapStr))
|
|
}
|
|
|
|
// 根据订单类型和状态处理逻辑
|
|
func handleOrderByType(db *gorm.DB, preOrder *DbModels.LinePreOrder, orderStatus int) {
|
|
switch {
|
|
// 主单成交
|
|
case preOrder.OrderType == 0 && orderStatus == 6:
|
|
handleMainOrderFilled(db, preOrder)
|
|
|
|
//主单减仓完毕
|
|
case preOrder.OrderCategory == 1 && preOrder.OrderType == 4 && orderStatus == 6:
|
|
handleMainReduceFilled(db, preOrder)
|
|
//主单取消
|
|
case preOrder.OrderType == 0 && preOrder.Pid == 0 && orderStatus == 4:
|
|
handleMainOrderCancel(preOrder)
|
|
// 止盈成交
|
|
case preOrder.OrderType == 1 && orderStatus == 6:
|
|
handleSpotTakeProfitFilled(db, preOrder)
|
|
//平仓单
|
|
case preOrder.OrderType == 3 && orderStatus == 6:
|
|
handleMainOrderClosePosition(db, preOrder)
|
|
|
|
//主单止损回调
|
|
case preOrder.OrderType == 2 && orderStatus == 6:
|
|
removeSpotLossAndAddPosition(preOrder)
|
|
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("id =?", preOrder.MainId).Update("status", 9).Error; err != nil {
|
|
logger.Errorf("主单止损回调 订单号:%s 修改主单状态失败:%v", preOrder.OrderSn, err)
|
|
}
|
|
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("main_id =? AND status =0", preOrder.MainId).Update("status", 4).Error; err != nil {
|
|
logger.Errorf("主单止损回调 订单号:%s 修改主单状态失败:%v", preOrder.OrderSn, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 主单减仓完毕
|
|
func handleMainReduceFilled(db *gorm.DB, preOrder *DbModels.LinePreOrder) {
|
|
apiUserInfo, _ := GetApiInfo(preOrder.ApiId)
|
|
|
|
if apiUserInfo.Id == 0 {
|
|
logger.Errorf("handleMainReduceFilled 获取api信息失败,订单号:%s", preOrder.OrderSn)
|
|
return
|
|
}
|
|
|
|
tradeSet, err := GetTradeSet(preOrder.Symbol, 0)
|
|
|
|
if err != nil {
|
|
logger.Errorf("handleMainReduceFilled 获取交易对设置失败,订单号:%s", preOrder.OrderSn)
|
|
return
|
|
}
|
|
|
|
price := utility.StrToDecimal(preOrder.Price)
|
|
parentOrder, _ := GetOrderById(db, preOrder.Pid)
|
|
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 {
|
|
num, err := getSpotPositionNum(apiUserInfo, preOrder, tradeSet)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
takeProfitOrder := models.LinePreOrder{}
|
|
copier.Copy(&takeProfitOrder, &preOrder)
|
|
takeProfitOrder.Id = 0
|
|
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 = "100"
|
|
takeProfitOrder.SignPrice = preOrder.Price
|
|
takeProfitOrder.CreatedAt = time.Now()
|
|
takeProfitOrder.BuyPrice = "0"
|
|
takeProfitOrder.MainOrderType = "LIMIT"
|
|
takeProfitOrder.Num = num.String()
|
|
orders = append(orders, takeProfitOrder)
|
|
|
|
//有止损单
|
|
if ext.ReduceStopLossRatio.Cmp(decimal.Zero) > 0 {
|
|
var stoploss models.LinePreOrder
|
|
|
|
copier.Copy(&stoploss, &preOrder)
|
|
stoploss.Id = 0
|
|
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 = "100"
|
|
stoploss.MainOrderType = "LIMIT"
|
|
stoploss.Num = num.String()
|
|
|
|
orders = append(orders, stoploss)
|
|
}
|
|
|
|
if err := db.Create(&orders).Error; err != nil {
|
|
logger.Errorf("handleMainReduceFilled 创建止盈止损单失败:%v", err)
|
|
return
|
|
}
|
|
|
|
spotApi := SpotRestApi{}
|
|
paramsMap := OrderPlacementService{
|
|
ApiId: takeProfitOrder.ApiId,
|
|
Symbol: takeProfitOrder.Symbol,
|
|
Side: takeProfitOrder.Site,
|
|
Type: "LIMIT",
|
|
TimeInForce: "GTC",
|
|
Price: utility.StrToDecimal(takeProfitOrder.Price),
|
|
Quantity: num,
|
|
NewClientOrderId: takeProfitOrder.OrderSn,
|
|
StopPrice: utility.StrToDecimal(takeProfitOrder.Price),
|
|
}
|
|
if err := spotApi.OrderPlace(db, paramsMap); err != nil {
|
|
logger.Errorf("减仓后重下止盈失败 减仓order_sn:%s err:%v", preOrder.OrderSn, err)
|
|
|
|
if err2 := db.Model(&takeProfitOrder).Updates(map[string]interface{}{"status": 2, "": err.Error()}).Error; err2 != nil {
|
|
logger.Errorf("handleMainReduceFilled 更新止盈单失败:%v", err2)
|
|
}
|
|
}
|
|
}
|
|
|
|
//计算实际亏损
|
|
if parentOrder.Price != "" {
|
|
parentPrice := utility.StrToDecimal(parentOrder.Price)
|
|
reduceNum := utility.StrToDecimal(preOrder.Num)
|
|
lossAmountU := price.Sub(parentPrice).Abs().Mul(reduceNum).Truncate(2)
|
|
|
|
if err := db.Model(&parentOrder).Update("loss_amount", lossAmountU).Error; err != nil {
|
|
logger.Errorf("修改主单实际亏损失败 订单号:%s err:%v", parentOrder.OrderSn, err)
|
|
}
|
|
}
|
|
|
|
//加仓待触发
|
|
addPositionOrder := DbModels.LinePreOrder{}
|
|
|
|
if err := db.Model(&addPositionOrder).Where("main_id =? AND order_category=3 AND status=0", preOrder.MainId).First(addPositionOrder).Error; err != nil {
|
|
logger.Errorf("handleMainReduceFilled 获取加仓单失败,订单号:%s err:%v", preOrder.OrderSn, err)
|
|
return
|
|
}
|
|
|
|
keySpotAddPosition := fmt.Sprintf(rediskey.SpotAddPositionList, global.EXCHANGE_BINANCE)
|
|
|
|
addPositionData := AddPositionList{
|
|
MainId: addPositionOrder.MainId,
|
|
Pid: addPositionOrder.Pid,
|
|
Price: utility.StrToDecimal(addPositionOrder.Price),
|
|
ApiId: addPositionOrder.ApiId,
|
|
Symbol: addPositionOrder.Symbol,
|
|
Side: addPositionOrder.Site,
|
|
SymbolType: addPositionOrder.SymbolType,
|
|
}
|
|
|
|
addVal, err := sonic.MarshalString(addPositionData)
|
|
|
|
if err != nil {
|
|
logger.Errorf("handleMainReduceFilled 序列化加仓单失败,订单号:%s err:%v", preOrder.OrderSn, err)
|
|
return
|
|
}
|
|
|
|
if err := helper.DefaultRedis.RPushList(keySpotAddPosition, addVal); err != nil {
|
|
logger.Errorf("handleMainReduceFilled 添加加仓单失败,订单号:%s err:%v", preOrder.OrderSn, err)
|
|
}
|
|
}
|
|
|
|
// 获取现货持仓数量
|
|
func getSpotPositionNum(apiUserInfo DbModels.LineApiUser, preOrder *DbModels.LinePreOrder, tradeSet models2.TradeSet) (decimal.Decimal, error) {
|
|
client := GetClient(&apiUserInfo)
|
|
|
|
resp, _, err := client.SendSpotAuth("/api/v3/account", "GET", map[string]interface{}{
|
|
"omitZeroBalances": true,
|
|
})
|
|
|
|
if err != nil {
|
|
logger.Errorf("api_id:%d 交易对:%s 查询用户信息失败:%v", apiUserInfo.Id, preOrder.Symbol, err)
|
|
return decimal.Decimal{}, err
|
|
}
|
|
|
|
var balanceInfo SpotAccountInfo
|
|
sonic.Unmarshal(resp, &balanceInfo)
|
|
suffix := utility.ReplaceSuffix(preOrder.Symbol, preOrder.QuoteSymbol, "")
|
|
var num decimal.Decimal
|
|
|
|
for _, item := range balanceInfo.Balances {
|
|
if item.Asset == suffix {
|
|
if utility.StrToDecimal(item.Free).Cmp(decimal.Zero) <= 0 {
|
|
logger.Error("handleMainReduceFilled 账户余额不足,交易对：%s 订单号:%s", preOrder.Symbol, preOrder.OrderSn)
|
|
return decimal.Decimal{}, errors.New("账户余额不足")
|
|
}
|
|
|
|
num = utility.StrToDecimal(item.Free).Mul(decimal.NewFromFloat(0.998)).Truncate(int32(tradeSet.AmountDigit))
|
|
break
|
|
}
|
|
}
|
|
return num, nil
|
|
}
|
|
|
|
// 主单取消
|
|
func handleMainOrderCancel(preOrder *DbModels.LinePreOrder) {
|
|
preOrderKey := fmt.Sprintf(rediskey.PreSpotOrderList, global.EXCHANGE_BINANCE)
|
|
preSpotOrders, _ := helper.DefaultRedis.GetAllList(preOrderKey)
|
|
preOrderCache := DbModels.LinePreOrder{}
|
|
|
|
for _, v := range preSpotOrders {
|
|
if v == "" {
|
|
continue
|
|
}
|
|
|
|
sonic.Unmarshal([]byte(v), &preOrderCache)
|
|
if preOrderCache.Pid == preOrder.Id {
|
|
_, err := helper.DefaultRedis.LRem(preOrderKey, v)
|
|
|
|
if err != nil {
|
|
logger.Errorf("订单回调失败, 回调订单号：%s 删除缓存失败:%v", preOrder.OrderSn, err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 平仓单
|
|
func handleMainOrderClosePosition(db *gorm.DB, preOrder *DbModels.LinePreOrder) {
|
|
//主单平仓
|
|
if preOrder.OrderCategory == 1 {
|
|
ids := []int{preOrder.Pid, preOrder.MainId}
|
|
db.Transaction(func(tx *gorm.DB) error {
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("id IN ? AND status=6", ids).Update("status", 9).Error; err != nil {
|
|
logger.Errorf("平仓订单回调失败, 回调订单号：%s 更新主单失败:%v", preOrder.OrderSn, err)
|
|
return err
|
|
}
|
|
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("main_id =? AND status=0", preOrder.MainId).Update("status", 4).Error; err != nil {
|
|
logger.Errorf("平仓订单回调失败, 回调订单号：%s 更新子单失败:%v", preOrder.OrderSn, err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
}
|
|
|
|
removeSpotLossAndAddPosition(preOrder)
|
|
}
|
|
|
|
// 止盈成交
|
|
func handleSpotTakeProfitFilled(db *gorm.DB, preOrder *DbModels.LinePreOrder) {
|
|
removeSpotLossAndAddPosition(preOrder)
|
|
|
|
db.Transaction(func(tx *gorm.DB) error {
|
|
ids := []int{preOrder.Pid, preOrder.MainId}
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("id IN ? AND status =6", ids).Update("status", 9).Error; err != nil {
|
|
logger.Errorf("止盈订单回调失败, 回调订单号：%s 更新主单失败:%v", preOrder.OrderSn, err)
|
|
|
|
return err
|
|
}
|
|
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("main_id =? AND status=0").Update("status", 4).Error; err != nil {
|
|
logger.Errorf("止盈订单回调失败, 回调订单号：%s 更新取消状态失败:%v", preOrder.OrderSn, err)
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
}
|
|
|
|
func removeSpotLossAndAddPosition(preOrder *DbModels.LinePreOrder) {
|
|
stoplossKey := fmt.Sprintf(rediskey.SpotStopLossList, global.EXCHANGE_BINANCE)
|
|
stoplossVal, _ := helper.DefaultRedis.GetAllList(stoplossKey)
|
|
addPositionKey := fmt.Sprintf(rediskey.SpotAddPositionList, global.EXCHANGE_BINANCE)
|
|
addPositionVal, _ := helper.DefaultRedis.GetAllList(addPositionKey)
|
|
reduceKey := fmt.Sprintf(rediskey.SpotReduceList, global.EXCHANGE_BINANCE)
|
|
reduceVal, _ := helper.DefaultRedis.GetAllList(reduceKey)
|
|
stoploss := dto.StopLossRedisList{}
|
|
addPosition := AddPositionList{}
|
|
reduce := ReduceListItem{}
|
|
|
|
//止损缓存
|
|
for _, v := range stoplossVal {
|
|
sonic.Unmarshal([]byte(v), &stoploss)
|
|
if stoploss.MainId == preOrder.MainId {
|
|
_, err := helper.DefaultRedis.LRem(stoplossKey, v)
|
|
|
|
if err != nil {
|
|
logger.Errorf("订单回调失败, 回调订单号：%s 删除止损缓存失败:%v", preOrder.OrderSn, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
//加仓缓存
|
|
for _, v := range addPositionVal {
|
|
sonic.Unmarshal([]byte(v), &addPosition)
|
|
if addPosition.MainId == preOrder.MainId {
|
|
_, err := helper.DefaultRedis.LRem(addPositionKey, v)
|
|
|
|
if err != nil {
|
|
logger.Errorf("订单回调失败, 回调订单号：%s 删除加仓缓存失败:%v", preOrder.OrderSn, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
//减仓缓存
|
|
for _, v := range reduceVal {
|
|
sonic.Unmarshal([]byte(v), &reduce)
|
|
if reduce.MainId == preOrder.MainId {
|
|
_, err := helper.DefaultRedis.LRem(reduceKey, v)
|
|
|
|
if err != nil {
|
|
logger.Errorf("订单回调失败, 回调订单号：%s 删除减仓缓存失败:%v", preOrder.OrderSn, err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 主单成交
|
|
func handleMainOrderFilled(db *gorm.DB, preOrder *DbModels.LinePreOrder) {
|
|
processTakeProfitAndStopLossOrders(db, preOrder)
|
|
tradeSet, _ := GetTradeSet(preOrder.Symbol, 0)
|
|
|
|
if tradeSet.Coin == "" {
|
|
logger.Errorf("获取交易对配置失败, 回调订单号：%s", preOrder.OrderSn)
|
|
return
|
|
}
|
|
|
|
//主单类型
|
|
if preOrder.OrderCategory == 1 {
|
|
//预生成加仓单
|
|
orderExts, err := GetOrderExts(db, preOrder.Id)
|
|
|
|
if err != nil {
|
|
logger.Errorf("预生成加仓单失败, 回调订单号：%s 获取主单拓展配置失败:%v", preOrder.OrderSn, err)
|
|
return
|
|
}
|
|
|
|
price := utility.StrToDecimal(preOrder.Price)
|
|
for _, v := range orderExts {
|
|
if v.OrderId == 0 {
|
|
var data DbModels.LinePreOrder
|
|
|
|
copier.Copy(&data, &v)
|
|
|
|
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
|
|
var percentage decimal.Decimal
|
|
|
|
if data.Site == "BUY" {
|
|
percentage = decimal.NewFromInt(1).Add(v.AddPositionPriceRatio)
|
|
} else {
|
|
percentage = decimal.NewFromInt(1).Sub(v.AddPositionPriceRatio)
|
|
}
|
|
data.Price = price.Mul(percentage).Truncate(int32(tradeSet.PriceDigit)).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
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 解析订单状态
|
|
// 5:委托中 6:完全成交 4:取消 9:已平仓
|
|
func parseOrderStatus(status interface{}, mapData map[string]interface{}) (int, string) {
|
|
reason, _ := mapData["r"].(string)
|
|
|
|
if strings.ToLower(reason) == "none" {
|
|
reason = ""
|
|
}
|
|
|
|
switch status {
|
|
case "NEW": // 未成交
|
|
return 5, reason
|
|
case "FILLED": // 完全成交
|
|
return 6, reason
|
|
case "CANCELED", "EXPIRED": // 取消
|
|
return 4, reason
|
|
default:
|
|
return 0, reason
|
|
}
|
|
}
|
|
|
|
// 更新订单状态
|
|
func updateOrderStatus(db *gorm.DB, preOrder *models.LinePreOrder, status int, reason string, isSpot bool, mapData map[string]interface{}) error {
|
|
params := map[string]interface{}{"status": strconv.Itoa(status), "desc": reason}
|
|
|
|
switch isSpot {
|
|
case true:
|
|
total := decimal.Zero
|
|
totalAmount := decimal.Zero
|
|
if totalStr, ok := mapData["Z"].(string); ok {
|
|
total, _ = decimal.NewFromString(totalStr)
|
|
}
|
|
if totalAmountStr, ok := mapData["z"].(string); ok {
|
|
totalAmount, _ = decimal.NewFromString(totalAmountStr)
|
|
}
|
|
|
|
//主单 修改单价 和成交数量
|
|
if total.Cmp(decimal.Zero) > 0 && totalAmount.Cmp(decimal.Zero) > 0 {
|
|
num := totalAmount.Div(decimal.NewFromFloat(100)).Mul(decimal.NewFromFloat(99.8))
|
|
params["num"] = num
|
|
params["price"] = total.Div(totalAmount)
|
|
preOrder.Num = num.String()
|
|
preOrder.Price = total.Div(totalAmount).String()
|
|
}
|
|
case false:
|
|
status, _ := mapData["X"].(string)
|
|
|
|
if status == "FILLED" {
|
|
num, _ := decimal.NewFromString(mapData["z"].(string))
|
|
params["num"] = num.Mul(decimal.NewFromFloat(0.998)).String()
|
|
price, _ := mapData["ap"].(string)
|
|
|
|
params["price"] = price
|
|
preOrder.Num = num.String()
|
|
preOrder.Price = price
|
|
}
|
|
}
|
|
|
|
return db.Model(&DbModels.LinePreOrder{}).Where("order_sn = ? AND status < 6", preOrder.OrderSn).
|
|
Updates(params).Error
|
|
}
|
|
|
|
// 主单成交 处理止盈止损订单
|
|
// preOrder 主单
|
|
func processTakeProfitAndStopLossOrders(db *gorm.DB, preOrder *models.LinePreOrder) {
|
|
orders := []models.LinePreOrder{}
|
|
tradeSet, _ := GetTradeSet(preOrder.Symbol, 0)
|
|
|
|
if tradeSet.Coin == "" {
|
|
logger.Error("获取交易对失败")
|
|
return
|
|
}
|
|
|
|
if err := db.Model(&DbModels.LinePreOrder{}).
|
|
Where("pid = ? AND order_category = 1 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 = makeSpotTakeAndReduce(preOrder, db, tradeSet, orders)
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
spotApi := SpotRestApi{}
|
|
num, _ := decimal.NewFromString(preOrder.Num)
|
|
|
|
for i, order := range orders {
|
|
if i >= 2 { // 最多处理 2 个订单
|
|
break
|
|
}
|
|
|
|
if err := db.Model(&order).Update("num", num).Error; err != nil {
|
|
logger.Errorf("修改止盈止损数量失败 订单号:%s err:%v", order.OrderSn, err)
|
|
}
|
|
|
|
switch order.OrderType {
|
|
case 1: // 止盈
|
|
processTakeProfitOrder(db, spotApi, order, num)
|
|
case 2: // 止损
|
|
processStopLossOrder(order)
|
|
case 4: //减仓
|
|
processSpotReduceOrder(order, num)
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
// 构建现货止盈、减仓单
|
|
func makeSpotTakeAndReduce(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, err := getSpotPositionNum(apiInfo, preOrder, tradeSet)
|
|
|
|
if err != nil {
|
|
logger.Error("订单回调查询持仓数量失败:", err)
|
|
return nil, errors.New("订单回调查询持仓数量失败")
|
|
}
|
|
|
|
//止盈单
|
|
if ext.TakeProfitRatio.Cmp(decimal.Zero) > 0 {
|
|
profitOrder := models.LinePreOrder{}
|
|
copier.Copy(&profitOrder, preOrder)
|
|
|
|
profitOrder.OrderSn = strconv.FormatInt(snowflakehelper.GetOrderId(), 10)
|
|
profitOrder.Pid = preOrder.Id
|
|
profitOrder.OrderType = 1
|
|
profitOrder.Status = 0
|
|
profitOrder.Rate = ext.TakeProfitRatio.String()
|
|
profitOrder.MainId = ext.MainOrderId
|
|
profitOrder.Num = num.String()
|
|
|
|
//止盈需要累加之前的亏损
|
|
if totalLossAmountU.Cmp(decimal.Zero) > 0 {
|
|
profitOrder.Rate = totalLossAmountU.Div(num).Div(price).Sub(decimal.NewFromInt(1)).Abs().Add(ext.TakeProfitRatio).Truncate(2).String()
|
|
} else {
|
|
profitOrder.Rate = ext.TakeProfitRatio.String()
|
|
}
|
|
|
|
orders = append(orders, profitOrder)
|
|
}
|
|
|
|
//减仓单
|
|
if ext.ReducePriceRatio.Cmp(decimal.Zero) > 0 {
|
|
stopOrder := models.LinePreOrder{}
|
|
copier.Copy(&stopOrder, preOrder)
|
|
|
|
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()
|
|
|
|
orders = append(orders, stopOrder)
|
|
}
|
|
|
|
for index := range orders {
|
|
if strings.ToUpper(preOrder.Site) == "BUY" {
|
|
orders[index].Site = "SELL"
|
|
orders[index].Price = utility.StrToDecimal(preOrder.Price).Mul(decimal.NewFromInt(1).Sub(ext.ReducePriceRatio.Div(decimal.NewFromInt(100)))).Truncate(int32(tradeSet.PriceDigit)).String()
|
|
} else {
|
|
orders[index].Site = "BUY"
|
|
orders[index].Price = utility.StrToDecimal(preOrder.Price).Mul(decimal.NewFromInt(1).Add(ext.ReducePriceRatio.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 processSpotReduceOrder(preOrder DbModels.LinePreOrder, num decimal.Decimal) {
|
|
key := fmt.Sprintf(rediskey.SpotReduceList, global.EXCHANGE_BINANCE)
|
|
item := ReduceListItem{
|
|
Id: preOrder.Id,
|
|
ApiId: preOrder.ApiId,
|
|
Pid: preOrder.Pid,
|
|
MainId: preOrder.MainId,
|
|
Price: utility.StrToDecimal(preOrder.Price),
|
|
Num: num,
|
|
Side: preOrder.Site,
|
|
Symbol: preOrder.Symbol,
|
|
OrderSn: preOrder.OrderSn,
|
|
}
|
|
|
|
itemVal, err := sonic.MarshalString(&item)
|
|
|
|
if err != nil {
|
|
logger.Errorf("序列化失败 err:%v", err)
|
|
return
|
|
}
|
|
|
|
if err := helper.DefaultRedis.RPushList(key, itemVal); err != nil {
|
|
logger.Errorf("减仓单写入缓存失败 err:%v", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// 处理止盈订单
|
|
func processTakeProfitOrder(db *gorm.DB, spotApi SpotRestApi, order models.LinePreOrder, num decimal.Decimal) {
|
|
tradeSet, _ := GetTradeSet(order.Symbol, 0)
|
|
|
|
if tradeSet.Coin == "" {
|
|
logger.Error("获取交易对失败")
|
|
return
|
|
}
|
|
|
|
price, _ := decimal.NewFromString(order.Price)
|
|
|
|
params := OrderPlacementService{
|
|
ApiId: order.ApiId,
|
|
Symbol: order.Symbol,
|
|
Side: order.Site,
|
|
Price: price.Truncate(int32(tradeSet.PriceDigit)),
|
|
Quantity: num.Truncate(int32(tradeSet.AmountDigit)),
|
|
Type: "TAKE_PROFIT_LIMIT",
|
|
TimeInForce: "GTC",
|
|
StopPrice: price.Truncate(int32(tradeSet.PriceDigit)),
|
|
NewClientOrderId: order.OrderSn,
|
|
}
|
|
|
|
err := spotApi.OrderPlace(db, params)
|
|
|
|
if err != nil {
|
|
for x := 0; x < 5; x++ {
|
|
if strings.Contains(err.Error(), "LOT_SIZE") {
|
|
break
|
|
}
|
|
|
|
err = spotApi.OrderPlace(db, params)
|
|
|
|
if err == nil {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if err != nil {
|
|
logger.Error("现货止盈下单失败:", order.OrderSn, " err:", err)
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("id = ?", order.Id).
|
|
Updates(map[string]interface{}{"status": "2", "desc": err.Error(), "num": params.Quantity}).Error; err != nil {
|
|
logger.Error("现货止盈下单失败,更新状态失败:", order.OrderSn, " err:", err)
|
|
}
|
|
} else {
|
|
if err := db.Model(&DbModels.LinePreOrder{}).Where("id = ? and status ='0'", order.Id).
|
|
Updates(map[string]interface{}{"status": "1", "num": num.String()}).Error; err != nil {
|
|
logger.Error("现货止盈下单成功,更新状态失败:", order.OrderSn, " err:", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 处理止损订单
|
|
// order 止损单
|
|
func processStopLossOrder(order models.LinePreOrder) error {
|
|
price := utility.StrToDecimal(order.Price)
|
|
stopLoss := dto.StopLossRedisList{
|
|
Id: order.Id,
|
|
PId: order.Pid,
|
|
ApiId: order.ApiId,
|
|
OrderTye: order.OrderType,
|
|
OrderCategory: order.OrderCategory,
|
|
Price: price,
|
|
SymbolType: order.SymbolType,
|
|
Symbol: order.Symbol,
|
|
Site: order.Site,
|
|
}
|
|
|
|
stoplossVal, _ := sonic.MarshalString(&stopLoss)
|
|
stoplossKey := fmt.Sprintf(rediskey.SpotStopLossList, global.EXCHANGE_BINANCE)
|
|
|
|
if stoplossVal == "" {
|
|
return errors.New("stoplossVal is empty")
|
|
}
|
|
|
|
if err := helper.DefaultRedis.RPushList(stoplossKey, stoplossVal); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetSystemSetting(db *gorm.DB) (models.LineSystemSetting, error) {
|
|
key := fmt.Sprintf(rediskey.SystemSetting)
|
|
val, _ := helper.DefaultRedis.GetString(key)
|
|
setting := models.LineSystemSetting{}
|
|
|
|
if val != "" {
|
|
sonic.UnmarshalString(val, &setting)
|
|
}
|
|
|
|
if setting.Id > 0 {
|
|
return setting, nil
|
|
}
|
|
|
|
var err error
|
|
setting, err = ResetSystemSetting(db)
|
|
if err != nil {
|
|
return setting, err
|
|
}
|
|
|
|
return setting, nil
|
|
}
|
|
|
|
func ResetSystemSetting(db *gorm.DB) (DbModels.LineSystemSetting, error) {
|
|
setting := DbModels.LineSystemSetting{}
|
|
if err := db.Model(&setting).First(&setting).Error; err != nil {
|
|
return setting, err
|
|
}
|
|
|
|
settVal, _ := sonic.MarshalString(&setting)
|
|
|
|
if settVal != "" {
|
|
if err := helper.DefaultRedis.SetString(rediskey.SystemSetting, settVal); err != nil {
|
|
logger.Error("redis添加系统设置失败", err)
|
|
}
|
|
}
|
|
return DbModels.LineSystemSetting{}, nil
|
|
}
|
|
|
|
// NEW
|
|
// PENDING_NEW
|
|
// PARTIALLY_FILLED
|
|
// FILLED
|
|
// CANCELED
|
|
// PENDING_CANCEL
|
|
// REJECTED
|
|
// EXPIRED
|
|
// EXPIRED_IN_MATCH
|