datalake_service_client.hpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
 
#pragma once
 
#include "azure/storage/files/datalake/datalake_options.hpp"
#include "azure/storage/files/datalake/datalake_responses.hpp"
 
#include <azure/core/credentials/credentials.hpp>
#include <azure/core/internal/http/pipeline.hpp>
#include <azure/core/response.hpp>
#include <azure/storage/blobs/blob_service_client.hpp>
#include <azure/storage/common/storage_credential.hpp>
 
#include <memory>
#include <string>
 
namespace Azure { namespace Storage { namespace Files { namespace DataLake {
 
  class DataLakeFileSystemClient;
 
  class DataLakeServiceClient final {
  public:
    static DataLakeServiceClient CreateFromConnectionString(
        const std::string& connectionString,
        const DataLakeClientOptions& options = DataLakeClientOptions());
 
    explicit DataLakeServiceClient(
        const std::string& serviceUrl,
        std::shared_ptr<StorageSharedKeyCredential> credential,
        const DataLakeClientOptions& options = DataLakeClientOptions());
 
    explicit DataLakeServiceClient(
        const std::string& serviceUrl,
        std::shared_ptr<const Core::Credentials::TokenCredential> credential,
        const DataLakeClientOptions& options = DataLakeClientOptions());
 
    explicit DataLakeServiceClient(
        const std::string& serviceUrl,
        const DataLakeClientOptions& options = DataLakeClientOptions());
 
    DataLakeFileSystemClient GetFileSystemClient(const std::string& fileSystemName) const;
 
    std::string GetUrl() const { return m_blobServiceClient.GetUrl(); }
 
    ListFileSystemsPagedResponse ListFileSystems(
        const ListFileSystemsOptions& options = ListFileSystemsOptions(),
        const Azure::Core::Context& context = Azure::Core::Context()) const;
 
    Azure::Response<Models::UserDelegationKey> GetUserDelegationKey(
        const Azure::DateTime& expiresOn,
        const GetUserDelegationKeyOptions& options = GetUserDelegationKeyOptions(),
        const Azure::Core::Context& context = Azure::Core::Context()) const
    {
      return m_blobServiceClient.GetUserDelegationKey(expiresOn, options, context);
    }
 
    Azure::Response<Models::SetServicePropertiesResult> SetProperties(
        Models::DataLakeServiceProperties properties,
        const SetServicePropertiesOptions& options = SetServicePropertiesOptions(),
        const Azure::Core::Context& context = Azure::Core::Context()) const
    {
      return m_blobServiceClient.SetProperties(properties, options, context);
    }
 
    Azure::Response<Models::DataLakeServiceProperties> GetProperties(
        const GetServicePropertiesOptions& options = GetServicePropertiesOptions(),
        const Azure::Core::Context& context = Azure::Core::Context()) const
    {
      return m_blobServiceClient.GetProperties(options, context);
    }
 
  private:
    Azure::Core::Url m_serviceUrl;
    Blobs::BlobServiceClient m_blobServiceClient;
    std::shared_ptr<Azure::Core::Http::_internal::HttpPipeline> m_pipeline;
    _detail::DatalakeClientConfiguration m_clientConfiguration;
  };
}}}} // namespace Azure::Storage::Files::DataLake