mariadb-columnstore-engine/tools/brmtest/locks.cpp

/*
* $Id: locks.cpp 282 2007-10-28 02:18:55Z rdempsey $
*/

#include <iostream>
#include <stdexcept>
using namespace std;

#include "sessionmanager.h"
#include "calpontsystemcatalog.h"
using namespace execplan;

#include "shmkeys.h"
#include "rwlock.h"
#include "mastersegmenttable.h"
using namespace BRM;

int query_locks()
{
    ShmKeys keys;
    RWLock* rwlock[MasterSegmentTable::nTables];
    int RWLockKeys[MasterSegmentTable::nTables];
    int i;

    RWLockKeys[0] = keys.KEYRANGE_EXTENTMAP_BASE;
    RWLockKeys[1] = keys.KEYRANGE_EMFREELIST_BASE;
    RWLockKeys[2] = keys.KEYRANGE_VBBM_BASE;
    RWLockKeys[3] = keys.KEYRANGE_VSS_BASE;
    RWLockKeys[4] = keys.KEYRANGE_CL_BASE;

    for (i = 0; i < MasterSegmentTable::nTables; i++)
        rwlock[i] = new RWLock(RWLockKeys[i]);

    for (i = 0; i < MasterSegmentTable::nTables; i++)
        if (rwlock[i]->getWriting() > 0 ||
                //rwlock[i]->getReading() > 0 ||
                rwlock[i]->getWritersWaiting() > 0 ||
                rwlock[i]->getReadersWaiting() > 0)
            return 1;

    for (i = 0; i < MasterSegmentTable::nTables; i++)
        delete rwlock[i];

    return 0;
}

int reset_locks()
{
    ShmKeys keys;
    RWLock* rwlock[MasterSegmentTable::nTables];
    int RWLockKeys[MasterSegmentTable::nTables];
    int i;
    SessionManager sm(true);

    RWLockKeys[0] = keys.KEYRANGE_EXTENTMAP_BASE;
    RWLockKeys[1] = keys.KEYRANGE_EMFREELIST_BASE;
    RWLockKeys[2] = keys.KEYRANGE_VBBM_BASE;
    RWLockKeys[3] = keys.KEYRANGE_VSS_BASE;
    RWLockKeys[4] = keys.KEYRANGE_CL_BASE;

    for (i = 0; i < MasterSegmentTable::nTables; i++)
        rwlock[i] = new RWLock(RWLockKeys[i]);

    for (i = 0; i < MasterSegmentTable::nTables; i++)
        rwlock[i]->reset();

    for (i = 0; i < MasterSegmentTable::nTables; i++)
        delete rwlock[i];

    try
    {
        sm.reset();
    }
    catch (exception& e)
    {
        cout << e.what() << endl;
        return -1;
    }

    return 0;
}