fix(observability): preserve inbound trace context
CI / Rust Checks (push) Failing after 9m30s
CI / UI Checks (push) Has been skipped
CI / Frontend E2E (push) Has been skipped
CI / Community Image Smoke (push) Has been skipped
CI / Deploy (push) Has been skipped

This commit is contained in:
2026-08-30 00:43:46 +03:00
parent c3188637b3
commit 501e0931c5
3 changed files with 62 additions and 7 deletions
@@ -154,17 +154,19 @@ impl<'a> MakeWriter<'a> for SharedLogWriter {
fn make_writer(&'a self) -> Self::Writer {
SharedLogGuard {
buffer: Arc::clone(&self.buffer),
pending: Vec::new(),
}
}
}
struct SharedLogGuard {
buffer: Arc<Mutex<Vec<u8>>>,
pending: Vec<u8>,
}
impl io::Write for SharedLogGuard {
fn write(&mut self, bytes: &[u8]) -> io::Result<usize> {
self.buffer.lock().unwrap().extend_from_slice(bytes);
self.pending.extend_from_slice(bytes);
Ok(bytes.len())
}
@@ -173,6 +175,24 @@ impl io::Write for SharedLogGuard {
}
}
impl Drop for SharedLogGuard {
fn drop(&mut self) {
self.buffer.lock().unwrap().extend_from_slice(&self.pending);
}
}
#[test]
fn shared_log_writer_publishes_complete_records_only() {
let writer = SharedLogWriter::default();
let mut guard = writer.make_writer();
io::Write::write_all(&mut guard, b"message").unwrap();
assert!(writer.output().is_empty());
io::Write::write_all(&mut guard, b" stage=\"fetch\"\n").unwrap();
drop(guard);
assert_eq!(writer.output(), "message stage=\"fetch\"\n");
}
#[derive(Clone, Debug)]
struct CapturingExporter(Arc<Mutex<Vec<SpanData>>>);