s3_upload_manager.cpp

Go to the documentation of this file.

/*
 * Copyright 2019 Amazon.com, Inc. or its affiliates. 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.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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.
 */

#include <boost/function.hpp>
#include <string>

#include <aws/core/Aws.h>
#include <aws/core/utils/logging/LogMacros.h>

#include <s3_common/s3_common_error_codes.h>
#include <s3_common/s3_facade.h>
#include <s3_common/s3_upload_manager.h>


namespace Aws
{
namespace S3
{

S3UploadManager::S3UploadManager(const bool enable_encryption):
  S3UploadManager(std::make_unique<S3Facade>(enable_encryption))
{
}

S3UploadManager::S3UploadManager(const bool enable_encryption, const Aws::Client::ClientConfiguration & config):
  S3UploadManager(std::make_unique<S3Facade>(enable_encryption, config))
{
}

S3UploadManager::S3UploadManager(std::unique_ptr<S3Facade> s3_facade):
  manager_status_(S3UploadManagerState::AVAILABLE),
  s3_facade_(std::move(s3_facade))
{
}

void S3UploadManager::CancelUpload()
{
  std::lock_guard<std::recursive_mutex> lock(mutex_);
  if (!IsAvailable()) {
    manager_status_ = S3UploadManagerState::CANCELLING;
  }
}

Model::PutObjectOutcome S3UploadManager::UploadFiles(
    const std::vector<UploadDescription> & upload_descriptions,
    const std::string & bucket,
    const boost::function<void (const std::vector<UploadDescription>&)>& feedback_callback)
{
  {
    std::lock_guard<std::recursive_mutex> lock(mutex_);
    if (!IsAvailable()) {
      return Model::PutObjectOutcome(Aws::Client::AWSError<S3Errors>(S3Errors::INVALID_ACTION,
                     "INVALID_ACTION", "UploadFiles aborted. UploadFiles request already active.", false));
    }
    manager_status_ = S3UploadManagerState::UPLOADING;
  }
  std::vector<UploadDescription> completed_uploads;
  // If no files were provided then upload was successful.
  Model::PutObjectResult default_result;
  Model::PutObjectOutcome upload_outcome(default_result);
  std::vector<UploadDescription> uploaded_files;
  for (const auto& upload_description: upload_descriptions) {
    {
      std::lock_guard<std::recursive_mutex> lock(mutex_);
      if(manager_status_ == S3UploadManagerState::CANCELLING) {
        break;
      }
    }
    auto file_path = upload_description.file_path;
    auto object_key = upload_description.object_key;
    //bucket comes from config
    AWS_LOGSTREAM_INFO(__func__,"Uploading file " << file_path << " to " << object_key);
    upload_outcome = s3_facade_->PutObject(file_path, bucket, object_key);
    if (!upload_outcome.IsSuccess()) {
      break;
    }
    completed_uploads.push_back(upload_description);
    feedback_callback(completed_uploads);
  }
  {
    std::lock_guard<std::recursive_mutex> lock(mutex_);
    manager_status_ = S3UploadManagerState::AVAILABLE;
  }
  return upload_outcome;
}

bool S3UploadManager::IsAvailable() const
{
  std::lock_guard<std::recursive_mutex> lock(mutex_);
  return manager_status_ == S3UploadManagerState::AVAILABLE;
}

}  // namespace S3
}  // namespace Aws