forked from c9s/bbgo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
risk_controls.go
74 lines (56 loc) · 2.04 KB
/
risk_controls.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package bbgo
import (
"context"
"github.com/sirupsen/logrus"
"github.com/c9s/bbgo/pkg/types"
)
type SymbolBasedRiskController struct {
BasicRiskController *BasicRiskController `json:"basic,omitempty" yaml:"basic,omitempty"`
}
type RiskControlOrderExecutor struct {
*ExchangeOrderExecutor
// Symbol => Executor config
BySymbol map[string]*SymbolBasedRiskController `json:"bySymbol,omitempty" yaml:"bySymbol,omitempty"`
}
func (e *RiskControlOrderExecutor) SubmitOrders(ctx context.Context, orders ...types.SubmitOrder) (retOrders types.OrderSlice, err error) {
var symbolOrders = groupSubmitOrdersBySymbol(orders)
for symbol, orders := range symbolOrders {
if controller, ok := e.BySymbol[symbol]; ok && controller != nil {
var riskErrs []error
orders, riskErrs = controller.BasicRiskController.ProcessOrders(e.Session, orders...)
for _, riskErr := range riskErrs {
// use logger from ExchangeOrderExecutor
logrus.Warnf("RISK ERROR: %s", riskErr.Error())
}
}
formattedOrders, err := e.Session.FormatOrders(orders)
if err != nil {
return retOrders, err
}
retOrders2, err := e.ExchangeOrderExecutor.SubmitOrders(ctx, formattedOrders...)
if err != nil {
return retOrders, err
}
retOrders = append(retOrders, retOrders2...)
}
return
}
type SessionBasedRiskControl struct {
OrderExecutor *RiskControlOrderExecutor `json:"orderExecutor,omitempty" yaml:"orderExecutor"`
}
func (control *SessionBasedRiskControl) SetBaseOrderExecutor(executor *ExchangeOrderExecutor) {
if control.OrderExecutor == nil {
return
}
control.OrderExecutor.ExchangeOrderExecutor = executor
}
func groupSubmitOrdersBySymbol(orders []types.SubmitOrder) map[string][]types.SubmitOrder {
var symbolOrders = make(map[string][]types.SubmitOrder, len(orders))
for _, order := range orders {
symbolOrders[order.Symbol] = append(symbolOrders[order.Symbol], order)
}
return symbolOrders
}
type RiskControls struct {
SessionBasedRiskControl map[string]*SessionBasedRiskControl `json:"sessionBased,omitempty" yaml:"sessionBased,omitempty"`
}