我需要简单地(并且危险地 - 为简洁起见省略错误处理)获取当前的可执行文件名称。我让它工作了,但我的函数将&str
转换为String
只是为了稍后调用as_str()
进行模式匹配。
fn binary_name() -> String {
std::env::current_exe().unwrap().file_name().unwrap().to_str().unwrap().to_string()
}
据我了解,std::env::current_exe()
赋予我PathBuf
的所有权,我可以通过归还它来转让它。就目前而言,我借用它来将其转换为&str
.从那里,返回字符串的唯一方法是在删除PathBuf
之前克隆它。
有没有办法避免这种&OsStr -> &str -> String -> &str
循环?
有没有办法在将
PathBuf
转换为String
时避免克隆?
绝对。但是,这不是您正在做的事情。您正在通过file_name
获取PathBuf
的一部分并将其转换。不能取得字符串部分的所有权。
如果您没有采用子集,则可以通过转换为OsString
然后转换为String
来完成整个PathBuf
。在这里,我忽略特定的错误,只返回成功或失败:
use std::path::PathBuf;
fn exe_name() -> Option<String> {
std::env::current_exe()
.ok()
.map(PathBuf::into_os_string)
.and_then(|exe| exe.into_string().ok())
}
有没有办法避免这种
&OsStr -> &str -> String -> &str
循环?
否,因为您正在方法中创建String
(或OsString
或PathBuf
,以所有权,具体取决于代码的变体)。请查看将本地字符串作为切片 (&str) 返回,了解为什么无法返回对堆栈分配的项目(包括字符串)的引用。
正如问答中所述,如果你想有参考文献,拥有数据的东西必须比参考文献活得更久:
use std::env;
use std::path::Path;
use std::ffi::OsStr;
fn binary_name(path: &Path) -> Option<&str> {
path.file_name().and_then(OsStr::to_str)
}
fn main() {
let exe = env::current_exe().ok();
match exe.as_ref().and_then(|e| binary_name(e)) {
Some("cat") => println!("Called as cat"),
Some("dog") => println!("Called as dog"),
Some(other) => println!("Why did you call me {}?", other),
None => println!("Not able to figure out what I was called as"),
}
}
您的原始代码可以编写为不会轻易因错误而崩溃
fn binary_name() -> Option<String> {
let exe = std::env::current_exe();
exe.ok()
.as_ref()
.and_then(|p| p.file_name())
.and_then(|s| s.to_str())
.map(String::from)
}