Add wrapper for uv_pipe_t

This commit is contained in:
patrick96 2021-09-11 11:46:50 +02:00 committed by Patrick Ziegler
parent 9d9fe8bffb
commit 88cd525dc8

View File

@ -31,16 +31,21 @@ struct cb_helper {
}
};
template <typename H, typename... Args>
struct UVHandle {
UVHandle(std::function<void(Args...)> fun) {
template <typename H, typename I, typename... Args>
struct UVHandleGeneric {
UVHandleGeneric(std::function<void(Args...)> fun) {
handle = std::make_unique<H>();
cb = cb_helper<H, Args...>{fun};
cb = cb_helper<I, Args...>{fun};
handle->data = &cb;
}
std::unique_ptr<H> handle;
cb_helper<H, Args...> cb;
cb_helper<I, Args...> cb;
};
template <typename H, typename... Args>
struct UVHandle : public UVHandleGeneric<H, H, Args...> {
UVHandle(std::function<void(Args...)> fun) : UVHandleGeneric<H, H, Args...>(fun) {}
};
struct SignalHandle : public UVHandle<uv_signal_t, int> {
@ -75,6 +80,47 @@ struct FSEventHandle : public UVHandle<uv_fs_event_t, const char*, int, int> {
}
};
struct PipeHandle : public UVHandleGeneric<uv_pipe_t, uv_stream_t, ssize_t, const uv_buf_t*> {
PipeHandle(uv_loop_t* loop, std::function<void(const string)> fun)
: UVHandleGeneric([&](ssize_t nread, const uv_buf_t* buf) {
if (nread > 0) {
string payload = string(buf->base, nread);
logger::make().notice("Bytes read: %d: '%s'", nread, payload);
fun(std::move(payload));
} else if (nread < 0) {
if (nread != UV_EOF) {
logger::make().err("Read error: %s", uv_err_name(nread));
uv_close((uv_handle_t*)handle.get(), nullptr);
} else {
UV(
uv_read_start, (uv_stream_t*)handle.get(),
[](uv_handle_t*, size_t, uv_buf_t* buf) {
buf->base = new char[BUFSIZ];
buf->len = BUFSIZ;
},
&cb.callback);
}
}
if (buf->base) {
delete[] buf->base;
}
}) {
UV(uv_pipe_init, loop, handle.get(), false);
}
void start(int fd) {
UV(uv_pipe_open, handle.get(), fd);
UV(
uv_read_start, (uv_stream_t*)handle.get(),
[](uv_handle_t*, size_t, uv_buf_t* buf) {
buf->base = new char[BUFSIZ];
buf->len = BUFSIZ;
},
&cb.callback);
}
};
class eventloop {
public:
eventloop();