cmd/bpa-operator/vendor/github.com/json-iterator/go/reflect_json_number.go

   1 package jsoniter
   2
   3 import (
   4         "encoding/json"
   5         "github.com/modern-go/reflect2"
   6         "strconv"
   7         "unsafe"
   8 )
   9
  10 type Number string
  11
  12 // String returns the literal text of the number.
  13 func (n Number) String() string { return string(n) }
  14
  15 // Float64 returns the number as a float64.
  16 func (n Number) Float64() (float64, error) {
  17         return strconv.ParseFloat(string(n), 64)
  18 }
  19
  20 // Int64 returns the number as an int64.
  21 func (n Number) Int64() (int64, error) {
  22         return strconv.ParseInt(string(n), 10, 64)
  23 }
  24
  25 func CastJsonNumber(val interface{}) (string, bool) {
  26         switch typedVal := val.(type) {
  27         case json.Number:
  28                 return string(typedVal), true
  29         case Number:
  30                 return string(typedVal), true
  31         }
  32         return "", false
  33 }
  34
  35 var jsonNumberType = reflect2.TypeOfPtr((*json.Number)(nil)).Elem()
  36 var jsoniterNumberType = reflect2.TypeOfPtr((*Number)(nil)).Elem()
  37
  38 func createDecoderOfJsonNumber(ctx *ctx, typ reflect2.Type) ValDecoder {
  39         if typ.AssignableTo(jsonNumberType) {
  40                 return &jsonNumberCodec{}
  41         }
  42         if typ.AssignableTo(jsoniterNumberType) {
  43                 return &jsoniterNumberCodec{}
  44         }
  45         return nil
  46 }
  47
  48 func createEncoderOfJsonNumber(ctx *ctx, typ reflect2.Type) ValEncoder {
  49         if typ.AssignableTo(jsonNumberType) {
  50                 return &jsonNumberCodec{}
  51         }
  52         if typ.AssignableTo(jsoniterNumberType) {
  53                 return &jsoniterNumberCodec{}
  54         }
  55         return nil
  56 }
  57
  58 type jsonNumberCodec struct {
  59 }
  60
  61 func (codec *jsonNumberCodec) Decode(ptr unsafe.Pointer, iter *Iterator) {
  62         switch iter.WhatIsNext() {
  63         case StringValue:
  64                 *((*json.Number)(ptr)) = json.Number(iter.ReadString())
  65         case NilValue:
  66                 iter.skipFourBytes('n', 'u', 'l', 'l')
  67                 *((*json.Number)(ptr)) = ""
  68         default:
  69                 *((*json.Number)(ptr)) = json.Number([]byte(iter.readNumberAsString()))
  70         }
  71 }
  72
  73 func (codec *jsonNumberCodec) Encode(ptr unsafe.Pointer, stream *Stream) {
  74         number := *((*json.Number)(ptr))
  75         if len(number) == 0 {
  76                 stream.writeByte('0')
  77         } else {
  78                 stream.WriteRaw(string(number))
  79         }
  80 }
  81
  82 func (codec *jsonNumberCodec) IsEmpty(ptr unsafe.Pointer) bool {
  83         return len(*((*json.Number)(ptr))) == 0
  84 }
  85
  86 type jsoniterNumberCodec struct {
  87 }
  88
  89 func (codec *jsoniterNumberCodec) Decode(ptr unsafe.Pointer, iter *Iterator) {
  90         switch iter.WhatIsNext() {
  91         case StringValue:
  92                 *((*Number)(ptr)) = Number(iter.ReadString())
  93         case NilValue:
  94                 iter.skipFourBytes('n', 'u', 'l', 'l')
  95                 *((*Number)(ptr)) = ""
  96         default:
  97                 *((*Number)(ptr)) = Number([]byte(iter.readNumberAsString()))
  98         }
  99 }
 100
 101 func (codec *jsoniterNumberCodec) Encode(ptr unsafe.Pointer, stream *Stream) {
 102         number := *((*Number)(ptr))
 103         if len(number) == 0 {
 104                 stream.writeByte('0')
 105         } else {
 106                 stream.WriteRaw(string(number))
 107         }
 108 }
 109
 110 func (codec *jsoniterNumberCodec) IsEmpty(ptr unsafe.Pointer) bool {
 111         return len(*((*Number)(ptr))) == 0
 112 }