asyncchannel.hh

Go to the documentation of this file.

#pragma once
#include "channel.hh"
#include "../messages/message.hh"
#include "netobserver.hh"

#include <string>
#include <queue>
#include <atomic>
#include <boost/asio.hpp>
#include <boost/asio/spawn.hpp>
#include <boost/asio/error.hpp>
#include <memory>
#include <mutex>

namespace eclipse {
namespace network {

using boost::asio::ip::tcp;
class AsyncChannel: public Channel, public std::enable_shared_from_this<AsyncChannel> {
  public:
    AsyncChannel(NetObserver* node_);
    virtual ~AsyncChannel() ;

    void do_write(messages::Message*) override; 

    void do_write(std::shared_ptr<std::string>&); 

    void do_write_buffer(); 

    void do_read();

    tcp::socket& get_socket();

    void commit(std::shared_ptr<std::string>&);

  private:
    void on_write(const boost::system::error_code&, size_t); 
    void do_write_impl(); 

    void read_coroutine(boost::asio::yield_context);

    NetObserver* node = nullptr;
    tcp::socket socket; 
    std::queue<std::shared_ptr<std::string>> messages_queue;
    std::atomic<bool> is_writing;
    boost::asio::io_service& iosvc;

    std::mutex queue_mutex;

    std::string host;
    uint32_t port;
};

}
}