diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-29 13:20:00 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-29 13:20:00 +0800 |
| commit | d193950d204ed82bdb3d6a3a111511d189f084bb (patch) | |
| tree | 5331dd37911dd991fc4c6060c7e5427bf4f850fa /src/utils/request.rs | |
| parent | 9416cbd8b81c0029572db854257cfc25991a12cd (diff) | |
| download | aichat-d193950d204ed82bdb3d6a3a111511d189f084bb.tar.gz | |
refactor: the name of temp file (#670)
Diffstat (limited to 'src/utils/request.rs')
| -rw-r--r-- | src/utils/request.rs | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/src/utils/request.rs b/src/utils/request.rs index bc5388c..fadd3b8 100644 --- a/src/utils/request.rs +++ b/src/utils/request.rs @@ -3,7 +3,7 @@ use super::*; use anyhow::{bail, Result}; use http::header::CONTENT_TYPE; use lazy_static::lazy_static; -use std::{collections::HashMap, env, time::Duration}; +use std::{collections::HashMap, time::Duration}; use tokio::io::AsyncWriteExt; pub const URL_LOADER: &str = "url"; @@ -48,10 +48,7 @@ pub async fn fetch(loaders: &HashMap<String, String>, path: &str) -> Result<(Str } let result = match loaders.get(&extension) { Some(loader_command) => { - let save_path = env::temp_dir() - .join(format!("aichat-download-{}.{extension}", sha256(path))) - .display() - .to_string(); + let save_path = temp_file("-download-", "").display().to_string(); let mut save_file = tokio::fs::File::create(&save_path).await?; let mut size = 0; while let Some(chunk) = res.chunk().await? { |
