S3ExpressIdentityProvider.h

 
#pragma once
 
#include <aws/core/auth/signer/AWSAuthSignerBase.h>
#include <aws/core/utils/ConcurrentCache.h>
#include <aws/core/utils/memory/stl/AWSSet.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/s3-crt/S3ExpressIdentity.h>
#include <smithy/identity/resolver/AwsIdentityResolverBase.h>
 
#include <condition_variable>
#include <thread>
 
namespace Aws {
namespace Http {
struct ServiceSpecificParameters;
}
 
namespace S3Crt {
class S3CrtClient;
class S3ExpressIdentityProvider : public smithy::IdentityResolverBase<S3ExpressIdentity> {
 public:
  explicit S3ExpressIdentityProvider(const S3CrtClient& s3Client) : m_s3Client(s3Client) {}
 
  virtual S3ExpressIdentity GetS3ExpressIdentity(
      const std::shared_ptr<Aws::Http::ServiceSpecificParameters>& serviceSpecificParameters) = 0;
 
  ResolveIdentityFutureOutcome getIdentity(const IdentityProperties& identityProperties,
                                           const AdditionalParameters& additionalParameters) override;
  S3ExpressIdentity getIdentity(const Aws::String& bucketName);
 
  virtual ~S3ExpressIdentityProvider() {}
 
 protected:
  std::shared_ptr<std::mutex> GetMutexForBucketName(const Aws::String& bucketName);
 
 private:
  const S3CrtClient& m_s3Client;
  mutable std::mutex m_bucketNameMapMutex;
  Aws::Map<Aws::String, std::shared_ptr<std::mutex>> m_bucketNameMutex;
};
 
class DefaultS3ExpressIdentityProvider : public S3ExpressIdentityProvider {
 public:
  explicit DefaultS3ExpressIdentityProvider(const S3CrtClient& m_s3Client);
 
  DefaultS3ExpressIdentityProvider(const S3CrtClient& s3Client,
                                   std::shared_ptr<Utils::ConcurrentCache<Aws::String, S3ExpressIdentity>> credentialsCache);
 
  DefaultS3ExpressIdentityProvider(const DefaultS3ExpressIdentityProvider& other) = delete;
  DefaultS3ExpressIdentityProvider(DefaultS3ExpressIdentityProvider&& other) noexcept = delete;
  DefaultS3ExpressIdentityProvider& operator=(const DefaultS3ExpressIdentityProvider& other) = delete;
  DefaultS3ExpressIdentityProvider& operator=(DefaultS3ExpressIdentityProvider&& other) noexcept = delete;
 
  virtual ~DefaultS3ExpressIdentityProvider() override = default;
 
  S3ExpressIdentity GetS3ExpressIdentity(const std::shared_ptr<Aws::Http::ServiceSpecificParameters>& serviceSpecificParameters) override;
 
 private:
  mutable std::shared_ptr<Aws::Utils::ConcurrentCache<Aws::String, S3ExpressIdentity>> m_credentialsCache;
};
 
class DefaultAsyncS3ExpressIdentityProvider : public S3ExpressIdentityProvider {
 public:
  explicit DefaultAsyncS3ExpressIdentityProvider(const S3CrtClient& m_s3Client,
                                                 std::chrono::minutes refreshPeriod = std::chrono::minutes(1));
 
  DefaultAsyncS3ExpressIdentityProvider(const S3CrtClient& s3Client,
                                        std::shared_ptr<Utils::ConcurrentCache<Aws::String, S3ExpressIdentity>> credentialsCache,
                                        std::chrono::minutes refreshPeriod = std::chrono::minutes(1));
 
  DefaultAsyncS3ExpressIdentityProvider(const DefaultAsyncS3ExpressIdentityProvider& other) = delete;
  DefaultAsyncS3ExpressIdentityProvider(DefaultAsyncS3ExpressIdentityProvider&& other) noexcept = delete;
  DefaultAsyncS3ExpressIdentityProvider& operator=(const DefaultAsyncS3ExpressIdentityProvider& other) = delete;
  DefaultAsyncS3ExpressIdentityProvider& operator=(DefaultAsyncS3ExpressIdentityProvider&& other) noexcept = delete;
 
  virtual ~DefaultAsyncS3ExpressIdentityProvider() override;
 
  S3ExpressIdentity GetS3ExpressIdentity(const std::shared_ptr<Aws::Http::ServiceSpecificParameters>& serviceSpecificParameters) override;
 
 private:
  void refreshIdentities(std::chrono::minutes refreshPeriod);
  void threadSafeKeyInsert(const Aws::String& key);
  bool threadSafeKeyHas(const Aws::String& key);
  void threadSafeKeyEmpty();
 
  mutable std::shared_ptr<Aws::Utils::ConcurrentCache<Aws::String, S3ExpressIdentity>> m_credentialsCache;
  Aws::Set<Aws::String> m_keysUsed;
  mutable std::mutex m_keysUsedMutex;
  mutable bool m_shouldStopBackgroundRefresh;
  Aws::UniquePtr<std::thread> m_backgroundRefreshThread;
  mutable std::mutex m_shutDownMutex;
  mutable std::condition_variable m_shutdownCondition;
};
}  // namespace S3Crt
}  // namespace Aws