BatchJobHelperImpl.java

// Copyright 2017 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.api.ads.adwords.jaxws.utils.v201702.batchjob;

import com.google.api.ads.adwords.jaxws.v201702.cm.ApiError;
import com.google.api.ads.adwords.jaxws.v201702.cm.BatchJob;
import com.google.api.ads.adwords.jaxws.v201702.cm.Operand;
import com.google.api.ads.adwords.jaxws.v201702.cm.Operation;
import com.google.api.ads.adwords.lib.utils.BatchJobException;
import com.google.api.ads.adwords.lib.utils.BatchJobHelperInterface;
import com.google.api.ads.adwords.lib.utils.BatchJobUploadResponse;
import com.google.api.ads.adwords.lib.utils.BatchJobUploadStatus;
import com.google.api.ads.adwords.lib.utils.BatchJobUploader;
import com.google.api.ads.adwords.lib.utils.logging.BatchJobLogger;
import com.google.api.ads.common.lib.soap.jaxb.JaxBDeserializer;
import com.google.inject.Inject;
import java.io.IOException;
import java.net.URI;
import java.net.URL;
import javax.xml.transform.stream.StreamSource;

/** Utility for uploading operations and downloading results for a {@link BatchJob}. */
class BatchJobHelperImpl
    implements BatchJobHelperInterface<
        Operation, Operand, ApiError, MutateResult, BatchJobMutateResponse> {
  private final BatchJobUploader uploader;
  private final BatchJobLogger batchJobLogger;

  @Inject
  BatchJobHelperImpl(BatchJobUploader uploader, BatchJobLogger batchJobLogger) {
    this.uploader = uploader;
    this.batchJobLogger = batchJobLogger;
  }

  @Override
  public BatchJobUploadResponse uploadBatchJobOperations(
      Iterable<Operation> operations, String uploadUrl) throws BatchJobException {
    // All uploads must go through the incremental upload workflow.
    return uploadIncrementalBatchJobOperations(
        operations, true, new BatchJobUploadStatus(0, URI.create(uploadUrl)));
  }

  @Override
  public BatchJobMutateResponse downloadBatchJobMutateResponse(String downloadUrl)
      throws BatchJobException {
    JaxBDeserializer<BatchJobMutateResponse> deserializer =
        new JaxBDeserializer<BatchJobMutateResponse>(BatchJobMutateResponse.class);
    MutateResult[] mutateResults;
    try {
      mutateResults =
          deserializer
              .deserialize(new StreamSource(new URL(downloadUrl).openStream()))
              .getMutateResults();
    } catch (IOException e) {
      batchJobLogger.logDownload(downloadUrl, null, e);
      throw new BatchJobException(
          "Failed to download batch job mutate response from URL: " + downloadUrl, e);
    }

    BatchJobMutateResponse response = new BatchJobMutateResponse();
    response.setMutateResults(mutateResults);

    batchJobLogger.logDownload(downloadUrl, response, null);
    return response;
  }

  @Override
  public BatchJobUploadResponse uploadIncrementalBatchJobOperations(
      Iterable<? extends Operation> operations,
      boolean isLastRequest,
      BatchJobUploadStatus batchJobUploadStatus)
      throws BatchJobException {
    BatchJobMutateRequest request = new BatchJobMutateRequest();
    request.addOperations(operations);
    return uploader.uploadIncrementalBatchJobOperations(
        request, isLastRequest, batchJobUploadStatus);
  }
}