InitDataFileReader.h

Go to the documentation of this file.

#ifndef CATALOG_INITDATAFILEREADER_H
#define CATALOG_INITDATAFILEREADER_H
 
#include "tdb.h"
#include "catalog/BootstrapCatCache.h"
 
namespace taco {
 
enum class InitLineType {
    Eof,
    Error,
    Table,
    Data,
};
 
class InitDataFileReader {
public:
    InitDataFileReader(
        const std::string &init_datafile,
        BootstrapCatCache *catcache);
 
    InitLineType next(char **rec, FieldOffset *rec_size);
 
    size_t
    LastLineNo() const {
        return m_lineno;
    }
 
    Oid
    LastTabid() const {
        return m_tabid;
    }
 
    Schema*
    LastSchema() const {
        return m_schema.get();
    }
 
private:
    absl::string_view FindNextToken(
        const std::string &str,
        std::string::size_type &start_pos) const;
 
    std::string m_init_datafile;
    std::ifstream m_input;
    BootstrapCatCache *m_catcache;
 
    size_t m_lineno;
    InitLineType m_last_state;
    Oid m_tabid;
    maxaligned_char_buf m_buf;
 
    std::unique_ptr<Schema> m_schema;
    absl::flat_hash_map<Oid, FunctionInfo> m_typ_infuncs;
};
 
}
 
#endif  // CATALOG_INITDATAFILERADER_H