opendal.hpp

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
 
#pragma once
 
#include <cstring>
#include <iostream>
#include <memory>
#include <optional>
#include <string>
#include <unordered_map>
#include <vector>
 
#include "data_structure.hpp"
 
namespace opendal {
 
namespace ffi {
class Operator;
class Reader;
class Lister;
}  // namespace ffi
 
class Reader;
class Lister;
 
class Operator {
 public:
  Operator() noexcept;
 
  Operator(std::string_view scheme,
           const std::unordered_map<std::string, std::string> &config = {});
 
  // Disable copy and assign
  Operator(const Operator &) = delete;
  Operator &operator=(const Operator &) = delete;
 
  // Enable move
  Operator(Operator &&other) noexcept;
  Operator &operator=(Operator &&other) noexcept;
 
  ~Operator() noexcept;
 
  bool Available() const;
 
  std::string Read(std::string_view path);
 
  void Write(std::string_view path, std::string_view data);
 
  Reader GetReader(std::string_view path);
 
  [[deprecated("Use Exists() instead.")]]
  bool IsExist(std::string_view path);
 
  bool Exists(std::string_view path);
 
  void CreateDir(std::string_view path);
 
  void Copy(std::string_view src, std::string_view dst);
 
  void Rename(std::string_view src, std::string_view dst);
 
  void Remove(std::string_view path);
 
  Metadata Stat(std::string_view path);
 
  std::vector<Entry> List(std::string_view path);
 
  Lister GetLister(std::string_view path);
  Capability Info();
 
  private:
   void Destroy() noexcept;
 
   ffi::Operator *operator_{nullptr};
 };
 
class Reader {
 public:
  Reader(Reader &&other) noexcept;
 
  ~Reader() noexcept;
 
  std::streamsize Read(void *s, std::streamsize n);
 
  std::streampos Seek(std::streamoff off, std::ios_base::seekdir way);
 
 private:
  friend class Operator;
 
  Reader(ffi::Reader *pointer) noexcept;
 
  void Destroy() noexcept;
 
  ffi::Reader *reader_{nullptr};
};
 
class ReaderStream : public std::istream {
 public:
  class ReaderStreamBuf : public std::streambuf {
   public:
    ReaderStreamBuf(Reader &&reader)
        : reader_(std::move(reader)), buffer_start_pos_(0) {
      setg(buffer_, buffer_, buffer_);
    }
 
   protected:
    std::streamsize xsgetn(char *s, std::streamsize count) override {
      std::streamsize total_read = 0;
 
      while (total_read < count) {
        if (gptr() < egptr()) {
          std::streamsize available_bytes = egptr() - gptr();
          std::streamsize to_copy =
              std::min(available_bytes, count - total_read);
          std::memcpy(s + total_read, gptr(), to_copy);
          gbump(to_copy);
          total_read += to_copy;
          continue;
        }
 
        if (underflow() == traits_type::eof()) break;
      }
 
      return total_read;
    }
 
    int_type underflow() override {
      if (gptr() < egptr()) {
        return traits_type::to_int_type(*gptr());
      }
 
      // Update the buffer start position to current reader position
      buffer_start_pos_ += static_cast<std::streamoff>(egptr() - eback());
 
      std::streamsize n = reader_.Read(buffer_, sizeof(buffer_));
      if (n <= 0) {
        return traits_type::eof();
      }
 
      setg(buffer_, buffer_, buffer_ + n);
      return traits_type::to_int_type(*gptr());
    }
 
    int_type uflow() override {
      int_type result = underflow();
      if (result != traits_type::eof()) {
        gbump(1);
      }
      return result;
    }
 
    std::streampos seekoff(std::streamoff off, std::ios_base::seekdir dir,
                           std::ios_base::openmode which) override {
      if (dir == std::ios_base::cur && off == 0) {
        // tellg() case - return current position
        return buffer_start_pos_ + static_cast<std::streamoff>(gptr() - eback());
      }
 
      // Actual seek operation
      std::streampos new_pos = reader_.Seek(off, dir);
      if (new_pos != std::streampos(-1)) {
        buffer_start_pos_ = new_pos;
        setg(buffer_, buffer_, buffer_);
      }
      return new_pos;
    }
 
    std::streampos seekpos(std::streampos pos,
                           std::ios_base::openmode which) override {
      return seekoff(pos, std::ios_base::beg, which);
    }
 
   private:
    Reader reader_;
    char buffer_[8192];
    std::streampos buffer_start_pos_;
  };
 
  ReaderStream(Reader &&reader)
      : std::istream(&buf_), buf_(std::move(reader)) {}
 
 private:
  ReaderStreamBuf buf_;
};
 
class Lister {
 public:
  Lister(Lister &&other) noexcept;
 
  ~Lister() noexcept;
 
  class Iterator {
   public:
    using iterator_category = std::input_iterator_tag;
    using value_type = Entry;
    using difference_type = std::ptrdiff_t;
    using pointer = Entry *;
    using reference = Entry &;
 
    Iterator(Lister &lister) : lister_{lister} {
      current_entry_ = lister_.Next();
    }
 
    Entry operator*() { return current_entry_.value(); }
 
    Iterator &operator++() {
      if (current_entry_) {
        current_entry_ = lister_.Next();
      }
      return *this;
    }
 
    bool operator!=(const Iterator &other) const {
      return current_entry_ != std::nullopt ||
             other.current_entry_ != std::nullopt;
    }
 
   protected:
    // Only used for end iterator
    Iterator(Lister &lister, bool /*end*/) noexcept : lister_(lister) {}
 
   private:
    friend class Lister;
 
    Lister &lister_;
    std::optional<Entry> current_entry_;
  };
 
  std::optional<Entry> Next();
 
  Iterator begin() { return Iterator(*this); }
  Iterator end() { return Iterator(*this, true); }
 
 private:
  friend class Operator;
 
  Lister(ffi::Lister *pointer) noexcept;
 
  void Destroy() noexcept;
 
  ffi::Lister *lister_{nullptr};
};
 
}  // namespace opendal