---

persist.h

Go to the documentation of this file.

// Copyright (C) 1999-2001 Open Source Telecom Corporation.
//  
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software 
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
// 
// As a special exception to the GNU General Public License, permission is 
// granted for additional uses of the text contained in its release 
// of Common C++.
// 
// The exception is that, if you link the Common C++ library with other
// files to produce an executable, this does not by itself cause the
// resulting executable to be covered by the GNU General Public License.
// Your use of that executable is in no way restricted on account of
// linking the Common C++ library code into it.
// 
// This exception does not however invalidate any other reasons why
// the executable file might be covered by the GNU General Public License.
// 
// This exception applies only to the code released under the 
// name Common C++.  If you copy code from other releases into a copy of
// Common C++, as the General Public License permits, the exception does
// not apply to the code that you add in this way.  To avoid misleading
// anyone as to the status of such modified files, you must delete
// this exception notice from them.
// 
// If you write modifications of your own for Common C++, it is your choice
// whether to permit this exception to apply to your modifications.
// If you do not wish that, delete this exception notice.  

#ifndef __CCXX_PERSIST_H__
#define __CCXX_PERSIST_H__

#ifndef __CCXX_CONFIG_H__
#include <cc++/config.h>
#endif

#ifndef __CCXX_STRCHAR_H__
#include <cc++/persist.h>
#endif

#ifdef HAVE_ZLIB_H
#ifndef NO_COMPRESSION
#include <zlib.h>
#endif
#else
#define NO_COMPRESSION
#endif

#include <iostream>
#include <string>
#include <vector>
#include <map>

#ifdef __NAMESPACES__
namespace ost {
#endif

class PersistException
{
public:
        CCXX_MEMBER_EXPORT(CCXX_EMPTY) PersistException(const std::string& reason);
        CCXX_MEMBER_EXPORT(const std::string&) GetMessage() const;
        CCXX_MEMBER_EXPORT(virtual) ~PersistException();
protected:
        std::string myMessage;
};

// This typedef allows us to declare NewBaseObjectFunction now
typedef class BaseObject* (*NewBaseObjectFunction) (void);

class TypeManager
{
public:
        
        class Registration
        {
        public:
                Registration(const char* name, NewBaseObjectFunction func)
                        : myName(name) { TypeManager::Add(name,func); }
                virtual ~Registration() { TypeManager::Remove(myName.c_str()); }
        private:
                std::string myName;
        };
        
        CCXX_MEMBER_EXPORT(static void) Add(const char* name, NewBaseObjectFunction construction);
        
        CCXX_MEMBER_EXPORT(static void) Remove(const char* name);
        
        CCXX_EXPORT(static BaseObject*) CreateInstanceOf(const char* name);
        
        typedef std::map<std::string,NewBaseObjectFunction> StringFunctionMap;
};


/*
 * The following defines are used to declare and define the relevant code
 * to allow a class to use the Persistence::Engine code.
 */

#define DECLARE_PERSISTENCE(ClassType)                                  \
  public:                                                               \
    friend Engine& operator>>(Engine& ar, ClassType *&ob);              \
    friend Engine& operator<<(Engine& ar, ClassType const *&ob);        \
    friend BaseObject *CreateNew##ClassType();                          \
    virtual const char* GetPersistenceID() const;                       \
    static TypeManager::Registration RegistrationFor##ClassType;

#define IMPLEMENT_PERSISTENCE(ClassType,FullyQualifiedName)                   \
  BaseObject *CreateNew##ClassType() { return new ClassType; }                \
  const char* ClassType::GetPersistenceID()const {return FullyQualifiedName;} \
  Engine& operator>>(Engine& ar, ClassType *&ob)                              \
    { ar >> (BaseObject *&) ob; return ar; }                                  \
  Engine& operator<<(Engine& ar, ClassType const *ob)                         \
    { ar << (BaseObject const *)ob; return ar; }                              \
  TypeManager::Registration                                                   \
        ClassType::RegistrationFor##ClassType(FullyQualifiedName,             \
                                              CreateNew##ClassType);

class Engine;

class BaseObject
{
public:
        CCXX_MEMBER_EXPORT(CCXX_EMPTY) BaseObject();
        
        CCXX_MEMBER_EXPORT(virtual) ~BaseObject();
        
        CCXX_MEMBER_EXPORT(virtual const char*) GetPersistenceID() const;
        
        CCXX_MEMBER_EXPORT(virtual bool) Write(Engine& archive) const;

        CCXX_MEMBER_EXPORT(virtual bool) Read(Engine& archive);
};


class Engine
{
public:
        class Exception : public PersistException
        {
        public:
                CCXX_MEMBER_EXPORT(CCXX_EMPTY) Exception(const std::string &reason) : PersistException(reason) {}
        };
        
        enum EngineMode
        {
                modeRead,
                modeWrite
        };
        
