mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-12-26 23:43:34 +00:00
0545adfac3
Have fun!
204 lines
7.4 KiB
C++
204 lines
7.4 KiB
C++
#ifndef __TABLE_SERIALIZER_H
|
|
#define __TABLE_SERIALIZER_H
|
|
|
|
#include "RakMemoryOverride.h"
|
|
#include "DS_Table.h"
|
|
#include "Export.h"
|
|
|
|
namespace RakNet
|
|
{
|
|
class BitStream;
|
|
}
|
|
|
|
class RAK_DLL_EXPORT TableSerializer : public RakNet::RakMemoryOverride
|
|
{
|
|
public:
|
|
static void SerializeTable(DataStructures::Table *in, RakNet::BitStream *out);
|
|
static bool DeserializeTable(unsigned char *serializedTable, unsigned int dataLength, DataStructures::Table *out);
|
|
static bool DeserializeTable(RakNet::BitStream *in, DataStructures::Table *out);
|
|
static void SerializeColumns(DataStructures::Table *in, RakNet::BitStream *out);
|
|
static void SerializeColumns(DataStructures::Table *in, RakNet::BitStream *out, DataStructures::List<int> &skipColumnIndices);
|
|
static bool DeserializeColumns(RakNet::BitStream *in, DataStructures::Table *out);
|
|
static void SerializeRow(DataStructures::Table::Row *in, unsigned keyIn, DataStructures::List<DataStructures::Table::ColumnDescriptor> &columns, RakNet::BitStream *out);
|
|
static void SerializeRow(DataStructures::Table::Row *in, unsigned keyIn, DataStructures::List<DataStructures::Table::ColumnDescriptor> &columns, RakNet::BitStream *out, DataStructures::List<int> &skipColumnIndices);
|
|
static bool DeserializeRow(RakNet::BitStream *in, DataStructures::Table *out);
|
|
static void SerializeCell(RakNet::BitStream *out, DataStructures::Table::Cell *cell, DataStructures::Table::ColumnType columnType);
|
|
static bool DeserializeCell(RakNet::BitStream *in, DataStructures::Table::Cell *cell, DataStructures::Table::ColumnType columnType);
|
|
static void SerializeFilterQuery(RakNet::BitStream *in, DataStructures::Table::FilterQuery *query);
|
|
// Note that this allocates query->cell->c!
|
|
static bool DeserializeFilterQuery(RakNet::BitStream *out, DataStructures::Table::FilterQuery *query);
|
|
static void SerializeFilterQueryList(RakNet::BitStream *in, DataStructures::Table::FilterQuery *query, unsigned int numQueries, unsigned int maxQueries);
|
|
// Note that this allocates queries, cells, and query->cell->c!. Use DeallocateQueryList to free.
|
|
static bool DeserializeFilterQueryList(RakNet::BitStream *out, DataStructures::Table::FilterQuery **query, unsigned int *numQueries, unsigned int maxQueries, int allocateExtraQueries=0);
|
|
static void DeallocateQueryList(DataStructures::Table::FilterQuery *query, unsigned int numQueries);
|
|
};
|
|
|
|
#endif
|
|
|
|
// Test code for the table
|
|
/*
|
|
#include "LightweightDatabaseServer.h"
|
|
#include "LightweightDatabaseClient.h"
|
|
#include "TableSerializer.h"
|
|
#include "BitStream.h"
|
|
#include "StringCompressor.h"
|
|
#include "DS_Table.h"
|
|
void main(void)
|
|
{
|
|
DataStructures::Table table;
|
|
DataStructures::Table::Row *row;
|
|
unsigned int dummydata=12345;
|
|
|
|
// Add columns Name (string), IP (binary), score (int), and players (int).
|
|
table.AddColumn("Name", DataStructures::Table::STRING);
|
|
table.AddColumn("IP", DataStructures::Table::BINARY);
|
|
table.AddColumn("Score", DataStructures::Table::NUMERIC);
|
|
table.AddColumn("Players", DataStructures::Table::NUMERIC);
|
|
table.AddColumn("Empty Test Column", DataStructures::Table::STRING);
|
|
assert(table.GetColumnCount()==5);
|
|
row=table.AddRow(0);
|
|
assert(row);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
row->UpdateCell(2,5);
|
|
row->UpdateCell(3,10);
|
|
//row->UpdateCell(4,"should be unique");
|
|
|
|
row=table.AddRow(1);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
row->UpdateCell(2,5);
|
|
row->UpdateCell(3,15);
|
|
|
|
row=table.AddRow(2);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
row->UpdateCell(2,5);
|
|
row->UpdateCell(3,20);
|
|
|
|
row=table.AddRow(3);
|
|
assert(row);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
row->UpdateCell(2,15);
|
|
row->UpdateCell(3,5);
|
|
row->UpdateCell(4,"col index 4");
|
|
|
|
row=table.AddRow(4);
|
|
assert(row);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
//row->UpdateCell(2,25);
|
|
row->UpdateCell(3,30);
|
|
//row->UpdateCell(4,"should be unique");
|
|
|
|
row=table.AddRow(5);
|
|
assert(row);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
//row->UpdateCell(2,25);
|
|
row->UpdateCell(3,5);
|
|
//row->UpdateCell(4,"should be unique");
|
|
|
|
row=table.AddRow(6);
|
|
assert(row);
|
|
row->UpdateCell(0,"Kevin Jenkins");
|
|
row->UpdateCell(1,sizeof(dummydata), (char*)&dummydata);
|
|
row->UpdateCell(2,35);
|
|
//row->UpdateCell(3,40);
|
|
//row->UpdateCell(4,"should be unique");
|
|
|
|
row=table.AddRow(7);
|
|
assert(row);
|
|
row->UpdateCell(0,"Bob Jenkins");
|
|
|
|
row=table.AddRow(8);
|
|
assert(row);
|
|
row->UpdateCell(0,"Zack Jenkins");
|
|
|
|
// Test multi-column sorting
|
|
DataStructures::Table::Row *rows[30];
|
|
DataStructures::Table::SortQuery queries[4];
|
|
queries[0].columnIndex=0;
|
|
queries[0].operation=DataStructures::Table::QS_INCREASING_ORDER;
|
|
queries[1].columnIndex=1;
|
|
queries[1].operation=DataStructures::Table::QS_INCREASING_ORDER;
|
|
queries[2].columnIndex=2;
|
|
queries[2].operation=DataStructures::Table::QS_INCREASING_ORDER;
|
|
queries[3].columnIndex=3;
|
|
queries[3].operation=DataStructures::Table::QS_DECREASING_ORDER;
|
|
table.SortTable(queries, 4, rows);
|
|
unsigned i;
|
|
char out[256];
|
|
printf("Sort: Ascending except for column index 3\n");
|
|
for (i=0; i < table.GetRowCount(); i++)
|
|
{
|
|
table.PrintRow(out,256,',',true, rows[i]);
|
|
printf("%s\n", out);
|
|
}
|
|
|
|
// Test query:
|
|
// Don't return column 3, and swap columns 0 and 2
|
|
unsigned columnsToReturn[4];
|
|
columnsToReturn[0]=2;
|
|
columnsToReturn[1]=1;
|
|
columnsToReturn[2]=0;
|
|
columnsToReturn[3]=4;
|
|
DataStructures::Table resultsTable;
|
|
table.QueryTable(columnsToReturn,4,0,0,&resultsTable);
|
|
printf("Query: Don't return column 3, and swap columns 0 and 2:\n");
|
|
for (i=0; i < resultsTable.GetRowCount(); i++)
|
|
{
|
|
resultsTable.PrintRow(out,256,',',true, resultsTable.GetRowByIndex(i));
|
|
printf("%s\n", out);
|
|
}
|
|
|
|
// Test filter:
|
|
// Only return rows with column index 4 empty
|
|
DataStructures::Table::FilterQuery inclusionFilters[3];
|
|
inclusionFilters[0].columnIndex=4;
|
|
inclusionFilters[0].operation=DataStructures::Table::QF_IS_EMPTY;
|
|
// inclusionFilters[0].cellValue; // Unused for IS_EMPTY
|
|
table.QueryTable(0,0,inclusionFilters,1,&resultsTable);
|
|
printf("Filter: Only return rows with column index 4 empty:\n");
|
|
for (i=0; i < resultsTable.GetRowCount(); i++)
|
|
{
|
|
resultsTable.PrintRow(out,256,',',true, resultsTable.GetRowByIndex(i));
|
|
printf("%s\n", out);
|
|
}
|
|
|
|
// Column 5 empty and column 0 == Kevin Jenkins
|
|
inclusionFilters[0].columnIndex=4;
|
|
inclusionFilters[0].operation=DataStructures::Table::QF_IS_EMPTY;
|
|
inclusionFilters[1].columnIndex=0;
|
|
inclusionFilters[1].operation=DataStructures::Table::QF_EQUAL;
|
|
inclusionFilters[1].cellValue.Set("Kevin Jenkins");
|
|
table.QueryTable(0,0,inclusionFilters,2,&resultsTable);
|
|
printf("Filter: Column 5 empty and column 0 == Kevin Jenkins:\n");
|
|
for (i=0; i < resultsTable.GetRowCount(); i++)
|
|
{
|
|
resultsTable.PrintRow(out,256,',',true, resultsTable.GetRowByIndex(i));
|
|
printf("%s\n", out);
|
|
}
|
|
|
|
RakNet::BitStream bs;
|
|
printf("PreSerialize:\n");
|
|
for (i=0; i < table.GetRowCount(); i++)
|
|
{
|
|
table.PrintRow(out,256,',',true, table.GetRowByIndex(i));
|
|
printf("%s\n", out);
|
|
}
|
|
StringCompressor::AddReference();
|
|
TableSerializer::Serialize(&table, &bs);
|
|
TableSerializer::Deserialize(&bs, &table);
|
|
StringCompressor::RemoveReference();
|
|
printf("PostDeserialize:\n");
|
|
for (i=0; i < table.GetRowCount(); i++)
|
|
{
|
|
table.PrintRow(out,256,',',true, table.GetRowByIndex(i));
|
|
printf("%s\n", out);
|
|
}
|
|
int a=5;
|
|
}
|
|
*/
|