37 lines
990 B
Go
37 lines
990 B
Go
package rkt
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
plugin "github.com/hashicorp/go-plugin"
|
|
"github.com/hashicorp/nomad/drivers/shared/executor"
|
|
"github.com/hashicorp/nomad/helper/uuid"
|
|
"github.com/hashicorp/nomad/plugins/drivers"
|
|
)
|
|
|
|
func (d *Driver) recoverPre09Task(config *drivers.TaskConfig, reattach *plugin.ReattachConfig) error {
|
|
config.ID = fmt.Sprintf("pre09-%s", uuid.Generate())
|
|
exec, pluginClient, err := executor.ReattachToPre09Executor(reattach,
|
|
d.logger.With("task_name", config.Name, "alloc_id", config.AllocID))
|
|
if err != nil {
|
|
d.logger.Error("failed to reattach to executor", "error", err, "task_name", config.Name)
|
|
return fmt.Errorf("failed to reattach to executor: %v", err)
|
|
}
|
|
|
|
h := &taskHandle{
|
|
exec: exec,
|
|
pid: reattach.Pid,
|
|
pluginClient: pluginClient,
|
|
taskConfig: config,
|
|
procState: drivers.TaskStateRunning,
|
|
startedAt: time.Now(),
|
|
exitResult: &drivers.ExitResult{},
|
|
}
|
|
|
|
d.tasks.Set(config.ID, h)
|
|
|
|
go h.run()
|
|
return nil
|
|
}
|