blob: 52137b395c60d8dfd2a72907938f7bbbaa59c448 [file] [log] [blame]
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* See the License for the specific language governing permissions and
* limitations under the License.
#include <boost/asio.hpp>
#include <boost/bind.hpp>
#include <boost/lambda/bind.hpp>
#include <boost/lambda/lambda.hpp>
#include <iostream>
#include <istream>
#include <ostream>
#include <string>
#include "Logging.h"
#include "url.h"
using boost::asio::deadline_timer;
using boost::asio::ip::tcp;
using boost::lambda::var;
namespace {
void check_deadline(deadline_timer* deadline, tcp::socket* socket, const boost::system::error_code& ec) {
// Check whether the deadline has passed. We compare the deadline against
// the current time since a new asynchronous operation may have moved the
// deadline before this actor had a chance to run.
if (deadline->expires_at() <= deadline_timer::traits_type::now()) {
// The deadline has passed. The socket is closed so that any outstanding
// asynchronous operations are cancelled. This allows the blocked
// connect(), read_line() or write_line() functions to return.
boost::system::error_code ignored_ec;
// There is no longer an active deadline. The expiry is set to positive
// infinity so that the actor takes no action until a new deadline is set.
// Put the actor back to sleep.
deadline->async_wait(boost::bind(&check_deadline, deadline, socket, boost::asio::placeholders::error));
} // namespace
namespace rocketmq {
bool SyncfetchNsAddr(const Url& url_s, std::string& body) {
bool ret = true;
try {
boost::asio::io_service io_service;
// Get a list of endpoints corresponding to the server name.
tcp::resolver resolver(io_service);
tcp::resolver::query query(url_s.host_, url_s.port_);
tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);
boost::system::error_code ec = boost::asio::error::would_block;
deadline_timer deadline(io_service);
// TODO hardcode
boost::posix_time::seconds timeout(3);
// Try each endpoint until we successfully establish a connection.
tcp::socket socket(io_service);
boost::system::error_code deadline_ec;
check_deadline(&deadline, &socket, deadline_ec);
boost::asio::async_connect(socket, endpoint_iterator, boost::lambda::var(ec) = boost::lambda::_1);
do {
} while (ec == boost::asio::error::would_block);
if (ec || !socket.is_open()) {
LOG_ERROR("socket connect failure, connect timeout or connect failure");
return false;
// Form the request. We specify the "Connection: close" header so that the
// server will close the socket after transmitting the response. This will
// allow us to treat all data up until the EOF as the content.
boost::asio::streambuf request;
std::ostream request_stream(&request);
request_stream << "GET " << url_s.path_ << " HTTP/1.0\r\n";
request_stream << "Host: " << url_s.host_ << "\r\n";
request_stream << "Accept: */*\r\n";
request_stream << "Connection: close\r\n\r\n";
// Send the request.
boost::asio::write(socket, request);
// Read the response status line. The response streambuf will automatically
// grow to accommodate the entire line. The growth may be limited by passing
// a maximum size to the streambuf constructor.
boost::asio::streambuf response;
boost::asio::read_until(socket, response, "\r\n");
// Check that response is OK.
std::istream response_stream(&response);
std::string http_version;
response_stream >> http_version;
unsigned int status_code;
response_stream >> status_code;
std::string status_message;
std::getline(response_stream, status_message);
if (!response_stream || http_version.substr(0, 5) != "HTTP/") {
LOG_INFO("Invalid response %s\n", status_message.c_str());
return false;
if (status_code != 200) {
LOG_INFO("Response returned with status code %d ", status_code);
return false;
// Read the response headers, which are terminated by a blank line.
boost::asio::read_until(socket, response, "\r\n\r\n");
// Process the response headers.
std::string header;
while (std::getline(response_stream, header) && header != "\r")
// Write whatever content we already have to output.
if (response.size() > 0) {
boost::asio::streambuf::const_buffers_type cbt =;
body.insert(body.begin(), boost::asio::buffers_begin(cbt), boost::asio::buffers_end(cbt));
// Read until EOF, writing data to output as we go.
boost::system::error_code error;
while (boost::asio::read(socket, response, boost::asio::transfer_at_least(1), error))
std::cout << &response;
if (error != boost::asio::error::eof)
throw boost::system::system_error(error);
} catch (std::exception& e) {
LOG_ERROR("Exception: %s", e.what());
ret = false;
return ret;
} // namespace rocketmq