Skip to content

Commit c512303

Browse files
committed
Differenciate between RW and RO postgress connections
1 parent 213ffca commit c512303

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

crates/corro-pg/src/lib.rs

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -556,21 +556,23 @@ pub async fn start(
556556
let server = TcpListener::bind(pg.bind_addr).await?;
557557
let (tls_acceptor, ssl_required) = setup_tls(pg).await?;
558558
let local_addr = server.local_addr()?;
559+
let conn_gauge = persistent_gauge!("corro.api.active.streams",
560+
"source" => "postgres",
561+
"protocol" => "pg",
562+
"readonly" => readonly.to_string(),
563+
);
559564

560565
tokio::spawn(async move {
561566
loop {
562567
let (tcp_conn, remote_addr) = match server.accept().preemptible(&mut tripwire).await {
563568
Outcome::Completed(res) => res?,
564569
Outcome::Preempted(_) => break,
565570
};
566-
let mut conn = CountedTcpStream::wrap(
567-
tcp_conn,
568-
persistent_gauge!("corro.api.active.streams", "source" => "postgres", "protocol" => "pg"),
569-
);
571+
let mut conn = CountedTcpStream::wrap(tcp_conn, conn_gauge.clone());
570572
let tls_acceptor = tls_acceptor.clone();
571573
debug!("Accepted a PostgreSQL connection (from: {remote_addr})");
572574

573-
counter!("corro.api.connection.count", "protocol" => "pg").increment(1);
575+
counter!("corro.api.connection.count", "protocol" => "pg", "readonly" => readonly.to_string()).increment(1);
574576

575577
let agent = agent.clone();
576578
tokio::spawn(async move {

0 commit comments

Comments
 (0)