zhipuzi_pos_windows/lewaimai_dispatch/tool/CSqlite3.cpp

#include "../pch/pch.h"
#include "CSqlite3.h"

#include "CSetting.h"

CSqlite3::CSqlite3()
{
    //如果没有这个文件，这里会创建这个文件
    m_rc = sqlite3_open("db/pos.db", &m_db);

    if(m_rc)
    {
        LOG_INFO("Can't open database: " << sqlite3_errmsg(m_db));
        return;
    }

    else
    {
        LOG_INFO("Opened database successfully");
    }
}


CSqlite3::~CSqlite3()
{
	if (m_db != NULL)
	{
		sqlite3_close(m_db);
	}
}

bool CSqlite3::InitConfig()
{
    //检查有没有pos_config这个表，如果没有就创建
    std::string sql = "SELECT COUNT(*) FROM sqlite_master where type = 'table' and name = 'pos_config';";
    sqlite3_stmt * stmt = NULL;

    if(sqlite3_prepare_v2(m_db, sql.c_str(), -1, &stmt, NULL) == SQLITE_OK)
    {
        if(sqlite3_step(stmt) == SQLITE_ROW)
        {
            int count = sqlite3_column_int(stmt, 0);

            if(count == 0)
            {
                //说明没找到这个表，那么这个时候新建这个表，先释放前面的stmt
                sqlite3_finalize(stmt);
                stmt = NULL;

                sql = "CREATE TABLE pos_config("  \
                      "name  CHAR(100)  UNIQUE       NOT NULL," \
                      "value CHAR(2000)              NOT NULL);";

                if(sqlite3_prepare_v2(m_db, sql.c_str(), -1, &stmt, NULL) == SQLITE_OK)
                {
                    //执行该语句
                    if(sqlite3_step(stmt) != SQLITE_OK)
                    {
                        LOG_INFO("create table fail: " << sqlite3_errmsg(m_db));

                        sqlite3_finalize(stmt);
                        return false;
                    }

                    //走到这里就是表创建成功了
                    LOG_INFO("create table success");
                    sqlite3_finalize(stmt);
                }

                else
                {
                    LOG_INFO("create table prepare fail: " << sqlite3_errmsg(m_db));

                    sqlite3_finalize(stmt);

                    return false;
                }
            }

            else
            {
                //说明已经有这个表了，就不用再创建了
                sqlite3_finalize(stmt);
            }

            std::string sql = "SELECT * FROM pos_config;";
            sqlite3_stmt * stmt = NULL;

            if(sqlite3_prepare_v2(m_db, sql.c_str(), -1, &stmt, NULL) == SQLITE_OK)
            {
                while(sqlite3_step(stmt) == SQLITE_ROW)
                {
                    std::string name = (char*)sqlite3_column_text(stmt, 0);
                    std::string value = (char*)sqlite3_column_text(stmt, 1);

                    CSetting::SetParam(name, value, false);
                }

                sqlite3_finalize(stmt);

                CSetting::Init();
            }

            else
            {
                //异常情况
                sqlite3_finalize(stmt);
                return false;
            }
        }

        else
        {
            //异常情况
            sqlite3_finalize(stmt);
            return false;
        }
    }

    else
    {
        //异常情况
        sqlite3_finalize(stmt);
        return false;
    }

    return true;
}

bool CSqlite3::SaveParams(std::map<std::string, std::string>& params)
{
    int result = sqlite3_exec(m_db, "BEGIN;", 0, 0, 0);

    std::string sql = "delete from pos_config;";
    result = sqlite3_exec(m_db, sql.c_str(), 0, 0, 0);

    for(std::map<std::string, std::string>::iterator it = params.begin(); it != params.end(); it++)
    {
        std::string name = it->first;
        std::string value = it->second;

        sql = "INSERT INTO pos_config (name, value) VALUES ('" + name + "','" + value + "');";
        result = sqlite3_exec(m_db, sql.c_str(), 0, 0, 0);
    }

    result = sqlite3_exec(m_db, "COMMIT;", 0, 0, 0);

    if(result == SQLITE_OK)
    {
        LOG_INFO("save params success");
        return true;
    }

    LOG_INFO("save params fail");
    return false;
}