Refactor metrics servlet to be a Daggerized action

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=120235694

java/com/google/domain/registry/env/common/backend/WEB-INF/web.xml

@@ -13,13 +13,8 @@
     <load-on-startup>1</load-on-startup>
   </servlet>
 
-  <!-- Whitebox Metrics servlet. -->
-  <servlet>
-    <servlet-name>metrics</servlet-name>
-    <servlet-class>com.google.domain.registry.monitoring.whitebox.MetricsTaskServlet</servlet-class>
-  </servlet>
   <servlet-mapping>
-    <servlet-name>metrics</servlet-name>
+    <servlet-name>backend-servlet</servlet-name>
     <url-pattern>/_dr/task/metrics</url-pattern>
   </servlet-mapping>
 

java/com/google/domain/registry/module/backend/BackendRequestComponent.java

@@ -40,7 +40,9 @@ import com.google.domain.registry.flows.async.DeleteContactResourceAction;
 import com.google.domain.registry.flows.async.DeleteHostResourceAction;
 import com.google.domain.registry.flows.async.DnsRefreshForHostRenameAction;
 import com.google.domain.registry.mapreduce.MapreduceModule;
+import com.google.domain.registry.monitoring.whitebox.MetricsExportAction;
 import com.google.domain.registry.monitoring.whitebox.VerifyEntityIntegrityAction;
+import com.google.domain.registry.monitoring.whitebox.WhiteboxModule;
 import com.google.domain.registry.rde.BrdaCopyAction;
 import com.google.domain.registry.rde.RdeModule;
 import com.google.domain.registry.rde.RdeReportAction;
@@ -73,6 +75,7 @@ import dagger.Subcomponent;
         RequestModule.class,
         SheetModule.class,
         TmchModule.class,
