Default to THROUGHPUT_BASED autoscaling for classic templates (Google…

…CloudPlatform#1403)

* single commit off of master

* No-op comment change to kick copybara

* Revert "No-op comment change to kick copybara"

This reverts commit 928dc6b.

* Different no-op to kick copybara and avoid conflicts

* One more try to get copybara to merge

plugins/templates-maven-plugin/src/main/java/com/google/cloud/teleport/plugin/maven/TemplatesStageMojo.java

@@ -258,6 +258,11 @@ protected String stageClassicTemplate(
     arguments.add(element("argument", "--templateLocation=" + templatePath));
     arguments.add(element("argument", "--project=" + projectId));
     arguments.add(element("argument", "--region=" + region));
+    if (imageSpec.getMetadata().isStreaming()) {
+      // Default to THROUGHPUT_BASED autoscaling for streaming classic templates
+      arguments.add(element("argument", "--autoscalingAlgorithm=THROUGHPUT_BASED"));
+      arguments.add(element("argument", "--maxNumWorkers=5"));
+    }
 
     if (gcpTempLocation != null) {
       String gcpTempLocationPath = "gs://" + bucketNameOnly(gcpTempLocation) + "/temp/";

v1/src/test/java/com/google/cloud/teleport/templates/PubSubToSplunkIT.java

@@ -254,7 +254,8 @@ public void testPubSubToSplunkUsingAtleastOnceMode() throws IOException {
             .addParameter("token", splunkResourceManager.getHecToken())
             .addParameter("batchCount", "1")
             .addEnvironment("additionalExperiments", experiments)
-            .addEnvironment("enableStreamingEngine", true);
+            .addEnvironment("enableStreamingEngine", true)
+            .addEnvironment("maxWorkers", 10);
     testPubSubToSplunkMain(parameters, false);
   }
 }