Add connection pool

This commit is contained in:
fbraem 2015-11-20 18:31:33 +01:00
parent e336fd7a8b
commit 267cbb7931
3 changed files with 137 additions and 0 deletions

View File

@ -0,0 +1,111 @@
//
// PoolableConnectionFactory.h
//
// $Id$
//
// Library: Redis
// Package: Redis
// Module: PoolableConnectionFactory
//
// Definition of the PoolableConnectionFactory class.
//
// Copyright (c) 2012, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// SPDX-License-Identifier: BSL-1.0
//
#ifndef Redis_PoolableConnectionFactory_INCLUDED
#define Redis_PoolableConnectionFactory_INCLUDED
#include "Poco/Redis/Client.h"
#include "Poco/ObjectPool.h"
namespace Poco {
template<>
class PoolableObjectFactory<Redis::Client, Redis::Client::Ptr>
/// PoolableObjectFactory specialisation for Client. New connections
/// are created with the given address.
{
public:
PoolableObjectFactory(Net::SocketAddress& address)
: _address(address)
{
}
PoolableObjectFactory(const std::string& address)
: _address(address)
{
}
Redis::Client::Ptr createObject()
{
return new Redis::Client(_address);
}
bool validateObject(Redis::Client::Ptr pObject)
{
return true;
}
void activateObject(Redis::Client::Ptr pObject)
{
}
void deactivateObject(Redis::Client::Ptr pObject)
{
}
void destroyObject(Redis::Client::Ptr pObject)
{
}
private:
Net::SocketAddress _address;
};
namespace Redis {
class Redis_API PooledConnection
/// Helper class for borrowing and returning a connection automatically from a pool.
{
public:
PooledConnection(ObjectPool<Client, Client::Ptr>& pool, long timeoutMilliseconds = 0) : _pool(pool)
{
_client = _pool.borrowObject(timeoutMilliseconds);
}
virtual ~PooledConnection()
{
try
{
_pool.returnObject(_client);
}
catch (...)
{
poco_unexpected();
}
}
operator Client::Ptr ()
{
return _client;
}
private:
ObjectPool<Client, Client::Ptr>& _pool;
Client::Ptr _client;
};
} // namespace Redis
} // namespace Poco
#endif // Redis_PoolableConnectionFactory_INCLUDED

View File

@ -17,6 +17,7 @@
#include "RedisTest.h"
#include "Poco/Redis/AsyncReader.h"
#include "Poco/Redis/Command.h"
#include "Poco/Redis/PoolableConnectionFactory.h"
#include "CppUnit/TestCaller.h"
#include "CppUnit/TestSuite.h"
@ -2852,6 +2853,27 @@ void RedisTest::testRPUSH()
}
}
void RedisTest::testPool()
{
Poco::Net::SocketAddress sa(_host, _port);
Poco::PoolableObjectFactory<Client, Client::Ptr> factory(sa);
Poco::ObjectPool<Client, Client::Ptr> pool(factory, 10, 15);
delKey("mypoolkey");
PooledConnection pclient1(pool);
PooledConnection pclient2(pool);
assert(pool.size() == 2);
Command set = Command::set("mypoolkey", "Hello");
std::string result = ((Client::Ptr) pclient1)->execute<std::string>(set);
assert(result.compare("OK") == 0);
Array get;
get << "GET" << "mypoolkey";
BulkString keyValue = ((Client::Ptr) pclient2)->execute<BulkString>(get);
assert(keyValue.value().compare("Hello") == 0);
}
void RedisTest::delKey(const std::string& key)
{
@ -2930,5 +2952,7 @@ CppUnit::Test* RedisTest::suite()
CppUnit_addTest(pSuite, RedisTest, testRPOPLPUSH);
CppUnit_addTest(pSuite, RedisTest, testRPUSH);
CppUnit_addTest(pSuite, RedisTest, testPool);
return pSuite;
}

View File

@ -84,6 +84,8 @@ public:
void testRPOPLPUSH();
void testRPUSH();
void testPool();
void setUp();
void tearDown();