thrift_asio_client_transport.hpp

//
// Created by Benjamin Schulz on 15/03/15.
//

#ifndef _THRIFT_ASIO_THRIFT_ASIO_CLIENT_TRANSPORT_HPP_
#define _THRIFT_ASIO_THRIFT_ASIO_CLIENT_TRANSPORT_HPP_

#pragma once

#include "./thrift_asio_transport.hpp"
#include <boost/asio/connect.hpp>

namespace betabugs {
namespace networking {

class thrift_asio_client_transport : public thrift_asio_transport
{
  public:
    thrift_asio_client_transport(
        boost::asio::io_service& io_service, 
        const std::string& host_name,        
        const std::string& service_name,     
        event_handlers* event_handlers       
    ) : thrift_asio_transport(std::make_shared<boost::asio::ip::tcp::socket>(io_service), event_handlers)
        , host_name_(host_name)
        , service_name_(service_name)
        , resolver_(io_service)
    {
    }

    virtual void open() override
    {
        using boost::asio::ip::tcp;
        //this->close();
        state_ = RESOLVING;
        resolver_.cancel();
        resolver_.async_resolve
            (
                {host_name_, service_name_},
            [this]
                (const boost::system::error_code& ec, boost::asio::ip::tcp::resolver::iterator iterator)
            {
                if (ec)
                {
                    event_handlers_->on_error(ec);
                    close();
                }
                else
                {
                    state_ = CONNECTING;
                    boost::asio::async_connect
                        (
                            *socket_,
                            iterator,
                            [this]
                                (boost::system::error_code ec, tcp::resolver::iterator)
                            {
                                if (ec)
                                {
                                    event_handlers_->on_error(ec);
                                    this->close();
                                }
                                else
                                {
                                    thrift_asio_transport::open();
                                }
                            }
                        );
                }
            }
        );
    }


    void connect_to(const std::string& host_name, const std::string& service_name)
    {
        close();
        this->host_name_ = host_name;
        this->service_name_ = service_name;
        open();
    }

  private:
    std::string host_name_;
    std::string service_name_;
    boost::asio::ip::tcp::resolver resolver_;
};

}
}

#endif //_THRIFT_ASIO_THRIFT_ASIO_CLIENT_TRANSPORT_HPP_