Function MovePathWithDataLakeAPI

cpp/src/arrow/filesystem/azurefs.cc:3102–3220

  }

  Status MovePathWithDataLakeAPI(
      const DataLake::DataLakeFileSystemClient& src_adlfs_client,
      const AzureLocation& src, const AzureLocation& dest) {
    DCHECK(!src.container.empty() && !src.path.empty());
    DCHECK(!dest.container.empty() && !dest.path.empty());
    const auto src_path = std::string{internal::RemoveTrailingSlash(src.path)};
    const auto dest_path = std::string{internal::RemoveTrailingSlash(dest.path)};

    // Ensure that src exists and, if path has a trailing slash, that it's a directory.
    ARROW_ASSIGN_OR_RAISE(auto src_lease_client,
                          AcquireBlobLease(src, kLeaseDuration, /*allow_missing=*/false));
    LeaseGuard src_lease_guard{std::move(src_lease_client), kLeaseDuration};
    // It might be necessary to check src is a directory 0-3 times in this function,
    // so we use a lazy evaluation function to avoid redundant calls to GetFileInfo().
    std::optional<bool> src_is_dir_opt{};
    auto src_is_dir_lazy = [&]() -> Result<bool> {
      if (src_is_dir_opt.has_value()) {
        return *src_is_dir_opt;
      }
      ARROW_ASSIGN_OR_RAISE(
          auto src_info, GetFileInfo(src_adlfs_client, src, src_lease_guard.LeaseId()));
      src_is_dir_opt = src_info.type() == FileType::Directory;
      return *src_is_dir_opt;
    };
    // src must be a directory if it has a trailing slash.
    if (internal::HasTrailingSlash(src.path)) {
      ARROW_ASSIGN_OR_RAISE(auto src_is_dir, src_is_dir_lazy());
      if (!src_is_dir) {
        return NotADir(src);
      }
    }
    // The Azure SDK and the backend don't perform many important checks, so we have to
    // do them ourselves. Additionally, based on many tests on a default-configuration
    // storage account, if the destination is an empty directory, the rename operation
    // will most likely fail due to a timeout. Providing both leases -- to source and
    // destination -- seems to have made things work.
    ARROW_ASSIGN_OR_RAISE(auto dest_lease_client,
                          AcquireBlobLease(dest, kLeaseDuration, /*allow_missing=*/true));
    std::optional<LeaseGuard> dest_lease_guard;
    if (dest_lease_client) {
      dest_lease_guard.emplace(std::move(dest_lease_client), kLeaseDuration);
      // Perform all the checks on dest (and src) before proceeding with the rename.
      auto dest_adlfs_client = GetFileSystemClient(dest.container);
      ARROW_ASSIGN_OR_RAISE(auto dest_info, GetFileInfo(dest_adlfs_client, dest,
                                                        dest_lease_guard->LeaseId()));
      if (dest_info.type() == FileType::Directory) {
        ARROW_ASSIGN_OR_RAISE(auto src_is_dir, src_is_dir_lazy());
        if (!src_is_dir) {
          // If src is a regular file, complain that dest is a directory
          // like POSIX rename() does.
          return internal::IsADir(dest.all);
        }
      } else {
        if (internal::HasTrailingSlash(dest.path)) {
          return NotADir(dest);
        }
      }
    } else {