From 445fc589ac85d9404679e725b5455ed53fe44b1e Mon Sep 17 00:00:00 2001 From: Dan Rammer Date: Fri, 29 Sep 2023 09:38:42 -0500 Subject: [PATCH] using negative interruptibleFailureThreshold as complementary (#403) Signed-off-by: Daniel Rammer --- go/tasks/pluginmachinery/core/exec_metadata.go | 2 +- .../core/mocks/task_execution_metadata.go | 10 +++++----- go/tasks/plugins/array/k8s/subtask_exec_context.go | 2 +- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/go/tasks/pluginmachinery/core/exec_metadata.go b/go/tasks/pluginmachinery/core/exec_metadata.go index b3115a7a2..7133bb3ad 100644 --- a/go/tasks/pluginmachinery/core/exec_metadata.go +++ b/go/tasks/pluginmachinery/core/exec_metadata.go @@ -44,6 +44,6 @@ type TaskExecutionMetadata interface { GetSecurityContext() core.SecurityContext IsInterruptible() bool GetPlatformResources() *v1.ResourceRequirements - GetInterruptibleFailureThreshold() uint32 + GetInterruptibleFailureThreshold() int32 GetEnvironmentVariables() map[string]string } diff --git a/go/tasks/pluginmachinery/core/mocks/task_execution_metadata.go b/go/tasks/pluginmachinery/core/mocks/task_execution_metadata.go index 29f0055dc..14a545eb5 100644 --- a/go/tasks/pluginmachinery/core/mocks/task_execution_metadata.go +++ b/go/tasks/pluginmachinery/core/mocks/task_execution_metadata.go @@ -92,7 +92,7 @@ type TaskExecutionMetadata_GetInterruptibleFailureThreshold struct { *mock.Call } -func (_m TaskExecutionMetadata_GetInterruptibleFailureThreshold) Return(_a0 uint32) *TaskExecutionMetadata_GetInterruptibleFailureThreshold { +func (_m TaskExecutionMetadata_GetInterruptibleFailureThreshold) Return(_a0 int32) *TaskExecutionMetadata_GetInterruptibleFailureThreshold { return &TaskExecutionMetadata_GetInterruptibleFailureThreshold{Call: _m.Call.Return(_a0)} } @@ -107,14 +107,14 @@ func (_m *TaskExecutionMetadata) OnGetInterruptibleFailureThresholdMatch(matcher } // GetInterruptibleFailureThreshold provides a mock function with given fields: -func (_m *TaskExecutionMetadata) GetInterruptibleFailureThreshold() uint32 { +func (_m *TaskExecutionMetadata) GetInterruptibleFailureThreshold() int32 { ret := _m.Called() - var r0 uint32 - if rf, ok := ret.Get(0).(func() uint32); ok { + var r0 int32 + if rf, ok := ret.Get(0).(func() int32); ok { r0 = rf() } else { - r0 = ret.Get(0).(uint32) + r0 = ret.Get(0).(int32) } return r0 diff --git a/go/tasks/plugins/array/k8s/subtask_exec_context.go b/go/tasks/plugins/array/k8s/subtask_exec_context.go index 646118698..f3a372666 100644 --- a/go/tasks/plugins/array/k8s/subtask_exec_context.go +++ b/go/tasks/plugins/array/k8s/subtask_exec_context.go @@ -268,7 +268,7 @@ func NewSubTaskExecutionMetadata(taskExecutionMetadata pluginsCore.TaskExecution } subTaskExecutionID := NewSubTaskExecutionID(taskExecutionMetadata.GetTaskExecutionID(), executionIndex, retryAttempt) - interruptible := taskExecutionMetadata.IsInterruptible() && uint32(systemFailures) < taskExecutionMetadata.GetInterruptibleFailureThreshold() + interruptible := taskExecutionMetadata.IsInterruptible() && int32(systemFailures) < taskExecutionMetadata.GetInterruptibleFailureThreshold() return SubTaskExecutionMetadata{ taskExecutionMetadata, utils.UnionMaps(taskExecutionMetadata.GetAnnotations(), secretsMap),