82175d1328
The driver manager is modeled after the device manager and is started by the client. It's responsible for handling driver lifecycle and reattachment state, as well as processing the incomming fingerprint and task events from each driver. The mananger exposes a method for registering event handlers for task events that is used by the task runner to update the server when a task has been updated with an event. Since driver fingerprinting has been implemented by the driver manager, it is no longer needed in the fingerprint mananger and has been removed.
66 lines
2.4 KiB
Go
66 lines
2.4 KiB
Go
package state
|
|
|
|
import (
|
|
"github.com/hashicorp/nomad/client/allocrunner/taskrunner/state"
|
|
dmstate "github.com/hashicorp/nomad/client/devicemanager/state"
|
|
driverstate "github.com/hashicorp/nomad/client/pluginmanager/drivermanager/state"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
)
|
|
|
|
// StateDB implementations store and load Nomad client state.
|
|
type StateDB interface {
|
|
// Name of implementation.
|
|
Name() string
|
|
|
|
// GetAllAllocations returns all valid allocations and a map of
|
|
// allocation IDs to retrieval errors.
|
|
//
|
|
// If a single error is returned then both allocations and the map will be nil.
|
|
GetAllAllocations() ([]*structs.Allocation, map[string]error, error)
|
|
|
|
// PulAllocation stores an allocation or returns an error if it could
|
|
// not be stored.
|
|
PutAllocation(*structs.Allocation) error
|
|
|
|
// GetTaskRunnerState returns the LocalState and TaskState for a
|
|
// TaskRunner. Either state may be nil if it is not found, but if an
|
|
// error is encountered only the error will be non-nil.
|
|
GetTaskRunnerState(allocID, taskName string) (*state.LocalState, *structs.TaskState, error)
|
|
|
|
// PutTaskRunnerLocalTask stores the LocalState for a TaskRunner or
|
|
// returns an error.
|
|
PutTaskRunnerLocalState(allocID, taskName string, val *state.LocalState) error
|
|
|
|
// PutTaskState stores the TaskState for a TaskRunner or returns an
|
|
// error.
|
|
PutTaskState(allocID, taskName string, state *structs.TaskState) error
|
|
|
|
// DeleteTaskBucket deletes a task's state bucket if it exists. No
|
|
// error is returned if it does not exist.
|
|
DeleteTaskBucket(allocID, taskName string) error
|
|
|
|
// DeleteAllocationBucket deletes an allocation's state bucket if it
|
|
// exists. No error is returned if it does not exist.
|
|
DeleteAllocationBucket(allocID string) error
|
|
|
|
// GetDevicePluginState is used to retrieve the device manager's plugin
|
|
// state.
|
|
GetDevicePluginState() (*dmstate.PluginState, error)
|
|
|
|
// PutDevicePluginState is used to store the device manager's plugin
|
|
// state.
|
|
PutDevicePluginState(state *dmstate.PluginState) error
|
|
|
|
// GetDriverPluginState is used to retrieve the driver manager's plugin
|
|
// state.
|
|
GetDriverPluginState() (*driverstate.PluginState, error)
|
|
|
|
// PutDriverPluginState is used to store the driver manager's plugin
|
|
// state.
|
|
PutDriverPluginState(state *driverstate.PluginState) error
|
|
|
|
// Close the database. Unsafe for further use after calling regardless
|
|
// of return value.
|
|
Close() error
|
|
}
|