提交 024f15e0 编写于 作者: P Pavel Begunkov 提交者: Jens Axboe

io_uring: dedup io_run_task_work

We have an identical copy of io_run_task_work() for io-wq called
io_flush_signals(), deduplicate them.
Signed-off-by: NPavel Begunkov <asml.silence@gmail.com>
Link: https://lore.kernel.org/r/a157a4df5fa217b8bd03c73494f2fd0e24e44fbc.1655802465.git.asml.silence@gmail.comSigned-off-by: NJens Axboe <axboe@kernel.dk>
上级 a6b21fbb
...@@ -2,6 +2,8 @@ ...@@ -2,6 +2,8 @@
#ifndef IOU_FILE_TABLE_H #ifndef IOU_FILE_TABLE_H
#define IOU_FILE_TABLE_H #define IOU_FILE_TABLE_H
#include <linux/file.h>
struct io_ring_ctx; struct io_ring_ctx;
struct io_kiocb; struct io_kiocb;
......
...@@ -19,6 +19,7 @@ ...@@ -19,6 +19,7 @@
#include "io-wq.h" #include "io-wq.h"
#include "slist.h" #include "slist.h"
#include "io_uring.h"
#define WORKER_IDLE_TIMEOUT (5 * HZ) #define WORKER_IDLE_TIMEOUT (5 * HZ)
...@@ -519,23 +520,11 @@ static struct io_wq_work *io_get_next_work(struct io_wqe_acct *acct, ...@@ -519,23 +520,11 @@ static struct io_wq_work *io_get_next_work(struct io_wqe_acct *acct,
return NULL; return NULL;
} }
static bool io_flush_signals(void)
{
if (unlikely(test_thread_flag(TIF_NOTIFY_SIGNAL))) {
__set_current_state(TASK_RUNNING);
clear_notify_signal();
if (task_work_pending(current))
task_work_run();
return true;
}
return false;
}
static void io_assign_current_work(struct io_worker *worker, static void io_assign_current_work(struct io_worker *worker,
struct io_wq_work *work) struct io_wq_work *work)
{ {
if (work) { if (work) {
io_flush_signals(); io_run_task_work();
cond_resched(); cond_resched();
} }
...@@ -655,7 +644,7 @@ static int io_wqe_worker(void *data) ...@@ -655,7 +644,7 @@ static int io_wqe_worker(void *data)
last_timeout = false; last_timeout = false;
__io_worker_idle(wqe, worker); __io_worker_idle(wqe, worker);
raw_spin_unlock(&wqe->lock); raw_spin_unlock(&wqe->lock);
if (io_flush_signals()) if (io_run_task_work())
continue; continue;
ret = schedule_timeout(WORKER_IDLE_TIMEOUT); ret = schedule_timeout(WORKER_IDLE_TIMEOUT);
if (signal_pending(current)) { if (signal_pending(current)) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册