mariadb-columnstore-engine/versioning/BRM/copylocktest.cpp

#include <iostream>
#include "dbrm.h"
#include "IDBPolicy.h"

using namespace std;
using namespace BRM;

CopyLocks copylocks;
DBRM dbrm;

void lockRange()
{
  LBIDRange r;
  r.start = 12345;
  r.size = 5;
  copylocks.lock(CopyLocks::WRITE);

  if (copylocks.isLocked(r))
    cout << "already locked\n";
  else
  {
    copylocks.lockRange(r, -1);
    cout << "got the 1st lock\n";
  }

  r.start = 12349;

  if (copylocks.isLocked(r))
    cout << "2nd range is locked\n";
  else
    cout << "2nd range is NOT locked\n";

  r.start = 12350;

  if (copylocks.isLocked(r))
    cout << "3rd range is locked\n";
  else
  {
    copylocks.lockRange(r, -1);
    cout << "got the 3rd lock\n";
  }

  copylocks.release(CopyLocks::WRITE);
}

void dbrmLockRange()
{
  dbrm.lockLBIDRange(12345, 5);
  cout << "OK\n";
}

void dbrmReleaseRange()
{
  dbrm.releaseLBIDRange(12345, 5);
  cout << "OK\n";
}

void releaseRange()
{
  LBIDRange r;
  r.start = 12345;
  r.size = 5;
  copylocks.lock(CopyLocks::WRITE);
  copylocks.releaseRange(r);

  r.start = 12350;
  copylocks.releaseRange(r);
  copylocks.release(CopyLocks::WRITE);

  cout << "OK\n";
}

int main(int argc, char** argv)
{
  char cmd;

  if (argc < 2)
    goto usage;

  cmd = argv[1][0];

  idbdatafile::IDBPolicy::configIDBPolicy();

  switch (cmd)
  {
    case 'l': lockRange(); break;

    case 'r': releaseRange(); break;

    case 'L': dbrmLockRange(); break;

    case 'R': dbrmReleaseRange(); break;

    default: goto usage;
  }

  exit(0);
usage:
  cout << "Usage: " << argv[0] << " l | r | L | R. lock/release. Check the code for specifics. :P" << endl;
  exit(1);
}