diff options
Diffstat (limited to 'src/queue.c')
| -rw-r--r-- | src/queue.c | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/queue.c b/src/queue.c index 18040d5..470c3a2 100644 --- a/src/queue.c +++ b/src/queue.c @@ -53,6 +53,29 @@ int io_uring_peek_cqe(struct io_uring *ring, struct io_uring_cqe **cqe_ptr) } /* + * Fill in an array of IO completions up to count, if any are available. + * Returns the amount of IO completions filled. + */ +unsigned io_uring_peek_batch_cqe(struct io_uring *ring, struct io_uring_cqe **cqes, unsigned count) +{ + unsigned ready = io_uring_cq_ready(ring); + if (ready) { + count = count > ready ? ready : count; + unsigned head = *ring->cq.khead; + unsigned last = head + count; + unsigned mask = *ring->cq.kring_mask; + int i = 0; + for (;head!=last;head++,i++) { + cqes[i] = &ring->cq.cqes[head & mask]; + } + + return count; + } + + return 0; +} + +/* * Return an IO completion, waiting for it if necessary. Returns 0 with * cqe_ptr filled in on success, -errno on failure. */ |