mariadb-columnstore-engine/utils/idbdatafile/IDBFactory.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. */

#include "IDBFactory.h"

#include <boost/thread.hpp>
#include <sstream>
#include <dlfcn.h>
#include <string.h>
#include <errno.h>
#include <iostream>

#include "FileFactoryBase.h"
#include "BufferedFileFactory.h"
#include "UnbufferedFileFactory.h"
#include "PosixFileSystem.h"
#include "SMFileSystem.h"
#include "SMFileFactory.h"
#include "IDBLogger.h"

using namespace std;
namespace
{
boost::mutex fac_guard;
}

namespace idbdatafile
{
IDBFactory::FactoryMap IDBFactory::s_plugins;

bool IDBFactory::installDefaultPlugins()
{
  // protect these methods since we are changing our static data structure
  boost::mutex::scoped_lock lock(fac_guard);

  s_plugins.emplace(IDBDataFile::BUFFERED, FileFactoryEnt(IDBDataFile::BUFFERED, "buffered", new BufferedFileFactory(),
                                             new PosixFileSystem()));
  s_plugins.emplace(IDBDataFile::UNBUFFERED, FileFactoryEnt(IDBDataFile::UNBUFFERED, "unbuffered",
                                             new UnbufferedFileFactory(), new PosixFileSystem()));

  return false;
}

bool IDBFactory::installPlugin(const std::string& plugin)
{
  // protect these methods since we are changing our static data structure
  boost::mutex::scoped_lock lock(fac_guard);

  void* handle = dlopen(plugin.c_str(), RTLD_LAZY);

  if (handle == NULL)
  {
    std::ostringstream oss;
    oss << "IDBFactory::installPlugin: dlopen for " << plugin << " failed: " << dlerror();
    IDBLogger::syslog(oss.str(), logging::LOG_TYPE_ERROR);
    return false;
  }

  void* functor = dlsym(handle, "plugin_instance");

  if (functor == NULL)
  {
    std::ostringstream oss;
    oss << "IDBFactory::installPlugin: dlsym for plugin_instance() failed. plugin " << plugin << dlerror();
    IDBLogger::syslog(oss.str(), logging::LOG_TYPE_ERROR);
    return false;
  }

  FileFactoryEnt ent = (*(FileFactoryEntryFunc)functor)();
  s_plugins.emplace(ent.type, std::move(ent));

  std::ostringstream oss;
  oss << "IDBFactory::installPlugin: installed filesystem plugin " << plugin;
  IDBLogger::syslog(oss.str(), logging::LOG_TYPE_DEBUG);
  return true;
}

vector<IDBDataFile::Types> IDBFactory::listPlugins()
{
  vector<IDBDataFile::Types> ret;
  for (FactoryMap::iterator it = s_plugins.begin(); it != s_plugins.end(); ++it)
    ret.push_back(it->first);
  return ret;
}

IDBDataFile* IDBFactory::open(IDBDataFile::Types type, const char* fname, const char* mode, unsigned opts,
                              unsigned colWidth)
{
  if (s_plugins.find(type) == s_plugins.end())
  {
    ostringstream oss;
    oss << "Cannot find factory plugin type " << (int)type << " to open file: " << fname;
    throw std::runtime_error(oss.str());
  }

  return s_plugins.at(type).factory->open(fname, mode, opts, colWidth);
}

IDBFileSystem& IDBFactory::getFs(IDBDataFile::Types type)
{
  if (s_plugins.find(type) == s_plugins.end())
  {
    ostringstream oss;
    oss << "Cannot find filesystem for plugin type " << (int)type;
    throw std::runtime_error(oss.str());
  }

  return *(s_plugins.at(type).filesystem);
}

FileFactoryEnt::~FileFactoryEnt()
{
  delete filesystem;
  delete factory;
}


}  // namespace idbdatafile