controllers

package
v1.5.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: May 6, 2026 License: Apache-2.0 Imports: 55 Imported by: 0

Documentation

Overview

Code generated by MockGen. DO NOT EDIT. Source: device_config_reconciler.go

Generated by this command: 

mockgen -source=device_config_reconciler.go -package=controllers -destination=mock_device_config_reconciler.go deviceConfigReconcilerHelperAPI

Package controllers is a generated GoMock package.

Code generated by MockGen. DO NOT EDIT. Source: remediation_handler.go

Generated by this command: 

mockgen -source=remediation_handler.go -package=controllers -destination=mock_remediation_handler.go remediationMgrHelperAPI

Package controllers is a generated GoMock package.

Code generated by MockGen. DO NOT EDIT. Source: upgrademgr.go

Generated by this command: 

mockgen -source=upgrademgr.go -package=controllers -destination=mock_upgrademgr.go upgradeMgrHelperAPI

Package controllers is a generated GoMock package.

Index

Constants

View Source 
const (
	RemediationTaintKey        = "amd-gpu-unhealthy"
	DefaultConfigMapSuffix     = "default-conditional-workflow-mappings"
	DefaultTemplate            = "default-template"
	DefaultTestRunnerImage     = "docker.io/rocm/test-runner:v1.4.1"
	TestRunnerServiceAccount   = "amd-gpu-operator-test-runner"
	AmdGpuRemediationRequired  = "amd-gpu-remediation-required"
	AmdGpuRemediationSucceeded = "amd-gpu-remediation-succeeded"
	AmdGpuRemediationFailed    = "amd-gpu-remediation-failed"
	DefaultUtilityImage        = "docker.io/rocm/gpu-operator-utils:latest"
	// DefaultRecoveryPolicyWindowSize - defines the time window size for recovery policy
	DefaultRecoveryPolicyWindowSize = "300m"
	// DefaultRecoveryPolicyMaxRunsPerWindow - defines the max allowed runs per window for recovery policy
	// If a specific node condition is hit more than this number of times within the window size, no new remediation workflows will be scheduled
	DefaultRecoveryPolicyMaxRunsPerWindow = 3
	DefaultTimeFormatLayout               = "2006-01-02 15:04:05 UTC"
	DefaultStatusCRCleanupWindowSize      = "72h"
	// DefaultRebootTimeout - default total time the waitfornodeready
	// step waits for a node to reboot and remain Ready before timing out.
	DefaultRebootTimeout = "15m"
	// Below is the label and value needed to be added to node to force resume a suspended workflow
	ForceResumeWorkflowLabelKey   = "operator.amd.com/gpu-force-resume-workflow"
	ForceResumeWorkflowLabelValue = "true"
	// Below is the label and value needed to be added to node to abort an ongoing workflow
	AbortWorkflowLabelKey           = "operator.amd.com/gpu-abort-workflow"
	AbortWorkflowLabelValue         = "true"
	RemediationFilesPath            = "/remediation"
	DefaultInitContainerImage       = "busybox:1.36"
	ConfigMapImageJobSuffix         = "remediation-configmap-job"
	ConfigMapImageAnnotationKey     = "operator.amd.com/configmap-source-image"
	ArgoWorkflowControllerConfigMap = "amd-gpu-operator-workflow-controller-config"
	ArgoWorkflowInstaceIDLabelKey   = "workflows.argoproj.io/controller-instanceid"
	ArgoWorkflowInstaceIDLabelValue = "amd-gpu-operator-remediation-workflow"
)
View Source 
const (
	DeviceConfigReconcilerName = "DriverAndPluginReconciler"
)

Variables

This section is empty.

Functions

This section is empty.

Types

type ConditionWorkflowMapping added in v1.5.0 

