snapshot of https://github.com/centrifugal/centrifugo.git (master, 2026-09-16; blobs > 131072B omitted)
This commit is contained in:
65
internal/proxy/refresh_grpc.go
Normal file
65
internal/proxy/refresh_grpc.go
Normal file
@@ -0,0 +1,65 @@
|
||||
package proxy
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/centrifugal/centrifugo/v6/internal/proxyproto"
|
||||
|
||||
"google.golang.org/grpc"
|
||||
)
|
||||
|
||||
// GRPCRefreshProxy ...
|
||||
type GRPCRefreshProxy struct {
|
||||
config Config
|
||||
client proxyproto.CentrifugoProxyClient
|
||||
}
|
||||
|
||||
var _ RefreshProxy = (*GRPCRefreshProxy)(nil)
|
||||
|
||||
// NewGRPCRefreshProxy ...
|
||||
func NewGRPCRefreshProxy(name string, p Config) (*GRPCRefreshProxy, 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 &GRPCRefreshProxy{
|
||||
config: p,
|
||||
client: proxyproto.NewCentrifugoProxyClient(conn),
|
||||
}, nil
|
||||
}
|
||||
|
||||
// ProxyRefresh proxies refresh to application backend.
|
||||
func (p *GRPCRefreshProxy) ProxyRefresh(ctx context.Context, req *proxyproto.RefreshRequest) (*proxyproto.RefreshResponse, error) {
|
||||
ctx, cancel := context.WithTimeout(ctx, p.config.Timeout.ToDuration())
|
||||
defer cancel()
|
||||
return p.client.Refresh(grpcRequestContext(ctx, p.config), req)
|
||||
}
|
||||
|
||||
// Name ...
|
||||
func (p *GRPCRefreshProxy) Name() string {
|
||||
return "default"
|
||||
}
|
||||
|
||||
// Protocol ...
|
||||
func (p *GRPCRefreshProxy) Protocol() string {
|
||||
return "grpc"
|
||||
}
|
||||
|
||||
// UseBase64 ...
|
||||
func (p *GRPCRefreshProxy) UseBase64() bool {
|
||||
return p.config.BinaryEncoding
|
||||
}
|
||||
|
||||
// IncludeMeta ...
|
||||
func (p *GRPCRefreshProxy) IncludeMeta() bool {
|
||||
return p.config.IncludeConnectionMeta
|
||||
}
|
||||
Reference in New Issue
Block a user