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              : }