golang 可以做类似 C++ 中#define 的事情吗？

在IT行业这个发展更新速度很快的行业，只有不停止的学习，才不会被行业所淘汰。如果你是Golang学习者，那么本文《golang 可以做类似 C++ 中#define 的事情吗？》就很适合你！本篇内容主要包括##content_title##，希望对大家的知识积累有所帮助，助力实战开发！

我使用 protobuf 定义了 3 种消息类型。 （消息a、消息b、消息c）

message msga {
    string content;
    int64 a;
};
message msgb {
    string content;
    char b;
};
message msgc {
    string content;
    double c;
};

我定义了一个msgtype来指示消息是msga/msgb/msgc

message msgtype {
    string type; // indicate msga/ msgb/ msgc
};

然后，我生成了一些消息并以这种格式存储在内存映射文件中：

|msgtype|msga/msgb/msgc|some end marker|

当我从缓冲区读取时，我想做类似的事情：

msgtype := &msgtype{}
err := proto.unmarshal(bytearrayformsgtype, msgtype)
...
switch msgtype.gettype() {
case "msga":
    a := &msga{}
    err := prto.unmarshal(bytearrayformsg, a)
    ...
case "msgb":
    b := &msgb{}
    err := prto.unmarshal(bytearrayformsg, b)
    ...
case "msgc":
    c := &msgc{}
    err := prto.unmarshal(bytearrayformsg, c)
    ...
}

问题是这样的： 由于每种情况都非常相似，所以我想做一些类似于 c++ 的事情

#define CASE(MsgType)\
    case #MsgType:\
    msg := createObject();\
    ...
switch type {
    CASE(MsgA);
    CASE(MsgB);
    CASE(MsgC);
}

其实消息类型有很多种，不只是a、b、c。每个案例部分都会有重复的代码。 go中有没有方法可以做与c++类似的事情？

解决方案

带有反射

您可以使用映射来存储从类型名称映射的类型描述符（例如 msgtype.type 字段）。类型描述符可能是reflect.Type。

因此，您可以通过简单的映射查找来获取类型描述符，并且可以使用 reflect.New() 来获取指向该类型的新的归零值的指针。

例如：

var registry = map[string]reflect.type{
    "msga" : reflect.typeof(msga{}),
    "msgb" : reflect.typeof(msgb{}),
    "msgc" : reflect.typeof(msgc{}),
}

以及阅读消息时的常用代码：

typetoread := registry[msgtype.gettype()]

msg := reflect.new(typetoread).interface()
err := prto.unmarshal(bytearrayformsg, msg.(proto.message))

注意：msg 将是静态类型 interface{}，它包装一个指向您的消息类型之一的指针，例如存储它的具体值的类型可能是 *msga，正是您必须传递给 proto.unmarshal() 的类型。

带有构造函数

获取新消息值的另一种方法可能是使用构造函数，因此不需要反射。

这就是它的样子：

var registry = map[string]func() proto.message{
    "msga" : func() proto.message { return new(msga) },
    "msgb" : func() proto.message { return new(msgb) },
    "msgc" : func() proto.message { return new(msgc) },
}

并使用它：

creator := registry[msgType.GetType()]

msg := creator()
err := prto.Unmarshal(byteArrayforMsg, msg)

今天关于《golang 可以做类似 C++ 中#define 的事情吗？》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！