Vendor executor module so that we can patch it

1 files changed, 55 insertions, 0 deletions

diff --git a/executor/z_executor_cmd.go b/executor/z_executor_cmd.go
new file mode 100644
index 0000000..5a5f13b
--- /dev/null
+++ b/executor/z_executor_cmd.go
@@ -0,0 +1,55 @@
+package executor
+
+import (
+	"encoding/json"
+	"os"
+
+	hclog "github.com/hashicorp/go-hclog"
+	plugin "github.com/hashicorp/go-plugin"
+
+	"github.com/hashicorp/nomad/plugins/base"
+)
+
+// Install a plugin cli handler to ease working with tests
+// and external plugins.
+// This init() must be initialized last in package required by the child plugin
+// process. It's recommended to avoid any other `init()` or inline any necessary calls
+// here. See eeaa95d commit message for more details.
+func init() {
+	if len(os.Args) > 1 && os.Args[1] == "executor" {
+		if len(os.Args) != 3 {
+			hclog.L().Error("json configuration not provided")
+			os.Exit(1)
+		}
+
+		config := os.Args[2]
+		var executorConfig ExecutorConfig
+		if err := json.Unmarshal([]byte(config), &executorConfig); err != nil {
+			os.Exit(1)
+		}
+
+		f, err := os.OpenFile(executorConfig.LogFile, os.O_CREATE|os.O_RDWR|os.O_APPEND, 0666)
+		if err != nil {
+			hclog.L().Error(err.Error())
+			os.Exit(1)
+		}
+
+		// Create the logger
+		logger := hclog.New(&hclog.LoggerOptions{
+			Level:      hclog.LevelFromString(executorConfig.LogLevel),
+			JSONFormat: true,
+			Output:     f,
+		})
+
+		plugin.Serve(&plugin.ServeConfig{
+			HandshakeConfig: base.Handshake,
+			Plugins: GetPluginMap(
+				logger,
+				executorConfig.FSIsolation,
+			),
+			GRPCServer: plugin.DefaultGRPCServer,
+			Logger:     logger,
+		})
+		os.Exit(0)
+	}
+}