perf: 字符串拼接按操作数字节长预分配容量,免几何 realloc 二次拷贝

This commit is contained in:
爱特每羊队
2026-08-10 17:40:39 +08:00
parent b70e334bce
commit 61898f8983
3 changed files with 14 additions and 10 deletions
+14 -6
View File
@@ -16,12 +16,16 @@ impl Vm {
let lhs = self.coerce_primitive_bounded(lv, false)?;
let rhs = self.coerce_primitive_bounded(rv, false)?;
if lhs.is_string() || rhs.is_string() {
let mut buf = std::mem::take(&mut self.string_buf);
buf.clear();
let lbytes = if lhs.is_string() { unsafe { (*lhs.as_string_ptr()).len() } } else { 0 };
let rbytes = if rhs.is_string() { unsafe { (*rhs.as_string_ptr()).len() } } else { 0 };
// 预分配精确容量:字符串操作数 O(1) 取字节长,一次分配写齐,免 push_str
// 几何 realloc 的二次拷贝(字符串拼接热路径的主要额外成本)。
// 32 字节余量覆盖数字/布尔等格式化文本(f64 文本最长约 24 字节),
// 避免追加非字符串操作数时二次扩容。
let mut buf = String::with_capacity(lbytes + rbytes + 32);
coercion::push_to_string(lhs, &mut buf);
coercion::push_to_string(rhs, &mut buf);
let result = self.new_string_owned(buf);
self.string_buf = String::new();
self.regs[rd] = result;
} else {
let ln = coercion::to_number(lhs);
@@ -58,12 +62,16 @@ impl Vm {
let lhs = self.coerce_primitive_bounded(lv, false)?;
let rhs = self.coerce_primitive_bounded(rv, false)?;
if lhs.is_string() || rhs.is_string() {
let mut buf = std::mem::take(&mut self.string_buf);
buf.clear();
let lbytes = if lhs.is_string() { unsafe { (*lhs.as_string_ptr()).len() } } else { 0 };
let rbytes = if rhs.is_string() { unsafe { (*rhs.as_string_ptr()).len() } } else { 0 };
// 预分配精确容量:字符串操作数 O(1) 取字节长,一次分配写齐,免 push_str
// 几何 realloc 的二次拷贝(字符串拼接热路径的主要额外成本)。
// 32 字节余量覆盖数字/布尔等格式化文本(f64 文本最长约 24 字节),
// 避免追加非字符串操作数时二次扩容。
let mut buf = String::with_capacity(lbytes + rbytes + 32);
coercion::push_to_string(lhs, &mut buf);
coercion::push_to_string(rhs, &mut buf);
let result = self.new_string_owned(buf);
self.string_buf = String::new();
self.regs[rd] = result;
} else {
let ln = coercion::to_number(lhs);
-2
View File
@@ -348,8 +348,6 @@ pub struct Vm {
/// 分组保存 inline cache 与指令计数器。
pub(crate) profiling: ProfilingState,
pub(crate) cell_stack: Vec<Vec<*mut Cell>>,
/// 可复用字符串缓冲区,避免每次 `+` 拼接都分配。
pub(crate) string_buf: String,
}
impl Vm {
-2
View File
@@ -97,7 +97,6 @@ impl Vm {
ic_misses: std::cell::Cell::new(0),
instruction_count: 0,
},
string_buf: String::new(),
cell_stack: Vec::new(),
};
vm.init_generator_intrinsics();
@@ -190,7 +189,6 @@ impl Vm {
ic_misses: std::cell::Cell::new(0),
instruction_count: 0,
},
string_buf: String::new(),
cell_stack: Vec::new(),
};
vm.init_generator_intrinsics();