DbReadDriverMySql.cpp

gehe zur Dokumentation dieser Datei

// Copyright (C) 2004 Bernhard Assmann <bernie@tuxomania.net>
//  
// 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.
//  

#include <iostream>
#include <sstream>
#include <map>

#include "psfa/DbReadDriverMySql.hpp"
#include "psfa/DbDriverMySqlHelper.hpp"


psfa::DbReadDriverMySql::DbReadDriverMySql() 
{
  bool success = myHelper_.connect();
  if( !success ) {
    std::cout << "Es konnte keine Verbindung zur Datenbank hergestellt werden."
         << std::endl;
    exit(1);
  }
  
}



psfa::DbReadDriverMySql::~DbReadDriverMySql() 
{
  bool success = myHelper_.disconnect();
  if( !success ) {
    std::cerr << "Die Verbindung zur DB konnte nicht erfolgreich abgebaut werden."
         << std::endl;
  }
  
}



void
psfa::DbReadDriverMySql::getPools( sVec& result ) 
{
  std::ostringstream sql;
  sql << "SELECT p_name FROM pool ORDER BY p_name";
  myHelper_.getValueVector( sql, result );
  return;
}



void
psfa::DbReadDriverMySql::getSearchTags( std::string poolName,
               sVec& result ) 
{
  std::ostringstream sql;
  sql << "SELECT pool_id FROM pool WHERE p_name = '"
      << poolName
      << "'";
  int poolId = myHelper_.getSingleValueInt( sql );
  sql.str("");
  sql << "SELECT s_name FROM search WHERE pool_id = '"
      << poolId
      << "' ORDER BY s_name";
  myHelper_.getValueVector( sql, result );
  return;
}


void
psfa::DbReadDriverMySql::getKontextTags( std::string poolName, 
                sVec& result ) 
{
  std::ostringstream sql;
  sql << "SELECT pool_id FROM pool WHERE p_name = '"
      << poolName
      << "'";
  int poolId = myHelper_.getSingleValueInt( sql );
  sql.str("");
  sql << "SELECT k_name FROM kontext WHERE pool_id = '"
      << poolId
      << "' ORDER BY k_name";
  myHelper_.getValueVector( sql, result );
  return;
}




void
psfa::DbReadDriverMySql::getUnits( std::string poolName, sVec& result ) 
{
  std::ostringstream sql;
  sql << "SELECT pool_id FROM pool WHERE p_name = '"
      << poolName
      << "'";
  int poolId = myHelper_.getSingleValueInt( sql );
  sql.str("");
  sql << "SELECT u_name FROM unit WHERE pool_id = '"
      << poolId
      << "' ORDER BY u_name";
  myHelper_.getValueVector( sql, result );
  return;
}



void
psfa::DbReadDriverMySql::getFiles( std::string poolName, sVec& result ) 
{
  std::ostringstream sql;
  sql << "SELECT pool_id FROM pool WHERE p_name = '"
      << poolName
      << "'";
  int poolId = myHelper_.getSingleValueInt( sql );
  sql.str("");
  sql << "SELECT f_name FROM file WHERE pool_id = '"
      << poolId
      << "' ORDER BY f_name";
  myHelper_.getValueVector( sql, result );
  return;
}




void
psfa::DbReadDriverMySql::getUniqueContent( std::string poolName,
                  sVec& result ) 
{
  std::ostringstream sql;
  sql << "SELECT pool_id FROM pool WHERE p_name = '"
      << poolName
      << "'";
  int poolId = myHelper_.getSingleValueInt( sql );
  sql.str("");
  sql << "SELECT DISTINCT e_content FROM entry WHERE pool_id = '"
      << poolId
      << "' ORDER BY e_content";
  myHelper_.getValueVector( sql, result );
  return;
}




