BufFix: fix the problem that AsyncIO fails to correctly handle the re…

…turn value when using SQ_POLL (#132)
trpc-group · May 27, 2024 · 00e845f · 00e845f
1 parent 5c32e3f
commit 00e845f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trpc/runtime/iomodel/async_io/async_io.cc b/trpc/runtime/iomodel/async_io/async_io.cc
@@ -108,15 +108,15 @@ trpc::Future<int32_t> AsyncIO::SubmitOne(F&& fill_sqe) {
   io_uring_sqe_set_data(sqe, user_data);
 
   int ret = io_uring_submit(ring);
-  if (TRPC_UNLIKELY(ret != 1)) {
+  if (TRPC_UNLIKELY(ret <= 0)) {
     trpc::object_pool::Delete<IOURingUserData>(user_data);
     if (ret < 0) {
       return MakeExceptionFuture<int32_t>(AsyncIOError(ret));
     }
     std::string msg = "Submit success but return not one, ret:" + std::to_string(ret);
     return MakeExceptionFuture<int32_t>(AsyncIOError(AsyncIOError::SUBMIT_FAIL, msg));
   }
-  submitted_++;
+  submitted_ += ret;
 
   return user_data->pr.GetFuture();
 }