bbgo/pkg/bbgo/stop_ema.go

43 lines
1.1 KiB
Go

package bbgo
import (
"github.com/sirupsen/logrus"
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
"git.qtrade.icu/lychiyu/bbgo/pkg/indicator"
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
)
type StopEMA struct {
types.IntervalWindow
Range fixedpoint.Value `json:"range"`
stopEWMA *indicator.EWMA
}
func (s *StopEMA) Bind(session *ExchangeSession, orderExecutor *GeneralOrderExecutor) {
symbol := orderExecutor.Position().Symbol
s.stopEWMA = session.StandardIndicatorSet(symbol).EWMA(s.IntervalWindow)
}
func (s *StopEMA) Allowed(closePrice fixedpoint.Value) bool {
ema := fixedpoint.NewFromFloat(s.stopEWMA.Last(0))
if ema.IsZero() {
logrus.Infof("stopEMA protection: value is zero, skip")
return false
}
emaStopShortPrice := ema.Mul(fixedpoint.One.Sub(s.Range))
if closePrice.Compare(emaStopShortPrice) < 0 {
Notify("stopEMA %s protection: close price %f less than stopEMA %f = EMA(%f) * (1 - RANGE %f)",
s.IntervalWindow.String(),
closePrice.Float64(),
emaStopShortPrice.Float64(),
ema.Float64(),
s.Range.Float64())
return false
}
return true
}