mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-30 12:34:22 +01:00
da4d89bd9a
Upgrade vendor ttrpc to fix the issue of fd leak. Fixes: #2000 0e0f228 Handle ok status 8c74fe8 Update to go 1.12x on travis 17f4d32 Client.Call(): do not return error if no Status is set(gRPC v1.23 and up) 271238a Fix method full name generation 694de9d metadata as KeyValue type 3afb82b Fix error handling with server shutdown f3eb35b Refactor close handling for ttrpc clients de8faac Add godocs for interceptors e409d7d Add example binary for testing the example service 819653f Add client and server unary interceptors 04523b9 Rename headers to metadata 5926a92 Support headers 911c9cd Improve connection error handling 96dcf73 Handle EOF to prevent file descriptor leak ba15956 Make onclose an option. Signed-off-by: lifupan <lifupan@gmail.com>
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
/*
|
|
Copyright The containerd 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 ttrpc
|
|
|
|
import "github.com/pkg/errors"
|
|
|
|
type serverConfig struct {
|
|
handshaker Handshaker
|
|
interceptor UnaryServerInterceptor
|
|
}
|
|
|
|
// ServerOpt for configuring a ttrpc server
|
|
type ServerOpt func(*serverConfig) error
|
|
|
|
// WithServerHandshaker can be passed to NewServer to ensure that the
|
|
// handshaker is called before every connection attempt.
|
|
//
|
|
// Only one handshaker is allowed per server.
|
|
func WithServerHandshaker(handshaker Handshaker) ServerOpt {
|
|
return func(c *serverConfig) error {
|
|
if c.handshaker != nil {
|
|
return errors.New("only one handshaker allowed per server")
|
|
}
|
|
c.handshaker = handshaker
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// WithUnaryServerInterceptor sets the provided interceptor on the server
|
|
func WithUnaryServerInterceptor(i UnaryServerInterceptor) ServerOpt {
|
|
return func(c *serverConfig) error {
|
|
if c.interceptor != nil {
|
|
return errors.New("only one interceptor allowed per server")
|
|
}
|
|
c.interceptor = i
|
|
return nil
|
|
}
|
|
}
|