mysql_connect.hpp

浏览该文件的文档.

#ifndef NEFORCE_DATABASE_MYSQL_CONNECT_HPP__
#define NEFORCE_DATABASE_MYSQL_CONNECT_HPP__

 
#ifdef NEFORCE_SUPPORT_MYSQL
#    include <mysql/mysql.h>
#    include "NeForce/db/sql_connect_base.hpp"
NEFORCE_BEGIN_NAMESPACE__



struct NEFORCE_API mysql_connect final : sql_connect_base<mysql_connect> {
protected:
    ::MYSQL* link_ = nullptr; 
    friend sql_connect_base<mysql_connect>;
 
private:
    mutable string last_error_;       
    mutable uint32_t last_errno_ = 0; 
 
public:
    mysql_connect() { link_ = ::mysql_init(nullptr); }

    ~mysql_connect() noexcept override { close(); }

    NEFORCE_NODISCARD bool connect(const db_config& config) override;

    NEFORCE_NODISCARD bool reconnect(const db_config& config) override;

    void close() noexcept override;

    NEFORCE_NODISCARD bool set_character_set(const string& encoding) noexcept override;

    NEFORCE_NODISCARD bool set_options(::mysql_option option, const string& str) const noexcept;

    NEFORCE_NODISCARD string_view get_character_set() const noexcept override;

    NEFORCE_NODISCARD string_view get_error() const noexcept override;

    NEFORCE_NODISCARD uint32_t get_errno() const noexcept override;

    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 noexcept override { return link_ != nullptr; }

    NEFORCE_NODISCARD bool is_valid() const noexcept override { return ::mysql_ping(link_) == 0; }
 
    size_t batch_insert(const string& table, const vector<string>& columns,
                        const vector<vector<string>>& rows) override;
 
private:
    string begin_sql() { return "START TRANSACTION"; }
    string commit_sql() { return "COMMIT"; }
    string rollback_sql() { return "ROLLBACK"; }
    string table_exists_query(const string& table) const { return "SHOW TABLES LIKE '" + table + "'"; }
};

class NEFORCE_API mysql_factory final : public idb_factory {
public:
    explicit mysql_factory(db_config config) noexcept :
    idb_factory(move(config)) {}
 
    mysql_factory(const mysql_factory&) = delete;
    mysql_factory& operator=(const mysql_factory&) = delete;
 
    mysql_factory(mysql_factory&&) noexcept = default;
    mysql_factory& operator=(mysql_factory&&) noexcept = default;
 
    ~mysql_factory() noexcept override = default;

    idb_connect* create_connect() override;

    idb_result* create_result(void* native_result) override;
};
 // MySQL
 // Database
 
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_SUPPORT_MYSQL
#endif // NEFORCE_DATABASE_MYSQL_CONNECT_HPP__