处理响应的动态方法

{
  "error": "this is bad request"
}

{
  "error": {
    "message": "this is error message"
  }
}

我在 stackoverflow 上尝试了各种答案。但无法达成可能的解决方案。使用我目前拥有的代码，它适用于一个 - error type 2。

对于 error 类型 1，我收到错误：

panic: reflect: elem of invalid type string

goroutine 1 [running]:
reflect.(*rtype).elem(0x100a88a80)
        /usr/local/go/src/reflect/type.go:954 +0x194
main.main()
        /users/username/documents/dev/go/prac.go:31 +0x104
exit status 2

对于 error type 2，我得到了完美的输出：

this is error message

package main

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

func main() {

    // Error Type: 1

    str1 := "{\"error\":\"This is bad request\"}"
    var errResp interface{}
    json.Unmarshal([]byte(str1), &errResp)

    // ------------------------------------------------------------------------

    // Error Type: 2

    // str2 := "{\"error\": {\"message\": \"This is error message\"}}"
    // var errResp interface{}
    // json.Unmarshal([]byte(str2), &errResp)

    // ------------------------------------------------------------------------

    // Handling both type dynamically

    errResponse := errResp.(map[string]interface{})
    realErr := errResponse["error"]
    if reflect.TypeOf(realErr).Elem().Kind() == reflect.String {
        // I have used fmt.Sprint here, because I will have to
        // return it as string in my original implementation.
        fmt.Println(fmt.Sprint(realErr))
    }
    lastRealErr := realErr.(map[string]interface{})
    fmt.Println(fmt.Sprint(lastRealErr["message"]))
}

这里^，我已经组合了两种错误类型的代码。要测试它，您只需注释其中一个并取消注释另一个即可进行测试。

你能帮我一下吗？谢谢。

正确答案

声明错误的自定义类型并让它实现 json.Unmarshaler 接口。在实现中，您可以查看原始数据，并根据数据似乎包含的内容决定如何对其进行解组。

type response struct {
    Error responseError
}

type responseError struct {
    Message string
}

func (e *responseError) UnmarshalJSON(data []byte) error {
    if len(data) == 0 || string(data) == `null` { // nothing?
        return nil
    }
    if data[0] == '"' && data[len(data)-1] == '"' { // string?
        return json.Unmarshal(data, &e.Message)
    }

    // avoid infinite recursion by declaring a new type
    // with same structure as responseError but no methods.
    type E responseError 
    return json.Unmarshal(data, (*E)(e))
}

当 encoding/json 包在目标值中遇到 json.unmarshaler 的有效实例时，它将自动调用 unmarshaljson 方法。

https://go.dev/play/p/jWzRSsSuN65

到这里，我们也就讲完了《处理响应的动态方法》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！