Import code from internal repository to git

java/com/google/domain/registry/tools/BigqueryCommand.java

@@ -0,0 +1,46 @@
+// 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.tools;
+
+import com.google.domain.registry.bigquery.BigqueryConnection;
+
+import com.beust.jcommander.ParametersDelegate;
+
+/** A {@link Command} that uses the bigquery client API. */
+abstract class BigqueryCommand implements Command {
+
+  /** Parameter delegate for encapsulating flags needed to set up the {@link BigqueryConnection}. */
+  @ParametersDelegate
+  private BigqueryParameters bigqueryParameters = new BigqueryParameters();
+
+  /** Connection object for interacting with the Bigquery API. */
+  private BigqueryConnection bigquery;
+
+  @Override
+  public void run() throws Exception {
+    try (BigqueryConnection autoClosingBigquery = bigqueryParameters.newConnection()) {
+      bigquery = autoClosingBigquery;
+      runWithBigquery();
+    }
+  }
+
+  /** Returns the {@link BigqueryConnection} object that has been initialized for use. */
+  BigqueryConnection bigquery() {
+    return bigquery;
+  }
+
+  /** Subclasses must override this to define command behavior. */
+  abstract void runWithBigquery() throws Exception;
+}