1 files changed, 33 insertions, 0 deletions

diff --git a/vendor/github.com/hashicorp/go-plugin/plugin.go b/vendor/github.com/hashicorp/go-plugin/plugin.go
index 37c8fd6..79d9674 100644
--- a/vendor/github.com/hashicorp/go-plugin/plugin.go
+++ b/vendor/github.com/hashicorp/go-plugin/plugin.go
@@ -9,7 +9,11 @@
 package plugin
 
 import (
+        "context"
+        "errors"
         "net/rpc"
+
+        "google.golang.org/grpc"
 )
 
 // Plugin is the interface that is implemented to serve/connect to an
@@ -23,3 +27,32 @@ type Plugin interface {
         // serving that communicates to the server end of the plugin.
         Client(*MuxBroker, *rpc.Client) (interface{}, error)
 }
+
+// GRPCPlugin is the interface that is implemented to serve/connect to
+// a plugin over gRPC.
+type GRPCPlugin interface {
+        // GRPCServer should register this plugin for serving with the
+        // given GRPCServer. Unlike Plugin.Server, this is only called once
+        // since gRPC plugins serve singletons.
+        GRPCServer(*GRPCBroker, *grpc.Server) error
+
+        // GRPCClient should return the interface implementation for the plugin
+        // you're serving via gRPC. The provided context will be canceled by
+        // go-plugin in the event of the plugin process exiting.
+        GRPCClient(context.Context, *GRPCBroker, *grpc.ClientConn) (interface{}, error)
+}
+
+// NetRPCUnsupportedPlugin implements Plugin but returns errors for the
+// Server and Client functions. This will effectively disable support for
+// net/rpc based plugins.
+//
+// This struct can be embedded in your struct.
+type NetRPCUnsupportedPlugin struct{}
+
+func (p NetRPCUnsupportedPlugin) Server(*MuxBroker) (interface{}, error) {
+        return nil, errors.New("net/rpc plugin protocol not supported")
+}
+
+func (p NetRPCUnsupportedPlugin) Client(*MuxBroker, *rpc.Client) (interface{}, error) {
+        return nil, errors.New("net/rpc plugin protocol not supported")
+}