mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 20:34:37 +03:00
30 lines
473 B
Go
30 lines
473 B
Go
package graphql
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
)
|
|
|
|
func MarshalInt(i int) Marshaler {
|
|
return WriterFunc(func(w io.Writer) {
|
|
io.WriteString(w, strconv.Itoa(i))
|
|
})
|
|
}
|
|
|
|
func UnmarshalInt(v interface{}) (int, error) {
|
|
switch v := v.(type) {
|
|
case string:
|
|
return strconv.Atoi(v)
|
|
case int:
|
|
return v, nil
|
|
case int64:
|
|
return int(v), nil
|
|
case json.Number:
|
|
return strconv.Atoi(string(v))
|
|
default:
|
|
return 0, fmt.Errorf("%T is not an int", v)
|
|
}
|
|
}
|