2020-04-21 09:16:41 -05:00
|
|
|
package pluginextensionv2
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
|
|
|
|
"github.com/hashicorp/go-plugin"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
)
|
|
|
|
|
|
|
|
type RendererPlugin interface {
|
|
|
|
RendererClient
|
2022-07-07 06:32:18 -05:00
|
|
|
SanitizerClient
|
2020-04-21 09:16:41 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
type RendererGRPCPlugin struct {
|
|
|
|
plugin.NetRPCUnsupportedPlugin
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *RendererGRPCPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *RendererGRPCPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
|
2022-07-07 06:32:18 -05:00
|
|
|
return &RendererGRPCClient{NewRendererClient(c), NewSanitizerClient(c)}, nil
|
2020-04-21 09:16:41 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
type RendererGRPCClient struct {
|
|
|
|
RendererClient
|
2022-07-07 06:32:18 -05:00
|
|
|
SanitizerClient
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m *RendererGRPCClient) Sanitize(ctx context.Context, req *SanitizeRequest, opts ...grpc.CallOption) (*SanitizeResponse, error) {
|
|
|
|
return m.SanitizerClient.Sanitize(ctx, req, opts...)
|
2020-04-21 09:16:41 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
func (m *RendererGRPCClient) Render(ctx context.Context, req *RenderRequest, opts ...grpc.CallOption) (*RenderResponse, error) {
|
|
|
|
return m.RendererClient.Render(ctx, req)
|
|
|
|
}
|
|
|
|
|
|
|
|
var _ RendererClient = &RendererGRPCClient{}
|
2022-07-07 06:32:18 -05:00
|
|
|
var _ SanitizerClient = &RendererGRPCClient{}
|
2020-04-21 09:16:41 -05:00
|
|
|
var _ plugin.GRPCPlugin = &RendererGRPCPlugin{}
|