src/DAL/TableImpl.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2008 by Sverre Rabbelier                                *
   3  *   sverre@rabbelier.nl                                                   *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 3 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "TableImpl.h"
  22 #include "SqliteMgr.h"
  23 #include "FieldImpl.h"
  24 #include "FieldDef.h"
  25 #include "KeyImpl.h"
  26 #include "SavableManager.h"
  27 #include "SavableManagers.h"
  28 #include "SelectionMask.h"
  29 #include "StringUtilities.h"
  30
  31 TableImpl::TableImpl(cstring name) :
  32 m_name(name),
  33 m_primarykeysize(0)
  34 {
  35
  36 }
  37
  38 cstring TableImpl::getName() const
  39 {
  40         return m_name;
  41 }
  42
  43 Strings TableImpl::tableList(SelectionMaskPtr mask)
  44 {
  45         std::vector<Strings> result;
  46
  47         SqliteMgr::Get()->doSelectMulti(mask.get());
  48         SavableManagersPtr managers = mask->getResult();
  49
  50         if(!managers->size())
  51         {
  52                 Strings emptyResult;
  53                 emptyResult.push_back("None.");
  54                 return emptyResult;
  55         }
  56
  57         Strings headerfields = managers->getHeader();
  58
  59         for(SavableManagerVector::const_iterator it = managers->begin(); it != managers->end(); it++)
  60         {
  61                 SavableManagerPtr manager = *it;
  62                 result.push_back(manager->fieldList());
  63         }
  64
  65         return String::Get()->createTable(result, headerfields);
  66 }
  67
  68 KeyImplPtr TableImpl::firstkey() const
  69 {
  70         Assert(m_primarykeysize == 1);
  71         KeyImplPtr key;
  72
  73         for(FieldImplVector::const_iterator it = m_fields.begin(); it != m_fields.end(); it++)
  74         {
  75                 FieldImplPtr field = *it;
  76                 Assert(field);
  77
  78                 if(!field->isPrimaryKey())
  79                         continue;
  80
  81                 key = boost::static_pointer_cast<KeyImpl>(field);
  82         }
  83
  84         Assert(key);
  85         return key;
  86 }
  87
  88 void TableImpl::updateFieldCount()
  89 {
  90         for(FieldImplVector::const_iterator it = m_fields.begin(); it != m_fields.end(); it++)
  91         {
  92                 FieldImplPtr field = *it;
  93                 Assert(field);
  94
  95                 if(field->isPrimaryKey())
  96                         m_primarykeysize++;
  97         }
  98 }