diff options
Diffstat (limited to 'exec2/driver.go')
| -rw-r--r-- | exec2/driver.go | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/exec2/driver.go b/exec2/driver.go index f035d15..4e52062 100644 --- a/exec2/driver.go +++ b/exec2/driver.go @@ -1,4 +1,4 @@ -package exec +package exec2 import ( "context" @@ -27,7 +27,7 @@ import ( const ( // pluginName is the name of the plugin - pluginName = "exec" + pluginName = "exec2" // fingerprintPeriod is the interval at which the driver will send fingerprint responses fingerprintPeriod = 30 * time.Second @@ -45,13 +45,6 @@ var ( PluginType: base.PluginTypeDriver, } - // PluginConfig is the exec driver factory function registered in the - // plugin catalog. - PluginConfig = &loader.InternalPluginConfig{ - Config: map[string]interface{}{}, - Factory: func(ctx context.Context, l hclog.Logger) interface{} { return NewExecDriver(ctx, l) }, - } - // pluginInfo is the response returned for the PluginInfo RPC pluginInfo = &base.PluginInfoResponse{ Type: base.PluginTypeDriver, @@ -125,6 +118,10 @@ type Driver struct { // coordinate shutdown ctx context.Context + // signalShutdown is called when the driver is shutting down and cancels + // the ctx passed to any subsystems + signalShutdown context.CancelFunc + // logger will log to the Nomad agent logger hclog.Logger @@ -233,13 +230,15 @@ type TaskState struct { StartedAt time.Time } -// NewExecDriver returns a new DrivePlugin implementation -func NewExecDriver(ctx context.Context, logger hclog.Logger) drivers.DriverPlugin { +// NewPlugin returns a new DrivePlugin implementation +func NewPlugin(logger hclog.Logger) drivers.DriverPlugin { + ctx, cancel := context.WithCancel(context.Background()) logger = logger.Named(pluginName) return &Driver{ eventer: eventer.NewEventer(ctx, logger), tasks: newTaskStore(), ctx: ctx, + signalShutdown: cancel, logger: logger, } } |