io_uring: refactor io_wake_function