mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 09:11:55 +00:00
36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
package riskcontrol
|
|
|
|
import (
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
indicatorv2 "github.com/c9s/bbgo/pkg/indicator/v2"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type OrderPriceRiskControl struct {
|
|
referencePrice *indicatorv2.EWMAStream
|
|
lossThreshold fixedpoint.Value
|
|
}
|
|
|
|
func NewOrderPriceRiskControl(referencePrice *indicatorv2.EWMAStream, threshold fixedpoint.Value) *OrderPriceRiskControl {
|
|
return &OrderPriceRiskControl{
|
|
referencePrice: referencePrice,
|
|
lossThreshold: threshold,
|
|
}
|
|
}
|
|
|
|
func (r *OrderPriceRiskControl) IsSafe(side types.SideType, price fixedpoint.Value, quantity fixedpoint.Value) bool {
|
|
refPrice := fixedpoint.NewFromFloat(r.referencePrice.Last(0))
|
|
// calculate profit
|
|
var profit fixedpoint.Value
|
|
if side == types.SideTypeBuy {
|
|
profit = refPrice.Sub(price).Mul(quantity)
|
|
} else if side == types.SideTypeSell {
|
|
profit = price.Sub(refPrice).Mul(quantity)
|
|
} else {
|
|
log.Warnf("OrderPriceRiskControl: unsupported side type: %s", side)
|
|
return false
|
|
}
|
|
return profit.Compare(r.lossThreshold) > 0
|
|
}
|