61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package proxy
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/centrifugal/centrifugo/v6/internal/proxyproto"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// GRPCRPCProxy ...
|
|
type GRPCRPCProxy struct {
|
|
config Config
|
|
client proxyproto.CentrifugoProxyClient
|
|
}
|
|
|
|
var _ RPCProxy = (*GRPCRPCProxy)(nil)
|
|
|
|
// NewGRPCRPCProxy ...
|
|
func NewGRPCRPCProxy(name string, p Config) (*GRPCRPCProxy, error) {
|
|
host, err := getGrpcHost(p.Endpoint)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting grpc host: %v", err)
|
|
}
|
|
dialOpts, err := getDialOpts(name, p)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating GRPC dial options: %v", err)
|
|
}
|
|
conn, err := grpc.NewClient(host, dialOpts...)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error connecting to GRPC proxy server: %v", err)
|
|
}
|
|
return &GRPCRPCProxy{
|
|
config: p,
|
|
client: proxyproto.NewCentrifugoProxyClient(conn),
|
|
}, nil
|
|
}
|
|
|
|
// ProxyRPC ...
|
|
func (p *GRPCRPCProxy) ProxyRPC(ctx context.Context, req *proxyproto.RPCRequest) (*proxyproto.RPCResponse, error) {
|
|
ctx, cancel := context.WithTimeout(ctx, p.config.Timeout.ToDuration())
|
|
defer cancel()
|
|
return p.client.RPC(grpcRequestContext(ctx, p.config), req)
|
|
}
|
|
|
|
// Protocol ...
|
|
func (p *GRPCRPCProxy) Protocol() string {
|
|
return "grpc"
|
|
}
|
|
|
|
// UseBase64 ...
|
|
func (p *GRPCRPCProxy) UseBase64() bool {
|
|
return p.config.BinaryEncoding
|
|
}
|
|
|
|
// IncludeMeta ...
|
|
func (p *GRPCRPCProxy) IncludeMeta() bool {
|
|
return p.config.IncludeConnectionMeta
|
|
}
|