mariadb-columnstore-engine/tools/clearShm/main.cpp

/* Copyright (C) 2014 InfiniDB, Inc.

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation; version 2 of
   the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA. */

// $Id: main.cpp 2101 2013-01-21 14:12:52Z rdempsey $

#include "mcsconfig.h"

#include <iostream>
#include <sys/types.h>
#include <cstring>
#include <cerrno>
#include <unistd.h>
#include <cstdlib>
#include <cstdio>
#include <sstream>
#include <iomanip>
#include <string>
#include <mutex>
using namespace std;

#include <boost/interprocess/shared_memory_object.hpp>
#include <boost/interprocess/sync/named_semaphore.hpp>
namespace bi = boost::interprocess;

#include <boost/thread/thread.hpp>
using namespace boost;

#include "shmkeys.h"
using namespace BRM;

namespace
{
bool vFlg;
bool nFlg;
std::mutex coutMutex;

void shmDoit(key_t shm_key, const string& label)
{
  string key_name = ShmKeys::keyToName(shm_key);

  if (vFlg)
  {
    try
    {
      bi::shared_memory_object memObj(bi::open_only, key_name.c_str(), bi::read_only);
      bi::offset_t memSize = 0;
      memObj.get_size(memSize);
      std::lock_guard<std::mutex> lk(coutMutex);
      cout << label << ": shm|sem_key: " << shm_key << "; key_name: " << key_name << "; size: " << memSize
           << endl;
    }
    catch (...)
    {
    }
  }

  if (!nFlg)
  {
    bi::shared_memory_object::remove(key_name.c_str());
  }
}

void semDoit(key_t sem_key, const string& label)
{
  shmDoit(sem_key, label);
}

void shmDoitRange(key_t shm_key, const string& label)
{
  if (shm_key == 0)
    return;

  unsigned shm_key_cnt;

  for (shm_key_cnt = 0; shm_key_cnt < ShmKeys::KEYRANGE_SIZE; shm_key_cnt++, shm_key++)
  {
    shmDoit(shm_key, label);
  }
}

void usage()
{
  cout << "usage: clearShm [-cvnh]" << endl;
  cout << "   delete all ColumnStore shared memory data" << endl;
  cout << "   -h display this help" << endl;
  cout << "   -c only clear ColumnStore Engine data, leave OAM intact" << endl;
  cout << "   -v verbose output" << endl;
  cout << "   -n don't actually delete anything (implies -v)" << endl;
}

class ThdFunc
{
 public:
  ThdFunc() : fShm_key(0)
  {
  }
  ThdFunc(key_t shm_key, const string& label) : fShm_key(shm_key), fLabel(label)
  {
  }

  ~ThdFunc()
  {
  }

  void operator()() const
  {
    shmDoitRange(fShm_key, fLabel);
  }

 private:
  // ThdFunc(const ThdFunc& rhs);
  // ThdFunc& operator=(const ThdFunc& rhs);

  key_t fShm_key;
  string fLabel;
};

}  // namespace

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

  int c;
  opterr = 0;
  bool cFlg = false;
  vFlg = false;
  nFlg = false;

  while ((c = getopt(argc, argv, "cvnh")) != EOF)
    switch (c)
    {
      case 'c': cFlg = true; break;

      case 'v': vFlg = true; break;

      case 'n': nFlg = true; break;

      case 'h':
      default:
        usage();
        return (c == 'h' ? 0 : 1);
        break;
    }

  if (nFlg)
    vFlg = true;

  ShmKeys BrmKeys;

  boost::thread_group tg;
  boost::thread* tp = 0;
  tp = new boost::thread(ThdFunc(BrmKeys.KEYRANGE_CL_BASE, "COPYLOCK   "));
  tg.add_thread(tp);
  tp = new boost::thread(ThdFunc(BrmKeys.KEYRANGE_EXTENTMAP_BASE, "EXTMAP     "));
  tg.add_thread(tp);
  tp = new boost::thread(ThdFunc(BrmKeys.KEYRANGE_EMFREELIST_BASE, "EXTMAP_FREE"));
  tg.add_thread(tp);
  tp = new boost::thread(ThdFunc(BrmKeys.KEYRANGE_VBBM_BASE, "VBBM       "));
  tg.add_thread(tp);
  tp = new boost::thread(ThdFunc(BrmKeys.KEYRANGE_VSS_BASE, "VSS        "));
  tg.add_thread(tp);
  tp = new boost::thread(ThdFunc(BrmKeys.KEYRANGE_EXTENTMAP_INDEX_BASE, "EXTMAP_INDX"));
  tg.add_thread(tp);
  tg.join_all();

  shmDoit(BrmKeys.MST_SYSVKEY, "MST        ");

  if (!cFlg)
  {
    shmDoit(BrmKeys.PROCESSSTATUS_SYSVKEY, "PROC_STAT  ");
    shmDoit(BrmKeys.SYSTEMSTATUS_SYSVKEY, "SYS_STAT   ");
    shmDoit(BrmKeys.SWITCHSTATUS_SYSVKEY, "SW_STAT    ");
    shmDoit(BrmKeys.NICSTATUS_SYSVKEY, "NIC_STAT   ");
    shmDoit(BrmKeys.DBROOTSTATUS_SYSVKEY, "DBROOT_STAT");
  }

  shmDoit(BrmKeys.DECOMSVRMUTEX_SYSVKEY, "DCMSVRMUTEX");

  semDoit(BrmKeys.KEYRANGE_CL_BASE, "COPYLOCK   ");
  semDoit(BrmKeys.KEYRANGE_EXTENTMAP_BASE, "EXTMAP     ");
  semDoit(BrmKeys.KEYRANGE_EMFREELIST_BASE, "EXTMAP_FREE");
  semDoit(BrmKeys.KEYRANGE_VBBM_BASE, "VBBM       ");
  semDoit(BrmKeys.KEYRANGE_VSS_BASE, "VSS        ");
  semDoit(BrmKeys.KEYRANGE_EXTENTMAP_INDEX_BASE, "EXTMAP_INDX");
  semDoit(BrmKeys.MST_SYSVKEY, "MST        ");

  if (!cFlg)
  {
    semDoit(BrmKeys.PROCESSSTATUS_SYSVKEY, "PROC_STAT  ");
    semDoit(BrmKeys.SYSTEMSTATUS_SYSVKEY, "SYS_STAT   ");
    semDoit(BrmKeys.SWITCHSTATUS_SYSVKEY, "SW_STAT    ");
    semDoit(BrmKeys.NICSTATUS_SYSVKEY, "NIC_STAT   ");
    shmDoit(BrmKeys.DBROOTSTATUS_SYSVKEY, "DBROOT_STAT");
  }

  return 0;
}