open-nomad/drivers/qemu/handle.go

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

80 lines
1.7 KiB
Go
Raw Normal View History

2018-10-17 16:43:15 +00:00
package qemu
import (
2018-12-05 16:04:18 +00:00
"context"
2018-10-31 18:53:37 +00:00
"strconv"
2018-10-17 16:43:15 +00:00
"sync"
"time"
2019-01-15 19:46:12 +00:00
hclog "github.com/hashicorp/go-hclog"
plugin "github.com/hashicorp/go-plugin"
2018-12-07 01:54:14 +00:00
"github.com/hashicorp/nomad/drivers/shared/executor"
2018-10-17 16:43:15 +00:00
"github.com/hashicorp/nomad/plugins/drivers"
)
2018-10-31 18:53:37 +00:00
type taskHandle struct {
2018-12-07 01:54:14 +00:00
exec executor.Executor
2018-10-17 16:43:15 +00:00
pid int
pluginClient *plugin.Client
logger hclog.Logger
monitorPath string
2018-10-17 16:43:15 +00:00
// stateLock syncs access to all fields below
stateLock sync.RWMutex
taskConfig *drivers.TaskConfig
procState drivers.TaskState
startedAt time.Time
completedAt time.Time
exitResult *drivers.ExitResult
}
2018-10-31 18:53:37 +00:00
func (h *taskHandle) TaskStatus() *drivers.TaskStatus {
h.stateLock.RLock()
defer h.stateLock.RUnlock()
2018-10-31 18:53:37 +00:00
return &drivers.TaskStatus{
ID: h.taskConfig.ID,
Name: h.taskConfig.Name,
State: h.procState,
StartedAt: h.startedAt,
CompletedAt: h.completedAt,
ExitResult: h.exitResult,
DriverAttributes: map[string]string{
"pid": strconv.Itoa(h.pid),
},
}
2018-10-17 16:43:15 +00:00
}
2018-10-31 18:53:37 +00:00
func (h *taskHandle) IsRunning() bool {
h.stateLock.RLock()
defer h.stateLock.RUnlock()
return h.procState == drivers.TaskStateRunning
}
2018-10-17 16:43:15 +00:00
2018-10-31 18:53:37 +00:00
func (h *taskHandle) run() {
h.stateLock.Lock()
2018-10-17 16:43:15 +00:00
if h.exitResult == nil {
h.exitResult = &drivers.ExitResult{}
}
h.stateLock.Unlock()
2018-10-17 16:43:15 +00:00
2018-12-05 16:04:18 +00:00
ps, err := h.exec.Wait(context.Background())
2018-10-17 16:43:15 +00:00
h.stateLock.Lock()
defer h.stateLock.Unlock()
if err != nil {
h.exitResult.Err = err
h.procState = drivers.TaskStateUnknown
h.completedAt = time.Now()
return
}
h.procState = drivers.TaskStateExited
h.exitResult.ExitCode = ps.ExitCode
h.exitResult.Signal = ps.Signal
h.completedAt = ps.Time
// TODO: detect if the taskConfig OOMed
}