在Go中创建一个带约束的自定义类型

时间:2013-09-24 22:06:51

标签: go

如何在Go中创建只接受有效值的自定义类型? 例如,我想创建一个名为“Names”的类型,其底层类型是一个字符串。但是它只能接受“John”,“Rob”或“Paul”这两个值。任何其他值都将返回错误。 我以非常简单的方式创建了以下程序,只是为了表示我想要实现的目标。

http://play.golang.org/p/jzZwALsiXz

编写此代码的最佳方法是什么?

1 个答案:

答案 0 :(得分:8)

你可以这样做(http://play.golang.org/p/JaIr_0a5_-):

type Name struct {
    string
}

func (n *Name) String() string {
    return n.string
}

func NewName(name string) (*Name, error) {
    switch name {
    case "John":
    case "Paul":
    case "Rob":
    default:
        return nil, fmt.Errorf("Wrong value")
    }
    return &Name{string: name}, nil
}

Golang不提供运算符重载,因此您无法在转换或影响值时进行检查。

根据您要执行的操作,您可能希望执行以下操作(http://play.golang.org/p/uXtnHKNRxk):

type Name string

func (n Name) String() string {
    switch n {
    case "John":
    case "Paul":
    case "Rob":
    default:
        return "Error: Wrong value"
    }
    return string(n)
}