16 changed files with 511 additions and 642 deletions
@ -1,209 +0,0 @@
|
||||
package backtest |
||||
|
||||
import ( |
||||
"sig-pub/pkg/trade" |
||||
"sig-pub/pkg/types" |
||||
"sig-pub/pkg/types/decimals" |
||||
) |
||||
|
||||
// CloseManager 管理持仓平仓逻辑：stoploss/takeprofit 与 基于信号的平仓
|
||||
// 1.交易信号和持单方向相反时是否进行平仓
|
||||
// 2.计算止盈止损时是否包含手续费
|
||||
type CloseManager struct { |
||||
StopLossPct float64 // static stoploss
|
||||
TakeProfitPct float64 // static take profit
|
||||
TrailingPct float64 // trailing stop percent (e.g. 0.02 = 2%)；移动止损百分比（例如 0.02 表示从最高价回撤 2% 时触发追踪止损）。
|
||||
MinProfitToTrail float64 // minimum profit (fraction) before trailing activates (e.g. 0.01 = 1%)；启动移动止损的最小盈利阈值（例如达到 1% 后才开始追踪）。
|
||||
ProfitRetracePct float64 // close when profit retraces more than this fraction of peak profit；基于最高利润回撤触发平仓（例如从最高利润回撤超过 30% 则平仓）。
|
||||
} |
||||
|
||||
func NewCloseManager(stopLossPct, takeProfitPct float64) *CloseManager { |
||||
return &CloseManager{StopLossPct: stopLossPct, TakeProfitPct: takeProfitPct} |
||||
} |
||||
|
||||
func (m *CloseManager) SetDynamicParams(trailingPct, minProfitToTrail, profitRetracePct float64) { |
||||
m.TrailingPct = trailingPct |
||||
m.MinProfitToTrail = minProfitToTrail |
||||
m.ProfitRetracePct = profitRetracePct |
||||
} |
||||
|
||||
func (m *CloseManager) Init(param trade.CloseStrategyParam) { |
||||
|
||||
} |
||||
|
||||
// OnKline 根据最新 kline 检查是否触发 stoploss 或 takeprofit，触发则平仓（市价）
|
||||
// 返回发生的平仓成交记录
|
||||
func (m *CloseManager) OnKline(k types.Kline, acct *Account) (trades []*Trade) { |
||||
if acct == nil { |
||||
return |
||||
} |
||||
if (m.StopLossPct <= 0) && (m.TakeProfitPct <= 0) { |
||||
return |
||||
} |
||||
|
||||
// collect indices to close to avoid modifying slice during iteration
|
||||
type closeTask struct { |
||||
tradeId int64 |
||||
cause string |
||||
} |
||||
var toClose []closeTask |
||||
closePrice := decimals.MustToFloat64(k.Close) |
||||
|
||||
for _, p := range acct.Positions { |
||||
if p == nil { |
||||
continue |
||||
} |
||||
entry := p.EntryPx |
||||
// update peak px
|
||||
switch p.Side { |
||||
case types.SideLong: |
||||
if closePrice > p.PeakPx { |
||||
p.PeakPx = closePrice |
||||
} |
||||
case types.SideShort: |
||||
if closePrice < p.PeakPx { |
||||
p.PeakPx = closePrice |
||||
} |
||||
} |
||||
// 固定止盈止损
|
||||
if p.Side == types.SideLong { |
||||
// stoploss
|
||||
if m.StopLossPct > 0 && closePrice <= entry*(1-m.StopLossPct) { |
||||
acct.Stat.StoplossTimes++ |
||||
toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "stoploss"}) |
||||
continue |
||||
} |
||||
// takeprofit
|
||||
// if m.TakeProfitPct > 0 && priceHigh >= entry*(1+m.TakeProfitPct) {
|
||||
// acct.Stat.TakeprofitTimes++
|
||||
// toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "takeprofit"})
|
||||
// continue
|
||||
// }
|
||||
// dynamic trailing stop based on peak price
|
||||
if m.TrailingPct > 0 && m.MinProfitToTrail > 0 { |
||||
// peak profit fraction
|
||||
peakProfit := (p.PeakPx - entry) / entry |
||||
if peakProfit >= m.MinProfitToTrail { // 最高盈利百分比
|
||||
// trailing level
|
||||
// trailLevel := p.PeakPx * (1 - m.TrailingPct)
|
||||
// if closePrice <= trailLevel {
|
||||
trail := (p.PeakPx - entry) * (1 - m.TrailingPct) |
||||
if (closePrice - entry) < trail { |
||||
acct.Stat.TrailingTimes++ |
||||
toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "trailing"}) |
||||
continue |
||||
} |
||||
} |
||||
} |
||||
// profit retrace rule: if peakProfit>0 and current retrace > ProfitRetracePct
|
||||
// if m.ProfitRetracePct > 0 {
|
||||
// peakProfit := (p.PeakPx - entry) / entry
|
||||
// curProfit := (closePrice - entry) / entry
|
||||
// if peakProfit > 0 {
|
||||
// retrace := (peakProfit - curProfit) / peakProfit
|
||||
// if retrace >= m.ProfitRetracePct {
|
||||
// acct.Stat.RetraceTimes++
|
||||
// toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "retrace"})
|
||||
// continue
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
} else if p.Side == types.SideShort { |
||||
// short: stoploss if high >= entry*(1+stop), takeprofit if low <= entry*(1-tp)
|
||||
if m.StopLossPct > 0 && closePrice >= entry*(1+m.StopLossPct) { |
||||
acct.Stat.StoplossTimes++ |
||||
toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "stoploss"}) |
||||
continue |
||||
} |
||||
// if m.TakeProfitPct > 0 && priceLow <= entry*(1-m.TakeProfitPct) {
|
||||
// acct.Stat.TakeprofitTimes++
|
||||
// toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "takeprofit"})
|
||||
// continue
|
||||
// }
|
||||
// update trailing for short based on PeakPx (lower is better for short)
|
||||
if m.TrailingPct > 0 && m.MinProfitToTrail > 0 { |
||||
peakProfit := (entry - p.PeakPx) / entry |
||||
if peakProfit >= m.MinProfitToTrail { |
||||
// trailLevel := p.PeakPx * (1 + m.TrailingPct)
|
||||
// if closePrice >= trailLevel {
|
||||
trail := (entry - p.PeakPx) * (1 - m.TrailingPct) |
||||
if (entry - closePrice) < trail { |
||||
acct.Stat.TrailingTimes++ |
||||
toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "trailing"}) |
||||
continue |
||||
} |
||||
} |
||||
} |
||||
// if m.ProfitRetracePct > 0 {
|
||||
// peakProfit := (entry - p.PeakPx) / entry
|
||||
// curProfit := (entry - closePrice) / entry
|
||||
// if peakProfit > 0 {
|
||||
// retrace := (peakProfit - curProfit) / peakProfit
|
||||
// if retrace >= m.ProfitRetracePct {
|
||||
// acct.Stat.RetraceTimes++
|
||||
// toClose = append(toClose, closeTask{tradeId: p.TradeId, cause: "retrace"})
|
||||
// continue
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
} |
||||
} |
||||
|
||||
// close collected positions (process from high index to low to safely remove)
|
||||
for j := len(toClose) - 1; j >= 0; j-- { |
||||
tradeId := toClose[j].tradeId |
||||
cause := toClose[j].cause |
||||
var pos *Position |
||||
var index int |
||||
for i, p := range acct.Positions { |
||||
if p.TradeId == tradeId { |
||||
pos = p |
||||
index = i |
||||
break |
||||
} |
||||
} |
||||
if pos == nil { |
||||
return |
||||
} |
||||
|
||||
// perform market close: side opposite
|
||||
tr, ok := acct.ClosePosition(index, pos, k, k.Ts, cause) |
||||
if ok { |
||||
trades = append(trades, tr) |
||||
} |
||||
} |
||||
return |
||||
} |
||||
|
||||
// CloseBySignal 根据策略信号尝试平掉相反方向的仓位。例如策略返回 SELL 时，尝试平掉所有 BUY 持仓
|
||||
func (m *CloseManager) CloseBySignal(sigSide types.Side, acct *Account, k types.Kline) (trades []*Trade) { |
||||
if acct == nil { |
||||
return |
||||
} |
||||
// determine which positions to close: positions with opposite side to sigSide
|
||||
type closeTask struct { |
||||
idx int |
||||
cause string |
||||
pos *Position |
||||
} |
||||
var toClose []closeTask |
||||
for i, p := range acct.Positions { |
||||
if p == nil { |
||||
continue |
||||
} |
||||
if sigSide == types.SideLong && p.Side == types.SideShort { |
||||
toClose = append(toClose, closeTask{idx: i, cause: "signal", pos: p}) |
||||
} else if sigSide == types.SideShort && p.Side == types.SideLong { |
||||
toClose = append(toClose, closeTask{idx: i, cause: "signal", pos: p}) |
||||
} |
||||
} |
||||
for j := len(toClose) - 1; j >= 0; j-- { |
||||
idx := toClose[j].idx |
||||
cause := toClose[j].cause |
||||
tr, ok := acct.ClosePosition(idx, toClose[j].pos, k, k.Ts, cause) |
||||
if ok { |
||||
trades = append(trades, tr) |
||||
} |
||||
} |
||||
return |
||||
} |
||||
Loading…
Reference in new issue