string_list_writer.h

Go to the documentation of this file.

#ifndef META_STRING_LIST_WRITER_H_
#define META_STRING_LIST_WRITER_H_

#include <fstream>
#include <mutex>
#include <string>

#if !META_HAS_STREAM_MOVE
#include <memory>
#include "util/shim.h"
#endif

#include "util/disk_vector.h"

namespace meta
{
namespace index
{

class string_list_writer
{
  public:
    string_list_writer(const std::string& path, uint64_t size);

    string_list_writer(string_list_writer&&);

    string_list_writer& operator=(string_list_writer&&);

    void insert(uint64_t idx, const std::string& elem);

  private:
#if META_HAS_STREAM_MOVE
    using ofstream = std::ofstream;
    std::ofstream& file()
    {
        return string_file_;
    }
    ofstream make_file(const std::string& path)
    {
        return std::ofstream{path};
    }
#else
    using ofstream = std::unique_ptr<std::ofstream>;
    std::ofstream& file()
    {
        return *string_file_;
    }
    ofstream make_file(const std::string& path)
    {
        return make_unique<std::ofstream>(path);
    }
#endif

    std::mutex mutex_;

    ofstream string_file_;

    uint64_t write_pos_;

    util::disk_vector<uint64_t> index_;
};
}
}

#endif