/*

* 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

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* 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;

socket->close(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.

deadline->expires_at(boost::posix_time::pos_infin);

}

// 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);

deadline.expires_from_now(timeout);

// 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 {

io_service.run_one();

} 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 = response.data();

body.clear();

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