storage_switch_to_secondary_policy.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include "azure/storage/common/dll_import_export.hpp"
 
#include <azure/core/http/policies/policy.hpp>
 
#include <memory>
#include <string>
 
namespace Azure { namespace Storage { namespace _internal {
 
  AZ_STORAGE_COMMON_DLLEXPORT extern const Azure::Core::Context::Key SecondaryHostReplicaStatusKey;
 
  inline Azure::Core::Context WithReplicaStatus(const Azure::Core::Context& context)
  {
    return context.WithValue(SecondaryHostReplicaStatusKey, std::make_shared<bool>(true));
  }
 
  class StorageSwitchToSecondaryPolicy final : public Azure::Core::Http::Policies::HttpPolicy {
  public:
    explicit StorageSwitchToSecondaryPolicy(std::string primaryHost, std::string secondaryHost)
        : m_primaryHost(std::move(primaryHost)), m_secondaryHost(std::move(secondaryHost))
    {
    }
 
    std::unique_ptr<Azure::Core::Http::Policies::HttpPolicy> Clone() const override
    {
      return std::make_unique<StorageSwitchToSecondaryPolicy>(*this);
    }
 
    std::unique_ptr<Azure::Core::Http::RawResponse> Send(
        Azure::Core::Http::Request& request,
        Azure::Core::Http::Policies::NextHttpPolicy nextPolicy,
        const Azure::Core::Context& context) const override;
 
  private:
    std::string m_primaryHost;
    std::string m_secondaryHost;
  };
 
}}} // namespace Azure::Storage::_internal