db_helper.hpp

Go to the documentation of this file.

#ifndef QNLP_DBHELPER
#define QNLP_DBHELPER

#include <iostream> 
#include <sqlite3.h> 
#include <string> 

//#############################################################################
 
//Return-type checking macro for SQL statements; compares the result x with that of y, and exits with the respective error-code if they differ. Macro wraps inline to return linenumber and filename for resultant cause.
#define SQL_RETURN_CHECK(x,y) sql_return_check(x, y, __FILE__, __LINE__)
inline void sql_return_check(   const int ret_code, 
                                const int expected_code, 
                                const char* fileName, 
                                const std::size_t lineNum){
#ifdef SQL_ERR_CHECK
    if ( ret_code != expected_code ) { 
        std::cerr << "SQL error: FILE: " << fileName << "\t LINE: " << lineNum << "\t RETURN_CODE: " << ret_code << std::endl; 
        std::exit(ret_code); 
    }
#endif
}

//#############################################################################

namespace QNLP{
    class DBHelper{
    public:

        DBHelper();
        DBHelper(const std::string filename) ;
        ~DBHelper();

        int openDB(const std::string filename);

        void closeDB();

        sqlite3* getDBRef();

        std::string getFilename();

    private:
        //Pointer to database object
        sqlite3* db_ptr;
        std::string db_filename;
    };
}
#endif