com.google.photos.library.v1.upload.PhotosLibraryUploadUnaryCallable Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of google-photos-library-client Show documentation
Google Photos Library API Client Library for Java
The newest version!
/*
 * Copyright 2018 Google LLC
 *
 * 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
 *
 *     https://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.photos.library.v1.upload;

import com.google.api.core.ApiFuture;
import com.google.api.core.ApiFutures;
import com.google.api.gax.rpc.ApiCallContext;
import com.google.api.gax.rpc.ClientContext;
import com.google.api.gax.rpc.StatusCode;
import com.google.api.gax.rpc.UnaryCallSettings;
import com.google.api.gax.rpc.UnaryCallable;
import com.google.common.util.concurrent.ListenableFutureTask;
import java.util.Set;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

final class PhotosLibraryUploadUnaryCallable
    extends UnaryCallable {

  private final ClientContext clientContext;
  private final UnaryCallSettings uploadSettings;
  private final Set retryableCodes;

  PhotosLibraryUploadUnaryCallable(
      ClientContext clientContext,
      UnaryCallSettings uploadSettings,
      Set retryableCodes) {
    this.clientContext = clientContext;
    this.uploadSettings = uploadSettings;
    this.retryableCodes = retryableCodes;
  }

  @Override
  public ApiFuture futureCall(
      UploadMediaItemRequest request, ApiCallContext unusedContext) {
    PhotosLibraryUploadCallable uploadCallable =
        new PhotosLibraryUploadCallable(request, clientContext, uploadSettings);

    // Map exceptions from the PhotosLibraryUploadCallable to an APIException with a retry URL if
    // possible.
    // This also determines if this request can be retried based on the list of retryableCodes.
    return ApiFutures.catching(
        PhotosLibraryUploadApiFuture.create(uploadCallable, clientContext),
        Throwable.class,
        new PhotosLibraryUploadExceptionMappingFn(
            retryableCodes, uploadCallable.getAtomicResumeUrl()),
        clientContext.getExecutor());
  }

  private static final class PhotosLibraryUploadApiFuture
      implements ApiFuture {

    private final ListenableFutureTask futureTask;

    private PhotosLibraryUploadApiFuture(ListenableFutureTask futureTask) {
      this.futureTask = futureTask;
    }

    public static final PhotosLibraryUploadApiFuture create(
        PhotosLibraryUploadCallable uploadCallable, ClientContext clientContext) {
      PhotosLibraryUploadApiFuture future =
          new PhotosLibraryUploadApiFuture(ListenableFutureTask.create(uploadCallable));
      clientContext.getExecutor().execute(future.futureTask);
      return future;
    }

    @Override
    public void addListener(Runnable listener, Executor executor) {
      futureTask.addListener(listener, executor);
    }

    @Override
    public boolean cancel(boolean b) {
      return futureTask.cancel(b);
    }

    @Override
    public boolean isCancelled() {
      return futureTask.isCancelled();
    }

    @Override
    public boolean isDone() {
      return futureTask.isDone();
    }

    @Override
    public UploadMediaItemResponse get() throws InterruptedException, ExecutionException {
      return futureTask.get();
    }

    @Override
    public UploadMediaItemResponse get(long l, TimeUnit timeUnit)
        throws InterruptedException, ExecutionException, TimeoutException {
      return futureTask.get(l, timeUnit);
    }
  }
}