databuild/examples/basic_graph/SumConfigure.java
2025-04-17 17:27:55 -07:00

31 lines
No EOL
1.1 KiB
Java

package com.databuild.examples.basic_graph;
import java.util.ArrayList;
import java.util.Arrays;
/**
* Configure class for generating a random number.
* This class creates a job configuration for generating a random number based on the partition ref.
*/
public class SumConfigure {
public static void main(String[] args) {
if (args.length != 1) {
System.err.println("Error: Must provide exactly one partition ref as an argument");
System.exit(1);
}
String partitionRef = args[0];
String[] upstreams = partitionRef.split("_");
// Create a list of quoted upstream values
ArrayList<String> quotedUpstreams = new ArrayList<>();
Arrays.stream(upstreams).forEach(s -> quotedUpstreams.add("\"" + s + "\""));
// Create a job config for generating a random number
String config = String.format(
"{\"outputs\":[\"%s\"],\"inputs\":[],\"args\":[%s],\"env\":{}}",
partitionRef, String.join(",", quotedUpstreams)
);
System.out.println(config);
}
}