написать пользовательский Marshall и Unmarshaller для типа прото буф в Голанге - PullRequest
Новая
       21

написать пользовательский Marshall и Unmarshaller для типа прото буф в Голанге

3 голосов
/ 02 июня 2019

У меня есть пользовательский тип, который я написал для своих собственных Marshall и Unmarshaller. И проблема в том, что я хочу сделать то же самое с помощью protobuf

Я просто хочу реализовать то же самое с помощью protobuf, чтобы я мог реализовать свой собственныйМаршалл и Унмаршаллер 

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
}
...