go - 为 golang 中的 proto buf 类型编写自定义 Marshall 和 Unmarshaller

Question

我有自定义类型，我编写了自己的 Marshall 和 Unmarshaller
问题是我想使用 protobuf 做同样的事情

我只想使用 protobuf 实现相同的功能，以便我可以实现自己的 Marshall 和 Unmarshaller

syntax="proto3";

package main;

message NullInt64{
    bool Valid = 1;
    int64 Int64 = 2;
}

如果 Valid 值为 false 则返回 null字符串

type NullInt64 struct {
    Int64 int64
    Valid bool
}

// MarshalJSON try to marshaling to json
func (nt NullInt64) MarshalJSON() ([]byte, error) {
    if nt.Valid {
        return []byte(fmt.Sprintf(`%d`, nt.Int64)), nil
    }

    return []byte("null"), nil
}

// UnmarshalJSON try to unmarshal dae from input
func (nt *NullInt64) UnmarshalJSON(b []byte) error {
    text := strings.ToLower(string(b))
    if text == "null" {
        nt.Valid = false

        return nil
    }

    err := json.Unmarshal(b, &nt.Int64)
    if err != nil {
        return err
    }

    nt.Valid = true
    return nil
}

Answer 1

Protoc 不会生成 MarshalJSON和 UnmarshalJSON职能。
你可以:

使用不同的 protobuf 生成器(参见 gogo/protobuf 和很多 extensions 或 fork golang/protobuf 来更改它们的 generator )

将您自己的函数插入 proto通过将文件添加到该文件夹​​来打包。您可以手写或代码生成这些函数。