mapnik/src/datasource_cache.cpp

222 lines
6.5 KiB
C++
Raw Normal View History

/*****************************************************************************
2012-02-02 02:53:35 +01:00
*
2006-03-31 12:32:02 +02:00
* This file is part of Mapnik (c++ mapping toolkit)
2005-06-14 17:06:59 +02:00
*
* Copyright (C) 2011 Artem Pavlenko
2005-06-14 17:06:59 +02:00
*
2006-03-31 12:32:02 +02:00
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
2005-06-14 17:06:59 +02:00
* but WITHOUT ANY WARRANTY; without even the implied warranty of
2006-03-31 12:32:02 +02:00
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
2005-06-14 17:06:59 +02:00
*
2006-03-31 12:32:02 +02:00
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
*****************************************************************************/
2005-06-14 17:06:59 +02:00
2007-10-08 19:42:41 +02:00
// mapnik
2012-04-08 02:20:56 +02:00
#include <mapnik/debug.hpp>
2007-10-08 19:42:41 +02:00
#include <mapnik/datasource_cache.hpp>
#include <mapnik/config_error.hpp>
// boost
#include <boost/make_shared.hpp>
2005-06-14 17:06:59 +02:00
#include <boost/filesystem/operations.hpp>
#include <boost/algorithm/string.hpp>
2007-10-08 19:42:41 +02:00
2007-03-22 11:55:43 +01:00
// ltdl
#include <ltdl.h>
2005-06-14 17:06:59 +02:00
2007-10-08 19:42:41 +02:00
// stl
#include <algorithm>
#include <stdexcept>
namespace mapnik {
2012-02-02 02:53:35 +01:00
2010-06-02 13:03:30 +02:00
bool is_input_plugin (std::string const& filename)
{
return boost::algorithm::ends_with(filename,std::string(".input"));
}
2012-02-02 02:53:35 +01:00
2005-06-14 17:06:59 +02:00
datasource_cache::datasource_cache()
2010-06-02 13:03:30 +02:00
{
if (lt_dlinit()) throw std::runtime_error("lt_dlinit() failed");
}
2005-06-14 17:06:59 +02:00
datasource_cache::~datasource_cache()
2010-06-02 13:03:30 +02:00
{
lt_dlexit();
}
datasource_ptr datasource_cache::create(const parameters& params)
2010-06-02 13:03:30 +02:00
{
boost::optional<std::string> type = params.get<std::string>("type");
if ( ! type)
{
2011-06-24 02:53:00 +02:00
throw config_error(std::string("Could not create datasource. Required ") +
2010-06-02 13:03:30 +02:00
"parameter 'type' is missing");
}
#ifdef MAPNIK_THREADSAFE
mutex::scoped_lock lock(mutex_);
#endif
2012-04-08 02:20:56 +02:00
2010-06-02 13:03:30 +02:00
datasource_ptr ds;
2011-06-24 02:53:00 +02:00
std::map<std::string,boost::shared_ptr<PluginInfo> >::iterator itr=plugins_.find(*type);
2010-06-02 13:03:30 +02:00
if ( itr == plugins_.end() )
{
std::string s("Could not create datasource for type: '");
s += *type + "'";
if (plugin_directories_.empty())
{
s + " (no datasource plugin directories have been successfully registered)";
}
else
{
s + " (searched for datasource plugins in '" + plugin_directories() + "')";
}
throw config_error(s);
2010-06-02 13:03:30 +02:00
}
2012-04-08 02:20:56 +02:00
2010-06-02 13:03:30 +02:00
if ( ! itr->second->handle())
{
2011-06-24 02:53:00 +02:00
throw std::runtime_error(std::string("Cannot load library: ") +
2010-06-02 13:03:30 +02:00
lt_dlerror());
}
2012-04-08 02:20:56 +02:00
// http://www.mr-edd.co.uk/blog/supressing_gcc_warnings
2012-02-02 02:53:35 +01:00
#ifdef __GNUC__
__extension__
2012-02-02 02:53:35 +01:00
#endif
create_ds* create_datasource =
reinterpret_cast<create_ds*>(lt_dlsym(itr->second->handle(), "create"));
2010-06-02 13:03:30 +02:00
2012-04-08 02:20:56 +02:00
if (! create_datasource)
2010-06-02 13:03:30 +02:00
{
2011-06-24 02:53:00 +02:00
throw std::runtime_error(std::string("Cannot load symbols: ") +
2010-06-02 13:03:30 +02:00
lt_dlerror());
}
2012-04-08 02:20:56 +02:00
#ifdef MAPNIK_LOG
MAPNIK_LOG_DEBUG(datasource_cache) << "datasource_cache: Size=" << params.size();
2012-04-08 02:20:56 +02:00
2010-06-02 13:03:30 +02:00
parameters::const_iterator i = params.begin();
2012-04-08 02:20:56 +02:00
for (; i != params.end(); ++i)
2010-06-02 13:03:30 +02:00
{
MAPNIK_LOG_DEBUG(datasource_cache) << "datasource_cache: -- " << i->first << "=" << i->second;
2010-06-02 13:03:30 +02:00
}
#endif
ds = datasource_ptr(create_datasource(params), datasource_deleter());
2012-04-08 02:20:56 +02:00
MAPNIK_LOG_DEBUG(datasource_cache) << "datasource_cache: Datasource=" << ds << " type=" << type;
2010-06-02 13:03:30 +02:00
return ds;
}
bool datasource_cache::insert(std::string const& type,const lt_dlhandle module)
2010-06-02 13:03:30 +02:00
{
return plugins_.insert(make_pair(type,boost::make_shared<PluginInfo>
(type,module))).second;
2010-06-02 13:03:30 +02:00
}
std::string datasource_cache::plugin_directories()
{
return boost::algorithm::join(plugin_directories_,", ");
}
std::vector<std::string> datasource_cache::plugin_names()
2010-06-02 13:03:30 +02:00
{
std::vector<std::string> names;
std::map<std::string,boost::shared_ptr<PluginInfo> >::const_iterator itr;
for (itr = plugins_.begin();itr!=plugins_.end();++itr)
{
names.push_back(itr->first);
}
return names;
}
2012-02-02 02:53:35 +01:00
void datasource_cache::register_datasources(std::string const& str)
2012-02-02 02:53:35 +01:00
{
2008-02-04 17:12:13 +01:00
#ifdef MAPNIK_THREADSAFE
mutex::scoped_lock lock(mutex_);
2008-02-04 17:12:13 +01:00
#endif
boost::filesystem::path path(str);
// TODO - only push unique paths
plugin_directories_.push_back(str);
boost::filesystem::directory_iterator end_itr;
2012-02-02 02:53:35 +01:00
2010-06-02 13:03:30 +02:00
if (exists(path) && is_directory(path))
{
for (boost::filesystem::directory_iterator itr(path);itr!=end_itr;++itr )
2010-06-02 13:03:30 +02:00
{
2012-02-02 02:53:35 +01:00
#if (BOOST_FILESYSTEM_VERSION == 3)
if (!is_directory( *itr ) && is_input_plugin(itr->path().filename().string()))
#else // v2
2012-02-02 02:53:35 +01:00
if (!is_directory( *itr ) && is_input_plugin(itr->path().leaf()))
#endif
2010-06-02 13:03:30 +02:00
{
2012-02-02 02:53:35 +01:00
#if (BOOST_FILESYSTEM_VERSION == 3)
if (register_datasource(itr->path().string().c_str()))
#else // v2
if (register_datasource(itr->string().c_str()))
#endif
{
registered_ = true;
}
2010-06-02 13:03:30 +02:00
}
}
}
2010-06-02 13:03:30 +02:00
}
bool datasource_cache::register_datasource(std::string const& str)
{
bool success = false;
try
{
lt_dlhandle module = lt_dlopen(str.c_str());
if (module)
{
// http://www.mr-edd.co.uk/blog/supressing_gcc_warnings
#ifdef __GNUC__
__extension__
#endif
datasource_name* ds_name =
reinterpret_cast<datasource_name*>(lt_dlsym(module, "datasource_name"));
if (ds_name && insert(ds_name(),module))
{
MAPNIK_LOG_DEBUG(datasource_cache) << "datasource_cache: Registered=" << ds_name();
success = true;
}
else if (!ds_name)
{
MAPNIK_LOG_ERROR(datasource_cache)
<< "Problem loading plugin library '"
<< str << "' (plugin is lacking compatible interface)";
}
}
else
{
MAPNIK_LOG_ERROR(datasource_cache)
<< "Problem loading plugin library: " << str
<< " (dlopen failed - plugin likely has an unsatisfied dependency or incompatible ABI)";
}
}
catch (...) {
MAPNIK_LOG_ERROR(datasource_cache)
<< "Exception caught while loading plugin library: " << str;
}
return success;
}
}