forked from toolshed/abra
.gitea
cli
cmd
pkg
scripts
tests
vendor
coopcloud.tech
dario.cat
git.coopcloud.tech
github.com
go.opentelemetry.io
golang.org
google.golang.org
genproto
grpc
attributes
backoff
balancer
binarylog
channelz
codes
connectivity
credentials
encoding
gzip
proto
encoding.go
encoding_v2.go
experimental
grpclog
health
internal
keepalive
mem
metadata
peer
resolver
serviceconfig
stats
status
tap
AUTHORS
CODE-OF-CONDUCT.md
CONTRIBUTING.md
GOVERNANCE.md
LICENSE
MAINTAINERS.md
Makefile
NOTICE.txt
README.md
SECURITY.md
backoff.go
balancer_wrapper.go
call.go
clientconn.go
codec.go
dialoptions.go
doc.go
interceptor.go
picker_wrapper.go
preloader.go
resolver_wrapper.go
rpc_util.go
server.go
service_config.go
stream.go
stream_interfaces.go
trace.go
trace_notrace.go
trace_withtrace.go
version.go
protobuf
gopkg.in
gotest.tools
modules.txt
.dockerignore
.drone.yml
.envrc.sample
.gitignore
.goreleaser.yml
AUTHORS.md
Dockerfile
LICENSE
Makefile
README.md
go.mod
go.sum
renovate.json
132 lines
4.9 KiB
Go
132 lines
4.9 KiB
Go
/*
|
|
*
|
|
* Copyright 2017 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 encoding defines the interface for the compressor and codec, and
|
|
// functions to register and retrieve compressors and codecs.
|
|
//
|
|
// # Experimental
|
|
//
|
|
// Notice: This package is EXPERIMENTAL and may be changed or removed in a
|
|
// later release.
|
|
package encoding
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
|
|
"google.golang.org/grpc/internal/grpcutil"
|
|
)
|
|
|
|
// Identity specifies the optional encoding for uncompressed streams.
|
|
// It is intended for grpc internal use only.
|
|
const Identity = "identity"
|
|
|
|
// Compressor is used for compressing and decompressing when sending or
|
|
// receiving messages.
|
|
//
|
|
// If a Compressor implements `DecompressedSize(compressedBytes []byte) int`,
|
|
// gRPC will invoke it to determine the size of the buffer allocated for the
|
|
// result of decompression. A return value of -1 indicates unknown size.
|
|
type Compressor interface {
|
|
// Compress writes the data written to wc to w after compressing it. If an
|
|
// error occurs while initializing the compressor, that error is returned
|
|
// instead.
|
|
Compress(w io.Writer) (io.WriteCloser, error)
|
|
// Decompress reads data from r, decompresses it, and provides the
|
|
// uncompressed data via the returned io.Reader. If an error occurs while
|
|
// initializing the decompressor, that error is returned instead.
|
|
Decompress(r io.Reader) (io.Reader, error)
|
|
// Name is the name of the compression codec and is used to set the content
|
|
// coding header. The result must be static; the result cannot change
|
|
// between calls.
|
|
Name() string
|
|
}
|
|
|
|
var registeredCompressor = make(map[string]Compressor)
|
|
|
|
// RegisterCompressor registers the compressor with gRPC by its name. It can
|
|
// be activated when sending an RPC via grpc.UseCompressor(). It will be
|
|
// automatically accessed when receiving a message based on the content coding
|
|
// header. Servers also use it to send a response with the same encoding as
|
|
// the request.
|
|
//
|
|
// NOTE: this function must only be called during initialization time (i.e. in
|
|
// an init() function), and is not thread-safe. If multiple Compressors are
|
|
// registered with the same name, the one registered last will take effect.
|
|
func RegisterCompressor(c Compressor) {
|
|
registeredCompressor[c.Name()] = c
|
|
if !grpcutil.IsCompressorNameRegistered(c.Name()) {
|
|
grpcutil.RegisteredCompressorNames = append(grpcutil.RegisteredCompressorNames, c.Name())
|
|
}
|
|
}
|
|
|
|
// GetCompressor returns Compressor for the given compressor name.
|
|
func GetCompressor(name string) Compressor {
|
|
return registeredCompressor[name]
|
|
}
|
|
|
|
// Codec defines the interface gRPC uses to encode and decode messages. Note
|
|
// that implementations of this interface must be thread safe; a Codec's
|
|
// methods can be called from concurrent goroutines.
|
|
type Codec interface {
|
|
// Marshal returns the wire format of v.
|
|
Marshal(v any) ([]byte, error)
|
|
// Unmarshal parses the wire format into v.
|
|
Unmarshal(data []byte, v any) error
|
|
// Name returns the name of the Codec implementation. The returned string
|
|
// will be used as part of content type in transmission. The result must be
|
|
// static; the result cannot change between calls.
|
|
Name() string
|
|
}
|
|
|
|
var registeredCodecs = make(map[string]any)
|
|
|
|
// RegisterCodec registers the provided Codec for use with all gRPC clients and
|
|
// servers.
|
|
//
|
|
// The Codec will be stored and looked up by result of its Name() method, which
|
|
// should match the content-subtype of the encoding handled by the Codec. This
|
|
// is case-insensitive, and is stored and looked up as lowercase. If the
|
|
// result of calling Name() is an empty string, RegisterCodec will panic. See
|
|
// Content-Type on
|
|
// https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md#requests for
|
|
// more details.
|
|
//
|
|
// NOTE: this function must only be called during initialization time (i.e. in
|
|
// an init() function), and is not thread-safe. If multiple Codecs are
|
|
// registered with the same name, the one registered last will take effect.
|
|
func RegisterCodec(codec Codec) {
|
|
if codec == nil {
|
|
panic("cannot register a nil Codec")
|
|
}
|
|
if codec.Name() == "" {
|
|
panic("cannot register Codec with empty string result for Name()")
|
|
}
|
|
contentSubtype := strings.ToLower(codec.Name())
|
|
registeredCodecs[contentSubtype] = codec
|
|
}
|
|
|
|
// GetCodec gets a registered Codec by content-subtype, or nil if no Codec is
|
|
// registered for the content-subtype.
|
|
//
|
|
// The content-subtype is expected to be lowercase.
|
|
func GetCodec(contentSubtype string) Codec {
|
|
c, _ := registeredCodecs[contentSubtype].(Codec)
|
|
return c
|
|
}
|