diff options
Diffstat (limited to 'src/components/policy/policy_external/test/include/policy/driver_dbms.h')
-rw-r--r-- | src/components/policy/policy_external/test/include/policy/driver_dbms.h | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/src/components/policy/policy_external/test/include/policy/driver_dbms.h b/src/components/policy/policy_external/test/include/policy/driver_dbms.h new file mode 100644 index 0000000000..e438c87b03 --- /dev/null +++ b/src/components/policy/policy_external/test/include/policy/driver_dbms.h @@ -0,0 +1,159 @@ +/* + * Copyright (c) 2016, Ford Motor Company + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following + * disclaimer in the documentation and/or other materials provided with the + * distribution. + * + * Neither the name of the Ford Motor Company nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ +#ifndef SRC_COMPONENTS_POLICY_POLICY_EXTERNAL_TEST_INCLUDE_POLICY_DRIVER_DBMS_H_ +#define SRC_COMPONENTS_POLICY_POLICY_EXTERNAL_TEST_INCLUDE_POLICY_DRIVER_DBMS_H_ + +#ifdef __QNX__ +#include <qdb/qdb.h> +#else // __QNX__ +#include <sqlite3.h> +#endif // __QNX__ + +namespace test { +namespace components { +namespace policy_test { + +#ifdef __QNX__ +class DBMS { + public: + explicit DBMS(std::string db_name) : db_name_(db_name), conn_(0) {} + ~DBMS() { + Close(); + } + bool Open() { + conn_ = qdb_connect(db_name_.c_str(), 0); + return conn_ != NULL; + } + void Close() { + qdb_disconnect(conn_); + } + bool Exec(const char* query) { + return -1 != qdb_statement(conn_, query); + } + int FetchOneInt(const char* query) { + int stmt = qdb_stmt_init(conn_, query, strlen(query) + 1); + qdb_stmt_exec(conn_, stmt, NULL, 0); + qdb_result_t* res = qdb_getresult(conn_); + void* ret = qdb_cell(res, 0, 0); + int value = 0; + if (ret) { + value = *static_cast<int*>(ret); + } + qdb_stmt_free(conn_, stmt); + return value; + } + double FetchOneDouble(const char* query) { + int stmt = qdb_stmt_init(conn_, query, strlen(query) + 1); + qdb_stmt_exec(conn_, stmt, NULL, 0); + qdb_result_t* res = qdb_getresult(conn_); + void* ret = qdb_cell(res, 0, 0); + double value = 0.0; + if (ret) { + value = *static_cast<double*>(ret); + } + qdb_stmt_free(conn_, stmt); + + return value; + } + std::string FetchOneString(const char* query) { + int stmt = qdb_stmt_init(conn_, query, strlen(query) + 1); + qdb_stmt_exec(conn_, stmt, NULL, 0); + qdb_result_t* res = qdb_getresult(conn_); + void* ret = qdb_cell(res, 0, 0); + std::string value = ""; + if (ret) { + value = std::string(static_cast<const char*>(ret)); + } + qdb_stmt_free(conn_, stmt); + + return value; + } + + private: + std::string db_name_; + qdb_hdl_t* conn_; +}; + +#else // __QNX__ +class DBMS { + public: + explicit DBMS(std::string file_name) : file_name_(file_name), conn_(0) {} + ~DBMS() { + Close(); + } + bool Open() { + return SQLITE_OK == sqlite3_open(file_name_.c_str(), &conn_); + } + void Close() { + sqlite3_close(conn_); + remove(file_name_.c_str()); + } + bool Exec(const char* query) { + return SQLITE_OK == sqlite3_exec(conn_, query, NULL, NULL, NULL); + } + int FetchOneInt(const char* query) { + sqlite3_stmt* statement; + sqlite3_prepare(conn_, query, -1, &statement, NULL); + sqlite3_step(statement); + int value = sqlite3_column_int(statement, 0); + sqlite3_finalize(statement); + return value; + } + double FethcOneDouble(const char* query) { + sqlite3_stmt* statement; + sqlite3_prepare(conn_, query, -1, &statement, NULL); + sqlite3_step(statement); + double value = sqlite3_column_double(statement, 0); + sqlite3_finalize(statement); + return value; + } + std::string FetchOneString(const char* query) { + sqlite3_stmt* statement; + sqlite3_prepare(conn_, query, -1, &statement, NULL); + sqlite3_step(statement); + const unsigned char* txt = sqlite3_column_text(statement, 0); + std::string value = std::string(reinterpret_cast<const char*>(txt)); + sqlite3_finalize(statement); + return value; + } + + private: + std::string file_name_; + sqlite3* conn_; +}; +#endif // __QNX__ + +} // namespace policy_test +} // namespace components +} // namespace test + +#endif // SRC_COMPONENTS_POLICY_POLICY_EXTERNAL_TEST_INCLUDE_POLICY_DRIVER_DBMS_H_ |