Add support for Compute's operations

gcloud-java-compute/src/main/java/com/google/gcloud/compute/ComputeImpl.java

@@ -35,6 +35,25 @@
 
 final class ComputeImpl extends BaseService<ComputeOptions> implements Compute {
 
+  private static class GlobalOperationPageFetcher implements NextPageFetcher<Operation> {
+
+    private static final long serialVersionUID = -2488912172182315364L;
+    private final Map<ComputeRpc.Option, ?> requestOptions;
+    private final ComputeOptions serviceOptions;
+
+    GlobalOperationPageFetcher(ComputeOptions serviceOptions, String cursor,
+        Map<ComputeRpc.Option, ?> optionMap) {
+      this.requestOptions =
+          PageImpl.nextRequestOptions(ComputeRpc.Option.PAGE_TOKEN, cursor, optionMap);
+      this.serviceOptions = serviceOptions;
+    }
+
+    @Override
+    public Page<Operation> nextPage() {
+      return listGlobalOperations(serviceOptions, requestOptions);
+    }
+  }
+
   private static class DiskTypePageFetcher implements NextPageFetcher<DiskType> {
 
     private static final long serialVersionUID = -5253916264932522976L;
@@ -153,6 +172,48 @@ public Page<Zone> nextPage() {
     }
   }
 
+  private static class RegionOperationPageFetcher implements NextPageFetcher<Operation> {
+
+    private static final long serialVersionUID = 4111705358926164078L;
+    private final Map<ComputeRpc.Option, ?> requestOptions;
+    private final ComputeOptions serviceOptions;
+    private final String region;
+
+    RegionOperationPageFetcher(String region, ComputeOptions serviceOptions, String cursor,
+        Map<ComputeRpc.Option, ?> optionMap) {
+      this.requestOptions =
+          PageImpl.nextRequestOptions(ComputeRpc.Option.PAGE_TOKEN, cursor, optionMap);
+      this.serviceOptions = serviceOptions;
+      this.region = region;
+    }
+
+    @Override
+    public Page<Operation> nextPage() {
+      return listRegionOperations(region, serviceOptions, requestOptions);
+    }
+  }
+
+  private static class ZoneOperationPageFetcher implements NextPageFetcher<Operation> {
+
+    private static final long serialVersionUID = -9012504536518197793L;
+    private final Map<ComputeRpc.Option, ?> requestOptions;
+    private final ComputeOptions serviceOptions;
+    private final String zone;
+
+    ZoneOperationPageFetcher(String zone, ComputeOptions serviceOptions, String cursor,
+        Map<ComputeRpc.Option, ?> optionMap) {
+      this.requestOptions =
+          PageImpl.nextRequestOptions(ComputeRpc.Option.PAGE_TOKEN, cursor, optionMap);
+      this.serviceOptions = serviceOptions;
+      this.zone = zone;
+    }
+
+    @Override
+    public Page<Operation> nextPage() {
+      return listZoneOperations(zone, serviceOptions, requestOptions);
+    }
+  }
+
   private final ComputeRpc computeRpc;
 
   ComputeImpl(ComputeOptions options) {
@@ -464,6 +525,159 @@ public com.google.api.services.compute.model.License call() {
     }
   }
 
+  @Override
+  public Operation get(final OperationId operationId, OperationOption... options) {
+    final Map<ComputeRpc.Option, ?> optionsMap = optionMap(options);
+    try {
+      com.google.api.services.compute.model.Operation answer =
+          runWithRetries(new Callable<com.google.api.services.compute.model.Operation>() {
+            @Override
+            public com.google.api.services.compute.model.Operation call() {
+              if (operationId instanceof RegionOperationId) {
+                RegionOperationId regionOperationId = (RegionOperationId) operationId;
+                return computeRpc.getRegionOperation(regionOperationId.region(),
+                    regionOperationId.operation(), optionsMap);
+              } else if (operationId instanceof ZoneOperationId) {
+                ZoneOperationId zoneOperationId = (ZoneOperationId) operationId;
+                return computeRpc.getZoneOperation(zoneOperationId.zone(),
+                    zoneOperationId.operation(), optionsMap);
+              } else {
+                return computeRpc.getGlobalOperation(operationId.operation(), optionsMap);
+              }
+            }
+          }, options().retryParams(), EXCEPTION_HANDLER);
+      return answer == null ? null : Operation.fromPb(this, answer);
+    } catch (RetryHelper.RetryHelperException e) {
+      throw ComputeException.translateAndThrow(e);
+    }
+  }
+
+  @Override
+  public Page<Operation> listGlobalOperations(OperationListOption... options) {
+    return listGlobalOperations(options(), optionMap(options));
+  }
+
+  private static Page<Operation> listGlobalOperations(final ComputeOptions serviceOptions,
+      final Map<ComputeRpc.Option, ?> optionsMap) {
+    try {
+      ComputeRpc.Tuple<String, Iterable<com.google.api.services.compute.model.Operation>> result =
+          runWithRetries(new Callable<ComputeRpc.Tuple<String,
+              Iterable<com.google.api.services.compute.model.Operation>>>() {
+            @Override
+            public ComputeRpc.Tuple<String,
+                Iterable<com.google.api.services.compute.model.Operation>> call() {
+              return serviceOptions.rpc().listGlobalOperations(optionsMap);
+            }
+          }, serviceOptions.retryParams(), EXCEPTION_HANDLER);
+      String cursor = result.x();
+      Iterable<Operation> operations = Iterables.transform(
+          result.y() == null ? ImmutableList.<com.google.api.services.compute.model.Operation>of()
+              : result.y(),
+          new Function<com.google.api.services.compute.model.Operation, Operation>() {
+            @Override
+            public Operation apply(com.google.api.services.compute.model.Operation operation) {
+              return Operation.fromPb(serviceOptions.service(), operation);
+            }
+          });
+      return new PageImpl<>(new GlobalOperationPageFetcher(serviceOptions, cursor, optionsMap),
+          cursor, operations);
+    } catch (RetryHelper.RetryHelperException e) {
+      throw ComputeException.translateAndThrow(e);
+    }
+  }
+
+  @Override
+  public Page<Operation> listRegionOperations(String region, OperationListOption... options) {
+    return listRegionOperations(region, options(), optionMap(options));
+  }
+
+  private static Page<Operation> listRegionOperations(final String region,
+      final ComputeOptions serviceOptions, final Map<ComputeRpc.Option, ?> optionsMap) {
+    try {
+      ComputeRpc.Tuple<String, Iterable<com.google.api.services.compute.model.Operation>> result =
+          runWithRetries(new Callable<ComputeRpc.Tuple<String,
+              Iterable<com.google.api.services.compute.model.Operation>>>() {
+            @Override
+            public ComputeRpc.Tuple<String,
+                Iterable<com.google.api.services.compute.model.Operation>> call() {
+              return serviceOptions.rpc().listRegionOperations(region, optionsMap);
+            }
+          }, serviceOptions.retryParams(), EXCEPTION_HANDLER);
+      String cursor = result.x();
+      Iterable<Operation> operations = Iterables.transform(
+          result.y() == null ? ImmutableList.<com.google.api.services.compute.model.Operation>of()
+              : result.y(),
+          new Function<com.google.api.services.compute.model.Operation, Operation>() {
+            @Override
+            public Operation apply(com.google.api.services.compute.model.Operation operation) {
+              return Operation.fromPb(serviceOptions.service(), operation);
+            }
+          });
+      return new PageImpl<>(new RegionOperationPageFetcher(region, serviceOptions, cursor,
+          optionsMap), cursor, operations);
+    } catch (RetryHelper.RetryHelperException e) {
+      throw ComputeException.translateAndThrow(e);
+    }
+  }
+
+  @Override
+  public Page<Operation> listZoneOperations(String zone, OperationListOption... options) {
+    return listZoneOperations(zone, options(), optionMap(options));
+  }
+
+  private static Page<Operation> listZoneOperations(final String zone,
+      final ComputeOptions serviceOptions, final Map<ComputeRpc.Option, ?> optionsMap) {
+    try {
+      ComputeRpc.Tuple<String, Iterable<com.google.api.services.compute.model.Operation>> result =
+          runWithRetries(new Callable<ComputeRpc.Tuple<String,
+              Iterable<com.google.api.services.compute.model.Operation>>>() {
+            @Override
+            public ComputeRpc.Tuple<String,
+                Iterable<com.google.api.services.compute.model.Operation>> call() {
+              return serviceOptions.rpc().listZoneOperations(zone, optionsMap);
+            }
+          }, serviceOptions.retryParams(), EXCEPTION_HANDLER);
+      String cursor = result.x();
+      Iterable<Operation> operations = Iterables.transform(
+          result.y() == null ? ImmutableList.<com.google.api.services.compute.model.Operation>of()
+              : result.y(),
+          new Function<com.google.api.services.compute.model.Operation, Operation>() {
+            @Override
+            public Operation apply(com.google.api.services.compute.model.Operation operation) {
+              return Operation.fromPb(serviceOptions.service(), operation);
+            }
+          });
+      return new PageImpl<>(new ZoneOperationPageFetcher(zone, serviceOptions, cursor, optionsMap),
+          cursor, operations);
+    } catch (RetryHelper.RetryHelperException e) {
+      throw ComputeException.translateAndThrow(e);
+    }
+  }
+
+  @Override
+  public boolean delete(final OperationId operation) {
+    try {
+      return runWithRetries(new Callable<Boolean>() {
+        @Override
+        public Boolean call() {
+          if (operation instanceof RegionOperationId) {
+            RegionOperationId regionOperationId = (RegionOperationId) operation;
+            return computeRpc.deleteRegionOperation(regionOperationId.region(),
+                regionOperationId.operation());
+          } else if (operation instanceof ZoneOperationId) {
+            ZoneOperationId zoneOperationId = (ZoneOperationId) operation;
+            return computeRpc.deleteZoneOperation(zoneOperationId.zone(),
+                zoneOperationId.operation());
+          } else {
+            return computeRpc.deleteGlobalOperation(operation.operation());
+          }
+        }
+      }, options().retryParams(), EXCEPTION_HANDLER);
+    } catch (RetryHelper.RetryHelperException e) {
+      throw ComputeException.translateAndThrow(e);
+    }
+  }
+
   private Map<ComputeRpc.Option, ?> optionMap(Option... options) {
     Map<ComputeRpc.Option, Object> optionMap = Maps.newEnumMap(ComputeRpc.Option.class);
     for (Option option : options) {

gcloud-java-compute/src/main/java/com/google/gcloud/compute/GlobalOperationId.java

@@ -0,0 +1,107 @@
+/*
+ * 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.gcloud.compute;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+
+import com.google.common.base.MoreObjects;
+
+import java.util.Objects;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+/**
+ * Identity for a Google Compute Engine global operation.
+ */
+public class GlobalOperationId extends ResourceId implements OperationId {
+
+  private static final String REGEX = ResourceId.REGEX + "global/operations/([^/]+)";
+  private static final Pattern PATTERN = Pattern.compile(REGEX);
+  private static final long serialVersionUID = 3945756772641577962L;
+
+  private final String operation;
+
+  private GlobalOperationId(String project, String operation) {
+    super(project);
+    this.operation = checkNotNull(operation);
+  }
+
+  @Override
+  public String operation() {
+    return operation;
+  }
+
+  @Override
+  public String selfLink() {
+    return super.selfLink() + "/global/operations/" + operation;
+  }
+
+  @Override
+  MoreObjects.ToStringHelper toStringHelper() {
+    return super.toStringHelper().add("operation", operation);
+  }
+
+  @Override
+  public int hashCode() {
+    return Objects.hash(baseHashCode(), operation);
+  }
+
+  @Override
+  public boolean equals(Object obj) {
+    return obj instanceof GlobalOperationId
+        && baseEquals((GlobalOperationId) obj)
+        && Objects.equals(operation, ((GlobalOperationId) obj).operation);
+  }
+
+  @Override
+  GlobalOperationId setProjectId(String projectId) {
+    if (project() != null) {
+      return this;
+    }
+    return GlobalOperationId.of(projectId, operation);
+  }
+
+  /**
+   * Returns a global operation identity given the operation name.
+   */
+  public static GlobalOperationId of(String operation) {
+    return new GlobalOperationId(null, operation);
+  }
+
+  /**
+   * Returns a global operation identity given project and operation names.
+   */
+  public static GlobalOperationId of(String project, String operation) {
+    return new GlobalOperationId(project, operation);
+  }
+
+  /**
+   * Returns {@code true} if the provided string matches the expected format of a global operation
+   * URL. Returns {@code false} otherwise.
+   */
+  static boolean matchesUrl(String url) {
+    return PATTERN.matcher(url).matches();
+  }
+
+  static GlobalOperationId fromUrl(String url) {
+    Matcher matcher = PATTERN.matcher(url);
+    if (!matcher.matches()) {
+      throw new IllegalArgumentException(url + " is not a valid global operation URL");
+    }
+    return GlobalOperationId.of(matcher.group(1), matcher.group(2));
+  }
+}