make margin order side effect json unmarshallable

This commit is contained in:
c9s 2021-02-15 01:26:23 +08:00
parent f8378957ee
commit bea750ca97

View File

@ -1,8 +1,11 @@
package types package types
import ( import (
"encoding/json"
"fmt" "fmt"
"strings"
"github.com/pkg/errors"
"github.com/slack-go/slack" "github.com/slack-go/slack"
"github.com/c9s/bbgo/pkg/datatype" "github.com/c9s/bbgo/pkg/datatype"
@ -24,6 +27,32 @@ var (
SideEffectTypeAutoRepay MarginOrderSideEffectType = "AUTO_REPAY" SideEffectTypeAutoRepay MarginOrderSideEffectType = "AUTO_REPAY"
) )
func (t *MarginOrderSideEffectType) UnmarshalJSON(data []byte) error {
var s string
var err = json.Unmarshal(data, &s)
if err != nil {
return errors.Wrapf(err, "unable to unmarshal side effect type: %s", data)
}
switch strings.ToUpper(s) {
case string(SideEffectTypeNoSideEffect), "":
*t = SideEffectTypeNoSideEffect
return nil
case string(SideEffectTypeMarginBuy):
*t = SideEffectTypeMarginBuy
return nil
case string(SideEffectTypeAutoRepay):
*t = SideEffectTypeAutoRepay
return nil
}
return fmt.Errorf("invalid side effect type: %s", data)
}
// OrderType define order type // OrderType define order type
type OrderType string type OrderType string