mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-18 21:44:02 +03:00
2e1394149b
* Fixes of bugs from ASAN warnings, part one * MQC as static library, with nifty counter for global map and mutex * Switch clang to 16 * link messageqcpp to execplan
133 lines
3.8 KiB
C++
133 lines
3.8 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>
|
|
#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
|