Line data Source code
1 : //! Enum-dispatch to the `OpenOptions` type of the respective [`super::IoEngineKind`];
2 :
3 : use super::io_engine::IoEngine;
4 : use std::{os::fd::OwnedFd, path::Path};
5 :
6 : #[derive(Debug, Clone)]
7 : pub enum OpenOptions {
8 : StdFs(std::fs::OpenOptions),
9 : #[cfg(target_os = "linux")]
10 : TokioEpollUring(tokio_epoll_uring::ops::open_at::OpenOptions),
11 : }
12 :
13 : impl Default for OpenOptions {
14 12614 : fn default() -> Self {
15 12614 : match super::io_engine::get() {
16 0 : IoEngine::NotSet => panic!("io engine not set"),
17 6300 : IoEngine::StdFs => Self::StdFs(std::fs::OpenOptions::new()),
18 : #[cfg(target_os = "linux")]
19 : IoEngine::TokioEpollUring => {
20 6314 : Self::TokioEpollUring(tokio_epoll_uring::ops::open_at::OpenOptions::new())
21 : }
22 : }
23 12614 : }
24 : }
25 :
26 : impl OpenOptions {
27 12614 : pub fn new() -> OpenOptions {
28 12614 : Self::default()
29 12614 : }
30 :
31 8396 : pub fn read(&mut self, read: bool) -> &mut OpenOptions {
32 8396 : match self {
33 4198 : OpenOptions::StdFs(x) => {
34 4198 : let _ = x.read(read);
35 4198 : }
36 : #[cfg(target_os = "linux")]
37 4198 : OpenOptions::TokioEpollUring(x) => {
38 4198 : let _ = x.read(read);
39 4198 : }
40 : }
41 8396 : self
42 8396 : }
43 :
44 6838 : pub fn write(&mut self, write: bool) -> &mut OpenOptions {
45 6838 : match self {
46 3412 : OpenOptions::StdFs(x) => {
47 3412 : let _ = x.write(write);
48 3412 : }
49 : #[cfg(target_os = "linux")]
50 3426 : OpenOptions::TokioEpollUring(x) => {
51 3426 : let _ = x.write(write);
52 3426 : }
53 : }
54 6838 : self
55 6838 : }
56 :
57 17860 : pub fn create(&mut self, create: bool) -> &mut OpenOptions {
58 17860 : match self {
59 8916 : OpenOptions::StdFs(x) => {
60 8916 : let _ = x.create(create);
61 8916 : }
62 : #[cfg(target_os = "linux")]
63 8944 : OpenOptions::TokioEpollUring(x) => {
64 8944 : let _ = x.create(create);
65 8944 : }
66 : }
67 17860 : self
68 17860 : }
69 :
70 13382 : pub fn create_new(&mut self, create_new: bool) -> &mut OpenOptions {
71 13382 : match self {
72 6684 : OpenOptions::StdFs(x) => {
73 6684 : let _ = x.create_new(create_new);
74 6684 : }
75 : #[cfg(target_os = "linux")]
76 6698 : OpenOptions::TokioEpollUring(x) => {
77 6698 : let _ = x.create_new(create_new);
78 6698 : }
79 : }
80 13382 : self
81 13382 : }
82 :
83 15248 : pub fn truncate(&mut self, truncate: bool) -> &mut OpenOptions {
84 15248 : match self {
85 7610 : OpenOptions::StdFs(x) => {
86 7610 : let _ = x.truncate(truncate);
87 7610 : }
88 : #[cfg(target_os = "linux")]
89 7638 : OpenOptions::TokioEpollUring(x) => {
90 7638 : let _ = x.truncate(truncate);
91 7638 : }
92 : }
93 15248 : self
94 15248 : }
95 :
96 390696 : pub(in crate::virtual_file) async fn open(&self, path: &Path) -> std::io::Result<OwnedFd> {
97 390696 : match self {
98 196959 : OpenOptions::StdFs(x) => x.open(path).map(|file| file.into()),
99 : #[cfg(target_os = "linux")]
100 193737 : OpenOptions::TokioEpollUring(x) => {
101 193737 : let system = super::io_engine::tokio_epoll_uring_ext::thread_local_system().await;
102 193737 : system.open(path, x).await.map_err(|e| match e {
103 0 : tokio_epoll_uring::Error::Op(e) => e,
104 0 : tokio_epoll_uring::Error::System(system) => {
105 0 : std::io::Error::new(std::io::ErrorKind::Other, system)
106 : }
107 193737 : })
108 : }
109 : }
110 390696 : }
111 : }
112 :
113 : impl std::os::unix::prelude::OpenOptionsExt for OpenOptions {
114 0 : fn mode(&mut self, mode: u32) -> &mut OpenOptions {
115 0 : match self {
116 0 : OpenOptions::StdFs(x) => {
117 0 : let _ = x.mode(mode);
118 0 : }
119 : #[cfg(target_os = "linux")]
120 0 : OpenOptions::TokioEpollUring(x) => {
121 0 : let _ = x.mode(mode);
122 0 : }
123 : }
124 0 : self
125 0 : }
126 :
127 0 : fn custom_flags(&mut self, flags: i32) -> &mut OpenOptions {
128 0 : match self {
129 0 : OpenOptions::StdFs(x) => {
130 0 : let _ = x.custom_flags(flags);
131 0 : }
132 : #[cfg(target_os = "linux")]
133 0 : OpenOptions::TokioEpollUring(x) => {
134 0 : let _ = x.custom_flags(flags);
135 0 : }
136 : }
137 0 : self
138 0 : }
139 : }
|