queue: Set noexcept for io_uring_peek_batch_cqe as it does not raise

exception, just return `0` if there is no ready entries.
YoSTEALTH · Apr 21, 2024 · e9c1ce0 · e9c1ce0
1 parent 4630e42
commit e9c1ce0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/liburing/queue.pxd b/src/liburing/queue.pxd
@@ -56,7 +56,7 @@ cpdef int io_uring_ring_dontfork(io_uring ring) nogil
 cpdef int io_uring_queue_exit(io_uring ring) nogil
 cpdef unsigned int io_uring_peek_batch_cqe(io_uring ring,
                                            io_uring_cqe cqes,
-                                           unsigned int count) nogil
+                                           unsigned int count) noexcept nogil
 cpdef int io_uring_wait_cqes(io_uring ring,
                              io_uring_cqe cqe_ptr,
                              unsigned int wait_nr,

diff --git a/src/liburing/queue.pyx b/src/liburing/queue.pyx
@@ -255,7 +255,7 @@ cpdef int io_uring_queue_exit(io_uring ring) nogil:
 
 cpdef unsigned int io_uring_peek_batch_cqe(io_uring ring,
                                            io_uring_cqe cqes,
-                                           unsigned int count) nogil:
+                                           unsigned int count) noexcept nogil:
     return __io_uring_peek_batch_cqe(&ring.ptr, &cqes.ptr, count)
 
 cpdef int io_uring_wait_cqes(io_uring ring,
@@ -426,3 +426,10 @@ cpdef inline io_uring_sqe io_uring_get_sqe(io_uring ring):
     cdef io_uring_sqe sqe = io_uring_sqe(0)
     sqe.ptr = __io_uring_get_sqe(&ring.ptr)
     return sqe
+
+
+cpdef inline unsigned int io_uring_for_each_cqe(io_uring ring,
+                                                io_uring_cqe cqe,
+                                                unsigned int head=0) nogil:
+    __io_uring_for_each_cqe(&ring.ptr, &head, cqe.ptr)
+    return head