rcpp_framework/modules/drogon/trantor/net/Channel.cc

/**
 *
 *  Channel.cc
 *  An Tao
 *
 *  Public header file in trantor lib.
 *
 *  Copyright 2018, An Tao.  All rights reserved.
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the License file.
 *
 *
 */

#include "Channel.h"
#include <trantor/net/EventLoop.h>
#ifdef _WIN32
#include "Wepoll.h"
#define POLLIN EPOLLIN
#define POLLPRI EPOLLPRI
#define POLLOUT EPOLLOUT
#define POLLHUP EPOLLHUP
#define POLLNVAL 0
#define POLLERR EPOLLERR
#else
#include <poll.h>
#endif
#include <iostream>
namespace trantor {
const int Channel::kNoneEvent = 0;

const int Channel::kReadEvent = POLLIN | POLLPRI;
const int Channel::kWriteEvent = POLLOUT;

Channel::Channel(EventLoop *loop, int fd) :
		loop_(loop), fd_(fd), events_(0), revents_(0), index_(-1), tied_(false) {
}

void Channel::remove() {
	assert(events_ == kNoneEvent);
	addedToLoop_ = false;
	loop_->removeChannel(this);
}

void Channel::update() {
	loop_->updateChannel(this);
}

void Channel::handleEvent() {
	// LOG_TRACE<<"revents_="<<revents_;
	if (tied_) {
		std::shared_ptr<void> guard = tie_.lock();
		if (guard) {
			handleEventSafely();
		}
	} else {
		handleEventSafely();
	}
}
void Channel::handleEventSafely() {
	if (eventCallback_) {
		eventCallback_();
		return;
	}
	if ((revents_ & POLLHUP) && !(revents_ & POLLIN)) {
		// LOG_TRACE<<"handle close";
		if (closeCallback_)
			closeCallback_();
	}
	if (revents_ & (POLLNVAL | POLLERR)) {
		// LOG_TRACE<<"handle error";
		if (errorCallback_)
			errorCallback_();
	}
#ifdef __linux__
	if (revents_ & (POLLIN | POLLPRI | POLLRDHUP))
#else
	if (revents_ & (POLLIN | POLLPRI))
#endif
	{
		// LOG_TRACE<<"handle read";
		if (readCallback_)
			readCallback_();
	}
#ifdef _WIN32
	if ((revents_ & POLLOUT) && !(revents_ & POLLHUP))
#else
	if (revents_ & POLLOUT)
#endif
	{
		// LOG_TRACE<<"handle write";
		if (writeCallback_)
			writeCallback_();
	}
}

} // namespace trantor