如何将Golang Protobuf枚举从int32转换为字符串（无需修改Proto）

针对你提出的需求（不修改Protobuf定义，服务间保留原结构体通信，仅网关层将枚举值转为字符串返回前端），以下是两种可行的Golang实现方案：

方案一：定义DTO并自定义JSON序列化 #

这种方法通过在网关层定义与Protobuf结构体对应的DTO，重写JSON序列化逻辑，将枚举值映射为字符串：

1. 定义DTO结构体 #

// 与Protobuf的CacheDuration结构对应，仅Units字段改为string类型
type CacheDurationDTO struct {
    Value uint32 `json:"Value"`
    Units string `json:"Units"`
}

// 从Protobuf结构体转换为DTO的方法
func (dto *CacheDurationDTO) FromProto(pb *yourproto.CacheDuration) {
    dto.Value = pb.Value
    // 映射枚举值到字符串
    switch pb.Units {
    case yourproto.TimeUnit_seconds:
        dto.Units = "seconds"
    case yourproto.TimeUnit_minutes:
        dto.Units = "minutes"
    case yourproto.TimeUnit_hours:
        dto.Units = "hours"
    case yourproto.TimeUnit_days:
        dto.Units = "days"
    case yourproto.TimeUnit_months:
        dto.Units = "months"
    default:
        dto.Units = "unknown"
    }
}

2. 网关层转换逻辑 #

当网关接收到后端返回的Protobuf CacheDuration 结构体后，转换为DTO再序列化返回前端：

// 假设gatewayHandler是网关的请求处理函数
func gatewayHandler(w http.ResponseWriter, r *http.Request) {
    // 调用后端服务，获取Protobuf结构体
    pbResp, err := backendService.GetCacheDuration(r.Context())
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 转换为DTO
    var dto CacheDurationDTO
    dto.FromProto(pbResp)

    // 序列化为JSON返回前端
    w.Header().Set("Content-Type", "application/json")
    if err := json.NewEncoder(w).Encode(dto); err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
    }
}

方案二：利用Protobuf反射动态转换枚举 #

如果需要处理多个枚举类型，可通过Protobuf的反射包（google.golang.org/protobuf/reflect/protoreflect）实现通用转换，避免重复写映射逻辑：

1. 通用枚举转字符串函数 #

import (
    "google.golang.org/protobuf/reflect/protoreflect"
)

// 将Protobuf枚举值转为对应的字符串
func EnumValueToString(enum protoreflect.EnumValueDescriptor) string {
    return string(enum.Name())
}

// 转换CacheDuration的Units字段为字符串
func ConvertCacheDurationUnitsToString(pb *yourproto.CacheDuration) map[string]interface{} {
    // 将Protobuf结构体转为map，方便修改字段
    result := make(map[string]interface{})
    result["Value"] = pb.Value
    // 通过反射获取枚举的字符串名称
    enumDesc := pb.Units.Descriptor()
    enumValue := enumDesc.Values().ByNumber(protoreflect.EnumNumber(pb.Units))
    result["Units"] = EnumValueToString(enumValue)
    return result
}

2. 网关层使用示例 #

func gatewayHandler(w http.ResponseWriter, r *http.Request) {
    pbResp, err := backendService.GetCacheDuration(r.Context())
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 转换为带字符串枚举的map
    respMap := ConvertCacheDurationUnitsToString(pbResp)

    w.Header().Set("Content-Type", "application/json")
    if err := json.NewEncoder(w).Encode(respMap); err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
    }
}

方案说明 #

• 两种方案都无需修改原Protobuf定义，服务间通信仍使用生成的Protobuf结构体。
• 方案一适合单一类型的转换，逻辑简单直观，性能略优。
• 方案二更通用，适合需要处理多个枚举类型的场景，减少重复代码。

内容的提问来源于stack exchange，提问作者Dũng