type ConditionWorkflowMapping struct {
	NodeCondition            string                 `json:"nodeCondition" yaml:"nodeCondition"`
	WorkflowTemplate         string                 `json:"workflowTemplate" yaml:"workflowTemplate"`
	ValidationTests          ValidationTestsProfile `json:"validationTestsProfile" yaml:"validationTestsProfile"`
	PhysicalActionNeeded     bool                   `json:"physicalActionNeeded" yaml:"physicalActionNeeded"`
	NotifyRemediationMessage string                 `json:"notifyRemediationMessage" yaml:"notifyRemediationMessage"`
	NotifyTestFailureMessage string                 `json:"notifyTestFailureMessage" yaml:"notifyTestFailureMessage"`
	RecoveryPolicy           RecoveryPolicyConfig   `json:"recoveryPolicy" yaml:"recoveryPolicy"`
	SkipRebootStep           bool                   `json:"skipRebootStep" yaml:"skipRebootStep"`
}

ConditionWorkflowMapping defines a single condition-to-workflow mapping. This is used when parsing the ConfigMap specified in the DeviceConfig.

type DeviceConfigReconciler 

type DeviceConfigReconciler struct {
	client.Client
	// contains filtered or unexported fields
}

ModuleReconciler reconciles a Module object

func NewDeviceConfigReconciler 

func NewDeviceConfigReconciler(
	k8sConfig *rest.Config,
	client client.Client,
	apiReader client.Reader,
	kmmHandler kmmmodule.KMMModuleAPI,
	dpHandler plugin.DevicePluginAPI,
	nlHandler nodelabeller.NodeLabeller,
	metricsHandler metricsexporter.MetricsExporter,
	testrunnerHandler testrunner.TestRunner,
	configmanagerHandler configmanager.ConfigManager,
	workerMgr workermgr.WorkerMgrAPI,
	isOpenShift bool,
	kmmWatchEnabled bool) *DeviceConfigReconciler

func (*DeviceConfigReconciler) Reconcile 

func (r *DeviceConfigReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error)

func (*DeviceConfigReconciler) SetupWithManager 

func (r *DeviceConfigReconciler) SetupWithManager(mgr ctrl.Manager) error

SetupWithManager sets up the controller with the Manager.

type MockdeviceConfigReconcilerHelperAPI 

type MockdeviceConfigReconcilerHelperAPI struct {
	// contains filtered or unexported fields
}

MockdeviceConfigReconcilerHelperAPI is a mock of deviceConfigReconcilerHelperAPI interface.

func NewMockdeviceConfigReconcilerHelperAPI 

func NewMockdeviceConfigReconcilerHelperAPI(ctrl *gomock.Controller) *MockdeviceConfigReconcilerHelperAPI

NewMockdeviceConfigReconcilerHelperAPI creates a new mock instance.

func (*MockdeviceConfigReconcilerHelperAPI) EXPECT 

func (m *MockdeviceConfigReconcilerHelperAPI) EXPECT() *MockdeviceConfigReconcilerHelperAPIMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

type MockdeviceConfigReconcilerHelperAPIMockRecorder 

type MockdeviceConfigReconcilerHelperAPIMockRecorder struct {
	// contains filtered or unexported fields
}

MockdeviceConfigReconcilerHelperAPIMockRecorder is the mock recorder for MockdeviceConfigReconcilerHelperAPI.

type MockremediationMgrAPI added in v1.5.0 

type MockremediationMgrAPI struct {
	// contains filtered or unexported fields
}

MockremediationMgrAPI is a mock of remediationMgrAPI interface.

func NewMockremediationMgrAPI added in v1.5.0 

func NewMockremediationMgrAPI(ctrl *gomock.Controller) *MockremediationMgrAPI

NewMockremediationMgrAPI creates a new mock instance.

func (*MockremediationMgrAPI) EXPECT added in v1.5.0 

func (m *MockremediationMgrAPI) EXPECT() *MockremediationMgrAPIMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockremediationMgrAPI) HandleDelete added in v1.5.0 