+        WhiteboxModule.class,
     })
 interface BackendRequestComponent {
   BigqueryPollJobAction bigqueryPollJobAction();
@@ -86,6 +89,7 @@ interface BackendRequestComponent {
   ExportCommitLogDiffAction exportCommitLogDiffAction();
   ExportDomainListsAction exportDomainListsAction();
   ExportReservedTermsAction exportReservedTermsAction();
+  MetricsExportAction metricsExportAction();
   NordnUploadAction nordnUploadAction();
   NordnVerifyAction nordnVerifyAction();
   PublishDnsUpdatesAction publishDnsUpdatesAction();

java/com/google/domain/registry/monitoring/whitebox/Metrics.java

@@ -66,7 +66,7 @@ public abstract class Metrics {
   public void export() {
     try {
       String hostname = modulesService.getVersionHostname("backend", null);
-      TaskOptions opts = withUrl(MetricsTaskServlet.PATH)
+      TaskOptions opts = withUrl(MetricsExportAction.PATH)
           .header("Host", hostname)
           .param("insertId", idGenerator.get())
           .param("startTime", toBigqueryTimestamp(startTimeMillis, TimeUnit.MILLISECONDS))

java/com/google/domain/registry/monitoring/whitebox/MetricsExportAction.java

@@ -14,7 +14,11 @@
 
 package com.google.domain.registry.monitoring.whitebox;
 
-import static com.google.domain.registry.util.HttpServletUtils.getRequiredParameterValue;
+import static com.google.common.base.Predicates.in;
+import static com.google.common.base.Predicates.not;
+import static com.google.common.collect.Multimaps.filterKeys;
+import static com.google.domain.registry.request.Action.Method.POST;
+import static com.google.domain.registry.util.FormattingLogger.getLoggerForCallerClass;
 
 import com.google.api.services.bigquery.Bigquery;
 import com.google.api.services.bigquery.model.TableDataInsertAllRequest;
@@ -24,69 +28,58 @@ import com.google.common.base.Function;
 import com.google.common.base.Joiner;
 import com.google.common.collect.FluentIterable;
 import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableListMultimap;
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
-import com.google.common.collect.Sets;
 import com.google.domain.registry.bigquery.BigqueryFactory;
-import com.google.domain.registry.config.RegistryEnvironment;
+import com.google.domain.registry.config.ConfigModule.Config;
+import com.google.domain.registry.request.Action;
+import com.google.domain.registry.request.Parameter;
+import com.google.domain.registry.request.ParameterMap;
 import com.google.domain.registry.util.FormattingLogger;
-import com.google.domain.registry.util.NonFinalForTesting;
 
 import java.io.IOException;
+import java.util.Map;
 import java.util.Set;
 
-import javax.servlet.http.HttpServlet;
-import javax.servlet.http.HttpServletRequest;
-import javax.servlet.http.HttpServletResponse;
+import javax.inject.Inject;
 
-/** Servlet for exporting metrics to BigQuery. */
-public class MetricsTaskServlet extends HttpServlet {
+/** Action for exporting metrics to BigQuery. */
+@Action(path = MetricsExportAction.PATH, method = POST)
+public class MetricsExportAction implements Runnable {
 
   public static final String PATH = "/_dr/task/metrics";
-
-  private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();
+  private static final FormattingLogger logger = getLoggerForCallerClass();
   private static final String DATASET_ID = "metrics";
-  private static final String PROJECT_ID = RegistryEnvironment.get().config().getProjectId();
-
   private static final Set<String> SPECIAL_PARAMS = ImmutableSet.of("tableId", "insertId");
 
-  @NonFinalForTesting
-  private static BigqueryFactory bigqueryFactory = new BigqueryFactory();
-
-  /** Returns a filtered {@link ImmutableMap} from an {@link HttpServletRequest} */
-  private static ImmutableMap<String, Object> getFilteredMapFromRequest(
-      HttpServletRequest req,
-      Set<String> filter) {
-    ImmutableMap.Builder<String, Object> b = new ImmutableMap.Builder<>();
-
-    @SuppressWarnings({"cast", "unchecked"})  // Return type is always a Set<String>.
-    Set<String> parameterKeys = req.getParameterMap().keySet();
-
-    for (String key : Sets.difference(parameterKeys, filter)) {
-      b.put(key, req.getParameter(key));
-    }
-
-    return b.build();
-  }
+  @Inject @Parameter("tableId") String tableId;
+  @Inject @Parameter("insertId") String insertId;
+  @Inject @Config("projectId") String projectId;
+  @Inject BigqueryFactory bigqueryFactory;
+  @Inject @ParameterMap ImmutableListMultimap<String, String> parameters;
+  @Inject MetricsExportAction() {}
 
   /** Exports metrics to BigQuery. */
   @Override
-  public void doPost(HttpServletRequest req, HttpServletResponse rsp) throws IOException {
+  public void run() {
     try {
-      final String tableId = getRequiredParameterValue(req, "tableId");
-      ImmutableMap<String, Object> fields = getFilteredMapFromRequest(req, SPECIAL_PARAMS);
-      Bigquery bigquery = bigqueryFactory.create(PROJECT_ID, DATASET_ID, tableId);
-
+      Bigquery bigquery = bigqueryFactory.create(projectId, DATASET_ID, tableId);
+      // Filter out the special parameters that the Action is called with.  Everything that's left
+      // is returned in a Map that is suitable to pass to Bigquery as row data.
+      Map<String, Object> jsonRows =
+          ImmutableMap.<String, Object>copyOf(
+              filterKeys(parameters, not(in(SPECIAL_PARAMS))).entries());
       TableDataInsertAllResponse response = bigquery.tabledata()
           .insertAll(
-              PROJECT_ID,
+              projectId,
               DATASET_ID,
               tableId,
               new TableDataInsertAllRequest()
                   .setRows(
                       ImmutableList.of(new TableDataInsertAllRequest.Rows()
-                          .setInsertId(req.getParameter("insertId"))
-                          .setJson(fields))))
+                          .setInsertId(insertId)
+                          .setJson(jsonRows))))
           .execute();
 
       if (response.getInsertErrors() != null && !response.getInsertErrors().isEmpty()) {

java/com/google/domain/registry/monitoring/whitebox/WhiteboxComponent.java

@@ -17,6 +17,7 @@ package com.google.domain.registry.monitoring.whitebox;
 import com.google.domain.registry.bigquery.BigqueryModule;
 import com.google.domain.registry.config.ConfigModule;
 import com.google.domain.registry.request.Modules.DatastoreServiceModule;
+import com.google.domain.registry.util.SystemSleeper.SystemSleeperModule;
 
 import dagger.Component;
 
@@ -29,6 +30,7 @@ import javax.inject.Singleton;
         BigqueryModule.class,
         ConfigModule.class,
         DatastoreServiceModule.class,
+        SystemSleeperModule.class,
         WhiteboxModule.class
     })
 interface WhiteboxComponent {

java/com/google/domain/registry/monitoring/whitebox/WhiteboxModule.java

@@ -14,21 +14,36 @@
 
 package com.google.domain.registry.monitoring.whitebox;
 
+import static com.google.domain.registry.request.RequestParameters.extractRequiredParameter;
+
 import com.google.common.base.Supplier;
-import com.google.domain.registry.util.Sleeper;
-import com.google.domain.registry.util.SystemSleeper;
+import com.google.domain.registry.request.Parameter;
 
 import dagger.Module;
 import dagger.Provides;
 
 import java.util.UUID;
 
+import javax.servlet.http.HttpServletRequest;
+
 /**
  * Dagger module for injecting common settings for Whitebox tasks.
  */
 @Module
 public class WhiteboxModule {
 
+  @Provides
+  @Parameter("tableId")
+  static String provideTableId(HttpServletRequest req) {
+    return extractRequiredParameter(req, "tableId");
+  }
+
+  @Provides
+  @Parameter("insertId")
+  static String provideInsertId(HttpServletRequest req) {
+    return extractRequiredParameter(req, "insertId");
+  }
+
   @Provides
   static Supplier<String> provideIdGenerator() {
     return new Supplier<String>() {
@@ -38,9 +53,4 @@ public class WhiteboxModule {
       }
     };
   }
-
-  @Provides
-  static Sleeper provideSleeper(SystemSleeper systemSleeper) {
-    return systemSleeper;
-  }
 }

java/com/google/domain/registry/util/HttpServletUtils.java

@@ -29,7 +29,10 @@ public final class HttpServletUtils {
   /**
    * Returns the value of the given request's first {@code name} parameter, or throws
    * {@code IllegalArgumentException} if the parameter is not present.
+   *
+   * @deprecated in favor of <code>RequestParameters.extractRequiredParameter</code>
    */
+  @Deprecated
   public static String getRequiredParameterValue(HttpServletRequest req, String name) {
     return checkArgumentNotNull(req.getParameter(name), "Missing required parameter: %s", name);
   }