48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package pluginutil
|
|
|
|
import (
|
|
context "context"
|
|
"fmt"
|
|
|
|
grpc "google.golang.org/grpc"
|
|
codes "google.golang.org/grpc/codes"
|
|
status "google.golang.org/grpc/status"
|
|
)
|
|
|
|
type PluginMultiplexingServerImpl struct {
|
|
UnimplementedPluginMultiplexingServer
|
|
|
|
Supported bool
|
|
}
|
|
|
|
func (pm PluginMultiplexingServerImpl) MultiplexingSupport(ctx context.Context, req *MultiplexingSupportRequest) (*MultiplexingSupportResponse, error) {
|
|
return &MultiplexingSupportResponse{
|
|
Supported: pm.Supported,
|
|
}, nil
|
|
}
|
|
|
|
func MultiplexingSupported(ctx context.Context, cc grpc.ClientConnInterface) (bool, error) {
|
|
if cc == nil {
|
|
return false, fmt.Errorf("client connection is nil")
|
|
}
|
|
|
|
req := new(MultiplexingSupportRequest)
|
|
resp, err := NewPluginMultiplexingClient(cc).MultiplexingSupport(ctx, req)
|
|
if err != nil {
|
|
|
|
// If the server does not implement the multiplexing server then we can
|
|
// assume it is not multiplexed
|
|
if status.Code(err) == codes.Unimplemented {
|
|
return false, nil
|
|
}
|
|
|
|
return false, err
|
|
}
|
|
if resp == nil {
|
|
// Somehow got a nil response, assume not multiplexed
|
|
return false, nil
|
|
}
|
|
|
|
return resp.Supported, nil
|
|
}
|