douxuan4556
2016-01-26 13:54
浏览 97

自定义JSON解组字符串编码的数字

I have a struct which contains various currency values, in cents (1/100 USD):

type CurrencyValues struct {
   v1 int `json:"v1,string"`
   v2 int `json:"v2,string"`
} 

I'd like to create a custom json Unmarshaller for currency values with thousand separators. These values are encoded as strings, with one or more thousand separators (,), and possibly a decimal point (.).

For this JSON {"v1": "10", "v2": "1,503.21"}, I'd like to JSON Unmarshal a CurrencyValues{v1: 1000, v2: 150321}.

Following a similar answer here: Golang: How to unmarshall both 0 and false as bool from JSON, I went ahead and created a custom type for my currency fields, which include a custom Unmarshalling function:

type ConvertibleCentValue int

func (cents *ConvertibleCentValue) UnmarshalJSON(data []byte) error {
    asString := string(data)

    // Remove thousands separators
    asString = strings.Replace(asString, ",", "", -1)

    // Parse to float, then convert dollars to cents
    if floatVal, err := strconv.ParseFloat(asString, 32); err == nil {
        *cents = ConvertibleCentValue(int(floatVal * 100.0))
        return nil
    } else {
        return err
    }
}

However, when writing unit tests:

func Test_ConvertibleCentValue_Unmarshal(t *testing.T) {
    var c ConvertibleCentValue
    assert.Nil(t, json.Unmarshal([]byte("1,500"), &c))
    assert.Equal(t, 150000, int(c))
}

I encounter this error:

Error:      Expected nil, but got: &json.SyntaxError{msg:"invalid character ',' after top-level value", Offset:2}

What am I missing here?

  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

1条回答 默认 最新

  • dongzha3058 2016-01-26 13:59
    已采纳

    You're trying to unmarshal the string 1,500 which is invalid in JSON. I think what you means is to unmarshal the JSON string "1,500":

    assert.Nil(t, json.Unmarshal([]byte(`"1,500"`), &c))
    

    Note the backticks. Here is a simplified example:

    b := []byte(`1,500`)
    var s string
    err := json.Unmarshal(b, &s)
    fmt.Println(s, err) // Prints error.
    
    b = []byte(`"1,500"`)
    err = json.Unmarshal(b, &s)
    fmt.Println(s, err) // Works fine.
    

    Playground: http://play.golang.org/p/uwayOSgmTv.

    打赏 评论

相关推荐 更多相似问题