jessequ
/
AMQP-CPP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add timer

This commit is contained in:
RPG-18 2015-05-20 01:53:13 +03:00
parent 44027dbb52
commit f094a65f02
2 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
rabbitmq_tutorials/asiohandler.cpp
View File

@ -59,6 +59,7 @@ private:
AsioHandler::AsioHandler(boost::asio::io_service& ioService) : AsioHandler::AsioHandler(boost::asio::io_service& ioService) :
_ioService(ioService), _ioService(ioService),
_socket(ioService), _socket(ioService),
_timer(ioService),
_asioInputBuffer(ASIO_INPUT_BUFFER_SIZE, 0), _asioInputBuffer(ASIO_INPUT_BUFFER_SIZE, 0),
_amqpBuffer(new AmqpBuffer(ASIO_INPUT_BUFFER_SIZE * 2)), _amqpBuffer(new AmqpBuffer(ASIO_INPUT_BUFFER_SIZE * 2)),
_connection(nullptr), _connection(nullptr),
@ -81,6 +82,15 @@ void AsioHandler::doConnect(const std::string& host, uint16_t port)
{ {
tcp::resolver::query query(host, std::to_string(port)); tcp::resolver::query query(host, std::to_string(port));
tcp::resolver::iterator iter = tcp::resolver(_ioService).resolve(query); tcp::resolver::iterator iter = tcp::resolver(_ioService).resolve(query);
_timer.expires_from_now(boost::posix_time::seconds(15));
_timer.async_wait([this](const boost::system::error_code& ec){
if(!ec && !_connected)
{
std::cerr<<"Connection timed out";
_socket.cancel();
exit(1);
}
});
boost::asio::async_connect(_socket, iter, boost::asio::async_connect(_socket, iter,
[this](boost::system::error_code ec, tcp::resolver::iterator) [this](boost::system::error_code ec, tcp::resolver::iterator)
@ -97,7 +107,8 @@ void AsioHandler::doConnect(const std::string& host, uint16_t port)
} }
else else
{ {
std::cerr<<ec<<std::endl; std::cerr<<"Connection error:"<<ec<<std::endl;
exit(1);
} }
}); });
@ -128,7 +139,7 @@ void AsioHandler::doRead()
} }
else else
{ {
std::cerr<<ec<<std::endl; std::cerr<<"Error reading:"<<ec<<std::endl;
exit(1); exit(1);
} }
}); });
@ -160,8 +171,9 @@ void AsioHandler::doWrite()
} }
else else
{ {
std::cerr<<ec<<std::endl; std::cerr<<"Error writing:"<<ec<<std::endl;
_socket.close(); _socket.close();
exit(1);
} }
}); });
} }
@ -188,7 +200,6 @@ void AsioHandler::parseData()
void AsioHandler::onConnected(AMQP::Connection *connection) void AsioHandler::onConnected(AMQP::Connection *connection)
{ {
//_connected = true;
} }
bool AsioHandler::connected() const bool AsioHandler::connected() const
{ {

1
rabbitmq_tutorials/asiohandler.h
View File

@ -31,6 +31,7 @@ private:
boost::asio::io_service& _ioService; boost::asio::io_service& _ioService;
boost::asio::ip::tcp::socket _socket; boost::asio::ip::tcp::socket _socket;
boost::asio::deadline_timer _timer;
std::vector<char> _asioInputBuffer; std::vector<char> _asioInputBuffer;
std::shared_ptr<AmqpBuffer> _amqpBuffer; std::shared_ptr<AmqpBuffer> _amqpBuffer;