#include <string>
#include <map>
#include <iostream>
#include <fstream>
#include <sstream>
#include <stdexcept>
#include <vector>
#include <set>
#include <algorithm>
#include <stdio.h>
#include <assert.h>

#include <mysql.h>

#include <util/errstream.hh>
#include <parser/embedmysql.hh>

using namespace std;

int
main(int ac, char **av)
{
    if (ac != 2) {
        cout << "Usage: " << av[0] << " db-dir < queries-file" << endl;
        exit(1);
    }

    embedmysql em(av[1]);

    stringstream ss;
    vector<string> queries;

    for (;;) {
        string s;
        getline(cin, s);
        if (!cin.good())
            break;

        if (s.substr(0, 4) == "USE ") {
            queries.push_back(ss.str());
            ss.str("");
        }
        ss << s << endl;
    }
    queries.push_back(ss.str());

    uint ndb = 0;
    for (const string &q: queries) {
        MYSQL *m = em.conn();

        if (mysql_query(m, q.c_str()))
            fatal() << "mysql_query: " << mysql_error(m);

        for (;;) {
            MYSQL_RES *r = mysql_store_result(m);
            if (r) {
                // cout << "got result.." << endl;
                mysql_free_result(r);
            } else if (mysql_field_count(m) == 0) {
                // cout << "rows affected: " << mysql_affected_rows(m) << endl;
            } else {
                fatal() << "could not retrieve result set";
            }

            int s = mysql_next_result(m);
            if (s > 0) {
                cout << "mysql_next_result: " << mysql_error(m) << endl
                     << "in batch: " << q << endl;
                break;
            }

            if (s < 0)
                break;
        }

        ndb++;
        if (!(ndb % 100))
            cout << "processed " << ndb << "/" << queries.size()
                 << " batches" << endl;
    }

    cout << "done" << endl;
    return 0;
}