mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-09 06:31:25 +02:00
6ccfdbaf1b
Fixes #4987 As machinery is supposed to be widely used project, and gRPC lacks proper support to override default codec easily, it might come into conflict with other projects. Instead, move codec to core talos, and register it explicitly in the server code (which covers machined, apid, trustd) and client code (talosctl). Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
// Package codec registers the gRPC for optimized marshaling.
|
|
//
|
|
// Package should be dummy imported to enable.
|
|
package codec
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"google.golang.org/grpc/encoding"
|
|
"google.golang.org/grpc/encoding/proto"
|
|
|
|
talosproto "github.com/talos-systems/talos/pkg/machinery/proto"
|
|
)
|
|
|
|
// gogoMessage is the interface for gogoproto additions.
|
|
//
|
|
// We use only a subset of that interface but include additional methods
|
|
// to prevent accidental successful type assertion for unrelated types.
|
|
type gogoMessage interface {
|
|
Marshal() ([]byte, error)
|
|
MarshalTo([]byte) (int, error)
|
|
MarshalToSizedBuffer([]byte) (int, error)
|
|
Unmarshal([]byte) error
|
|
}
|
|
|
|
// Codec provides protobuf encoding.Codec.
|
|
type Codec struct{}
|
|
|
|
// Marshal implements encoding.Codec.
|
|
func (Codec) Marshal(v interface{}) ([]byte, error) {
|
|
// some third-party types (like from etcd and containerd) implement gogoMessage
|
|
if gm, ok := v.(gogoMessage); ok {
|
|
return gm.Marshal()
|
|
}
|
|
|
|
// our types implement Message (with or without vtproto additions depending on build configuration)
|
|
if m, ok := v.(talosproto.Message); ok {
|
|
return talosproto.Marshal(m)
|
|
}
|
|
|
|
// no types implement protobuf API v1 only, so don't check for it
|
|
|
|
return nil, fmt.Errorf("failed to marshal %T", v)
|
|
}
|
|
|
|
// Unmarshal implements encoding.Codec.
|
|
func (Codec) Unmarshal(data []byte, v interface{}) error {
|
|
// some third-party types (like from etcd and containerd) implement gogoMessage
|
|
if gm, ok := v.(gogoMessage); ok {
|
|
return gm.Unmarshal(data)
|
|
}
|
|
|
|
// our types implement Message (with or without vtproto additions depending on build configuration)
|
|
if m, ok := v.(talosproto.Message); ok {
|
|
return talosproto.Unmarshal(data, m)
|
|
}
|
|
|
|
// no types implement protobuf API v1 only, so don't check for it
|
|
|
|
return fmt.Errorf("failed to unmarshal %T", v)
|
|
}
|
|
|
|
// Name implements encoding.Codec.
|
|
func (Codec) Name() string {
|
|
return proto.Name // overrides google.golang.org/grpc/encoding/proto codec
|
|
}
|
|
|
|
func init() {
|
|
encoding.RegisterCodec(Codec{})
|
|
}
|