rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-26 01:20:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Write traces to file

Browse Source
This commit is contained in:
Bojan Serafimov
2022-09-12 14:43:07 -04:00
parent 0bfc422eb3
commit 646e0f3581
2 changed files with 61 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
pageserver/src/page_service.rs
View File

@@ -118,6 +118,50 @@ pub struct PagestreamDbSizeResponse {
}
impl PagestreamFeMessage {
pub fn serialize(&self) -> Bytes {
let mut bytes = BytesMut::new();
match self {
Self::Exists(req) => {
bytes.put_u8(if req.latest { 1 } else { 0 });
bytes.put_u64(req.lsn.0);
bytes.put_u32(req.rel.spcnode);
bytes.put_u32(req.rel.dbnode);
bytes.put_u32(req.rel.relnode);
bytes.put_u8(req.rel.forknum);
}
Self::Nblocks(req) => {
bytes.put_u8(1);
bytes.put_u8(if req.latest { 1 } else { 0 });
bytes.put_u64(req.lsn.0);
bytes.put_u32(req.rel.spcnode);
bytes.put_u32(req.rel.dbnode);
bytes.put_u32(req.rel.relnode);
bytes.put_u8(req.rel.forknum);
}
Self::GetPage(req) => {
bytes.put_u8(2);
bytes.put_u8(if req.latest { 1 } else { 0 });
bytes.put_u64(req.lsn.0);
bytes.put_u32(req.rel.spcnode);
bytes.put_u32(req.rel.dbnode);
bytes.put_u32(req.rel.relnode);
bytes.put_u8(req.rel.forknum);
}
Self::DbSize(req) => {
bytes.put_u8(3);
bytes.put_u8(if req.latest { 1 } else { 0 });
bytes.put_u64(req.lsn.0);
bytes.put_u32(req.dbnode);
}
}
bytes.into()
}
fn parse(mut body: Bytes) -> anyhow::Result<PagestreamFeMessage> {
// TODO these gets can fail
@@ -479,15 +523,15 @@ impl PageServerHandler {
_ => continue,
};
// Trace request if needed
if let Some(t) = tracer.as_mut() {
t.trace(&copy_data_bytes)
}
let zenith_fe_msg = PagestreamFeMessage::parse(copy_data_bytes)?;
let tenant_id = tenant_id.to_string();
let timeline_id = timeline_id.to_string();
// Trace request if needed
if let Some(t) = tracer.as_mut() {
t.trace(&zenith_fe_msg)
}
let response = match zenith_fe_msg {
PagestreamFeMessage::Exists(req) => SMGR_QUERY_TIME
.with_label_values(&["get_rel_exists", &tenant_id, &timeline_id])

19
pageserver/src/trace.rs
View File

@@ -1,8 +1,12 @@
use crate::page_service::PagestreamFeMessage;
use std::{fs::File, io::Write, path::PathBuf};
use bytes::Bytes;
use std::{
fs::File,
io::{BufWriter, Write},
path::PathBuf,
};
pub struct Tracer {
output: File,
writer: BufWriter<File>,
}
impl Drop for Tracer {
@@ -13,16 +17,17 @@ impl Drop for Tracer {
impl Tracer {
pub fn new(path: PathBuf) -> Self {
let file = File::create(path).expect("failed to create trace file");
Tracer {
output: File::create(path).expect("failed to create trace file"),
writer: BufWriter::new(file),
}
}
pub fn trace(&mut self, _msg: &PagestreamFeMessage) {
// TODO(now) implement
pub fn trace(&mut self, msg: &Bytes) {
self.writer.write(msg).expect("failed to write trace");
}
pub fn flush(&mut self) {
self.output.flush().expect("failed to flush trace file");
self.writer.flush().expect("failed to flush trace file");
}
}