Jeff Mitchell
7bfe35da0d
* Don't assume it's non-empty or that this is an issue * Ensure it's empty if the response is coming back through the multihop service so that it simply uses what is in the config file, at least for now
68 lines
2.4 KiB
Go
68 lines
2.4 KiB
Go
package worker
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
|
|
pbs "github.com/hashicorp/boundary/internal/gen/controller/servers/services"
|
|
)
|
|
|
|
type workerProxyServiceServer struct {
|
|
pbs.UnimplementedServerCoordinationServiceServer
|
|
pbs.UnimplementedSessionServiceServer
|
|
|
|
scsClient *atomic.Value
|
|
ssClient *atomic.Value
|
|
}
|
|
|
|
func NewWorkerProxyServiceServer(
|
|
scsClient *atomic.Value,
|
|
ssClient *atomic.Value,
|
|
) *workerProxyServiceServer {
|
|
return &workerProxyServiceServer{
|
|
scsClient: scsClient,
|
|
ssClient: ssClient,
|
|
}
|
|
}
|
|
|
|
var (
|
|
_ pbs.ServerCoordinationServiceServer = &workerProxyServiceServer{}
|
|
_ pbs.SessionServiceServer = &workerProxyServiceServer{}
|
|
)
|
|
|
|
func (ws *workerProxyServiceServer) Status(ctx context.Context, req *pbs.StatusRequest) (*pbs.StatusResponse, error) {
|
|
resp, err := ws.scsClient.Load().(pbs.ServerCoordinationServiceClient).Status(ctx, req)
|
|
|
|
if resp != nil {
|
|
// We don't currently support distributing new addreses to workers
|
|
// multiple hops away so ensure they're stripped out
|
|
resp.Controllers = nil
|
|
}
|
|
|
|
return resp, err
|
|
}
|
|
|
|
func (ws *workerProxyServiceServer) LookupSession(ctx context.Context, req *pbs.LookupSessionRequest) (*pbs.LookupSessionResponse, error) {
|
|
return ws.ssClient.Load().(pbs.SessionServiceClient).LookupSession(ctx, req)
|
|
}
|
|
|
|
func (ws *workerProxyServiceServer) CancelSession(ctx context.Context, req *pbs.CancelSessionRequest) (*pbs.CancelSessionResponse, error) {
|
|
return ws.ssClient.Load().(pbs.SessionServiceClient).CancelSession(ctx, req)
|
|
}
|
|
|
|
func (ws *workerProxyServiceServer) ActivateSession(ctx context.Context, req *pbs.ActivateSessionRequest) (*pbs.ActivateSessionResponse, error) {
|
|
return ws.ssClient.Load().(pbs.SessionServiceClient).ActivateSession(ctx, req)
|
|
}
|
|
|
|
func (ws *workerProxyServiceServer) AuthorizeConnection(ctx context.Context, req *pbs.AuthorizeConnectionRequest) (*pbs.AuthorizeConnectionResponse, error) {
|
|
return ws.ssClient.Load().(pbs.SessionServiceClient).AuthorizeConnection(ctx, req)
|
|
}
|
|
|
|
func (ws *workerProxyServiceServer) ConnectConnection(ctx context.Context, req *pbs.ConnectConnectionRequest) (*pbs.ConnectConnectionResponse, error) {
|
|
return ws.ssClient.Load().(pbs.SessionServiceClient).ConnectConnection(ctx, req)
|
|
}
|
|
|
|
func (ws *workerProxyServiceServer) CloseConnection(ctx context.Context, req *pbs.CloseConnectionRequest) (*pbs.CloseConnectionResponse, error) {
|
|
return ws.ssClient.Load().(pbs.SessionServiceClient).CloseConnection(ctx, req)
|
|
}
|