go实战系列——枚举

枚举

现实场景

场景	备注
商城订单状态，返回的数据中状态值要调整为对应的中文状态描述	excel导出时会遇到

示例

//demand售后单类型 1退货 2退物退货 3换货，需要提取码值帮助类
switch v.Demand {
case 1:
    mData["demand"] = "退货"
case 2:
    mData["demand"] = "退物退货"
case 3:
    mData["demand"] = "换货"
default:
    mData["demand"] = "其他"
}

//还有订单状态
switch v.Status {
case "WAIT_SELLER_AGREE":
    mData["status"] = "卖家同意退款"
case "WAIT_BUYER_RETURN_GOODS":
    mData["status"] = "买家退货中"
case "WAIT_SELLER_CONFIRM_GOODS":
    mData["status"] = "卖家待收货"
case "SELLER_REFUSE_BUYER":
    mData["status"] = "卖家拒绝退款"
case "CLOSED":
    mData["status"] = "退款关闭"
case "SUCCESS":
    mData["status"] = "退款成功"
default:
    mData["status"] = "其他"
}

比较惨，容易很轻松写出百万行代码~

能不能一行代码就得到自己想要的字符串
我去尝试造轮子

package main

import "fmt"

//方法一

var Enum_DemandType = map[int]string{
    1: "退货",
    2: "退物退货",
}

var Enum_OrderType = map[int]string{
    1: "待付款",
    2: "支付中",
    3: "已支付",
}

func ToEnumValue(i int, c map[int]string) string {
    v, is_ok := c[i]
    if !is_ok {
        //弊端就是其他类型，需要自己再处理
        v = "其他"
    }
    return v
}

func main() {
    input_demand := 1   
    fmt.Println(ToEnumValue(input_demand, Enum_DemandType))
    fmt.Println(ToEnumValue(input_demand, Enum_OrderType ))
}

这是用枚举的方式去做出枚举的效果。

另一个是这种

package main

import "fmt"


type DemandType int

const (
    TuiHuo DemandType = iota + 1
    TuiWuTuiHuo
)

func (c DemandType) String() string {
    switch c {
    case TuiHuo:
        return "退货"
    case TuiWuTuiHuo:
        return "退物退货11"
    }
    return "N/A"
}

func (c DemandType) Enum_String() string {
    switch c {
    case TuiHuo:
        return TuiHuo.String() //也以直接赋值
    case TuiWuTuiHuo:
        return "退物退货"
    }
    return "其他"
}

func main() {
    input_demand := 1
    fmt.Println(DemandType(input_demand).Enum_String())
}

看上去似乎正规了那么一点，各位看官有没有更换的使用方式，可以分享让我学习一下哦~