[Caching] Add a cached label for cached pods (kubeflow#3623)

* Update mutation.go * Update mutation.go * Update mutation.go * Update mutation.go
Jeffwan · Dec 9, 2020 · 48aca5b · 48aca5b
1 parent 245915e
commit 48aca5b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/backend/src/cache/server/mutation.go b/backend/src/cache/server/mutation.go
@@ -34,6 +34,8 @@ import (
 const (
 	KFPCacheEnabledLabelKey   string = "pipelines.kubeflow.org/cache_enabled"
 	KFPCacheEnabledLabelValue string = "true"
+	KFPCachedLabelKey         string = "pipelines.kubeflow.org/reused_from_cache"
+	KFPCachedLabelValue       string = "true"
 	ArgoWorkflowNodeName      string = "workflows.argoproj.io/node-name"
 	ArgoWorkflowTemplate      string = "workflows.argoproj.io/template"
 	ExecutionKey              string = "pipelines.kubeflow.org/execution_cache_key"
@@ -124,6 +126,9 @@ func MutatePodIfCached(req *v1beta1.AdmissionRequest, clientMgr ClientManagerInt
 
 		annotations[ArgoWorkflowOutputs] = getValueFromSerializedMap(cachedExecution.ExecutionOutput, ArgoWorkflowOutputs)
 		labels[CacheIDLabelKey] = strconv.FormatInt(cachedExecution.ID, 10)
+		labels[KFPCachedLabelKey] = KFPCachedLabelValue // This label indicates the pod is taken from cache.
+
+		// These labels cache results for metadata-writer.
 		labels[MetadataExecutionIDKey] = getValueFromSerializedMap(cachedExecution.ExecutionOutput, MetadataExecutionIDKey)
 		labels[MetadataWrittenKey] = "true"