mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-02-07 21:37:14 +01:00
4a6bfc086c
Switched to POCO library for unified platform/library interface. Deprecated the external module API. It was creating more problems than solving. Removed most built-in libraries in favor of system libraries for easier maintenance. Cleaned and secured code with help from static analyzers.
129 lines
2.4 KiB
C++
129 lines
2.4 KiB
C++
//
|
|
// PoolableConnectionFactory.h
|
|
//
|
|
// Library: MongoDB
|
|
// Package: MongoDB
|
|
// Module: PoolableConnectionFactory
|
|
//
|
|
// Definition of the PoolableConnectionFactory class.
|
|
//
|
|
// Copyright (c) 2012, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#ifndef MongoDB_PoolableConnectionFactory_INCLUDED
|
|
#define MongoDB_PoolableConnectionFactory_INCLUDED
|
|
|
|
|
|
#include "Poco/MongoDB/Connection.h"
|
|
#include "Poco/ObjectPool.h"
|
|
|
|
|
|
namespace Poco {
|
|
|
|
|
|
template<>
|
|
class PoolableObjectFactory<MongoDB::Connection, MongoDB::Connection::Ptr>
|
|
/// PoolableObjectFactory specialisation for Connection. New connections
|
|
/// are created with the given address or URI.
|
|
///
|
|
/// If a Connection::SocketFactory is given, it must live for the entire
|
|
/// lifetime of the PoolableObjectFactory.
|
|
{
|
|
public:
|
|
PoolableObjectFactory(Net::SocketAddress& address):
|
|
_address(address),
|
|
_pSocketFactory(0)
|
|
{
|
|
}
|
|
|
|
PoolableObjectFactory(const std::string& address):
|
|
_address(address),
|
|
_pSocketFactory(0)
|
|
{
|
|
}
|
|
|
|
PoolableObjectFactory(const std::string& uri, MongoDB::Connection::SocketFactory& socketFactory):
|
|
_uri(uri),
|
|
_pSocketFactory(&socketFactory)
|
|
{
|
|
}
|
|
|
|
MongoDB::Connection::Ptr createObject()
|
|
{
|
|
if (_pSocketFactory)
|
|
return new MongoDB::Connection(_uri, *_pSocketFactory);
|
|
else
|
|
return new MongoDB::Connection(_address);
|
|
}
|
|
|
|
bool validateObject(MongoDB::Connection::Ptr pObject)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
void activateObject(MongoDB::Connection::Ptr pObject)
|
|
{
|
|
}
|
|
|
|
void deactivateObject(MongoDB::Connection::Ptr pObject)
|
|
{
|
|
}
|
|
|
|
void destroyObject(MongoDB::Connection::Ptr pObject)
|
|
{
|
|
}
|
|
|
|
private:
|
|
Net::SocketAddress _address;
|
|
std::string _uri;
|
|
MongoDB::Connection::SocketFactory* _pSocketFactory;
|
|
};
|
|
|
|
|
|
namespace MongoDB {
|
|
|
|
|
|
class PooledConnection
|
|
/// Helper class for borrowing and returning a connection automatically from a pool.
|
|
{
|
|
public:
|
|
PooledConnection(Poco::ObjectPool<Connection, Connection::Ptr>& pool) : _pool(pool)
|
|
{
|
|
_connection = _pool.borrowObject();
|
|
}
|
|
|
|
virtual ~PooledConnection()
|
|
{
|
|
try
|
|
{
|
|
if (_connection)
|
|
{
|
|
_pool.returnObject(_connection);
|
|
}
|
|
}
|
|
catch (...)
|
|
{
|
|
poco_unexpected();
|
|
}
|
|
}
|
|
|
|
operator Connection::Ptr ()
|
|
{
|
|
return _connection;
|
|
}
|
|
|
|
private:
|
|
Poco::ObjectPool<Connection, Connection::Ptr>& _pool;
|
|
Connection::Ptr _connection;
|
|
};
|
|
|
|
|
|
} } // namespace Poco::MongoDB
|
|
|
|
|
|
#endif // MongoDB_PoolableConnectionFactory_INCLUDED
|