        CCXX_MEMBER_EXPORT(CCXX_EMPTY) Engine(std::iostream& stream, EngineMode mode) THROWS (PersistException);
        
        CCXX_MEMBER_EXPORT(virtual) ~Engine();


        // Write operations
        CCXX_MEMBER_EXPORT(void) Write(const BaseObject *object) THROWS (Exception);
        /*
         * shortcut, to make the following more readable
         */
#define _ENGINEWRITE_REF(valref) WriteBinary((const uint8*)&valref,sizeof(valref))
        void Write(int8 i)   THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(uint8 i)  THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(int16 i)  THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(uint16 i) THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(int32 i)  THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(uint32 i) THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(int64 i)  THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(uint64 i) THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(float i)  THROWS (Exception) { _ENGINEWRITE_REF(i); }
        void Write(double i) THROWS (Exception) { _ENGINEWRITE_REF(i); }
#undef _ENGINEWRITE_REF

        CCXX_MEMBER_EXPORT(void) Write(const std::string& str) THROWS (Exception);
        // Every write operation boils down to one or more of theses
        CCXX_MEMBER_EXPORT(void) WriteBinary(const uint8* data, const uint32 size) THROWS (Exception);
        
        // Read Operations
        
        CCXX_MEMBER_EXPORT(void) Read(BaseObject *&object) THROWS (Exception);
#define _ENGINEREAD_REF(valref) ReadBinary((uint8*)&valref,sizeof(valref))
        void Read(int8& i)   THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(uint8& i)  THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(int16& i)  THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(uint16& i) THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(int32& i)  THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(uint32& i) THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(int64& i)  THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(uint64& i) THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(float& i)  THROWS (Exception) { _ENGINEREAD_REF(i); }
        void Read(double& i) THROWS (Exception) { _ENGINEREAD_REF(i); }
#undef _ENGINEREAD_REF

        CCXX_MEMBER_EXPORT(void) Read(std::string& str) THROWS (Exception);
        // Every read operation boild down to one or more of these
        CCXX_MEMBER_EXPORT(void) ReadBinary(uint8* data, uint32 size) THROWS (Exception);

private:
        std::iostream& myUnderlyingStream;
        
        EngineMode myOperationalMode;

        typedef std::vector<BaseObject*>           ArchiveVector;
        typedef std::map<BaseObject const*, int32> ArchiveMap;
        typedef std::vector<std::string>                ClassVector;
        typedef std::map<std::string, int32>            ClassMap;
        
        ArchiveVector myArchiveVector;
        ArchiveMap myArchiveMap;
        ClassVector myClassVector;
        ClassMap myClassMap;

        // Compression support
#ifndef NO_COMPRESSION
        z_stream myZStream;
        uint8* myCompressedDataBuffer;
        uint8* myUncompressedDataBuffer;
        uint8* myLastUncompressedDataRead;
#endif
};

// Standard >> and << stream operators for BaseObject
CCXX_EXPORT(Engine&) operator >>( Engine& ar, BaseObject *&ob)      THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, BaseObject const *ob) THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, int8& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, int8 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, uint8& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, uint8 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, int16& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, int16 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, uint16& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, uint16 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, int32& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, int32 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, uint32& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, uint32 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, int64& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, int64 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, uint64& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, uint64 ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, float& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, float ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, double& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, double ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, std::string& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, std::string ob)  THROWS (Engine::Exception);

CCXX_EXPORT(Engine&) operator >>( Engine& ar, bool& ob) THROWS (Engine::Exception);
CCXX_EXPORT(Engine&) operator <<( Engine& ar, bool ob)  THROWS (Engine::Exception);

template<class T>
Engine& operator <<( Engine& ar, std::vector<T> const& ob) THROWS (Engine::Exception)
{
        ar << (uint32)ob.size();
        for(uint i=0; i < ob.size(); ++i)
                ar << ob[i];
        return ar;
}

template<class T>
Engine& operator >>( Engine& ar, std::vector<T>& ob) THROWS (Engine::Exception)
{
        ob.clear();
        uint32 siz;
        ar >> siz;
        ob.resize(siz);
        for(uint32 i=0; i < siz; ++i)
                ar >> ob[i];
        return ar;
}

template<class Key, class Value>
Engine& operator <<( Engine& ar, std::map<Key,Value> const & ob) THROWS (Engine::Exception)
{
        ar << (uint32)ob.size();
        for(std::map<Key,Value>::const_iterator it = ob.begin();it != ob.end();++it)
                ar << it->first << it->second;
        return ar;
}

template<class Key, class Value>
Engine& operator >>( Engine& ar, std::map<Key,Value>& ob) THROWS (Engine::Exception)
{
        ob.clear();
        uint32 siz;
        ar >> siz;
        for(uint32 i=0; i < siz; ++i) {
                Key a;
                ar >> a;
                ar >> ob[a];
        }
        return ar;
}

#ifdef  __NAMESPACES__
};
#endif

#endif

---