自定义解组器 - 浮动的字符串

golang学习网今天将给大家带来《自定义解组器 - 浮动的字符串》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

我使用第 3 方 api，它以字符串形式提供价格，有 3 种风格：

//input
[
    {"id": "a", "price": "123"},
    {"id": "b", "price": "123.5"},
    {"id": "c", "price": ""}
]

//expected output

[
    a 123,
    b 123.5,
    c 0
]

我知道我可以在我的结构中添加 ,string 部分，但这不会处理订单 c。我发现了这个 github 问题，其中询问了有关 int 的问题。其中有一个演示代码，我根据自己的需要做了一些修改。

如果我将其更改为 float64 而不是 int，我会收到一个错误，订单 a 的日志现在为 0。

json: cannot unmarshal string into go struct field order.price of type float64
[{a 0}]

我认为这个 if string(data) == '""' { 部分代码将检查空字符串（case order c），如果是的话：设置为 0。但是 "123" 不是""。除了下面的部分之外，我理解自定义解组器的其余部分。有人能指出我正确的方向吗？

p := (*float64)(foe)
*p = f

完整代码也在这里

package main

import (
    "encoding/json"
    "fmt"
    "strings"
)

type stringtofloat float64

type order struct {
    id    string        `json:"id"`
    price stringtofloat `json:"price"`
}

func (foe *stringtofloat) unmarshaljson(data []byte) error {
    if string(data) == `""` {
        if foe != nil {
            *foe = 0
        }
        return nil
    }

    var f float64
    err := json.unmarshal(data, &f)
    if err != nil {
        return err
    }
    p := (*float64)(foe)
    *p = f
    return nil
}

func main() {
    jsonstring := `[
        {"id": "a", "price": "123"},
        {"id": "b", "price": "123.5"},
        {"id": "c", "price": ""}
    ]`
    var orders []order
    if err := json.newdecoder(strings.newreader(jsonstring)).decode(&orders); err != nil {
        fmt.println(err)
    }

    fmt.println(orders)
}

编辑：评论@mkopriva

我调整了解组器：

func (foe *stringtofloat) unmarshaljson(data []byte) {
    fmt.println(string(data))
    if string(data) == `""` {
        if foe != nil {
            *foe = 0
        }
    }

    n, err := strconv.parsefloat(string(data), 64)
    if err != nil {
        fmt.println(err)
    }

    *foe = stringtofloat(n)
}

什么给了我这个输出：

json: cannot unmarshal string into Go struct field Order.price of type main.StringToFloat
[{A 0} {B 0} {C 0}]

正确答案

试试这个

package main

import (
    "encoding/json"
    "fmt"
    "strconv"
    "strings"
)

type StringToFloat float64

type Order struct {
    Id    string        `json:"id"`
    Price StringToFloat `json:"price"`
}

func (foe *StringToFloat) UnmarshalJSON(data []byte) error {
    if string(data) == "\"\"" {
        if foe != nil {
            *foe = 0
        }
        return nil
    }
    num := strings.ReplaceAll(string(data), "\"", "")
    n, err := strconv.ParseFloat(num, 64)
    if err != nil {
        return err
    }
    *foe = StringToFloat(n)
    return nil
}

func main() {
    jsonString := `[
        {"id": "A", "price": "123"},
        {"id": "B", "price": "123.5"},
        {"id": "C", "price": ""}
    ]`
    var orders []Order
    if err := json.NewDecoder(strings.NewReader(jsonString)).Decode(&orders); err != nil {
        fmt.Println(err)
    }

    fmt.Println(orders)
}

go演示：https://go.dev/play/p/cHjNQ447eX9

注意事项：

• umarshaljson func 的函数签名需要与 decode 获取的函数签名完全相同。在更新后的代码中，缺少 error 返回类型
• string(data) 给出 "123" 因此需要在 parsefloat 之前删除引号
• 在 string = "" 检查后需要 return nil，因此不会进一步

理论要掌握，实操不能落！以上关于《自定义解组器 - 浮动的字符串》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！