[#98] rpc: Accept interface in place of ClientConn
All checks were successful
DCO action / DCO (pull_request) Successful in 1m35s
Tests and linters / Tests (1.19) (pull_request) Successful in 1m38s
Tests and linters / Tests (1.20) (pull_request) Successful in 1m48s
Tests and linters / Tests with -race (pull_request) Successful in 1m52s
Tests and linters / Lint (pull_request) Successful in 2m14s

gRPC client load-balancing API is ugly as f:
1. It is configured by pre-registering a balancer and the providing JSON
   configuration.
2. It doesn't allow different credentials for different endpoints
   (consider using "insecure" localhost and external endpoint).
3. To support frostfs usecase we also need to implement a resolver,
   which has its own difficulties.
4. https://github.com/grpc/grpc-go/issues/239#issuecomment-264548415

Using interface in place of grpc.ClientConn allows us to provide custom
implentation for it (load-balancing, circuit breaker etc.).

Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
Evgenii Stratonikov 2024-07-31 15:34:44 +03:00
parent c27b978770
commit e092ce4cf8
2 changed files with 10 additions and 2 deletions

View file

@ -2,8 +2,16 @@ package client
import ( import (
"io" "io"
"google.golang.org/grpc"
) )
// Conn is an interface for grpc client connection.
type Conn interface {
grpc.ClientConnInterface
io.Closer
}
// Conn returns underlying connection. // Conn returns underlying connection.
// //
// Returns non-nil result after the first Init() call // Returns non-nil result after the first Init() call

View file

@ -25,7 +25,7 @@ type cfg struct {
tlsCfg *tls.Config tlsCfg *tls.Config
grpcDialOpts []grpc.DialOption grpcDialOpts []grpc.DialOption
conn *grpc.ClientConn conn Conn
} }
const ( const (
@ -114,7 +114,7 @@ func WithTLSCfg(v *tls.Config) Option {
// WithGRPCConn returns option to specify // WithGRPCConn returns option to specify
// gRPC virtual connection. // gRPC virtual connection.
func WithGRPCConn(v *grpc.ClientConn) Option { func WithGRPCConn(v Conn) Option {
return func(c *cfg) { return func(c *cfg) {
if v != nil { if v != nil {
c.conn = v c.conn = v