Ejemplo n.º 1
0
	void on_read(mp::wavy::event& e)
	{
		try{
			while(true) {
				if(m_pac.execute()) {
					msgpack::object msg = m_pac.data();
					std::shared_ptr<msgpack::zone> z( m_pac.release_zone() );
					m_pac.reset();

					e.more();  //e.next();
					
					//DEBUG(std::cout << "object received: " << msg << "->");
					
					//DEBUG(lock_mut lock(&mut););
					event_handle(fd(), msg, &*z);
					return;
				}
			
				m_pac.reserve_buffer(8*1024);

				int read_len = ::read(fd(), m_pac.buffer(), m_pac.buffer_capacity());
				if(read_len <= 0) {
					if(read_len == 0) {throw mp::system_error(errno, "connection closed"); }
					if(errno == EAGAIN || errno == EINTR) { return; }
					else { perror("read"); throw mp::system_error(errno, "read error"); }
				}
				m_pac.buffer_consumed(read_len);
			}
		}catch(...){
			DEBUG_OUT("fd:%d ",fd());;
			perror("exception ");
			e.remove();
		}
	}
Ejemplo n.º 2
0
static void feed_file(msgpack::unpacker& pac, const char* path)
{
    std::ifstream fin(path);
    while(true) {
        pac.reserve_buffer(32*1024);
        fin.read(pac.buffer(), pac.buffer_capacity());
        if(fin.bad()) {
            throw std::runtime_error("read failed");
        }
        pac.buffer_consumed(static_cast<size_t>(fin.gcount()));
        if(fin.fail()) {
            break;
        }
    }
}
Ejemplo n.º 3
0
    void QueryChannel::produce(ResponseHeader &hdr, msgpack::unpacker &unpacker)
    {
        // Unpack the payload
        msgpack::unpacked unp;
        if (unpacker.next(&unp)) {
            msgpack::object obj = unp.get();

            NodeResponseBase type;

            try {
                obj.convert(type);
                if (type.Type == "ack") {
                    NodeAck ack;
                    obj.convert(ack);
                    m_listener->onQueryAck(hdr,ack);
                } else if (type.Type == "response") {
                    NodeResponse resp;
                    obj.convert(resp);
                    m_listener->onQueryResponse(hdr,resp);
                } else if (type.Type == "done") {
                    m_listener->onQueryComplete(hdr);

                    // deregister this channel
                    m_ioThread.removeChannel(hdr.Seq);
                }
            } catch (std::exception &e) {
                std::cout << "Unexpected node response type " << type.Type << " "
                          << e.what() << std::endl;
            }
        }

    }
Ejemplo n.º 4
0
    void LogChannel::produce(ResponseHeader &hdr, msgpack::unpacker &unpacker)
    {
        LogRecord record;

        // Unpack the payload
        msgpack::unpacked unp;
        if (unpacker.next(&unp)) {
            msgpack::object obj = unp.get();
            record = obj.as<LogRecord>();
        }

        m_listener->onLogRecord(hdr,record);
    }
Ejemplo n.º 5
0
    void EventChannel::produce(ResponseHeader &hdr, msgpack::unpacker &unpacker)
    {

        // Unpack the payload
        msgpack::unpacked unp;
        if (unpacker.next(&unp)) {
            msgpack::object obj = unp.get();
            RecordBase type;

            try {
                obj.convert(type);

                if (type.Event == "user") {
                    UserEventRecord rec;
                    obj.convert(rec);
	                m_listener->onUserEventRecord(hdr,rec);
                    return;
                } else if (type.Event == "query") {
                    QueryRecord rec;
                    obj.convert(rec);
                    m_listener->onQueryEventRecord(hdr,rec);
                    return;
                } else if (type.Event == "member-join" ||
                           type.Event == "member-leave" ||
                           type.Event == "member-update" ||
                           type.Event == "member-failed" ||
                           type.Event == "member-reap") {
                    MemberEventRecord rec;
                    obj.convert(rec);
                    m_listener->onMemberEventRecord(hdr,rec);
                    return;
                }
            } catch (std::exception &e) {
                std::cout << "Unexpected event type " << type.Event << " " << e.what() << std::endl;
            }
        }
    }