share_lease_client.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include "azure/storage/files/shares/share_client.hpp"
#include "azure/storage/files/shares/share_file_client.hpp"
 
#include <chrono>
#include <mutex>
#include <string>
 
namespace Azure { namespace Storage { namespace Files { namespace Shares {
 
  class ShareLeaseClient final {
  public:
    explicit ShareLeaseClient(ShareFileClient fileClient, std::string leaseId)
        : m_fileClient(std::move(fileClient)), m_leaseId(std::move(leaseId))
    {
    }
 
    explicit ShareLeaseClient(ShareClient shareClient, std::string leaseId)
        : m_shareClient(std::move(shareClient)), m_leaseId(std::move(leaseId))
    {
    }
 
    static std::string CreateUniqueLeaseId();
 
    AZ_STORAGE_FILES_SHARES_DLLEXPORT static const std::chrono::seconds InfiniteLeaseDuration;
 
    const std::string& GetLeaseId()
    {
      std::lock_guard<std::mutex> guard(m_mutex);
      return m_leaseId;
    }
 
    Azure::Response<Models::AcquireLeaseResult> Acquire(
        std::chrono::seconds duration,
        const AcquireLeaseOptions& options = AcquireLeaseOptions(),
        const Azure::Core::Context& context = Azure::Core::Context());
 
    Azure::Response<Models::RenewLeaseResult> Renew(
        const RenewLeaseOptions& options = RenewLeaseOptions(),
        const Azure::Core::Context& context = Azure::Core::Context());
 
    Azure::Response<Models::ReleaseLeaseResult> Release(
        const ReleaseLeaseOptions& options = ReleaseLeaseOptions(),
        const Azure::Core::Context& context = Azure::Core::Context());
 
    Azure::Response<Models::ChangeLeaseResult> Change(
        const std::string& proposedLeaseId,
        const ChangeLeaseOptions& options = ChangeLeaseOptions(),
        const Azure::Core::Context& context = Azure::Core::Context());
 
    Azure::Response<Models::BreakLeaseResult> Break(
        const BreakLeaseOptions& options = BreakLeaseOptions(),
        const Azure::Core::Context& context = Azure::Core::Context());
 
  private:
    Azure::Nullable<ShareFileClient> m_fileClient;
    Azure::Nullable<ShareClient> m_shareClient;
    std::mutex m_mutex;
    std::string m_leaseId;
  };
 
}}}} // namespace Azure::Storage::Files::Shares