enqueue jobs with pointset IDs when testing

opentripplanner · Jun 8, 2015 · b96c5db · b96c5db
1 parent a59f60e
commit b96c5db
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/main/java/org/opentripplanner/analyst/cluster/JobSimulator.java b/src/main/java/org/opentripplanner/analyst/cluster/JobSimulator.java
@@ -21,12 +21,11 @@
  */
 public class JobSimulator {
 
-    public static final String QUEUE_PREFIX = "analyst-dev_";
-
     public static void main(String[] args) {
 
         String prefix = args[0];
-        String graphId = args[1];
+        String pointSetId = args[1];
+        String graphId = pointSetId.split("_")[0];
         int nOrigins = Integer.parseInt(args[2]);
 
         String jobId = compactUUID();
@@ -40,17 +39,21 @@ public static void main(String[] args) {
         sqs.setRegion(awsRegion);
 
         // Create a queue for this job
-        String queueUrl = sqs.createQueue(String.join("_", prefix, "job", graphId, jobId)).getQueueUrl();
+        String taskQueueUrl = sqs.createQueue(String.join("_", prefix, "job", graphId, jobId)).getQueueUrl();
+        String resultQueueUrl = sqs.createQueue(String.join("_", prefix, "job", graphId, jobId)).getQueueUrl();
 
         ObjectMapper objectMapper = new ObjectMapper();
         IntStream.range(0, nOrigins).forEach(i -> {
             // Enqueue one fake origin
             ProfileRequest profileRequest = new ProfileRequest();
-            AnalystClusterRequest clusterRequest = new OneToManyProfileRequest(null, profileRequest, graphId);
+            AnalystClusterRequest clusterRequest = new OneToManyProfileRequest(pointSetId, profileRequest, graphId);
             clusterRequest.id = Integer.toString(i);
             clusterRequest.jobId = jobId;
+            clusterRequest.outputLocation = prefix + "_output";
+            clusterRequest.outputQueue = resultQueueUrl;
+            clusterRequest.destinationPointsetId = pointSetId;
             try {
-                sqs.sendMessage(queueUrl, objectMapper.writeValueAsString(clusterRequest));
+                sqs.sendMessage(taskQueueUrl, objectMapper.writeValueAsString(clusterRequest));
             } catch (JsonProcessingException e) {
                 e.printStackTrace();
             }