开发protoc插件时，如何用类型判断确定字段的消息类型？

我正在使用protogen包开发protoc生成插件，需要遍历消息的字段，判断某个字段是否属于几种特定的消息类型。

目前可以通过以下代码获取消息类型的全名：

field.Desc.Message().FullName() // 输出示例：mypackage.MyMessage

但这种基于字符串的switch判断方式容易出错：

switch field.Desc.Message().FullName(){
  case "mypackage.MyMessage":
    // 处理逻辑
  case "mypackage.MyMessage2":
    // 处理逻辑
}

我尝试用dynamicpb创建消息实例来做类型断言，但失败了：

mt := dynamicpb.NewMessage(field.Desc.Message())

msg, ok := mt.(*mypackage.MyMessage) // ok为false，尽管field.Desc.Message().FullName()返回mypackage.MyMessage

首先要明确：dynamicpb创建的消息实例本质是*dynamicpb.Message类型，和protoc生成的静态消息类型（比如*mypackage.MyMessage）是完全不同的Go类型，所以类型断言必然失败。

要实现类型安全的判断，避免字符串匹配的误差，可以用以下两种方法：

方法一：通过protogen.Message指针直接对比 #

在protogen插件中，你可以直接获取目标消息类型的protogen.Message实例，然后和字段对应的消息描述符对比：

假设你已经通过plugin.Messages拿到了目标类型的protogen.Message实例（比如myMsg和myMsg2），可以这样判断：

// 获取字段对应的消息描述符
fieldMsgDesc := field.Desc.Message()

// 直接对比描述符指针
switch fieldMsgDesc {
case myMsg.Desc:
    // 字段是mypackage.MyMessage类型
case myMsg2.Desc:
    // 字段是mypackage.MyMessage2类型
}

这种方式基于指针对比，无字符串匹配误差，类型安全且高效。

方法二：预存目标类型的描述符 #

如果无法直接拿到protogen.Message实例，可以提前通过protoregistry.GlobalTypes获取目标类型的描述符，再进行对比：

import "google.golang.org/protobuf/reflect/protoregistry"

// 预获取目标类型的描述符
myMsgDesc, _ := protoregistry.GlobalTypes.FindMessageByName("mypackage.MyMessage")
myMsg2Desc, _ := protoregistry.GlobalTypes.FindMessageByName("mypackage.MyMessage2")

// 对比字段的消息描述符
switch field.Desc.Message() {
case myMsgDesc:
    // 处理逻辑
case myMsg2Desc:
    // 处理逻辑
}

这种方式只需初始化时获取一次描述符，后续对比同样是指针级别的，避免了字符串匹配的问题。

内容的提问来源于stack exchange，提问作者F21