SimpleTbusdConn.h

//
// Created by capitalg on 2018/8/20.
//

#ifndef TENCENT_INTERN_SIMPLETBUSDCONN_H
#define TENCENT_INTERN_SIMPLETBUSDCONN_H
#include <unordered_map>
#include <set>
#include <map>
#include <boost/asio.hpp>
#include "structs/TbusMsg.h"
#include "structs/SimpleChannelInfo.h"
#include "NetworkUtil.h"
#include "SimpleChannel.h"


/**
 * SimpleTbusdConn需要改变tbus共享内存中对应通道的信息以及通道内容
 * 因此需要知道
 *          1.所有通道对应读写指针的地址，即通道->simplechannelInfo*的映射关系
 *          2.每个通道对应的共享内存地址, 即通道->通道共享内存地址 的映射关系
 */
class SimpleTbusdConn : public std::enable_shared_from_this<SimpleTbusdConn> {
public:
    SimpleTbusdConn(boost::asio::ip::tcp::socket socket,
                        std::map<std::pair<uint32_t, uint32_t>, std::unique_ptr<SimpleChannel>> &channels,
                        std::map<uint32_t, std::pair<uint32_t, uint32_t>> &process_id2endpoint,
                        std::map<uint32_t, std::shared_ptr<SimpleTbusdConn>> &local_conns,
                        std::map<std::pair<uint32_t, uint32_t>, std::shared_ptr<boost::asio::ip::tcp::socket>> &remote_endpoint2socket,
                        std::set<uint32_t> &local_proc_id);

    void start();

    void do_read_message_type();


    void do_read_tbusmsg();

    void do_read_local_proc_id();

    boost::asio::ip::tcp::socket &get_socket() { return socket_; }

private:
    boost::asio::ip::tcp::socket socket_;
    TbusMsg tbus_msg;
    uint32_t len, cur_read_len, message_type, buffer_proc_id;
    std::map<std::pair<uint32_t, uint32_t>, std::unique_ptr<SimpleChannel>> &channels;
    std::map<uint32_t, std::pair<uint32_t, uint32_t>> &process_id2endpoint;
    std::map<std::pair<uint32_t, uint32_t>, std::shared_ptr<boost::asio::ip::tcp::socket>> &remote_endpoint2socket;
    std::map<uint32_t, std::shared_ptr<SimpleTbusdConn>> &local_conns;
    std::set<uint32_t> &local_proc_id;

    inline bool _is_local(uint32_t proc_id) {
        return local_proc_id.find(proc_id) != local_proc_id.end();
    }
};


#endif //TENCENT_INTERN_SIMPLETBUSDCONN_H