Import code from internal repository to git

java/com/google/domain/registry/backup/BackupModule.java

@@ -0,0 +1,92 @@
+// Copyright 2016 Google Inc. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.domain.registry.backup;
+
+import static com.google.appengine.api.ThreadManager.currentRequestThreadFactory;
+import static com.google.common.util.concurrent.MoreExecutors.listeningDecorator;
+import static com.google.domain.registry.backup.ExportCommitLogDiffAction.LOWER_CHECKPOINT_TIME_PARAM;
+import static com.google.domain.registry.backup.ExportCommitLogDiffAction.UPPER_CHECKPOINT_TIME_PARAM;
+import static com.google.domain.registry.backup.RestoreCommitLogsAction.FROM_TIME_PARAM;
+import static com.google.domain.registry.request.RequestParameters.extractRequiredDatetimeParameter;
+import static com.google.domain.registry.request.RequestParameters.extractRequiredParameter;
+import static java.util.concurrent.Executors.newFixedThreadPool;
+
+import com.google.common.primitives.Ints;
+import com.google.common.util.concurrent.ListeningExecutorService;
+import com.google.domain.registry.cron.CommitLogFanoutAction;
+import com.google.domain.registry.request.HttpException.BadRequestException;
+import com.google.domain.registry.request.Parameter;
+
+import dagger.Module;
+import dagger.Provides;
+
+import org.joda.time.DateTime;
+
+import java.lang.annotation.Documented;
+
+import javax.inject.Qualifier;
+import javax.servlet.http.HttpServletRequest;
+
+/**
+ * Dagger module for backup package.
+ *
+ * @see "com.google.domain.registry.module.backend.BackendComponent"
+ */
+@Module
+public final class BackupModule {
+
+  /** Dagger qualifier for backups. */
+  @Qualifier
+  @Documented
+  public static @interface Backups {}
+
+  /** Number of threads in the threaded executor. */
+  private static final int NUM_THREADS = 10;
+
+  @Provides
+  @Parameter("bucket")
+  static int provideBucket(HttpServletRequest req) {
+    String param = extractRequiredParameter(req, CommitLogFanoutAction.BUCKET_PARAM);
+    Integer bucket = Ints.tryParse(param);
+    if (bucket == null) {
+      throw new BadRequestException("Bad bucket id");
+    }
+    return bucket;
+  }
+
+  @Provides
+  @Parameter(LOWER_CHECKPOINT_TIME_PARAM)
+  static DateTime provideLowerCheckpointKey(HttpServletRequest req) {
+    return extractRequiredDatetimeParameter(req, LOWER_CHECKPOINT_TIME_PARAM);
+  }
+
+  @Provides
+  @Parameter(UPPER_CHECKPOINT_TIME_PARAM)
+  static DateTime provideUpperCheckpointKey(HttpServletRequest req) {
+    return extractRequiredDatetimeParameter(req, UPPER_CHECKPOINT_TIME_PARAM);
+  }
+
+  @Provides
+  @Parameter(FROM_TIME_PARAM)
+  static DateTime provideFromTime(HttpServletRequest req) {
+    return extractRequiredDatetimeParameter(req, FROM_TIME_PARAM);
+  }
+
+  @Provides
+  @Backups
+  static ListeningExecutorService provideListeningExecutorService() {
+    return listeningDecorator(newFixedThreadPool(NUM_THREADS, currentRequestThreadFactory()));
+  }
+}