Skip to content

Commit ab64ab6

Browse files
author
Mansi Bhargava
committed
Add job id for feature row in warehouse store
1 parent 96d3173 commit ab64ab6

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

ingestion/src/main/java/feast/store/warehouse/bigquery/FeatureRowToBigQueryTableRowDoFn.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ public class FeatureRowToBigQueryTableRowDoFn extends DoFn<FeatureRowExtended, T
3636
private static final String ENTITY_KEY_COLUMN = "id";
3737
private static final String EVENT_TIMESTAMP_COLUMN = "event_timestamp";
3838
private static final String CREATED_TIMESTAMP_COLUMN = "created_timestamp";
39+
private static final String JOB_ID_COLUMN = "job_id";
40+
3941

4042
private Specs specs;
4143

@@ -59,6 +61,7 @@ public TableRow toTableRow(FeatureRowExtended featureRowExtended) {
5961
ValueBigQueryBuilder.bigQueryObjectOf(
6062
Value.newBuilder()
6163
.setTimestampVal(DateUtil.toTimestamp(DateTime.now(DateTimeZone.UTC)))));
64+
tableRow.set(JOB_ID_COLUMN, specs.getJobName());
6265

6366
for (Feature feature : featureRow.getFeaturesList()) {
6467
Object featureValue = ValueBigQueryBuilder.bigQueryObjectOf(feature.getValue());

0 commit comments

Comments
 (0)