Add optional run ID to job meta

Signed-off-by: wslulciuc <[email protected]>

Author: wslulciuc

src/main/java/marquez/client/models/Job.java

@@ -50,7 +50,7 @@ public Job(
       final Map<String, String> context,
       final String description,
       @Nullable final Run latestRun) {
-    super(type, inputs, outputs, location, context, description);
+    super(type, inputs, outputs, location, context, description, null);
     this.id = id;
     this.name = name;
     this.createdAt = createdAt;

src/main/java/marquez/client/models/JobMeta.java

@@ -36,20 +36,23 @@ public class JobMeta {
   @Nullable private final URL location;
   @Getter private final Map<String, String> context;
   @Nullable String description;
+  @Nullable String runId;
 
   public JobMeta(
       @NonNull final JobType type,
       @NonNull final Set<DatasetId> inputs,
       @NonNull final Set<DatasetId> outputs,
       @Nullable final URL location,
       @Nullable final Map<String, String> context,
-      @Nullable final String description) {
+      @Nullable final String description,
+      @Nullable String runId) {
     this.type = type;
     this.inputs = inputs;
     this.outputs = outputs;
     this.location = location;
     this.context = (context == null) ? ImmutableMap.of() : ImmutableMap.copyOf(context);
     this.description = description;
+    this.runId = runId;
   }
 
   public Optional<URL> getLocation() {
@@ -60,6 +63,10 @@ public Optional<String> getDescription() {
     return Optional.ofNullable(description);
   }
 
+  public Optional<String> getRunId() {
+    return Optional.ofNullable(runId);
+  }
+
   public String toJson() {
     return Utils.toJson(this);
   }
@@ -75,6 +82,7 @@ public static final class Builder {
     @Nullable private URL location;
     @Nullable private String description;
     @Nullable Map<String, String> context;
+    @Nullable String runId;
 
     private Builder() {
       this.inputs = ImmutableSet.of();
@@ -137,8 +145,13 @@ public Builder description(@Nullable String description) {
       return this;
     }
 
+    public Builder runId(@Nullable String runId) {
+      this.runId = runId;
+      return this;
+    }
+
     public JobMeta build() {
-      return new JobMeta(type, inputs, outputs, location, context, description);
+      return new JobMeta(type, inputs, outputs, location, context, description, runId);
     }
   }
 }

src/test/java/marquez/client/models/JsonGenerator.java

@@ -190,6 +190,7 @@ public static String newJsonFor(final JobMeta meta) {
     obj.put("location", meta.getLocation().map(URL::toString).orElse(null));
     obj.set("context", context);
     obj.put("description", meta.getDescription().orElse(null));
+    obj.put("runId", meta.getRunId().orElse(null));
 
     return obj.toString();
   }