LightweightDatabaseServer.h

Go to the documentation of this file.

#include "NativeFeatureIncludes.h"
#if _RAKNET_SUPPORT_LightweightDatabaseServer==1

#ifndef __LIGHTWEIGHT_DATABASE_SERVER_H
#define __LIGHTWEIGHT_DATABASE_SERVER_H

#include "Export.h"
#include "PluginInterface2.h"
#include "LightweightDatabaseCommon.h"
#include "DS_Map.h"

class RakPeerInterface;
struct Packet;

class RAK_DLL_EXPORT LightweightDatabaseServer : public PluginInterface2
        {
        public:
                // Constructor
                LightweightDatabaseServer();

                // Destructor
                virtual ~LightweightDatabaseServer();

                DataStructures::Table *GetTable(const char *tableName);

                DataStructures::Table* AddTable(const char *tableName,
                        bool allowRemoteQuery,
                        bool allowRemoteUpdate,
                        bool allowRemoteRemove,
                        const char *queryPassword,
                        const char *updatePassword,
                        const char *removePassword,
                        bool oneRowPerSystemAddress,
                        bool onlyUpdateOwnRows,
                        bool removeRowOnPingFailure,
                        bool removeRowOnDisconnect,
                        bool autogenerateRowIDs);

                bool RemoveTable(const char *tableName);

                void Clear(void);

                // Gets the next valid auto-generated rowId for a table and increments it.
                unsigned GetAndIncrementRowID(const char *tableName);

                DataStructures::Page<unsigned, DataStructures::Table::Row*, _TABLE_BPLUS_TREE_ORDER> *GetTableRows(const char *tableName);

                virtual void Update(void);
                virtual PluginReceiveResult OnReceive(Packet *packet);
                virtual void OnClosedConnection(SystemAddress systemAddress, RakNetGUID rakNetGUID, PI2_LostConnectionReason lostConnectionReason );

                struct DatabaseTable
                        {
                        bool allowRemoteUpdate;
                        bool allowRemoteQuery;
                        bool allowRemoteRemove;

                        char updatePassword[_SIMPLE_DATABASE_PASSWORD_LENGTH];
                        char queryPassword[_SIMPLE_DATABASE_PASSWORD_LENGTH];
                        char removePassword[_SIMPLE_DATABASE_PASSWORD_LENGTH];
                        bool oneRowPerSystemAddress;
                        bool onlyUpdateOwnRows;
                        bool removeRowOnPingFailure;
                        bool removeRowOnDisconnect;
                        bool autogenerateRowIDs;
                        char tableName[_SIMPLE_DATABASE_TABLE_NAME_LENGTH];

                        // Used if autogenerateRowIDs==true
                        unsigned nextRowId;

                        unsigned SystemAddressColumnIndex;
                        unsigned SystemGuidColumnIndex;
                        unsigned lastPingResponseColumnIndex;
                        unsigned nextPingSendColumnIndex;
                        RakNetTime nextRowPingCheck;
                        DataStructures::Table table;
                        };

                static int DatabaseTableComp( const char* const &key1, const char* const &key2 );

        protected:
                DataStructures::Map<const char *, LightweightDatabaseServer::DatabaseTable*, LightweightDatabaseServer::DatabaseTableComp> database;
                void OnQueryRequest(Packet *packet);
                void OnUpdateRow(Packet *packet);
                void OnRemoveRow(Packet *packet);
                void OnPong(Packet *packet);
                // mode 0 = query, mode 1 = update, mode 2 = remove
                DatabaseTable * DeserializeClientHeader(RakNet::BitStream *inBitstream, RakPeerInterface *peer, Packet *packet, int mode);
                DataStructures::Table::Row * GetRowFromIP(DatabaseTable *databaseTable, SystemAddress systemAddress, unsigned *rowId);
                bool RowHasIP(DataStructures::Table::Row *row, SystemAddress systemAddress, unsigned SystemAddressColumnIndex);
                DataStructures::Table::Row * AddRow(LightweightDatabaseServer::DatabaseTable *databaseTable, SystemAddress systemAddress, RakNetGUID guid, bool hasRowId, unsigned rowId);
                void RemoveRowsFromIP(SystemAddress systemAddress);

        };

#endif

#endif // _RAKNET_SUPPORT_*