diff --git a/ingestion/src/main/java/feast/ingestion/transform/metrics/WriteRowMetricsDoFn.java b/ingestion/src/main/java/feast/ingestion/transform/metrics/WriteRowMetricsDoFn.java index 8650285445..29597ae84f 100644 --- a/ingestion/src/main/java/feast/ingestion/transform/metrics/WriteRowMetricsDoFn.java +++ b/ingestion/src/main/java/feast/ingestion/transform/metrics/WriteRowMetricsDoFn.java @@ -25,9 +25,11 @@ import feast.proto.types.ValueProto.Value; import feast.proto.types.ValueProto.Value.ValCase; import java.time.Clock; +import java.util.Arrays; import java.util.HashMap; import java.util.Map; import java.util.Map.Entry; +import java.util.stream.Collectors; import javax.annotation.Nullable; import org.apache.beam.sdk.transforms.DoFn; import org.apache.beam.sdk.values.KV; @@ -192,11 +194,15 @@ public void processElement( } } + String[] split = c.getPipelineOptions().getJobName().split("-"); + String jobNameWithoutTimestamp = + Arrays.stream(split).limit(split.length - 1).collect(Collectors.joining("-")); + String[] tags = { STORE_TAG_KEY + ":" + getStoreName(), FEATURE_SET_PROJECT_TAG_KEY + ":" + featureSetProject, FEATURE_SET_NAME_TAG_KEY + ":" + featureSetName, - INGESTION_JOB_NAME_KEY + ":" + c.getPipelineOptions().getJobName(), + INGESTION_JOB_NAME_KEY + ":" + jobNameWithoutTimestamp, METRICS_NAMESPACE_KEY + ":" + getMetricsNamespace(), };