int
psfa::DbReadDriverMySql::getEntries( std::string poolName, 
                 SearchData& search,
                 std::vector< IndexData_SP >& result ) 
{
  using std::ostringstream;
  using std::string;
  using mysqlcppapi::Result_Store;
  typedef std::map< string, string > sMap;
  
  ostringstream sql;
  ostringstream tables;
  tables << " entry ";
  
  ostringstream ands;
  if( search.searchTag != "" ) {
    sql.str("");
    sql << "SELECT search_id FROM search WHERE s_name = '"
   << search.searchTag
   << "'";
    int id = myHelper_.getSingleValueInt( sql );
    ands << " AND search_id = '"
    << id
    << "'";
  }
  if( search.content != "" ) {
    ands << " AND e_content = '"
    << search.content
    << "'";
  }
  if( search.attKey != "" && search.attValue != "" ) {
    tables << ", attribut";
    ands << " AND entry.entry_id = attribut.entry_id"
    << " AND attribut.a_key = '"
    << search.attKey
    << "' AND attribut.a_value = '"
    << search.attValue
    << "'";
  }
  if( search.kontextTag != "" ) {
    sql.str("");
    sql << "SELECT kontext_id FROM kontext WHERE k_name = '"
   << search.kontextTag
   << "'";
    int id = myHelper_.getSingleValueInt( sql );
    ands << " AND kontext_id = '"
    << id
    << "'";
  }
  if( search.unit != "" ) {
    sql.str("");
    sql << "SELECT unit_id FROM unit WHERE u_name = '"
   << search.unit
   << "'";
    int id = myHelper_.getSingleValueInt( sql );
    ands << " AND unit_id = '"
    << id
    << "'";
  }

  sql.str("");
  sql << "SELECT pool_id FROM pool WHERE p_name = '"
      << poolName
      << "'";
  int poolId = myHelper_.getSingleValueInt( sql );
  sql.str("");
  sql << "SELECT * FROM"
      << tables.str()
      << " WHERE pool_id ='"
      << poolId
      << "'"
      << ands.str()
      << " ORDER BY e_content";

  Result_Store dbResult;
  myHelper_.queryDb( sql, dbResult );
  
  unsigned int size = dbResult.size();
  if( size == 0 ) {
    return 0;
  }

  sMap nameMap;
  sMap::iterator dbName;
  std::string name;
  ostringstream nameStream;
  
  result.reserve( size );
  IndexData_SP data;
  Result_Store::iterator i;
  Result_Store::iterator iEnd = dbResult.end();
  for( i = dbResult.begin(); i != iEnd; ++i ) {
    mysqlcppapi::Row row = *i;
    data = IndexData_SP( new IndexData );

    if( search.incSearch ) {
      nameStream.str("");
      nameStream << "search_id" << row[2];
      dbName = nameMap.find( nameStream.str() );
      if( dbName == nameMap.end() ) {
   sql.str("");
   sql << "SELECT s_name FROM search WHERE search_id = '"
       << row[2]
       << "'";
   name = myHelper_.getSingleValueString( sql );
   sMap::value_type v( nameStream.str(), name );
   nameMap.insert( v );
   data->tag = name;
      }
      else {
   data->tag = dbName->second;
      }
    }
 
    if( search.incKontext ) {
      nameStream.str("");
      nameStream << "kontext_id" << row[3];
      dbName = nameMap.find( nameStream.str() );
      if( dbName == nameMap.end() ) {
   sql.str("");
   sql << "SELECT k_name FROM kontext WHERE kontext_id = '"
       << row[3]
       << "'";
   name = myHelper_.getSingleValueString( sql );
   sMap::value_type v( nameStream.str(), name );
   nameMap.insert( v );
   data->kontext = name;
      }
      else {
   data->kontext = dbName->second;
      }
    }
    
    if( search.incUnit ) {
      nameStream.str("");
      nameStream << "unit_id" << row[4];
      dbName = nameMap.find( nameStream.str() );
      if( dbName == nameMap.end() ) {
   sql.str("");
   sql << "SELECT u_name FROM unit WHERE unit_id = '"
       << row[4]
       << "'";
   name = myHelper_.getSingleValueString( sql );
   sMap::value_type v( nameStream.str(), name );
   nameMap.insert( v );
      data->unit = name;
      }
      else {
   data->unit = dbName->second;
      }
    }
        
    if( search.incFileName ) {
      nameStream.str("");
      nameStream << "file_id" << row[5];
      dbName = nameMap.find( nameStream.str() );
      if( dbName == nameMap.end() ) {
   sql.str("");
   sql << "SELECT f_name FROM file WHERE file_id = '"
       << row[5]
       << "'";
   name = myHelper_.getSingleValueString( sql );
   sMap::value_type v( nameStream.str(), name );
   nameMap.insert( v );
   data->file = name;
      }
      else {
   data->file = dbName->second;
      }
    }

    data->content     = row[6];

    if( search.incPosData ) {
      data->lineStart   = row[7];
      data->columnStart = row[8];
      data->lineStop    = row[9];
      data->columnStop  = row[10];
      data->posStart    = row[11];
      data->posEnd      = row[12];
    }
    
    if( search.incAtts ) {
      sql.str("");
      sql << "SELECT a_key, a_value FROM attribut WHERE entry_id = '"
     << row[0]
     << "'";
      Result_Store attResult;
      myHelper_.queryDb( sql, attResult );
      unsigned int attSize = attResult.size();
      if( attSize > 0 ) {
   psfa::AttMap atts;
   Result_Store::iterator a;
   Result_Store::iterator aEnd = attResult.end();
   for( a = attResult.begin(); a != aEnd; ++a ) {
     mysqlcppapi::Row attRow = *a;
     AttMap::value_type v( attRow[0], attRow[1] );
     data->attributes.insert( v );
   }
      }
    }
    
    result.push_back( data );
  }
  return result.size();
}

---