mysql.hh

#ifndef MYSQL_HH
#define MYSQL_HH

extern "C"  {
#include <mysql/mysql.h>
}

#include "schema.hh"
#include "relmodel.hh"
#include "dut.hh"

#include <sys/time.h> // for gettimeofday

#define MYSQL_STMT_BLOCK_MS 100

struct mysql_connection {
    MYSQL mysql;
    string test_db;
    unsigned int test_port;
    mysql_connection(string db, unsigned int port);
    ~mysql_connection();
};

struct schema_mysql : schema, mysql_connection {
    schema_mysql(string db, unsigned int port);
    virtual void update_schema();
    virtual std::string quote_name(const std::string &id) {
        return id;
    }
};

struct dut_mysql : dut_base, mysql_connection {
    virtual void test(const string &stmt, 
        vector<vector<string>>* output = NULL, 
        int* affected_row_num = NULL,
        vector<string>* env_setting_stmts = NULL);
    virtual void reset(void);

    virtual void backup(void);
    virtual void reset_to_backup(void);
    
    virtual string commit_stmt();
    virtual string abort_stmt();
    virtual string begin_stmt();

    virtual void get_content(vector<string>& tables_name, map<string, vector<vector<string>>>& content);
    virtual string get_process_id();

    static pid_t fork_db_server();
    dut_mysql(string db, unsigned int port);

    static int save_backup_file(string testdb, string path);
    void block_test(const string &stmt, vector<vector<string>>* output = NULL, int* affected_row_num = NULL);
    bool check_whether_block(vector<unsigned long>& blocking_tids);
    bool has_sent_sql;
    string sent_sql;
    bool txn_abort;
    unsigned long thread_id;
};

#endif