func (m *MockremediationMgrAPI) HandleDelete(ctx context.Context, deviceConfig *v1alpha1.DeviceConfig, nodes *v1.NodeList) (controllerruntime.Result, error)

HandleDelete mocks base method.

func (*MockremediationMgrAPI) HandleRemediation added in v1.5.0 

func (m *MockremediationMgrAPI) HandleRemediation(ctx context.Context, deviceConfig *v1alpha1.DeviceConfig, nodes *v1.NodeList) (controllerruntime.Result, error)

HandleRemediation mocks base method.

type MockremediationMgrAPIMockRecorder added in v1.5.0 

type MockremediationMgrAPIMockRecorder struct {
	// contains filtered or unexported fields
}

MockremediationMgrAPIMockRecorder is the mock recorder for MockremediationMgrAPI.

func (*MockremediationMgrAPIMockRecorder) HandleDelete added in v1.5.0 

func (mr *MockremediationMgrAPIMockRecorder) HandleDelete(ctx, deviceConfig, nodes any) *gomock.Call

HandleDelete indicates an expected call of HandleDelete.

func (*MockremediationMgrAPIMockRecorder) HandleRemediation added in v1.5.0 

func (mr *MockremediationMgrAPIMockRecorder) HandleRemediation(ctx, deviceConfig, nodes any) *gomock.Call

HandleRemediation indicates an expected call of HandleRemediation.

type MockremediationMgrHelperAPI added in v1.5.0 

type MockremediationMgrHelperAPI struct {
	// contains filtered or unexported fields
}

MockremediationMgrHelperAPI is a mock of remediationMgrHelperAPI interface.

func NewMockremediationMgrHelperAPI added in v1.5.0 

func NewMockremediationMgrHelperAPI(ctrl *gomock.Controller) *MockremediationMgrHelperAPI

NewMockremediationMgrHelperAPI creates a new mock instance.

func (*MockremediationMgrHelperAPI) EXPECT added in v1.5.0 

func (m *MockremediationMgrHelperAPI) EXPECT() *MockremediationMgrHelperAPIMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

type MockremediationMgrHelperAPIMockRecorder added in v1.5.0 

type MockremediationMgrHelperAPIMockRecorder struct {
	// contains filtered or unexported fields
}

MockremediationMgrHelperAPIMockRecorder is the mock recorder for MockremediationMgrHelperAPI.

type MockupgradeMgrAPI 

type MockupgradeMgrAPI struct {
	// contains filtered or unexported fields
}

MockupgradeMgrAPI is a mock of upgradeMgrAPI interface.

func NewMockupgradeMgrAPI 

func NewMockupgradeMgrAPI(ctrl *gomock.Controller) *MockupgradeMgrAPI

NewMockupgradeMgrAPI creates a new mock instance.

func (*MockupgradeMgrAPI) EXPECT 

func (m *MockupgradeMgrAPI) EXPECT() *MockupgradeMgrAPIMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (*MockupgradeMgrAPI) GetNodeBootId added in v1.2.2 

func (m *MockupgradeMgrAPI) GetNodeBootId(nodeName string) string

GetNodeBootId mocks base method.

func (*MockupgradeMgrAPI) GetNodeStatus 

func (m *MockupgradeMgrAPI) GetNodeStatus(nodeName string) v1alpha1.UpgradeState

GetNodeStatus mocks base method.

func (*MockupgradeMgrAPI) GetNodeUpgradeStartTime 

func (m *MockupgradeMgrAPI) GetNodeUpgradeStartTime(nodeName string) string

GetNodeUpgradeStartTime mocks base method.

func (*MockupgradeMgrAPI) HandleDelete 

func (m *MockupgradeMgrAPI) HandleDelete(ctx context.Context, deviceConfig *v1alpha1.DeviceConfig, nodes *v1.NodeList) (controllerruntime.Result, error)

HandleDelete mocks base method.

