github.com/Ilhicas/nomad@v1.0.4-0.20210304152020-e86851182bc3/client/state/errdb.go (about)

     1  package state
     2  
     3  import (
     4  	"fmt"
     5  
     6  	"github.com/hashicorp/nomad/client/allocrunner/taskrunner/state"
     7  	dmstate "github.com/hashicorp/nomad/client/devicemanager/state"
     8  	"github.com/hashicorp/nomad/client/dynamicplugins"
     9  	driverstate "github.com/hashicorp/nomad/client/pluginmanager/drivermanager/state"
    10  	"github.com/hashicorp/nomad/nomad/structs"
    11  )
    12  
    13  // ErrDB implements a StateDB that returns errors on restore methods, used for testing
    14  type ErrDB struct {
    15  	// Allocs is a preset slice of allocations used in GetAllAllocations
    16  	Allocs []*structs.Allocation
    17  }
    18  
    19  func (m *ErrDB) Name() string {
    20  	return "errdb"
    21  }
    22  
    23  func (m *ErrDB) Upgrade() error {
    24  	return nil
    25  }
    26  
    27  func (m *ErrDB) GetAllAllocations() ([]*structs.Allocation, map[string]error, error) {
    28  	return m.Allocs, nil, nil
    29  }
    30  
    31  func (m *ErrDB) PutAllocation(alloc *structs.Allocation, opts ...WriteOption) error {
    32  	return fmt.Errorf("Error!")
    33  }
    34  
    35  func (m *ErrDB) GetDeploymentStatus(allocID string) (*structs.AllocDeploymentStatus, error) {
    36  	return nil, fmt.Errorf("Error!")
    37  }
    38  
    39  func (m *ErrDB) PutDeploymentStatus(allocID string, ds *structs.AllocDeploymentStatus) error {
    40  	return fmt.Errorf("Error!")
    41  }
    42  
    43  func (m *ErrDB) GetNetworkStatus(allocID string) (*structs.AllocNetworkStatus, error) {
    44  	return nil, fmt.Errorf("Error!")
    45  }
    46  
    47  func (m *ErrDB) PutNetworkStatus(allocID string, ns *structs.AllocNetworkStatus, opts ...WriteOption) error {
    48  	return fmt.Errorf("Error!")
    49  }
    50  
    51  func (m *ErrDB) GetTaskRunnerState(allocID string, taskName string) (*state.LocalState, *structs.TaskState, error) {
    52  	return nil, nil, fmt.Errorf("Error!")
    53  }
    54  
    55  func (m *ErrDB) PutTaskRunnerLocalState(allocID string, taskName string, val *state.LocalState) error {
    56  	return fmt.Errorf("Error!")
    57  }
    58  
    59  func (m *ErrDB) PutTaskState(allocID string, taskName string, state *structs.TaskState) error {
    60  	return fmt.Errorf("Error!")
    61  }
    62  
    63  func (m *ErrDB) DeleteTaskBucket(allocID, taskName string) error {
    64  	return fmt.Errorf("Error!")
    65  }
    66  
    67  func (m *ErrDB) DeleteAllocationBucket(allocID string, opts ...WriteOption) error {
    68  	return fmt.Errorf("Error!")
    69  }
    70  
    71  func (m *ErrDB) PutDevicePluginState(ps *dmstate.PluginState) error {
    72  	return fmt.Errorf("Error!")
    73  }
    74  
    75  func (m *ErrDB) GetDynamicPluginRegistryState() (*dynamicplugins.RegistryState, error) {
    76  	return nil, fmt.Errorf("Error!")
    77  }
    78  
    79  func (m *ErrDB) PutDynamicPluginRegistryState(state *dynamicplugins.RegistryState) error {
    80  	return fmt.Errorf("Error!")
    81  }
    82  
    83  // GetDevicePluginState stores the device manager's plugin state or returns an
    84  // error.
    85  func (m *ErrDB) GetDevicePluginState() (*dmstate.PluginState, error) {
    86  	return nil, fmt.Errorf("Error!")
    87  }
    88  
    89  func (m *ErrDB) GetDriverPluginState() (*driverstate.PluginState, error) {
    90  	return nil, fmt.Errorf("Error!")
    91  }
    92  
    93  func (m *ErrDB) PutDriverPluginState(ps *driverstate.PluginState) error {
    94  	return fmt.Errorf("Error!")
    95  }
    96  
    97  func (m *ErrDB) Close() error {
    98  	return fmt.Errorf("Error!")
    99  }
   100  
   101  // Ensure *ErrDB implements StateDB
   102  var _ StateDB = (*ErrDB)(nil)