mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-22 23:05:15 +00:00
Merge pull request #800 from c9s/fix/trailing-stop
fix: fix exit method for trailing stop
This commit is contained in:
commit
c20b3e991b
|
@ -75,4 +75,8 @@ func (m *ExitMethod) Bind(session *ExchangeSession, orderExecutor *GeneralOrderE
|
||||||
if m.CumulatedVolumeTakeProfit != nil {
|
if m.CumulatedVolumeTakeProfit != nil {
|
||||||
m.CumulatedVolumeTakeProfit.Bind(session, orderExecutor)
|
m.CumulatedVolumeTakeProfit.Bind(session, orderExecutor)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if m.TrailingStop != nil {
|
||||||
|
m.TrailingStop.Bind(session, orderExecutor)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,6 +10,48 @@ type StructFieldIterator func(tag string, ft reflect.StructField, fv reflect.Val
|
||||||
|
|
||||||
var ErrCanNotIterateNilPointer = errors.New("can not iterate struct on a nil pointer")
|
var ErrCanNotIterateNilPointer = errors.New("can not iterate struct on a nil pointer")
|
||||||
|
|
||||||
|
func IterateFields(obj interface{}, cb func(ft reflect.StructField, fv reflect.Value) error) error {
|
||||||
|
if obj == nil {
|
||||||
|
return errors.New("can not iterate field, given object is nil")
|
||||||
|
}
|
||||||
|
|
||||||
|
sv := reflect.ValueOf(obj)
|
||||||
|
st := reflect.TypeOf(obj)
|
||||||
|
|
||||||
|
if st.Kind() != reflect.Ptr {
|
||||||
|
return fmt.Errorf("f should be a pointer of a struct, %s given", st)
|
||||||
|
}
|
||||||
|
|
||||||
|
// for pointer, check if it's nil
|
||||||
|
if sv.IsNil() {
|
||||||
|
return ErrCanNotIterateNilPointer
|
||||||
|
}
|
||||||
|
|
||||||
|
// solve the reference
|
||||||
|
st = st.Elem()
|
||||||
|
sv = sv.Elem()
|
||||||
|
|
||||||
|
if st.Kind() != reflect.Struct {
|
||||||
|
return fmt.Errorf("f should be a struct, %s given", st)
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < sv.NumField(); i++ {
|
||||||
|
fv := sv.Field(i)
|
||||||
|
ft := st.Field(i)
|
||||||
|
|
||||||
|
// skip unexported fields
|
||||||
|
if !st.Field(i).IsExported() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := cb(ft, fv) ; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func IterateFieldsByTag(obj interface{}, tagName string, cb StructFieldIterator) error {
|
func IterateFieldsByTag(obj interface{}, tagName string, cb StructFieldIterator) error {
|
||||||
sv := reflect.ValueOf(obj)
|
sv := reflect.ValueOf(obj)
|
||||||
st := reflect.TypeOf(obj)
|
st := reflect.TypeOf(obj)
|
||||||
|
|
44
pkg/dynamic/iterate_test.go
Normal file
44
pkg/dynamic/iterate_test.go
Normal file
|
@ -0,0 +1,44 @@
|
||||||
|
package dynamic
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestIterateFields(t *testing.T) {
|
||||||
|
|
||||||
|
t.Run("basic", func(t *testing.T) {
|
||||||
|
var a = struct {
|
||||||
|
A int
|
||||||
|
B float64
|
||||||
|
C *os.File
|
||||||
|
}{}
|
||||||
|
|
||||||
|
cnt := 0
|
||||||
|
err := IterateFields(&a, func(ft reflect.StructField, fv reflect.Value) error {
|
||||||
|
cnt++
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, 3, cnt)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("non-ptr", func(t *testing.T) {
|
||||||
|
err := IterateFields(struct{}{}, func(ft reflect.StructField, fv reflect.Value) error {
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
assert.Error(t, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("nil", func(t *testing.T) {
|
||||||
|
err := IterateFields(nil, func(ft reflect.StructField, fv reflect.Value) error {
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
assert.Error(t, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user