operations_file.hpp

#pragma once
 
#include <chrono>
 
#include <boost/filesystem/operations.hpp>
 
#include <userver/fs/blocking/c_file.hpp>
#include <userver/utils/cpu_relax.hpp>
#include <userver/utils/fast_pimpl.hpp>
 
#include <userver/dump/factory.hpp>
#include <userver/dump/operations.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace dump {
 
/// A handle to a dump file. File operations block the thread.
class FileWriter final : public Writer {
 public:
  /// @brief Creates a new dump file and opens it
  /// @throws `Error` on a filesystem error
  explicit FileWriter(std::string path, boost::filesystem::perms perms,
                      tracing::ScopeTime& scope);
 
  void Finish() override;
 
 private:
  void WriteRaw(std::string_view data) override;
 
  fs::blocking::CFile file_;
  std::string final_path_;
  std::string path_;
  boost::filesystem::perms perms_;
  utils::StreamingCpuRelax cpu_relax_;
};
 
/// A handle to a dump file. File operations block the thread.
class FileReader final : public Reader {
 public:
  /// @brief Opens an existing dump file
  /// @throws `Error` on a filesystem error
  explicit FileReader(std::string path);
 
  void Finish() override;
 
 private:
  std::string_view ReadRaw(std::size_t max_size) override;
 
  fs::blocking::CFile file_;
  std::string path_;
  std::string curr_chunk_;
};
 
class FileOperationsFactory final : public OperationsFactory {
 public:
  explicit FileOperationsFactory(boost::filesystem::perms perms);
 
  std::unique_ptr<Reader> CreateReader(std::string full_path) override;
 
  std::unique_ptr<Writer> CreateWriter(std::string full_path,
                                       tracing::ScopeTime& scope) override;
 
 private:
  const boost::filesystem::perms perms_;
};
 
}  // namespace dump
 
USERVER_NAMESPACE_END