mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-28 10:31:57 +00:00
124 lines
3.3 KiB
C++
124 lines
3.3 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; }
|
|
|
|
#if defined(POCO_ENABLE_CPP11)
|
|
// Disable copy to prevent unwanted release of resources: C++11 way
|
|
PooledConnection(const PooledConnection &) = delete;
|
|
PooledConnection & operator=(const PooledConnection &) = delete;
|
|
|
|
// Enable move semantics
|
|
PooledConnection(PooledConnection && other) = default;
|
|
PooledConnection & operator=(PooledConnection &&) = default;
|
|
#endif
|
|
|
|
private:
|
|
#if !defined(POCO_ENABLE_CPP11)
|
|
// Disable copy to prevent unwanted release of resources: pre C++11 way
|
|
PooledConnection(const PooledConnection &);
|
|
PooledConnection & operator=(const PooledConnection &);
|
|
#endif
|
|
|
|
Poco::ObjectPool<Connection, Connection::Ptr> & _pool;
|
|
Connection::Ptr _connection;
|
|
};
|
|
|
|
|
|
}
|
|
} // namespace Poco::MongoDB
|
|
|
|
|
|
#endif // MongoDB_PoolableConnectionFactory_INCLUDED
|