/**
 * MXS-1506: Delayed query retry
 *
 * https://jira.mariadb.org/browse/MXS-1506
 */
#include "testconnections.h"
#include <functional>
#include <thread>
#include <iostream>
#include <vector>

using namespace std;

struct TestCase
{
    string description;
    function<void ()> pre;   // Called before master goes down
    function<void ()> block; // Executed in a separate thread before `main` is called
    function<void ()> main;  // Called after master goes down
    function<void ()> check; // Called after `main` and `block` are completed
};

int main(int argc, char** argv)
{
    TestConnections test(argc, argv);

    auto query = [&test](string q, int t = 0)
    {
        sleep(t);
        return execute_query_silent(test.maxscales->conn_rwsplit[0], q.c_str()) == 0;
    };

    auto compare = [&test](string q, string res)
    {
        auto rc = execute_query_check_one(test.maxscales->conn_rwsplit[0], q.c_str(), res.c_str()) == 0;
        test.assert(rc, "Query '%s' did not produce result of '%s'", q.c_str(), res.c_str());
    };

    auto check = [&test, &compare](string q, string res)
    {
        test.repl->sync_slaves();
        test.maxscales->connect();
        compare(q, res);
        test.maxscales->disconnect();
    };

    auto ok = [&test, &query](string q, int t = 0)
    {
        test.assert(query(q, t), "Query '%' should work: %s", q.c_str(), mysql_error(test.maxscales->conn_rwsplit[0]));
    };

    auto err = [&test, &query](string q, int t = 0)
    {
        test.assert(!query(q, t), "Query should fail: %s", q.c_str());
    };

    auto block = [&test](int pre = 0, int node = 0)
    {
        sleep(pre);
        test.repl->block_node(node);
        sleep(10);
        test.repl->unblock_node(node);
    };

    auto noop = [](){};

    vector<TestCase> tests(
    {
        {
            "Normal insert",
            noop,
            block,
            bind(ok, "INSERT INTO test.t1 VALUES (1)", 5),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 1", "1")
        },
        {
            "Insert with user variables",
            bind(ok, "SET @a = 2"),
            block,
            bind(ok, "INSERT INTO test.t1 VALUES (@a)", 5),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 2", "1")
        },
        {
            "Normal transaction",
            bind(ok, "START TRANSACTION"),
            block,
            bind(err, "INSERT INTO test.t1 VALUES (3)", 5),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 3", "0")
        },
        {
            "Read-only transaction",
            bind(ok, "START TRANSACTION READ ONLY"),
            block,
            bind(err, "INSERT INTO test.t1 VALUES (4)", 5),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 4", "0")
        },
        {
            "Insert with autocommit=0",
            bind(ok, "SET autocommit=0"),
            block,
            bind(err, "INSERT INTO test.t1 VALUES (5)", 5),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 5", "0")
        },
        {
            "Interrupted insert (should cause duplicate statement execution)",
            noop,
            bind(block, 5),
            bind(ok, "INSERT INTO test.t1 VALUES ((SELECT SLEEP(10) + 6))", 0),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 6", "2")
        },
        {
            "Interrupted insert with user variable (should cause duplicate statement execution)",
            bind(ok, "SET @b = 7"),
            bind(block, 5),
            bind(ok, "INSERT INTO test.t1 VALUES ((SELECT SLEEP(10) + @b))", 0),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 7", "2")
        },
        {
            "Interrupted insert in transaction",
            bind(ok, "START TRANSACTION"),
            bind(block, 5),
            bind(err, "INSERT INTO test.t1 VALUES ((SELECT SLEEP(10) + 8))", 0),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 8", "0")
        },
        {
            "Interrupted insert in read-only transaction",
            bind(ok, "START TRANSACTION READ ONLY"),
            bind(block, 5),
            bind(err, "INSERT INTO test.t1 VALUES ((SELECT SLEEP(10) + 9))", 0),
            bind(check, "SELECT COUNT(*) FROM test.t1 WHERE id = 9", "0")
        },
        {
            "Interrupted select",
            noop,
            bind(block, 5, 1),
            bind(compare, "SELECT SLEEP(10) + 10", "10"),
            noop
        },
        {
            "Interrupted select with user variable",
            bind(ok, "SET @c = 11"),
            bind(block, 5, 1),
            bind(compare, "SELECT SLEEP(10) + @c", "11"),
            noop
        },
        {
            "Interrupted select in transaction",
            bind(ok, "START TRANSACTION"),
            bind(block, 5, 0),
            bind(err, "SELECT SLEEP(10)"),
            noop
        },
        {
            "Interrupted select in read-only transaction",
            bind(ok, "START TRANSACTION READ ONLY"),
            bind(block, 5, 1),
            bind(err, "SELECT SLEEP(10)"),
            noop
        }
    });

    cout << "Create table for testing" << endl;
    test.maxscales->connect();
    ok("DROP TABLE IF EXISTS test.t1");
    ok("CREATE TABLE test.t1 (id INT)");
    test.maxscales->disconnect();

    for (auto a : tests)
    {
        cout << a.description << endl;
        test.maxscales->connect();
        a.pre();
        thread thr(a.block);
        a.main();
        test.maxscales->disconnect();
        thr.join();
        a.check();
    }

    test.maxscales->connect();
    query("DROP TABLE test.t1");
    test.maxscales->disconnect();

    return test.global_result;
}