mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-22 23:05:15 +00:00
38 lines
817 B
Go
38 lines
817 B
Go
|
package bbgo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func injectStrategyField(strategy SingleExchangeStrategy, rs reflect.Value, fieldName string, obj interface{}) error {
|
||
|
|
field := rs.FieldByName(fieldName)
|
||
|
|
if !field.IsValid() {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
logrus.Infof("found %s in strategy %T, injecting %T...", fieldName, strategy, obj)
|
||
|
|
|
||
|
|
if !field.CanSet() {
|
||
|
|
return errors.Errorf("field %s of strategy %T can not be set", fieldName, strategy)
|
||
|
|
}
|
||
|
|
|
||
|
|
rv := reflect.ValueOf(obj)
|
||
|
|
if field.Kind() == reflect.Ptr {
|
||
|
|
if field.Type() != rv.Type() {
|
||
|
|
return errors.Errorf("field type mismatches: %s != %s", field.Type(), rv.Type())
|
||
|
|
}
|
||
|
|
|
||
|
|
field.Set(rv)
|
||
|
|
} else if field.Kind() == reflect.Interface {
|
||
|
|
field.Set(rv)
|
||
|
|
} else {
|
||
|
|
// set as value
|
||
|
|
field.Set(rv.Elem())
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|