sqlite_connect.hpp

浏览该文件的文档.

#ifndef NEFORCE_DATABASE_SQLITE_CONNECT_HPP__
#define NEFORCE_DATABASE_SQLITE_CONNECT_HPP__

 
#ifdef NEFORCE_SUPPORT_SQLITE3
#    ifdef NEFORCE_SUPPORT_SQLCIPHER
#        include <sqlcipher/sqlite3.h>
 
#        include <utility>
#    else
#        include <sqlite3.h>
#    endif
#    include "NeForce/db/sql_connect_base.hpp"
NEFORCE_BEGIN_NAMESPACE__



struct NEFORCE_API sqlite_connect final : sql_connect_base<sqlite_connect> {
public:
#    ifdef NEFORCE_SUPPORT_SQLCIPHER
    enum class key_type : byte_t {
        PBKDF2, 
        RAW     
    };
#    endif
 
protected:
    ::sqlite3* link_ = nullptr; 
    friend sql_connect_base<sqlite_connect>;
 
private:
    mutable string last_error_;       
    mutable uint32_t last_errno_ = 0; 
 
public:
    sqlite_connect() noexcept { ::sqlite3_open(nullptr, &link_); }

    ~sqlite_connect() noexcept override { this->close(); }

    bool connect(const db_config& config) override;

    NEFORCE_NODISCARD bool reconnect(const db_config& config) override;
 
#    ifdef NEFORCE_SUPPORT_SQLCIPHER

    bool connect(const db_config& config, const string& encryption_key, key_type type = key_type::PBKDF2);

    NEFORCE_NODISCARD bool reconnect(const db_config& config, const string& encryption_key,
                                     key_type type = key_type::PBKDF2);

    NEFORCE_NODISCARD bool rekey(const string& new_key, key_type type = key_type::PBKDF2);
 
#    endif

    void close() noexcept override;

    NEFORCE_NODISCARD bool set_character_set(const string& encoding) override;

    NEFORCE_NODISCARD string_view get_character_set() const override;

    NEFORCE_NODISCARD string_view get_error() const override;

    NEFORCE_NODISCARD uint32_t get_errno() const override { return last_errno_; }

    NEFORCE_NODISCARD bool update(const string& sql) const override;

    NEFORCE_NODISCARD unique_ptr<idb_tb_result> query(const string& sql) const override;

    NEFORCE_NODISCARD unique_ptr<idb_prepared_statement> prepare_statement(const string& sql) const override;

    NEFORCE_NODISCARD bool connected() const override { return link_ != nullptr; }

    NEFORCE_NODISCARD bool is_valid() const override;
 
    NEFORCE_NODISCARD bool table_exists(const string& table) const override;
 
    size_t batch_insert(const string& table, const vector<string>& columns,
                        const vector<vector<string>>& rows) override;
 
private:
    string begin_sql() { return "BEGIN TRANSACTION"; }
    string commit_sql() { return "COMMIT"; }
    string rollback_sql() { return "ROLLBACK"; }
    string table_exists_query(const string& table) const {
        return "SELECT name FROM sqlite_master WHERE type='table' AND name='" + table + "'";
    }
};

class NEFORCE_API sqlite_factory final : public idb_factory {
#    ifdef NEFORCE_SUPPORT_SQLCIPHER
    string encryption_key_;                                                
    sqlite_connect::key_type key_type_ = sqlite_connect::key_type::PBKDF2; 
#    endif
 
public:
    explicit sqlite_factory(db_config config) :
    idb_factory(move(config)) {}
 
#    ifdef NEFORCE_SUPPORT_SQLCIPHER
    explicit sqlite_factory(db_config config, string encryption_key,
                            sqlite_connect::key_type type = sqlite_connect::key_type::PBKDF2) :
    idb_factory(move(config)),
    encryption_key_(move(encryption_key)),
    key_type_(type) {}
#    endif

    idb_connect* create_connect() override;

    idb_result* create_result(void* native_result) override;
};
 // SQLite3
 // Database
 
NEFORCE_END_NAMESPACE__
#endif
#endif // NEFORCE_DATABASE_SQLITE_CONNECT_HPP__