Team2890/allwpilib
4
0
Fork 0
mirror of https://github.com/wpilibsuite/allwpilib synced 2026-06-28 02:11:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

wpiutil: uv::Poll: add reuse functionality

Browse Source
This commit is contained in:
Peter Johnson
2018-09-05 23:04:33 -07:00
parent f0ac048645
commit 40a9fc44ff
2 changed files with 71 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
wpiutil/src/main/native/cpp/uv/Poll.cpp
View File

@@ -34,6 +34,44 @@ std::shared_ptr<Poll> Poll::CreateSocket(Loop& loop, uv_os_sock_t sock) {
return h;
}
void Poll::Reuse(int fd, std::function<void()> callback) {
if (IsClosing()) return;
if (!m_reuseData) m_reuseData = std::make_unique<ReuseData>();
m_reuseData->callback = callback;
m_reuseData->isSocket = false;
m_reuseData->fd = fd;
uv_close(GetRawHandle(), [](uv_handle_t* handle) {
Poll& h = *static_cast<Poll*>(handle->data);
if (!h.m_reuseData || h.m_reuseData->isSocket) return; // just in case
auto data = std::move(h.m_reuseData);
int err = uv_poll_init(h.GetLoopRef().GetRaw(), h.GetRaw(), data->fd);
if (err < 0) {
h.ReportError(err);
return;
}
data->callback();
});
}
void Poll::ReuseSocket(uv_os_sock_t sock, std::function<void()> callback) {
if (IsClosing()) return;
if (!m_reuseData) m_reuseData = std::make_unique<ReuseData>();
m_reuseData->callback = callback;
m_reuseData->isSocket = true;
m_reuseData->sock = sock;
uv_close(GetRawHandle(), [](uv_handle_t* handle) {
Poll& h = *static_cast<Poll*>(handle->data);
if (!h.m_reuseData || !h.m_reuseData->isSocket) return; // just in case
auto data = std::move(h.m_reuseData);
int err = uv_poll_init(h.GetLoopRef().GetRaw(), h.GetRaw(), data->sock);
if (err < 0) {
h.ReportError(err);
return;
}
data->callback();
});
}
void Poll::Start(int events) {
Invoke(&uv_poll_start, GetRaw(), events,
[](uv_poll_t* handle, int status, int events) {

33
wpiutil/src/main/native/include/wpi/uv/Poll.h
View File

@@ -68,6 +68,30 @@ class Poll final : public HandleImpl<Poll, uv_poll_t> {
return CreateSocket(*loop, sock);
}
/**
* Reuse this handle. This closes the handle, and after the close completes,
* reinitializes it (identically to Create) and calls the provided callback.
* Unlike Close(), it does NOT emit the closed signal, however, IsClosing()
* will return true until the callback is called. This does nothing if
* IsClosing() is true (e.g. if Close() was called).
*
* @param fd File descriptor
* @param callback Callback
*/
void Reuse(int fd, std::function<void()> callback);
/**
* Reuse this handle. This closes the handle, and after the close completes,
* reinitializes it (identically to CreateSocket) and calls the provided
* callback. Unlike Close(), it does NOT emit the closed signal, however,
* IsClosing() will return true until the callback is called. This does
* nothing if IsClosing() is true (e.g. if Close() was called).
*
* @param sock Socket descriptor.
* @param callback Callback
*/
void ReuseSocket(uv_os_sock_t sock, std::function<void()> callback);
/**
* Start polling the file descriptor.
*
@@ -85,6 +109,15 @@ class Poll final : public HandleImpl<Poll, uv_poll_t> {
* Signal generated when a poll event occurs.
*/
sig::Signal<int> pollEvent;
private:
struct ReuseData {
std::function<void()> callback;
bool isSocket;
int fd;
uv_os_sock_t sock;
};
std::unique_ptr<ReuseData> m_reuseData;
};
} // namespace uv