-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
f5b73c6
commit 3dbae80
Showing
10 changed files
with
278 additions
and
323 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,139 @@ | ||
| /* | ||
| +----------------------------------------------------------------------+ | ||
| | Swoole | | ||
| +----------------------------------------------------------------------+ | ||
| | This source file is subject to version 2.0 of the Apache license, | | ||
| | that is bundled with this package in the file LICENSE, and is | | ||
| | available through the world-wide-web at the following url: | | ||
| | http://www.apache.org/licenses/LICENSE-2.0.html | | ||
| | If you did not receive a copy of the Apache2.0 license and are unable| | ||
| | to obtain it through the world-wide-web, please send a note to | | ||
| | [email protected] so we can mail you a copy immediately. | | ||
| +----------------------------------------------------------------------+ | ||
| | Author: NathanFreeman <[email protected]> | | ||
| +----------------------------------------------------------------------+ | ||
| */ | ||
|
|
||
| #ifndef SWOOLE_SRC_SWOOLE_IOURING_H | ||
| #define SWOOLE_SRC_SWOOLE_IOURING_H | ||
|
|
||
| #ifdef SW_USE_IOURING | ||
| #include <liburing.h> | ||
|
|
||
| enum opcodes { | ||
| SW_IORING_OP_OPENAT = IORING_OP_OPENAT, | ||
| SW_IORING_OP_CLOSE = IORING_OP_CLOSE, | ||
| SW_IORING_OP_STATX = IORING_OP_STATX, | ||
| SW_IORING_OP_READ = IORING_OP_READ, | ||
| SW_IORING_OP_WRITE = IORING_OP_WRITE, | ||
| SW_IORING_OP_RENAMEAT = IORING_OP_RENAMEAT, | ||
| SW_IORING_OP_UNLINKAT = IORING_OP_UNLINKAT, | ||
| SW_IORING_OP_MKDIRAT = IORING_OP_MKDIRAT, | ||
|
|
||
| SW_IORING_OP_FSTAT = 1000, | ||
| SW_IORING_OP_LSTAT = 1001, | ||
| SW_IORING_OP_UNLINK_FILE = 1002, | ||
| SW_IORING_OP_UNLINK_DIR = 1003, | ||
| SW_IORING_OP_FSYNC = 1004, | ||
| SW_IORING_OP_FDATASYNC = 1005, | ||
| }; | ||
|
|
||
| namespace swoole { | ||
| struct IouringEvent { | ||
| int fd; | ||
| int flags; | ||
| int opcode; | ||
| mode_t mode; | ||
| uint64_t count; // share with offset | ||
| ssize_t result; | ||
| void *rbuf; | ||
| void *coroutine; | ||
| const void *wbuf; | ||
| const char *pathname; | ||
| const char *pathname2; | ||
| struct statx *statxbuf; | ||
| }; | ||
|
|
||
| class Iouring { | ||
| private: | ||
| int ring_fd; | ||
| uint64_t task_num = 0; | ||
| uint64_t entries = 8192; | ||
| struct io_uring ring; | ||
| std::queue<IouringEvent *> waiting_tasks; | ||
| network::Socket *iou_socket = nullptr; | ||
| Reactor *reactor = nullptr; | ||
|
|
||
| void add_event(); | ||
| void delete_event(); | ||
| bool wakeup(); | ||
| bool open(IouringEvent *event); | ||
| bool close(IouringEvent *event); | ||
| bool wr(IouringEvent *event); | ||
| bool statx(IouringEvent *event); | ||
| bool mkdir(IouringEvent *event); | ||
| bool unlink(IouringEvent *event); | ||
| bool rename(IouringEvent *event); | ||
| bool fsync(IouringEvent *event); | ||
|
|
||
| inline struct io_uring_sqe *get_iouring_sqe() { | ||
| struct io_uring_sqe *sqe = io_uring_get_sqe(&ring); | ||
| // We need to reset the values of each sqe structure so that they can be used in a loop. | ||
| if (sqe) { | ||
| memset(sqe, 0, sizeof(struct io_uring_sqe)); | ||
| } | ||
| return sqe; | ||
| } | ||
|
|
||
| inline bool submit_iouring_sqe(IouringEvent *event) { | ||
| int ret = io_uring_submit(&ring); | ||
|
|
||
| if (ret < 0) { | ||
| errno = -ret; | ||
| if (ret == -EAGAIN) { | ||
| waiting_tasks.push(event); | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| task_num++; | ||
| return true; | ||
| } | ||
|
|
||
| static Iouring *create_iouring(); | ||
|
|
||
| public: | ||
| Iouring(Reactor *reactor_); | ||
| ~Iouring(); | ||
|
|
||
| enum flags { | ||
| SW_IOURING_DEFAULT = 0, | ||
| SW_IOURING_SQPOLL = IORING_SETUP_SQPOLL, | ||
| }; | ||
|
|
||
| inline bool is_empty_waiting_tasks() { | ||
| return waiting_tasks.size() == 0; | ||
| } | ||
|
|
||
| inline uint64_t get_task_num() { | ||
| return task_num; | ||
| } | ||
|
|
||
| static int async(opcodes opcode, | ||
| int fd = 0, | ||
| uint64_t count = 0, | ||
| void *rbuf = nullptr, | ||
| const void *wbuf = nullptr, | ||
| struct statx *statxbuf = nullptr); | ||
| static int async(opcodes opcode, | ||
| const char *pathname = nullptr, | ||
| const char *pathname2 = nullptr, | ||
| struct statx *statxbuf = nullptr, | ||
| int flags = 0, | ||
| mode_t mode = 0); | ||
| static int callback(Reactor *reactor, Event *event); | ||
| }; | ||
| }; // namespace swoole | ||
| #endif | ||
| #endif |
Oops, something went wrong.