Beesnest/Adapters/BnSQLite/BnSQLite.cpp

Go to the documentation of this file.

#include "BnSQLite.h"


static int callback(void* p_this, int argc, char** argv, char** azColName)
{
    BnSQLite* pThis = (BnSQLite*)p_this;
    if (!pThis) return -1;

    /* Initialize fields names map if it is empty. */
    if (pThis->mLastQueryNames.empty ())
        for (int i = 0; i < argc; i++)
            pThis->mLastQueryNames [azColName[i]] = i;

    /* Add line to the results vector, and fill it with values. */
    pThis->vLastQuery.push_back (t_query_fields (argc));
    for(int i = 0; i < argc; i++)
        pThis->vLastQuery.back ()[i] = argv[i] ? argv[i] : "NULL";

    pThis->vLastQueryItr = pThis->vLastQuery.begin ();
    return 0;
}

BnSQLite::BnSQLite (const char* name, bool (*pGetParam)(const char*, char*, int))
:nLastErrCode (0), csErrMsg (NULL), isInitErr (false), db (NULL)
{
    sName = name;
    pfGetParam = pGetParam;

    char db_path[50] = {0};
    char path[50];
    sprintf (path, "Adapters->%s->DB", name);
    isInitErr = !pGetParam (path, db_path, 49);

    if (!isInitErr)
        isInitErr = sqlite3_open(db_path, &db);
}

BnSQLite::~BnSQLite ()
{
    if (csErrMsg)
        sqlite3_free(csErrMsg);
    sqlite3_close(db);
}

BufferHolder BnSQLite::Read (BufferHolder address) throw (BufferHolder)
{
    if (isInitErr)
    {   /* If Init failed throw an exception. */
        sRetMsg = "Cannot initialize BnSQLite: ";
        if (db)
            sRetMsg += sqlite3_errmsg(db);
        throw BufferHolder (sRetMsg.c_str ());
    }

    sRetMsg.clear ();
    if (address.empty ()) return
        BufferHolder (sRetMsg.c_str ()); /* Empty string */
    string cmd (address.pointer, address.size);

    if (cmd == "_LastError")
    {   /* Error message */
        if (nLastErrCode == SQLITE_OK)
            sRetMsg = "OK";
        else
        {   /* We do have an error */
            if (csErrMsg)
                sRetMsg = csErrMsg;
            char buff [40];
            sprintf (buff, " (%d)", nLastErrCode);
            sRetMsg += buff;
        }
    }
    else if (cmd == "_EOF")
    {   /* Is iterator in the end? */
        sRetMsg = (vLastQueryItr == vLastQuery.end () ? "True" : "False");
    }
    else if (cmd == "_FieldsList")
    {
        for (t_fields_names_itr itr = mLastQueryNames.begin ();
                itr != mLastQueryNames.end (); ++itr)
            sRetMsg += itr->first + ' ';
    }
    else if (vLastQueryItr != vLastQuery.end ())
    {   /* Read a field value */
        t_fields_names_itr itr = mLastQueryNames.find (cmd);
        if (itr != mLastQueryNames.end () && itr->second < vLastQueryItr->size ())
            sRetMsg = (*vLastQueryItr)[itr->second];
        else
        {   /* Or by field number */
            int num = atoi(cmd.c_str ());
            if ((num > 0 || cmd == "0") && num < vLastQueryItr->size ())
                sRetMsg = (*vLastQueryItr)[num];
        }
    }
    return BufferHolder (sRetMsg.c_str ());
}

bool BnSQLite::Write (BufferHolder address, BufferHolder value) throw (BufferHolder)
{
    if (isInitErr)
    {   /* If Init failed throw an eception. */
        sRetMsg = "Cannot initialize BnSQLite: ";
        if (db)
            sRetMsg += sqlite3_errmsg(db);
        throw BufferHolder (sRetMsg.c_str ());
    }

    if (address.empty ()) return false;
    string cmd (address.pointer, address.size);

    if (cmd == "Clear")
    {
        mLastQueryNames.clear ();
        for (t_query_itr itr = vLastQuery.begin (); itr != vLastQuery.end (); ++itr)
            itr->clear ();
        vLastQuery.clear ();
        return true;
    }
    else if (cmd == "MoveFirst")
    {
        vLastQueryItr = vLastQuery.begin ();
        return true;
    }
    else if (cmd == "MoveNext")
    {
        if (vLastQueryItr != vLastQuery.end ()) vLastQueryItr++;
        return true;
    }
    else if (cmd == "Exec")
    {
        if (value.empty ()) return false;
        if (csErrMsg) sqlite3_free(csErrMsg);
        string exec (value.pointer, value.size);
        nLastErrCode = sqlite3_exec(db, exec.c_str (), callback, this, &csErrMsg);
        return  nLastErrCode == SQLITE_OK;
    }
    return false;
}