mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-27 09:15:15 +00:00
types: add StrInt64 type for unmarshalling integer in string
This commit is contained in:
parent
10311f5c93
commit
d7183cb38f
38
pkg/types/strint.go
Normal file
38
pkg/types/strint.go
Normal file
|
@ -0,0 +1,38 @@
|
|||
package types
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type StrInt64 int64
|
||||
|
||||
func (s *StrInt64) UnmarshalJSON(body []byte) error {
|
||||
var arg interface{}
|
||||
if err := json.Unmarshal(body, &arg); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
switch ta := arg.(type) {
|
||||
case string:
|
||||
// parse string
|
||||
i, err := strconv.ParseInt(ta, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*s = StrInt64(i)
|
||||
|
||||
case int64:
|
||||
*s = StrInt64(ta)
|
||||
case int32:
|
||||
*s = StrInt64(ta)
|
||||
case int:
|
||||
*s = StrInt64(ta)
|
||||
|
||||
default:
|
||||
return fmt.Errorf("StrInt64 error: unsupported value type %T", ta)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
Loading…
Reference in New Issue
Block a user