| /* |
| * |
| * Copyright 2018 gRPC authors. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| // Package proto defines the protobuf codec. Importing this package will |
| // register the codec. |
| package proto |
| |
| import ( |
| "fmt" |
| |
| "google.golang.org/grpc/encoding" |
| "google.golang.org/protobuf/proto" |
| "google.golang.org/protobuf/protoadapt" |
| ) |
| |
| // Name is the name registered for the proto compressor. |
| const Name = "proto" |
| |
| func init() { |
| encoding.RegisterCodec(codec{}) |
| } |
| |
| // codec is a Codec implementation with protobuf. It is the default codec for gRPC. |
| type codec struct{} |
| |
| func (codec) Marshal(v any) ([]byte, error) { |
| vv := messageV2Of(v) |
| if vv == nil { |
| return nil, fmt.Errorf("failed to marshal, message is %T, want proto.Message", v) |
| } |
| |
| return proto.Marshal(vv) |
| } |
| |
| func (codec) Unmarshal(data []byte, v any) error { |
| vv := messageV2Of(v) |
| if vv == nil { |
| return fmt.Errorf("failed to unmarshal, message is %T, want proto.Message", v) |
| } |
| |
| return proto.Unmarshal(data, vv) |
| } |
| |
| func messageV2Of(v any) proto.Message { |
| switch v := v.(type) { |
| case protoadapt.MessageV1: |
| return protoadapt.MessageV2Of(v) |
| case protoadapt.MessageV2: |
| return v |
| } |
| |
| return nil |
| } |
| |
| func (codec) Name() string { |
| return Name |
| } |