2015-09-01 21:57:21 +00:00
|
|
|
package driver
|
|
|
|
|
|
|
|
import (
|
2015-09-02 16:41:25 +00:00
|
|
|
"bytes"
|
2017-02-01 00:43:57 +00:00
|
|
|
"context"
|
2015-12-23 00:10:30 +00:00
|
|
|
"encoding/json"
|
2015-09-01 21:57:21 +00:00
|
|
|
"fmt"
|
2015-12-23 00:10:30 +00:00
|
|
|
"log"
|
2016-10-07 19:37:52 +00:00
|
|
|
"os"
|
2015-09-01 21:57:21 +00:00
|
|
|
"os/exec"
|
2015-09-03 15:25:09 +00:00
|
|
|
"path/filepath"
|
2015-09-22 23:32:05 +00:00
|
|
|
"runtime"
|
2015-09-01 21:57:21 +00:00
|
|
|
"strings"
|
2015-09-22 23:32:05 +00:00
|
|
|
"syscall"
|
2015-09-01 21:57:21 +00:00
|
|
|
"time"
|
|
|
|
|
2016-02-09 02:51:11 +00:00
|
|
|
"github.com/hashicorp/go-multierror"
|
2016-02-04 01:46:45 +00:00
|
|
|
"github.com/hashicorp/go-plugin"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
|
|
|
2017-01-14 00:03:11 +00:00
|
|
|
"github.com/hashicorp/nomad/client/driver/env"
|
2016-02-05 00:03:17 +00:00
|
|
|
"github.com/hashicorp/nomad/client/driver/executor"
|
2016-06-12 03:15:50 +00:00
|
|
|
dstructs "github.com/hashicorp/nomad/client/driver/structs"
|
2015-11-05 21:46:02 +00:00
|
|
|
"github.com/hashicorp/nomad/client/fingerprint"
|
2016-06-12 03:15:50 +00:00
|
|
|
cstructs "github.com/hashicorp/nomad/client/structs"
|
2017-02-21 03:35:51 +00:00
|
|
|
"github.com/hashicorp/nomad/helper"
|
2016-04-09 22:38:42 +00:00
|
|
|
"github.com/hashicorp/nomad/helper/fields"
|
2015-09-01 21:57:21 +00:00
|
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
|
|
)
|
|
|
|
|
2016-04-01 01:11:27 +00:00
|
|
|
const (
|
|
|
|
// The key populated in Node Attributes to indicate presence of the Java
|
|
|
|
// driver
|
|
|
|
javaDriverAttr = "driver.java"
|
|
|
|
)
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
// JavaDriver is a simple driver to execute applications packaged in Jars.
|
|
|
|
// It literally just fork/execs tasks with the java command.
|
|
|
|
type JavaDriver struct {
|
2015-09-10 01:06:23 +00:00
|
|
|
DriverContext
|
2015-11-05 21:46:02 +00:00
|
|
|
fingerprint.StaticFingerprinter
|
2017-02-21 03:35:51 +00:00
|
|
|
|
|
|
|
// A tri-state boolean to know if the fingerprinting has happened and
|
|
|
|
// whether it has been successful
|
|
|
|
fingerprintSuccess *bool
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2015-11-14 04:22:49 +00:00
|
|
|
type JavaDriverConfig struct {
|
2017-01-14 00:03:11 +00:00
|
|
|
Class string `mapstructure:"class"`
|
|
|
|
ClassPath string `mapstructure:"class_path"`
|
|
|
|
JarPath string `mapstructure:"jar_path"`
|
|
|
|
JvmOpts []string `mapstructure:"jvm_options"`
|
|
|
|
Args []string `mapstructure:"args"`
|
2015-11-14 02:09:42 +00:00
|
|
|
}
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
// javaHandle is returned from Start/Open as a handle to the PID
|
|
|
|
type javaHandle struct {
|
2016-02-08 21:48:26 +00:00
|
|
|
pluginClient *plugin.Client
|
|
|
|
userPid int
|
|
|
|
executor executor.Executor
|
2016-06-12 03:15:50 +00:00
|
|
|
isolationConfig *dstructs.IsolationConfig
|
2017-02-01 00:43:57 +00:00
|
|
|
taskDir string
|
2016-02-04 01:46:45 +00:00
|
|
|
|
2016-03-03 17:21:21 +00:00
|
|
|
killTimeout time.Duration
|
|
|
|
maxKillTimeout time.Duration
|
|
|
|
version string
|
|
|
|
logger *log.Logger
|
2016-06-12 03:15:50 +00:00
|
|
|
waitCh chan *dstructs.WaitResult
|
2016-03-03 17:21:21 +00:00
|
|
|
doneCh chan struct{}
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// NewJavaDriver is used to create a new exec driver
|
2015-09-10 01:06:23 +00:00
|
|
|
func NewJavaDriver(ctx *DriverContext) Driver {
|
2015-11-05 21:46:02 +00:00
|
|
|
return &JavaDriver{DriverContext: *ctx}
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2016-04-09 22:38:42 +00:00
|
|
|
// Validate is used to validate the driver configuration
|
2016-04-08 20:19:43 +00:00
|
|
|
func (d *JavaDriver) Validate(config map[string]interface{}) error {
|
2016-04-09 22:38:42 +00:00
|
|
|
fd := &fields.FieldData{
|
|
|
|
Raw: config,
|
|
|
|
Schema: map[string]*fields.FieldSchema{
|
2017-09-26 22:26:33 +00:00
|
|
|
"class": {
|
2017-01-14 00:03:11 +00:00
|
|
|
Type: fields.TypeString,
|
|
|
|
},
|
2017-09-26 22:26:33 +00:00
|
|
|
"class_path": {
|
2017-01-14 00:03:11 +00:00
|
|
|
Type: fields.TypeString,
|
|
|
|
},
|
2017-09-26 22:26:33 +00:00
|
|
|
"jar_path": {
|
2017-01-14 00:03:11 +00:00
|
|
|
Type: fields.TypeString,
|
2016-04-09 22:38:42 +00:00
|
|
|
},
|
2017-09-26 22:26:33 +00:00
|
|
|
"jvm_options": {
|
2016-04-09 22:38:42 +00:00
|
|
|
Type: fields.TypeArray,
|
|
|
|
},
|
2017-09-26 22:26:33 +00:00
|
|
|
"args": {
|
2016-04-09 22:38:42 +00:00
|
|
|
Type: fields.TypeArray,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := fd.Validate(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2016-04-08 20:19:43 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2016-10-19 22:06:23 +00:00
|
|
|
func (d *JavaDriver) Abilities() DriverAbilities {
|
|
|
|
return DriverAbilities{
|
|
|
|
SendSignals: true,
|
2017-04-13 16:52:16 +00:00
|
|
|
Exec: true,
|
2016-10-19 22:06:23 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
func (d *JavaDriver) Fingerprint(req *cstructs.FingerprintRequest, resp *cstructs.FingerprintResponse) error {
|
2016-02-18 05:20:22 +00:00
|
|
|
// Only enable if we are root and cgroups are mounted when running on linux systems.
|
2018-01-24 14:09:53 +00:00
|
|
|
if runtime.GOOS == "linux" && (syscall.Geteuid() != 0 || !cgroupsMounted(req.Node)) {
|
2017-02-21 03:35:51 +00:00
|
|
|
if d.fingerprintSuccess == nil || *d.fingerprintSuccess {
|
2018-01-12 14:44:53 +00:00
|
|
|
d.logger.Printf("[DEBUG] driver.java: root privileges and mounted cgroups required on linux, disabling")
|
2016-04-01 09:22:17 +00:00
|
|
|
}
|
2017-02-21 03:35:51 +00:00
|
|
|
d.fingerprintSuccess = helper.BoolToPtr(false)
|
2018-01-26 19:31:37 +00:00
|
|
|
resp.RemoveAttribute(javaDriverAttr)
|
2018-01-24 14:09:53 +00:00
|
|
|
return nil
|
2015-09-22 23:32:05 +00:00
|
|
|
}
|
|
|
|
|
2015-09-02 16:41:25 +00:00
|
|
|
// Find java version
|
|
|
|
var out bytes.Buffer
|
|
|
|
var erOut bytes.Buffer
|
|
|
|
cmd := exec.Command("java", "-version")
|
|
|
|
cmd.Stdout = &out
|
|
|
|
cmd.Stderr = &erOut
|
|
|
|
err := cmd.Run()
|
|
|
|
if err != nil {
|
|
|
|
// assume Java wasn't found
|
2017-02-21 03:35:51 +00:00
|
|
|
d.fingerprintSuccess = helper.BoolToPtr(false)
|
2018-01-26 19:31:37 +00:00
|
|
|
resp.RemoveAttribute(javaDriverAttr)
|
2018-01-24 14:09:53 +00:00
|
|
|
return nil
|
2015-09-02 16:41:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// 'java -version' returns output on Stderr typically.
|
|
|
|
// Check stdout, but it's probably empty
|
|
|
|
var infoString string
|
|
|
|
if out.String() != "" {
|
|
|
|
infoString = out.String()
|
|
|
|
}
|
|
|
|
|
|
|
|
if erOut.String() != "" {
|
|
|
|
infoString = erOut.String()
|
|
|
|
}
|
|
|
|
|
|
|
|
if infoString == "" {
|
2017-02-21 03:35:51 +00:00
|
|
|
if d.fingerprintSuccess == nil || *d.fingerprintSuccess {
|
2016-04-01 09:22:17 +00:00
|
|
|
d.logger.Println("[WARN] driver.java: error parsing Java version information, aborting")
|
|
|
|
}
|
2017-02-21 03:35:51 +00:00
|
|
|
d.fingerprintSuccess = helper.BoolToPtr(false)
|
2018-01-26 19:31:37 +00:00
|
|
|
resp.RemoveAttribute(javaDriverAttr)
|
2018-01-24 14:09:53 +00:00
|
|
|
return nil
|
2015-09-02 16:41:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Assume 'java -version' returns 3 lines:
|
|
|
|
// java version "1.6.0_36"
|
|
|
|
// OpenJDK Runtime Environment (IcedTea6 1.13.8) (6b36-1.13.8-0ubuntu1~12.04)
|
|
|
|
// OpenJDK 64-Bit Server VM (build 23.25-b01, mixed mode)
|
|
|
|
// Each line is terminated by \n
|
|
|
|
info := strings.Split(infoString, "\n")
|
|
|
|
versionString := info[0]
|
|
|
|
versionString = strings.TrimPrefix(versionString, "java version ")
|
|
|
|
versionString = strings.Trim(versionString, "\"")
|
2018-01-26 16:21:07 +00:00
|
|
|
resp.AddAttribute(javaDriverAttr, "1")
|
|
|
|
resp.AddAttribute("driver.java.version", versionString)
|
|
|
|
resp.AddAttribute("driver.java.runtime", info[1])
|
|
|
|
resp.AddAttribute("driver.java.vm", info[2])
|
2017-02-21 03:35:51 +00:00
|
|
|
d.fingerprintSuccess = helper.BoolToPtr(true)
|
2018-01-30 17:57:37 +00:00
|
|
|
resp.Applicable = true
|
2015-09-02 16:41:25 +00:00
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
return nil
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2017-05-19 18:08:49 +00:00
|
|
|
func (d *JavaDriver) Prestart(*ExecContext, *structs.Task) (*PrestartResponse, error) {
|
2017-01-10 21:24:45 +00:00
|
|
|
return nil, nil
|
2016-11-30 00:39:36 +00:00
|
|
|
}
|
|
|
|
|
2017-05-17 00:33:50 +00:00
|
|
|
func NewJavaDriverConfig(task *structs.Task, env *env.TaskEnv) (*JavaDriverConfig, error) {
|
2015-11-14 04:22:49 +00:00
|
|
|
var driverConfig JavaDriverConfig
|
2015-11-14 02:09:42 +00:00
|
|
|
if err := mapstructure.WeakDecode(task.Config, &driverConfig); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2016-03-23 18:45:03 +00:00
|
|
|
|
2017-01-14 00:03:11 +00:00
|
|
|
// Interpolate everything
|
|
|
|
driverConfig.Class = env.ReplaceEnv(driverConfig.Class)
|
|
|
|
driverConfig.ClassPath = env.ReplaceEnv(driverConfig.ClassPath)
|
|
|
|
driverConfig.JarPath = env.ReplaceEnv(driverConfig.JarPath)
|
|
|
|
driverConfig.JvmOpts = env.ParseAndReplace(driverConfig.JvmOpts)
|
|
|
|
driverConfig.Args = env.ParseAndReplace(driverConfig.Args)
|
|
|
|
|
|
|
|
// Validate
|
|
|
|
jarSpecified := driverConfig.JarPath != ""
|
|
|
|
classSpecified := driverConfig.Class != ""
|
|
|
|
if !jarSpecified && !classSpecified {
|
|
|
|
return nil, fmt.Errorf("jar_path or class must be specified")
|
|
|
|
}
|
|
|
|
|
|
|
|
return &driverConfig, nil
|
|
|
|
}
|
|
|
|
|
2017-06-09 17:29:41 +00:00
|
|
|
func (d *JavaDriver) Start(ctx *ExecContext, task *structs.Task) (*StartResponse, error) {
|
2017-05-23 00:46:40 +00:00
|
|
|
driverConfig, err := NewJavaDriverConfig(task, ctx.TaskEnv)
|
2017-01-14 00:03:11 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2015-10-16 19:33:29 +00:00
|
|
|
args := []string{}
|
2017-01-14 00:03:11 +00:00
|
|
|
|
2015-10-16 18:32:37 +00:00
|
|
|
// Look for jvm options
|
2015-11-18 23:16:42 +00:00
|
|
|
if len(driverConfig.JvmOpts) != 0 {
|
2015-11-17 03:29:06 +00:00
|
|
|
d.logger.Printf("[DEBUG] driver.java: found JVM options: %s", driverConfig.JvmOpts)
|
2015-11-18 23:16:42 +00:00
|
|
|
args = append(args, driverConfig.JvmOpts...)
|
2015-10-16 18:32:37 +00:00
|
|
|
}
|
|
|
|
|
2017-01-14 00:03:11 +00:00
|
|
|
// Add the classpath
|
|
|
|
if driverConfig.ClassPath != "" {
|
|
|
|
args = append(args, "-cp", driverConfig.ClassPath)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Add the jar
|
|
|
|
if driverConfig.JarPath != "" {
|
|
|
|
args = append(args, "-jar", driverConfig.JarPath)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Add the class
|
|
|
|
if driverConfig.Class != "" {
|
|
|
|
args = append(args, driverConfig.Class)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Add any args
|
2015-11-18 23:16:42 +00:00
|
|
|
if len(driverConfig.Args) != 0 {
|
|
|
|
args = append(args, driverConfig.Args...)
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2017-01-06 00:02:20 +00:00
|
|
|
pluginLogFile := filepath.Join(ctx.TaskDir.Dir, "executor.out")
|
2017-01-12 19:50:49 +00:00
|
|
|
executorConfig := &dstructs.ExecutorConfig{
|
|
|
|
LogFile: pluginLogFile,
|
|
|
|
LogLevel: d.config.LogLevel,
|
2015-09-25 23:49:14 +00:00
|
|
|
}
|
|
|
|
|
2017-01-12 19:50:49 +00:00
|
|
|
execIntf, pluginClient, err := createExecutor(d.config.LogOutput, d.config, executorConfig)
|
2016-02-04 01:46:45 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2016-10-12 18:35:29 +00:00
|
|
|
|
|
|
|
// Set the context
|
2016-02-05 00:03:17 +00:00
|
|
|
executorCtx := &executor.ExecutorContext{
|
2017-05-23 00:46:40 +00:00
|
|
|
TaskEnv: ctx.TaskEnv,
|
2016-12-03 01:04:07 +00:00
|
|
|
Driver: "java",
|
|
|
|
Task: task,
|
|
|
|
TaskDir: ctx.TaskDir.Dir,
|
|
|
|
LogDir: ctx.TaskDir.LogDir,
|
2016-02-04 01:46:45 +00:00
|
|
|
}
|
2016-10-12 18:35:29 +00:00
|
|
|
if err := execIntf.SetContext(executorCtx); err != nil {
|
|
|
|
pluginClient.Kill()
|
|
|
|
return nil, fmt.Errorf("failed to set executor context: %v", err)
|
|
|
|
}
|
2016-02-18 05:20:22 +00:00
|
|
|
|
2016-03-16 02:22:40 +00:00
|
|
|
absPath, err := GetAbsolutePath("java")
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2017-12-07 15:07:00 +00:00
|
|
|
taskKillSignal, err := getTaskKillSignal(task.KillSignal)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2016-10-12 18:35:29 +00:00
|
|
|
execCmd := &executor.ExecCommand{
|
2016-03-17 09:53:31 +00:00
|
|
|
Cmd: absPath,
|
|
|
|
Args: args,
|
|
|
|
FSIsolation: true,
|
|
|
|
ResourceLimits: true,
|
2016-03-23 11:57:31 +00:00
|
|
|
User: getExecutorUser(task),
|
2017-12-07 15:07:00 +00:00
|
|
|
TaskKillSignal: taskKillSignal,
|
2016-10-12 18:35:29 +00:00
|
|
|
}
|
|
|
|
ps, err := execIntf.LaunchCmd(execCmd)
|
2016-02-04 01:46:45 +00:00
|
|
|
if err != nil {
|
|
|
|
pluginClient.Kill()
|
2016-03-19 19:18:10 +00:00
|
|
|
return nil, err
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
2016-02-06 02:07:06 +00:00
|
|
|
d.logger.Printf("[DEBUG] driver.java: started process with pid: %v", ps.Pid)
|
2015-09-01 21:57:21 +00:00
|
|
|
|
|
|
|
// Return a driver handle
|
2016-03-03 17:21:21 +00:00
|
|
|
maxKill := d.DriverContext.config.MaxKillTimeout
|
2015-09-01 21:57:21 +00:00
|
|
|
h := &javaHandle{
|
2016-02-08 21:48:26 +00:00
|
|
|
pluginClient: pluginClient,
|
2016-03-16 03:21:52 +00:00
|
|
|
executor: execIntf,
|
2016-02-08 21:48:26 +00:00
|
|
|
userPid: ps.Pid,
|
|
|
|
isolationConfig: ps.IsolationConfig,
|
2017-02-01 00:43:57 +00:00
|
|
|
taskDir: ctx.TaskDir.Dir,
|
2016-03-03 17:21:21 +00:00
|
|
|
killTimeout: GetKillTimeout(task.KillTimeout, maxKill),
|
|
|
|
maxKillTimeout: maxKill,
|
2017-08-16 22:42:15 +00:00
|
|
|
version: d.config.Version.VersionNumber(),
|
2016-02-08 21:48:26 +00:00
|
|
|
logger: d.logger,
|
|
|
|
doneCh: make(chan struct{}),
|
2016-06-12 03:15:50 +00:00
|
|
|
waitCh: make(chan *dstructs.WaitResult, 1),
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
go h.run()
|
2017-06-09 17:29:41 +00:00
|
|
|
return &StartResponse{Handle: h}, nil
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2017-01-14 00:46:08 +00:00
|
|
|
func (d *JavaDriver) Cleanup(*ExecContext, *CreatedResources) error { return nil }
|
2017-01-10 21:24:45 +00:00
|
|
|
|
2015-12-23 00:10:30 +00:00
|
|
|
type javaId struct {
|
2016-02-25 03:06:30 +00:00
|
|
|
Version string
|
2016-02-08 21:48:26 +00:00
|
|
|
KillTimeout time.Duration
|
2016-03-03 17:21:21 +00:00
|
|
|
MaxKillTimeout time.Duration
|
2016-02-09 20:59:05 +00:00
|
|
|
PluginConfig *PluginReattachConfig
|
2016-06-12 03:15:50 +00:00
|
|
|
IsolationConfig *dstructs.IsolationConfig
|
2017-02-01 00:43:57 +00:00
|
|
|
TaskDir string
|
2016-02-08 21:48:26 +00:00
|
|
|
UserPid int
|
2015-12-23 00:10:30 +00:00
|
|
|
}
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
func (d *JavaDriver) Open(ctx *ExecContext, handleID string) (DriverHandle, error) {
|
2015-12-23 00:10:30 +00:00
|
|
|
id := &javaId{}
|
|
|
|
if err := json.Unmarshal([]byte(handleID), id); err != nil {
|
|
|
|
return nil, fmt.Errorf("Failed to parse handle '%s': %v", handleID, err)
|
|
|
|
}
|
|
|
|
|
2016-02-04 01:46:45 +00:00
|
|
|
pluginConfig := &plugin.ClientConfig{
|
2016-02-05 18:49:54 +00:00
|
|
|
Reattach: id.PluginConfig.PluginConfig(),
|
2016-02-04 01:46:45 +00:00
|
|
|
}
|
2017-01-12 19:50:49 +00:00
|
|
|
exec, pluginClient, err := createExecutorWithConfig(pluginConfig, d.config.LogOutput)
|
2015-09-15 20:45:48 +00:00
|
|
|
if err != nil {
|
2016-02-09 02:51:11 +00:00
|
|
|
merrs := new(multierror.Error)
|
|
|
|
merrs.Errors = append(merrs.Errors, err)
|
2016-02-24 23:57:58 +00:00
|
|
|
d.logger.Println("[ERR] driver.java: error connecting to plugin so destroying plugin pid and user pid")
|
2016-02-04 21:53:30 +00:00
|
|
|
if e := destroyPlugin(id.PluginConfig.Pid, id.UserPid); e != nil {
|
2016-02-09 02:51:11 +00:00
|
|
|
merrs.Errors = append(merrs.Errors, fmt.Errorf("error destroying plugin and userpid: %v", e))
|
2016-02-04 21:53:30 +00:00
|
|
|
}
|
2016-02-08 21:48:26 +00:00
|
|
|
if id.IsolationConfig != nil {
|
2016-04-19 00:20:11 +00:00
|
|
|
ePid := pluginConfig.Reattach.Pid
|
2016-07-10 06:45:33 +00:00
|
|
|
if e := executor.ClientCleanup(id.IsolationConfig, ePid); e != nil {
|
|
|
|
merrs.Errors = append(merrs.Errors, fmt.Errorf("destroying resource container failed: %v", e))
|
2016-02-08 21:48:26 +00:00
|
|
|
}
|
2016-02-08 18:05:39 +00:00
|
|
|
}
|
|
|
|
|
2016-02-09 02:51:11 +00:00
|
|
|
return nil, fmt.Errorf("error connecting to plugin: %v", merrs.ErrorOrNil())
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2016-03-30 05:05:02 +00:00
|
|
|
ver, _ := exec.Version()
|
|
|
|
d.logger.Printf("[DEBUG] driver.java: version of executor: %v", ver.Version)
|
2016-03-29 23:27:31 +00:00
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
// Return a driver handle
|
|
|
|
h := &javaHandle{
|
2016-02-08 21:48:26 +00:00
|
|
|
pluginClient: pluginClient,
|
2016-02-09 00:08:29 +00:00
|
|
|
executor: exec,
|
2016-02-08 21:48:26 +00:00
|
|
|
userPid: id.UserPid,
|
|
|
|
isolationConfig: id.IsolationConfig,
|
|
|
|
logger: d.logger,
|
2016-02-25 03:06:30 +00:00
|
|
|
version: id.Version,
|
2016-02-08 21:48:26 +00:00
|
|
|
killTimeout: id.KillTimeout,
|
2016-03-03 17:21:21 +00:00
|
|
|
maxKillTimeout: id.MaxKillTimeout,
|
2016-02-08 21:48:26 +00:00
|
|
|
doneCh: make(chan struct{}),
|
2016-06-12 03:15:50 +00:00
|
|
|
waitCh: make(chan *dstructs.WaitResult, 1),
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
go h.run()
|
|
|
|
return h, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *javaHandle) ID() string {
|
2015-12-23 00:10:30 +00:00
|
|
|
id := javaId{
|
2016-02-25 03:06:30 +00:00
|
|
|
Version: h.version,
|
2016-02-08 21:48:26 +00:00
|
|
|
KillTimeout: h.killTimeout,
|
2016-03-03 17:21:21 +00:00
|
|
|
MaxKillTimeout: h.maxKillTimeout,
|
2016-02-09 20:59:05 +00:00
|
|
|
PluginConfig: NewPluginReattachConfig(h.pluginClient.ReattachConfig()),
|
2016-02-08 21:48:26 +00:00
|
|
|
UserPid: h.userPid,
|
|
|
|
IsolationConfig: h.isolationConfig,
|
2017-02-01 00:43:57 +00:00
|
|
|
TaskDir: h.taskDir,
|
2015-12-23 00:10:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
data, err := json.Marshal(id)
|
|
|
|
if err != nil {
|
|
|
|
h.logger.Printf("[ERR] driver.java: failed to marshal ID to JSON: %s", err)
|
|
|
|
}
|
|
|
|
return string(data)
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2016-06-12 03:15:50 +00:00
|
|
|
func (h *javaHandle) WaitCh() chan *dstructs.WaitResult {
|
2015-09-01 21:57:21 +00:00
|
|
|
return h.waitCh
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *javaHandle) Update(task *structs.Task) error {
|
2016-02-04 03:43:44 +00:00
|
|
|
// Store the updated kill timeout.
|
2016-03-03 17:21:21 +00:00
|
|
|
h.killTimeout = GetKillTimeout(task.KillTimeout, h.maxKillTimeout)
|
2016-03-17 09:53:31 +00:00
|
|
|
h.executor.UpdateTask(task)
|
2016-02-04 03:43:44 +00:00
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
// Update is not possible
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-02-01 00:43:57 +00:00
|
|
|
func (h *javaHandle) Exec(ctx context.Context, cmd string, args []string) ([]byte, int, error) {
|
2017-04-21 23:20:37 +00:00
|
|
|
deadline, ok := ctx.Deadline()
|
|
|
|
if !ok {
|
|
|
|
// No deadline set on context; default to 1 minute
|
|
|
|
deadline = time.Now().Add(time.Minute)
|
|
|
|
}
|
|
|
|
return h.executor.Exec(deadline, cmd, args)
|
2017-02-01 00:43:57 +00:00
|
|
|
}
|
|
|
|
|
2016-10-07 19:37:52 +00:00
|
|
|
func (h *javaHandle) Signal(s os.Signal) error {
|
2016-10-10 18:46:27 +00:00
|
|
|
return h.executor.Signal(s)
|
2016-10-07 19:37:52 +00:00
|
|
|
}
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
func (h *javaHandle) Kill() error {
|
2016-02-17 05:00:49 +00:00
|
|
|
if err := h.executor.ShutDown(); err != nil {
|
|
|
|
if h.pluginClient.Exited() {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return fmt.Errorf("executor Shutdown failed: %v", err)
|
|
|
|
}
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
select {
|
|
|
|
case <-h.doneCh:
|
2015-12-23 00:10:30 +00:00
|
|
|
case <-time.After(h.killTimeout):
|
2016-02-17 05:00:49 +00:00
|
|
|
if h.pluginClient.Exited() {
|
2016-12-03 01:04:07 +00:00
|
|
|
break
|
2016-02-17 05:00:49 +00:00
|
|
|
}
|
|
|
|
if err := h.executor.Exit(); err != nil {
|
|
|
|
return fmt.Errorf("executor Exit failed: %v", err)
|
|
|
|
}
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
2016-12-03 01:04:07 +00:00
|
|
|
return nil
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|
|
|
|
|
2016-04-28 23:06:01 +00:00
|
|
|
func (h *javaHandle) Stats() (*cstructs.TaskResourceUsage, error) {
|
|
|
|
return h.executor.Stats()
|
|
|
|
}
|
|
|
|
|
2015-09-01 21:57:21 +00:00
|
|
|
func (h *javaHandle) run() {
|
2016-11-04 21:58:55 +00:00
|
|
|
ps, werr := h.executor.Wait()
|
2015-09-01 21:57:21 +00:00
|
|
|
close(h.doneCh)
|
2016-11-04 21:58:55 +00:00
|
|
|
if ps.ExitCode == 0 && werr != nil {
|
2016-02-09 18:17:33 +00:00
|
|
|
if h.isolationConfig != nil {
|
2016-04-19 00:20:11 +00:00
|
|
|
ePid := h.pluginClient.ReattachConfig().Pid
|
2016-07-10 06:45:33 +00:00
|
|
|
if e := executor.ClientCleanup(h.isolationConfig, ePid); e != nil {
|
|
|
|
h.logger.Printf("[ERR] driver.java: destroying resource container failed: %v", e)
|
2016-02-09 18:17:33 +00:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if e := killProcess(h.userPid); e != nil {
|
2016-02-24 23:57:58 +00:00
|
|
|
h.logger.Printf("[ERR] driver.java: error killing user process: %v", e)
|
2016-02-09 18:17:33 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-03-23 20:19:45 +00:00
|
|
|
|
2016-11-04 21:58:55 +00:00
|
|
|
// Exit the executor
|
2016-03-18 19:04:11 +00:00
|
|
|
h.executor.Exit()
|
2016-02-04 01:46:45 +00:00
|
|
|
h.pluginClient.Kill()
|
2016-11-04 21:58:55 +00:00
|
|
|
|
|
|
|
// Send the results
|
|
|
|
h.waitCh <- &dstructs.WaitResult{ExitCode: ps.ExitCode, Signal: ps.Signal, Err: werr}
|
|
|
|
close(h.waitCh)
|
2015-09-01 21:57:21 +00:00
|
|
|
}
|