37 lines
709 B
Go
37 lines
709 B
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strconv"
|
|
)
|
|
|
|
// StringToInt this is a custom type that will allow our request bodies to support either a string OR an int.
|
|
// It has its own UnmarshalJSON method to handle both types automatically and it can return an `int`
|
|
// from the `Int` method.
|
|
type StringToInt int
|
|
|
|
func (s *StringToInt) UnmarshalJSON(data []byte) error {
|
|
var i int
|
|
err := json.Unmarshal(data, &i)
|
|
if err == nil {
|
|
*s = StringToInt(i)
|
|
return nil
|
|
}
|
|
|
|
var str string
|
|
err = json.Unmarshal(data, &str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tmp, err := strconv.Atoi(str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*s = StringToInt(tmp)
|
|
return nil
|
|
}
|
|
|
|
func (s *StringToInt) Int() int {
|
|
return int(*s)
|
|
}
|