dump.cpp

Go to the documentation of this file.

#include "joedb/ui/dump.h"
#include "joedb/interpreted/Database.h"
#include "joedb/Writable.h"
 
#include "joedb/Selective_Writable.h"
#include "joedb/Multiplexer.h"
#include "joedb/journal/Readonly_Journal.h"
 
namespace joedb
{
 /////////////////////////////////////////////////////////////////////////////
 void dump(const Readable &db, Writable &writable, bool schema_only)
 /////////////////////////////////////////////////////////////////////////////
 {
  //
  // Dump tables
  //
  std::map<Table_Id, Table_Id> table_map;
  {
   Table_Id mapped_tid = Table_Id(0);
   for (const auto &[tid, tname]: db.get_tables())
   {
    ++mapped_tid;
    table_map[tid] = mapped_tid;
    writable.create_table(tname);
   }
  }
 
  //
  // Dump fields
  //
  std::map<Table_Id, std::map<Field_Id, Field_Id>> field_maps;
  {
   for (const auto &[tid, tname]: db.get_tables())
   {
    Field_Id mapped_fid = Field_Id(0);
    for (const auto &[fid, fname]: db.get_fields(tid))
    {
     ++mapped_fid;
     Type type = db.get_field_type(tid, fid);
     if (type.get_type_id() == Type::Type_Id::reference)
      type = Type::reference(table_map[type.get_table_id()]);
     field_maps[tid][fid] = mapped_fid;
 
     writable.add_field(table_map[tid], fname, type);
    }
   }
  }
 
  //
  // Dump records
  //
  if (schema_only)
   return;
 
  for (const auto &[tid, tname]: db.get_tables())
  {
   const Record_Id table_size = db.get_size(tid);
 
   for (Record_Id record_id{0}; record_id < table_size;)
   {
    while (record_id < table_size && !db.is_used(tid, record_id))
     ++record_id;
 
    size_t size = 0;
 
    while
    (
     record_id + size < table_size &&
     db.is_used(tid, record_id + size)
    )
    {
     size++;
    }
 
    if (size)
    {
     writable.insert_vector(table_map[tid], record_id, size);
     record_id = record_id + size;
    }
   }
 
   for (const auto &[fid, fname]: db.get_fields(tid))
   {
    for (Record_Id record_id{0}; record_id < table_size; ++record_id)
    {
     if (db.is_used(tid, record_id))
     {
      Table_Id mapped_tid = table_map[tid];
      Field_Id mapped_fid = field_maps[tid][fid];
 
      switch (db.get_field_type(tid, fid).get_type_id())
      {
       case Type::Type_Id::null:
       break;
 
       #define TYPE_MACRO(type, return_type, type_id, R, W)\
       case Type::Type_Id::type_id:\
        writable.update_##type_id\
        (\
         mapped_tid, record_id, mapped_fid, db.get_##type_id(tid, record_id, fid)\
        );\
       break;
       #include "joedb/TYPE_MACRO.h"
      }
     }
    }
   }
  }
 }
 
 /////////////////////////////////////////////////////////////////////////////
 void dump_data(const Readable &db, Writable &writable)
 /////////////////////////////////////////////////////////////////////////////
 {
  for (const auto &[tid, tname]: db.get_tables())
  {
   const Freedom_Keeper &freedom_keeper = db.get_freedom(tid);
   const Record_Id table_size{freedom_keeper.get_size()};
 
   for (Record_Id record_id{0}; record_id < table_size;)
   {
    while
    (
     record_id < table_size &&
     !freedom_keeper.is_used(record_id)
    )
    {
     ++record_id;
    }
 
    size_t size = 0;
 
    while
    (
     record_id + size < table_size &&
     freedom_keeper.is_used(record_id + size)
    )
    {
     size++;
    }
 
    if (size > 1)
     writable.insert_vector(tid, record_id, size);
    else if (size == 1)
     writable.insert_into(tid, record_id);
 
    if (size > 0)
    {
     for (const auto &[fid, fname]: db.get_fields(tid))
     {
      switch(db.get_field_type(tid, fid).get_type_id())
      {
       case Type::Type_Id::null:
       break;
 
       #define TYPE_MACRO(type, return_type, type_id, R, W)\
       case Type::Type_Id::type_id:\
       {\
        const type &value = db.get_##type_id(tid, record_id, fid);\
        if (size > 1)\
         writable.update_vector_##type_id(tid, record_id, fid, size, &value);\
        else\
         writable.update_##type_id(tid, record_id, fid, value);\
       }\
       break;
       #include "joedb/TYPE_MACRO.h"
      }
     }
    }
 
    record_id = record_id + size;
   }
  }
  writable.soft_checkpoint();
 }
 
 /////////////////////////////////////////////////////////////////////////////
 void pack(Readonly_Journal &input_journal, Writable &writable)
 /////////////////////////////////////////////////////////////////////////////
 {
  Database db;
 
  {
   Selective_Writable schema_filter(writable, Selective_Writable::Mode::schema);
   Multiplexer multiplexer{db, schema_filter};
 
   input_journal.raw_play_until_checkpoint(multiplexer);
 
   if (schema_filter.has_blobs())
    throw Exception("can't pack blobs");
  }
 
  dump_data(db, writable);
 }
}