mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-14 00:40:55 +08:00
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
#pragma once
|
|
#include "../acl_cpp_define.hpp"
|
|
#include "../stdlib/noncopyable.hpp"
|
|
#include <map>
|
|
|
|
#if !defined(ACL_CLIENT_ONLY) && !defined(ACL_BEANSTALK_DISABLE)
|
|
|
|
namespace acl {
|
|
|
|
class beanstalk;
|
|
class locker;
|
|
|
|
/**
|
|
* beanstalk 客户端连接池，可以同时连接不同的 beanstalkd 服务器，
|
|
* 每个 beanstalkd 有多个连接，内部自动加锁；但不控制连接数限制，
|
|
* 用户应自行控制连接池的最大连接上限
|
|
*/
|
|
class ACL_CPP_API beanstalk_pool : public noncopyable
|
|
{
|
|
public:
|
|
beanstalk_pool();
|
|
~beanstalk_pool();
|
|
|
|
/**
|
|
* 从连接池中取得一个 beanstalkd 的客户端连接
|
|
* @param addr {const char*} beanstalkd 服务地址(domain:port)
|
|
* @param clean_watch {bool} 在取得连接对象后是否自动取消所有的
|
|
* 已关注队列
|
|
* @param conn_timeout {int} 连接 beanstalkd 的超时时间
|
|
* @return {beanstalk*} 返回非空表示正常，否则表示出错
|
|
*/
|
|
beanstalk* peek(const char* addr, bool clean_watch = true,
|
|
int conn_timeout = 60);
|
|
|
|
/**
|
|
* 将不用的 beanstalkd 连接放回到连接池中
|
|
* @param client {beanstalk*} beanstalkd 客户端连接
|
|
* @param clean_watch {bool} 是否取消已经关注的队列
|
|
* @param keep {bool} 如果为 true 则将 client 放回至连接池，
|
|
* 否则释放该连接
|
|
*/
|
|
void put(beanstalk* client, bool clean_watch = true,
|
|
bool keep = true);
|
|
private:
|
|
locker* lock_;
|
|
typedef std::multimap<string, beanstalk*> bspool;
|
|
typedef bspool::const_iterator pool_cit;
|
|
typedef bspool::iterator pool_it;
|
|
typedef std::pair<pool_it, pool_it> pool_range;
|
|
|
|
bspool pool_;
|
|
};
|
|
|
|
} // namespace acl
|
|
|
|
#endif // !defined(ACL_CLIENT_ONLY) && !defined(ACL_BEANSTALK_DISABLE)
|