pana_database.h

/* BEGIN_COPYRIGHT                                                        */
/*                                                                        */
/* Open Diameter: Open-source software for the Diameter and               */
/*                Diameter related protocols                              */
/*                                                                        */
/* Copyright (C) 2002-2004 Open Diameter Project                          */
/*                                                                        */
/* This library is free software; you can redistribute it and/or modify   */
/* it under the terms of the GNU Lesser General Public License as         */
/* published by the Free Software Foundation; either version 2.1 of the   */
/* License, or (at your option) any later version.                        */
/*                                                                        */
/* This library 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      */
/* Lesser General Public License for more details.                        */
/*                                                                        */
/* You should have received a copy of the GNU Lesser General Public       */
/* License along with this library; if not, write to the Free Software    */
/* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307    */
/* USA.                                                                   */
/*                                                                        */
/* In addition, when you copy and redistribute some or the entire part of */
/* the source code of this software with or without modification, you     */
/* MUST include this copyright notice in each copy.                       */
/*                                                                        */
/* If you make any changes that are appeared to be useful, please send    */
/* sources that include the changed part to                               */
/* diameter-developers@lists.sourceforge.net so that we can reflect your  */
/* changes to one unified version of this software.                       */
/*                                                                        */
/* END_COPYRIGHT                                                          */

#ifndef __PANA_DATABASE_H__
#define __PANA_DATABASE_H__

#include <string.h>
#include "pana_rbtree.h"

template<class ARG>
class PANA_SessionTree : public RbTreeTree
{
   protected:
     class PANA_DbEntry : public RbTreeData {
         public:
            PANA_DbEntry(std::string &index) : m_Index(index) { }
            void operator()() { }
            int operator<(RbTreeData &cmp) {
               PANA_DbEntry &compare = reinterpret_cast<PANA_DbEntry&>(cmp);
               return (m_Index < compare.Index());
            }
            int operator==(RbTreeData &cmp) {
               PANA_DbEntry &compare = reinterpret_cast<PANA_DbEntry&>(cmp);
               return (m_Index == compare.Index());
            }
            void clear(void *userData = 0) { }
            std::string &Index() { return m_Index; }
         protected:
            std::string m_Index;
     };

   public:
      PANA_SessionTree() { }
      virtual ~PANA_SessionTree() { }

      void Add(std::string &index, ARG &arg) {
         PANA_DbEntry *newEntry;
         ACE_NEW_NORETURN(newEntry, PANA_DbEntry(index));
         if (newEntry) {
            newEntry->payload = reinterpret_cast<void*>(&arg);
            if (Insert(newEntry)) {
               return;
            }
            throw (PANA_Exception(PANA_Exception::DATABASE_ERROR, 
                                 "Failed to insert new entry into rbtree"));
         }
         throw (PANA_Exception(PANA_Exception::NO_MEMORY, 
                               "Allocation error for new session entry"));
      }
      void Remove(std::string &index, ARG **arg = NULL) {
         PANA_DbEntry lookupEntry(index), *searchEntry;
         searchEntry = static_cast<PANA_DbEntry*>
             (RbTreeTree::Remove(&lookupEntry));
         if (searchEntry) {
            if (arg) {
               *arg = reinterpret_cast<ARG*>(searchEntry->payload);
            }
            return;
         }
         throw (PANA_Exception(PANA_Exception::DATABASE_ERROR, 
                              "Failed to delete entry from rbtree"));
      }
      ARG *Search(std::string &index) {
         PANA_DbEntry lookupEntry(index), *searchEntry;
         searchEntry = static_cast<PANA_DbEntry*>
             (RbTreeTree::Find(&lookupEntry));
         if (searchEntry) {
            return reinterpret_cast<ARG*>(searchEntry->payload);
         }
         throw (PANA_Exception(PANA_Exception::DATABASE_ERROR, 
                              "Could not find session entry"));
      }
};

#endif // __PANA_DATABASE_H__

---