mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-18 21:44:02 +03:00
122 lines
3.6 KiB
C++
122 lines
3.6 KiB
C++
/* 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>
|
|
#ifndef _MSC_VER
|
|
#include <dlfcn.h>
|
|
#endif
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <iostream>
|
|
|
|
#include "FileFactoryBase.h"
|
|
#include "BufferedFileFactory.h"
|
|
#include "UnbufferedFileFactory.h"
|
|
#include "PosixFileSystem.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[IDBDataFile::BUFFERED] = FileFactoryEnt(IDBDataFile::BUFFERED, "buffered", new BufferedFileFactory(), new PosixFileSystem());
|
|
s_plugins[IDBDataFile::UNBUFFERED] = FileFactoryEnt(IDBDataFile::UNBUFFERED, "unbuffered", new UnbufferedFileFactory(), new PosixFileSystem());
|
|
|
|
return false;
|
|
}
|
|
|
|
bool IDBFactory::installPlugin(const std::string& plugin)
|
|
{
|
|
#ifdef _MSC_VER
|
|
ostringstream oss;
|
|
oss << "InfiniDB for Windows does not support plugins: plugin = " << plugin;
|
|
throw std::runtime_error(oss.str());
|
|
#else
|
|
// 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[ent.type] = ent;
|
|
|
|
std::ostringstream oss;
|
|
oss << "IDBFactory::installPlugin: installed filesystem plugin " << plugin;
|
|
IDBLogger::syslog(oss.str(), logging::LOG_TYPE_DEBUG);
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
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[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[type].filesystem);
|
|
}
|
|
|
|
}
|