opendal/services/fs/
core.rs

1// Licensed to the Apache Software Foundation (ASF) under one
2// or more contributor license agreements.  See the NOTICE file
3// distributed with this work for additional information
4// regarding copyright ownership.  The ASF licenses this file
5// to you under the Apache License, Version 2.0 (the
6// "License"); you may not use this file except in compliance
7// with the License.  You may obtain a copy of the License at
8//
9//   http://www.apache.org/licenses/LICENSE-2.0
10//
11// Unless required by applicable law or agreed to in writing,
12// software distributed under the License is distributed on an
13// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14// KIND, either express or implied.  See the License for the
15// specific language governing permissions and limitations
16// under the License.
17
18use std::path::Path;
19use std::path::PathBuf;
20use std::sync::Arc;
21use uuid::Uuid;
22
23use crate::raw::*;
24use crate::*;
25
26#[derive(Debug)]
27pub struct FsCore {
28    pub info: Arc<AccessorInfo>,
29    pub root: PathBuf,
30    pub atomic_write_dir: Option<PathBuf>,
31    pub buf_pool: oio::PooledBuf,
32}
33
34impl FsCore {
35    // Synchronously build write path and ensure the parent dirs created
36    pub fn blocking_ensure_write_abs_path(&self, parent: &Path, path: &str) -> Result<PathBuf> {
37        let p = parent.join(path);
38
39        // Create dir before write path.
40        //
41        // TODO(xuanwo): There are many works to do here:
42        //   - Is it safe to create dir concurrently?
43        //   - Do we need to extract this logic as new util functions?
44        //   - Is it better to check the parent dir exists before call mkdir?
45        let parent = PathBuf::from(&p)
46            .parent()
47            .ok_or_else(|| {
48                Error::new(
49                    ErrorKind::Unexpected,
50                    "path should have parent but not, it must be malformed",
51                )
52                .with_context("input", p.to_string_lossy())
53            })?
54            .to_path_buf();
55
56        std::fs::create_dir_all(parent).map_err(new_std_io_error)?;
57
58        Ok(p)
59    }
60
61    // Build write path and ensure the parent dirs created
62    pub async fn ensure_write_abs_path(&self, parent: &Path, path: &str) -> Result<PathBuf> {
63        let p = parent.join(path);
64
65        // Create dir before write path.
66        //
67        // TODO(xuanwo): There are many works to do here:
68        //   - Is it safe to create dir concurrently?
69        //   - Do we need to extract this logic as new util functions?
70        //   - Is it better to check the parent dir exists before call mkdir?
71        let parent = PathBuf::from(&p)
72            .parent()
73            .ok_or_else(|| {
74                Error::new(
75                    ErrorKind::Unexpected,
76                    "path should have parent but not, it must be malformed",
77                )
78                .with_context("input", p.to_string_lossy())
79            })?
80            .to_path_buf();
81
82        tokio::fs::create_dir_all(&parent)
83            .await
84            .map_err(new_std_io_error)?;
85
86        Ok(p)
87    }
88}
89
90#[inline]
91pub fn tmp_file_of(path: &str) -> String {
92    let name = get_basename(path);
93    let uuid = Uuid::new_v4().to_string();
94
95    format!("{name}.{uuid}")
96}
opendal/services/fs/core.rs

opendal/services/fs/
core.rs
