@@ -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