如何使用json字符串值获取iota值？

当前位置： > > > > 如何使用json字符串值获取iota值？

来源：stackoverflow
2024-04-26 16:24:38
0浏览
收藏

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《如何使用json字符串值获取iota值？》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

问题内容

我有一个 constutil 包，其中使用 iota 定义了一些 const 值。

package constutil

type usertype uint

const (
    free usertype = iota + 1
    premium usertype
    ...
)

从 json 我将得到 {"user": "premium", ...}。现在我需要保存用户的值，例如 premium 它是 2。我试图获取这样的值：

constutil.(req.User)

但它不起作用，因为 req.user 返回 string，如："premium"。

我可以使用 map[string]uint 来完成。但是有没有办法使用 iota 来做到这一点？

解决方案

不要认为 iota 值和字符串之间有任何内置的映射方式。有一些工具可以生成用于进行映射的代码。

我也遇到过类似的情况，当我不想依赖生成器或其他工具时，我就做了这样的事情。希望它可以作为某些事情的开始。

package main

import (
    "encoding/json"
    "fmt"
)

type UserType uint

const (
    UserTypeFree UserType = iota
    UserTypePremium
)

var UserTypeToString = map[UserType]string{
    UserTypeFree:    "Free",
    UserTypePremium: "Premium",
}

var UserTypeFromString = map[string]UserType{
    "Free":    UserTypeFree,
    "Premium": UserTypePremium,
}

func (ut UserType) String() string {
    if s, ok := UserTypeToString[ut]; ok {
        return s
    }
    return "unknown"
}

func (ut UserType) MarshalJSON() ([]byte, error) {
    if s, ok := UserTypeToString[ut]; ok {
        return json.Marshal(s)
    }
    return nil, fmt.Errorf("unknown user type %d", ut)
}

func (ut *UserType) UnmarshalJSON(text []byte) error {
    var s string
    if err := json.Unmarshal(text, &s); err != nil {
        return err
    }
    var v UserType
    var ok bool
    if v, ok = UserTypeFromString[s]; !ok {
        return fmt.Errorf("unknown user type %s", s)
    }
    *ut = v
    return nil
}

func main() {
    var ut UserType

    json.Unmarshal([]byte(`"Free"`), &ut)

    fmt.Printf("%#v %v \n", ut, ut)

    b, _ := json.Marshal(ut)

    fmt.Printf("%v\n", string(b))

}

到这里，我们也就讲完了《如何使用json字符串值获取iota值？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注公众号，带你了解更多关于的知识点！