github-mirror/fish-shell
2
0
Fork 0
mirror of https://github.com/fish-shell/fish-shell.git synced 2024-11-25 09:39:52 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

builtins/printf: avoid string copies by formatting directly to buffer

Browse Source 
Closes #9765.
This commit is contained in:
Xiretza 2023-04-29 18:53:33 +00:00 committed by Mahmoud Al-Qudsi
parent 4771f25102
commit afe2e9d8db
1 changed files with 23 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
fish-rust/src/builtins/printf.rs
View File

@ -261,20 +261,19 @@ impl<'a> builtin_printf_state_t<'a> {
argument: &wstr,
) {
/// Printf macro helper which provides our locale.
macro_rules! sprintf_loc {
macro_rules! append_output_fmt {
(
$fmt:expr, // format string of type &wstr
$($arg:expr),* // arguments
) => {
{
let mut target = WString::new();
// Don't output if we're done.
if !self.early_exit {
sprintf_locale(
&mut target,
&mut self.buff,
$fmt,
&self.locale,
&[$($arg.to_arg()),*]
);
target
)
}
}
}
@ -307,15 +306,15 @@ impl<'a> builtin_printf_state_t<'a> {
let arg: i64 = string_to_scalar_type(argument, self);
if !have_field_width {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, arg));
append_output_fmt!(fmt, arg);
} else {
self.append_output_str(sprintf_loc!(fmt, precision, arg));
append_output_fmt!(fmt, precision, arg);
}
} else {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, field_width, arg));
append_output_fmt!(fmt, field_width, arg);
} else {
self.append_output_str(sprintf_loc!(fmt, field_width, precision, arg));
append_output_fmt!(fmt, field_width, precision, arg);
}
}
}
@ -323,15 +322,15 @@ impl<'a> builtin_printf_state_t<'a> {
let arg: u64 = string_to_scalar_type(argument, self);
if !have_field_width {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, arg));
append_output_fmt!(fmt, arg);
} else {
self.append_output_str(sprintf_loc!(fmt, precision, arg));
append_output_fmt!(fmt, precision, arg);
}
} else {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, field_width, arg));
append_output_fmt!(fmt, field_width, arg);
} else {
self.append_output_str(sprintf_loc!(fmt, field_width, precision, arg));
append_output_fmt!(fmt, field_width, precision, arg);
}
}
}
@ -340,39 +339,39 @@ impl<'a> builtin_printf_state_t<'a> {
let arg: f64 = string_to_scalar_type(argument, self);
if !have_field_width {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, arg));
append_output_fmt!(fmt, arg);
} else {
self.append_output_str(sprintf_loc!(fmt, precision, arg));
append_output_fmt!(fmt, precision, arg);
}
} else {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, field_width, arg));
append_output_fmt!(fmt, field_width, arg);
} else {
self.append_output_str(sprintf_loc!(fmt, field_width, precision, arg));
append_output_fmt!(fmt, field_width, precision, arg);
}
}
}
'c' => {
if !have_field_width {
self.append_output_str(sprintf_loc!(fmt, argument.char_at(0)));
append_output_fmt!(fmt, argument.char_at(0));
} else {
self.append_output_str(sprintf_loc!(fmt, field_width, argument.char_at(0)));
append_output_fmt!(fmt, field_width, argument.char_at(0));
}
}
's' => {
if !have_field_width {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, argument));
append_output_fmt!(fmt, argument);
} else {
self.append_output_str(sprintf_loc!(fmt, precision, argument));
append_output_fmt!(fmt, precision, argument);
}
} else {
if !have_precision {
self.append_output_str(sprintf_loc!(fmt, field_width, argument));
append_output_fmt!(fmt, field_width, argument);
} else {
self.append_output_str(sprintf_loc!(fmt, field_width, precision, argument));
append_output_fmt!(fmt, field_width, precision, argument);
}
}
}
@ -763,15 +762,6 @@ impl<'a> builtin_printf_state_t<'a> {
self.buff.push(c);
}
fn append_output_str<Str: AsRef<wstr>>(&mut self, s: Str) {
// Don't output if we're done.
if self.early_exit {
return;
}
self.buff.push_utfstr(&s);
}
}
/// The printf builtin.