sqlitepp.cpp

Go to the documentation of this file.

/*
  Phrasehunter - index and query text corpora
  Copyright (C) 2006  Torsten Marek (shlomme@gmx.de) &
  Armin Schmidt (armin.sch@gmail.com)
  
  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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

#include "sqlitepp/sqlitepp.h"

namespace SQLitePP {

sqlite3_stmt* prepareStatement(const char* sqlcode, sqlite3* db) 
{
    const char* tail;
    sqlite3_stmt* stmt;
    int rcode = sqlite3_prepare(db,
                                sqlcode, -1,
                                &stmt, &tail);
    if(rcode != SQLITE_OK) {
        showSqlite3Error(__PRETTY_FUNCTION__, db);            
    }
    return stmt;
}


void showSqlite3Error(const char* infunction, sqlite3* db) 
{
    std::cerr << infunction << ":" << sqlite3_errmsg(db) << std::endl;
    abort();
}

template<>
int ResultIterator::get<int>(int column) 
{
    return sqlite3_column_int(m_stmt, column);
}

template<>
unsigned int ResultIterator::get<unsigned int>(int column) 
{
    return sqlite3_column_int(m_stmt, column);
}

template<>
const char* ResultIterator::get<const char*>(int column)
{
    return reinterpret_cast<const char*>(sqlite3_column_text(m_stmt, column));
}

template<>
Blob ResultIterator::get<Blob>(int column) 
{
    return std::make_pair(sqlite3_column_blob(m_stmt, column), 
                          sqlite3_column_bytes(m_stmt, column));
}

template<>
std::string ResultIterator::get<std::string>(int column)
{
    std::string s;
    s.assign(reinterpret_cast<const char*>(sqlite3_column_blob(m_stmt, column)),
             sqlite3_column_bytes(m_stmt, column));
    return s;
}

template<>
schma::UnicodePtr ResultIterator::get<schma::UnicodePtr>(int column)
{
    return schma::UnicodePtr(new UnicodeString(reinterpret_cast<const char*>(sqlite3_column_text(m_stmt, column))));
}

SqliteDB::SqliteDB(const std::string& filename)
{
    int rcode = sqlite3_open(filename.c_str(), &m_dbhandle);
    if(rcode != SQLITE_OK) {
        std::cerr << sqlite3_errmsg(m_dbhandle) << std::endl;
        p_assert(rcode == SQLITE_OK, "db could not be opened");
    }
    sqlite3_stmt* stmt;
    const char* tail;
    sqlite3_prepare(m_dbhandle, "pragma synchronous = OFF", -1,
                    &stmt, &tail);
    sqlite3_step(stmt);
    sqlite3_finalize(stmt);
}

}

---