github.com/99designs/gqlgen@v0.17.45/graphql/id.go (about)

     1  package graphql
     2  
     3  import (
     4  	"encoding/json"
     5  	"fmt"
     6  	"io"
     7  	"strconv"
     8  )
     9  
    10  func MarshalID(s string) Marshaler {
    11  	return MarshalString(s)
    12  }
    13  
    14  func UnmarshalID(v interface{}) (string, error) {
    15  	switch v := v.(type) {
    16  	case string:
    17  		return v, nil
    18  	case json.Number:
    19  		return string(v), nil
    20  	case int:
    21  		return strconv.Itoa(v), nil
    22  	case int64:
    23  		return strconv.FormatInt(v, 10), nil
    24  	case float64:
    25  		return fmt.Sprintf("%f", v), nil
    26  	case bool:
    27  		if v {
    28  			return "true", nil
    29  		} else {
    30  			return "false", nil
    31  		}
    32  	case nil:
    33  		return "null", nil
    34  	default:
    35  		return "", fmt.Errorf("%T is not a string", v)
    36  	}
    37  }
    38  
    39  func MarshalIntID(i int) Marshaler {
    40  	return WriterFunc(func(w io.Writer) {
    41  		writeQuotedString(w, strconv.Itoa(i))
    42  	})
    43  }
    44  
    45  func UnmarshalIntID(v interface{}) (int, error) {
    46  	switch v := v.(type) {
    47  	case string:
    48  		return strconv.Atoi(v)
    49  	case int:
    50  		return v, nil
    51  	case int64:
    52  		return int(v), nil
    53  	case json.Number:
    54  		return strconv.Atoi(string(v))
    55  	default:
    56  		return 0, fmt.Errorf("%T is not an int", v)
    57  	}
    58  }