thrift_asio_connection_management_mixin.hpp

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

#ifndef _THRIFT_ASIO_THRIFT_ASIO_CONNECTION_MANAGEMENT_MIXIN_HPP_
#define _THRIFT_ASIO_THRIFT_ASIO_CONNECTION_MANAGEMENT_MIXIN_HPP_

namespace betabugs{
namespace networking{

template <typename ClientType>
class thrift_asio_connection_management_mixin
{
  public:
    virtual void on_client_connected(boost::shared_ptr<apache::thrift::protocol::TProtocol> output_protocol)
    {
        assert( clients_.find(output_protocol) == clients_.end() );

        clients_.insert(
            std::make_pair(
                output_protocol,
                std::make_shared<ClientType>(output_protocol)
            )
        );
    }

    virtual void on_client_disconnected(const boost::shared_ptr<apache::thrift::protocol::TProtocol>& output_protocol, const boost::system::error_code& ec)
    {
        (void)ec;
        assert( clients_.find(output_protocol) != clients_.end() );
        clients_.erase(output_protocol);
    }

    virtual void before_process(boost::shared_ptr<apache::thrift::protocol::TProtocol> output_protocol)
    {
        auto pos = clients_.find(output_protocol);
        assert(pos != clients_.end());
        current_client_ = pos->second;
    }

    virtual void after_process()
    {
        current_client_.reset();
    }
  protected:
    typedef boost::shared_ptr<apache::thrift::protocol::TProtocol> protocol_ptr;

    typedef std::shared_ptr<ClientType> client_ptr;

    typedef std::map<protocol_ptr, client_ptr> client_map;

    client_map clients_;

    client_ptr current_client_;
};

}
}

#endif //_THRIFT_ASIO_THRIFT_ASIO_CONNECTION_MANAGEMENT_MIXIN_HPP_