void read_cb(uv_fs_t *read_req) {
int err = 0;
if (read_req->result < 0) {
UV_CHECK(read_req->result, "uv_fs_read callback");
}
/* extracting our context from the read_req */
context_t *context = read_req->data;
/* 4. Report the contents of the buffer */
//log_report("%s", read_req->bufs->base);
lwlog_info("Read buf: %s", read_req->bufs->base);
free(read_req->bufs->base);
/* 5. Close the file descriptor (synchronously) */
uv_fs_t close_req;
err = uv_fs_close(uv_default_loop(), &close_req, context->open_req->result, NULL);
if (err < 0) {
UV_CHECK(abs(err), "uv_fs_close");
}
/* cleanup all requests and context */
uv_fs_req_cleanup(context->open_req);
uv_fs_req_cleanup(read_req);
uv_fs_req_cleanup(&close_req);
free(context);
return;
}
void init(uv_loop_t *loop) {
int err = 0;
/* No more globals, we need to malloc each request and pass it around for later cleanup */
uv_fs_t *open_req = malloc(sizeof(uv_fs_t));
uv_fs_t *read_req = malloc(sizeof(uv_fs_t));
context_t *context = malloc(sizeof(context_t));
context->open_req = open_req;
/* 1. Open file */
err = uv_fs_open(loop, open_req, filename, O_RDONLY, S_IRUSR, NULL);
if (err < 0) {
UV_CHECK(err, "uv_fs_open");
}
/* 2. Create buffer and initialize it to turn it into a a uv_buf_t */
size_t buf_len = sizeof(char) * BUF_SIZE;
char *buf = malloc(buf_len);
uv_buf_t iov = uv_buf_init(buf, buf_len);
/* allow us to access the context inside read_cb */
read_req->data = context;
/* 3. Read from the file into the buffer */
err = uv_fs_read(loop, read_req, open_req->result, &iov, 1, 0, read_cb);
if (err < 0) {
UV_CHECK(err, "uv_fs_read");
}
}
int main() {
int err = 0;
uv_loop_t *loop = uv_default_loop();
/* 1. Open file */
err = uv_fs_open(loop, &open_req, filename, O_RDONLY, S_IRUSR, NULL);
if (err < 0) {
UV_CHECK(err, "uv_fs_open");
}
/* 2. Create buffer and initialize it to turn it into a a uv_buf_t */
char buf[BUF_SIZE];
uv_buf_t iov = uv_buf_init(buf, sizeof(buf));
/* 3. Use the file descriptor (the .result of the open_req) to read from the file into the buffer */
uv_fs_t read_req;
err = uv_fs_read(loop, &read_req, open_req.result, &iov, 1, 0, read_cb);
if (err < 0) {
UV_CHECK(err, "uv_fs_read");
}
uv_run(loop, UV_RUN_DEFAULT);
return 0;
}
ThreadPool::ThreadPool() {
loop_ = (uv_loop_t*)malloc(sizeof *loop_);
UV_CHECK(uv_loop_init(loop_));
UV_CHECK(uv_async_init(loop_, &async_, thread_pool_async_cb));
async_.data = (void*)(this);
thread_.reset(new thread([=] () {
UV_CHECK(uv_run(loop_, UV_RUN_DEFAULT));
}));
}
Loop::Loop(int device) : device_(device), stop_(false) {
loop_ = (uv_loop_t*)malloc(sizeof *loop_);
UV_CHECK(uv_loop_init(loop_));
UV_CHECK(uv_async_init(loop_, &async_, async_cb));
async_.data = (void*)(this);
thread_.reset(new thread([=] () {
if (device_ >= 0) {
THREAD_SET_CUDA_DEVICE(device);
}
UV_CHECK(uv_run(loop_, UV_RUN_DEFAULT));
}));
}
Loop::~Loop() {
stop_ = true;
uv_async_send(&async_);
thread_->join();
UV_CHECK(uv_loop_close(loop_));
free(loop_);
}
void read_cb(uv_fs_t *read_req) {
int err = 0;
if (read_req->result < 0) {
UV_CHECK(read_req->result, "uv_fs_read callback");
}
/* 4. Report the contents of the buffer */
//log_report("%s", read_req->bufs->base);
lwlog_info("Read buf: %s", read_req->bufs->base);
/* 5. Close the file descriptor */
uv_fs_t close_req;
err = uv_fs_close(read_req->loop, &close_req, open_req.result, NULL);
if (err < 0) {
UV_CHECK(err, "uv_fs_close");
}
uv_fs_req_cleanup(&open_req);
uv_fs_req_cleanup(read_req);
uv_fs_req_cleanup(&close_req);
return;
}
