From 38e8a074a7bbd659d98afd3d07d2a047d4b5c6b5 Mon Sep 17 00:00:00 2001 From: Dan Rammer Date: Fri, 31 Mar 2023 13:36:30 -0500 Subject: [PATCH] mocking plugin state reader in maptask (#338) Signed-off-by: Daniel Rammer --- .../plugins/array/k8s/subtask_exec_context.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/go/tasks/plugins/array/k8s/subtask_exec_context.go b/go/tasks/plugins/array/k8s/subtask_exec_context.go index 0eb0a92e8..1f18b1009 100644 --- a/go/tasks/plugins/array/k8s/subtask_exec_context.go +++ b/go/tasks/plugins/array/k8s/subtask_exec_context.go @@ -50,6 +50,23 @@ func (s SubTaskExecutionContext) TaskReader() pluginsCore.TaskReader { return s.subtaskReader } +// pluginStateReader overrides the default PluginStateReader because the maptask does not persist +// state between task evaluations. +type pluginStateReader struct{} + +func (p pluginStateReader) GetStateVersion() uint8 { + return 0 +} + +func (p pluginStateReader) Get(t interface{}) (stateVersion uint8, err error) { + return 0, nil +} + +// PluginStateReader overrides the default behavior to return a custom pluginStateReader. +func (s SubTaskExecutionContext) PluginStateReader() pluginsCore.PluginStateReader { + return pluginStateReader{} +} + // NewSubtaskExecutionContext constructs a SubTaskExecutionContext using the provided parameters func NewSubTaskExecutionContext(ctx context.Context, tCtx pluginsCore.TaskExecutionContext, taskTemplate *core.TaskTemplate, executionIndex, originalIndex int, retryAttempt uint64, systemFailures uint64) (SubTaskExecutionContext, error) {