func (*MockupgradeMgrAPI) HandleUpgrade 

func (m *MockupgradeMgrAPI) HandleUpgrade(ctx context.Context, deviceConfig *v1alpha1.DeviceConfig, nodes *v1.NodeList) (controllerruntime.Result, error)

HandleUpgrade mocks base method.

type MockupgradeMgrAPIMockRecorder 

type MockupgradeMgrAPIMockRecorder struct {
	// contains filtered or unexported fields
}

MockupgradeMgrAPIMockRecorder is the mock recorder for MockupgradeMgrAPI.

func (*MockupgradeMgrAPIMockRecorder) GetNodeBootId added in v1.2.2 

func (mr *MockupgradeMgrAPIMockRecorder) GetNodeBootId(nodeName any) *gomock.Call

GetNodeBootId indicates an expected call of GetNodeBootId.

func (*MockupgradeMgrAPIMockRecorder) GetNodeStatus 

func (mr *MockupgradeMgrAPIMockRecorder) GetNodeStatus(nodeName any) *gomock.Call

GetNodeStatus indicates an expected call of GetNodeStatus.

func (*MockupgradeMgrAPIMockRecorder) GetNodeUpgradeStartTime 

func (mr *MockupgradeMgrAPIMockRecorder) GetNodeUpgradeStartTime(nodeName any) *gomock.Call

GetNodeUpgradeStartTime indicates an expected call of GetNodeUpgradeStartTime.

func (*MockupgradeMgrAPIMockRecorder) HandleDelete 

func (mr *MockupgradeMgrAPIMockRecorder) HandleDelete(ctx, deviceConfig, nodes any) *gomock.Call

HandleDelete indicates an expected call of HandleDelete.

func (*MockupgradeMgrAPIMockRecorder) HandleUpgrade 

func (mr *MockupgradeMgrAPIMockRecorder) HandleUpgrade(ctx, deviceConfig, nodes any) *gomock.Call

HandleUpgrade indicates an expected call of HandleUpgrade.

type MockupgradeMgrHelperAPI 

type MockupgradeMgrHelperAPI struct {
	// contains filtered or unexported fields
}

MockupgradeMgrHelperAPI is a mock of upgradeMgrHelperAPI interface.

func NewMockupgradeMgrHelperAPI 

func NewMockupgradeMgrHelperAPI(ctrl *gomock.Controller) *MockupgradeMgrHelperAPI

NewMockupgradeMgrHelperAPI creates a new mock instance.

func (*MockupgradeMgrHelperAPI) EXPECT 

func (m *MockupgradeMgrHelperAPI) EXPECT() *MockupgradeMgrHelperAPIMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

type MockupgradeMgrHelperAPIMockRecorder 

type MockupgradeMgrHelperAPIMockRecorder struct {
	// contains filtered or unexported fields
}

MockupgradeMgrHelperAPIMockRecorder is the mock recorder for MockupgradeMgrHelperAPI.

type RecoveryPolicyConfig added in v1.5.0 

type RecoveryPolicyConfig struct {
	MaxAllowedRunsPerWindow int    `json:"maxAllowedRunsPerWindow" yaml:"maxAllowedRunsPerWindow"`
	WindowSize              string `json:"windowSize" yaml:"windowSize"`
}

type ValidationTestsProfile added in v1.5.0 

type ValidationTestsProfile struct {
	Framework      string `json:"framework" yaml:"framework"`
	Recipe         string `json:"recipe" yaml:"recipe"`
	Iterations     int    `json:"iterations" yaml:"iterations"`
	StopOnFailure  bool   `json:"stopOnFailure" yaml:"stopOnFailure"`
	TimeoutSeconds int    `json:"timeoutSeconds" yaml:"timeoutSeconds"`
}

Source Files

View all Source files

Directories

Path Synopsis
Code generated by MockGen.
 Code generated by MockGen.
Code generated by MockGen.
 Code generated by MockGen.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.