stats

Author: levkk

CLAUDE.md

@@ -7,45 +7,15 @@
 
 # Code style
 
-Use standard Rust code style. Use `cargo fmt` to reformat code automatically after every edit.
-Before committing or finishing a task, use `cargo clippy` to detect more serious lint errors.
-
-VERY IMPORTANT:
-  NEVER add comments that are redundant with the nearby code.
-  ALWAYS be sure comments document "Why", not "What", the code is doing.
-  ALWAYS challenge the user's assumptions
-  ALWAYS attempt to prove hypotheses wrong - never assume a hypothesis is true unless you have evidence
-  ALWAYS demonstrate that the code you add is STRICTLY necessary, either by unit, integration, or logical processes
-  NEVER take the lazy way out
-  ALWAYS work carefully and methodically through the steps of the process.
-  NEVER use quick fixes. Always carefully work through the problem unless specifically asked.
-  ALWAYS Ask clarifying questions before implementing
-  ALWAYS Break large tasks into single-session chunks
-
-VERY IMPORTANT: you are to act as a detective, attempting to find ways to falsify the code or planning we've done by discovering gaps or inconsistencies. ONLY write code when it is absolutely required to pass tests, the build, or typecheck.
-
-VERY IMPORTANT: NEVER comment out code or skip tests unless specifically requested by the user
-
-## Principles
-- **Data first**: Define types before implementation
-- **Small Modules**: Try to keep files under 200 lines, unless required by implementation. NEVER allow files to exceed 1000 lines unless specifically instructed.
+- Use standard Rust code style.
+- Use `cargo fmt` to reformat code automatically after every edit.
+- Don't write functions with many arguments: create a struct and use that as input instead.
 
 # Workflow
 
 - Prefer to run individual tests with `cargo nextest run --test-threads=1 --no-fail-fast <name of the test here>`. This is much faster.
-- A local PostgreSQL server is required for some tests to pass. Ensure it is set up, and if necessary create a database called "pgdog", and create a user called "pgdog" with password "pgdog".
-- Focus on files in `./pgdog` and `./integration` - other files are LOWEST priority
-
-## Test-Driven Development (TDD) - STRICT ENFORCEMENT
-- **MANDATORY WORKFLOW - NO EXCEPTIONS:**
-  1. Write exactly ONE test that fails
-  2. Write ONLY the minimal code to make that test pass
-  3. Refactor if needed (tests must still pass)
-  4. Return to step 1 for next test
-- **CRITICAL RULES:**
-  - NO implementation code without a failing test first
-  - NO untested code is allowed to exist
-  - Every line of production code must be justified by a test
+- A local PostgreSQL server is required for some tests to pass. Assume it's running, if not, stop and ask the user to start it.
+- Coe
 
 # About the project
 

pgdog/src/admin/mod.rs

@@ -30,6 +30,7 @@ pub mod show_pools;
 pub mod show_prepared_statements;
 pub mod show_query_cache;
 pub mod show_replication;
+pub mod show_rewrite;
 pub mod show_server_memory;
 pub mod show_servers;
 pub mod show_stats;

pgdog/src/admin/parser.rs

@@ -7,7 +7,7 @@ use super::{
     show_client_memory::ShowClientMemory, show_clients::ShowClients, show_config::ShowConfig,
     show_instance_id::ShowInstanceId, show_lists::ShowLists, show_mirrors::ShowMirrors,
     show_peers::ShowPeers, show_pools::ShowPools, show_prepared_statements::ShowPreparedStatements,
-    show_query_cache::ShowQueryCache, show_replication::ShowReplication,
+    show_query_cache::ShowQueryCache, show_replication::ShowReplication, show_rewrite::ShowRewrite,
     show_server_memory::ShowServerMemory, show_servers::ShowServers, show_stats::ShowStats,
     show_transactions::ShowTransactions, show_version::ShowVersion, shutdown::Shutdown, Command,
     Error,
@@ -30,6 +30,7 @@ pub enum ParseResult {
     ShowStats(ShowStats),
     ShowTransactions(ShowTransactions),
     ShowMirrors(ShowMirrors),
+    ShowRewrite(ShowRewrite),
     ShowVersion(ShowVersion),
     ShowInstanceId(ShowInstanceId),
     SetupSchema(SetupSchema),
@@ -65,6 +66,7 @@ impl ParseResult {
             ShowStats(show_stats) => show_stats.execute().await,
             ShowTransactions(show_transactions) => show_transactions.execute().await,
             ShowMirrors(show_mirrors) => show_mirrors.execute().await,
+            ShowRewrite(show_rewrite) => show_rewrite.execute().await,
             ShowVersion(show_version) => show_version.execute().await,
             ShowInstanceId(show_instance_id) => show_instance_id.execute().await,
             SetupSchema(setup_schema) => setup_schema.execute().await,
@@ -100,6 +102,7 @@ impl ParseResult {
             ShowStats(show_stats) => show_stats.name(),
             ShowTransactions(show_transactions) => show_transactions.name(),
             ShowMirrors(show_mirrors) => show_mirrors.name(),
+            ShowRewrite(show_rewrite) => show_rewrite.name(),
             ShowVersion(show_version) => show_version.name(),
             ShowInstanceId(show_instance_id) => show_instance_id.name(),
             SetupSchema(setup_schema) => setup_schema.name(),
@@ -163,6 +166,7 @@ impl Parser {
                 "lists" => ParseResult::ShowLists(ShowLists::parse(&sql)?),
                 "prepared" => ParseResult::ShowPrepared(ShowPreparedStatements::parse(&sql)?),
                 "replication" => ParseResult::ShowReplication(ShowReplication::parse(&sql)?),
+                "rewrite" => ParseResult::ShowRewrite(ShowRewrite::parse(&sql)?),
                 command => {
                     debug!("unknown admin show command: '{}'", command);
                     return Err(Error::Syntax);

pgdog/src/admin/show_mirrors.rs

@@ -35,7 +35,7 @@ impl Command for ShowMirrors {
             let counts = {
                 let stats = cluster.stats();
                 let stats = stats.lock();
-                stats.counts
+                stats.mirrors
             };
 
             // Create a data row for this cluster

pgdog/src/admin/show_rewrite.rs

@@ -0,0 +1,49 @@
+//! SHOW REWRITE - per-cluster rewrite statistics
+
+use crate::backend::databases::databases;
+
+use super::prelude::*;
+
+pub struct ShowRewrite;
+
+#[async_trait]
+impl Command for ShowRewrite {
+    fn name(&self) -> String {
+        "SHOW REWRITE".into()
+    }
+
+    fn parse(_: &str) -> Result<Self, Error> {
+        Ok(Self)
+    }
+
+    async fn execute(&self) -> Result<Vec<Message>, Error> {
+        let fields = vec![
+            Field::text("database"),
+            Field::text("user"),
+            Field::numeric("parse"),
+            Field::numeric("bind"),
+            Field::numeric("simple"),
+        ];
+
+        let mut messages = vec![RowDescription::new(&fields).message()?];
+
+        for (user, cluster) in databases().all() {
+            let rewrite = {
+                let stats = cluster.stats();
+                let stats = stats.lock();
+                stats.rewrite
+            };
+
+            let mut dr = DataRow::new();
+            dr.add(user.database.as_str())
+                .add(user.user.as_str())
+                .add(rewrite.parse as i64)
+                .add(rewrite.bind as i64)
+                .add(rewrite.simple as i64);
+
+            messages.push(dr.message()?);
+        }
+
+        Ok(messages)
+    }
+}

pgdog/src/admin/tests/mod.rs

@@ -1,6 +1,6 @@
 use crate::admin::Command;
 use crate::backend::databases::{databases, from_config, replace_databases, Databases};
-use crate::backend::pool::mirror_stats::Counts;
+use crate::backend::pool::cluster_stats::MirrorStats;
 use crate::config::{self, ConfigAndUsers, Database, Role, User as ConfigUser};
 use crate::net::messages::{DataRow, DataType, FromBytes, Protocol, RowDescription};
 
@@ -226,7 +226,7 @@ async fn show_mirrors_reports_counts() {
     {
         let cluster_stats = cluster.stats();
         let mut stats = cluster_stats.lock();
-        stats.counts = Counts {
+        stats.mirrors = MirrorStats {
             total_count: 5,
             mirrored_count: 4,
             dropped_count: 1,

pgdog/src/backend/pool/cluster.rs

@@ -26,7 +26,7 @@ use crate::{
     net::{messages::BackendKeyData, Query},
 };
 
-use super::{Address, Config, Error, Guard, MirrorStats, Request, Shard, ShardConfig};
+use super::{Address, ClusterStats, Config, Error, Guard, Request, Shard, ShardConfig};
 use crate::config::LoadBalancingStrategy;
 
 #[derive(Clone, Debug, Default)]
@@ -53,7 +53,7 @@ pub struct Cluster {
     multi_tenant: Option<MultiTenant>,
     rw_strategy: ReadWriteStrategy,
     schema_admin: bool,
-    stats: Arc<Mutex<MirrorStats>>,
+    stats: Arc<Mutex<ClusterStats>>,
     cross_shard_disabled: bool,
     two_phase_commit: bool,
     two_phase_commit_auto: bool,
@@ -245,7 +245,7 @@ impl Cluster {
             multi_tenant: multi_tenant.clone(),
             rw_strategy,
             schema_admin,
-            stats: Arc::new(Mutex::new(MirrorStats::default())),
+            stats: Arc::new(Mutex::new(ClusterStats::default())),
             cross_shard_disabled,
             two_phase_commit: two_pc && shards.len() > 1,
             two_phase_commit_auto: two_pc_auto && shards.len() > 1,
@@ -409,7 +409,7 @@ impl Cluster {
         self.schema_admin = owner;
     }
 
-    pub fn stats(&self) -> Arc<Mutex<MirrorStats>> {
+    pub fn stats(&self) -> Arc<Mutex<ClusterStats>> {
         self.stats.clone()
     }
 

pgdog/src/backend/pool/mirror_stats.rs renamed to pgdog/src/backend/pool/cluster_stats.rs

@@ -5,17 +5,19 @@ use std::{
     ops::{Add, Div, Sub},
 };
 
+use crate::frontend::router::rewrite::stats::RewriteStats;
+
 #[derive(Debug, Clone, Default, Copy)]
-pub struct Counts {
+pub struct MirrorStats {
     pub total_count: usize,
     pub mirrored_count: usize,
     pub dropped_count: usize,
     pub error_count: usize,
     pub queue_length: usize,
 }
 
-impl Sub for Counts {
-    type Output = Counts;
+impl Sub for MirrorStats {
+    type Output = MirrorStats;
 
     fn sub(self, rhs: Self) -> Self::Output {
         Self {
@@ -28,8 +30,8 @@ impl Sub for Counts {
     }
 }
 
-impl Div<usize> for Counts {
-    type Output = Counts;
+impl Div<usize> for MirrorStats {
+    type Output = MirrorStats;
 
     fn div(self, rhs: usize) -> Self::Output {
         Self {
@@ -42,11 +44,11 @@ impl Div<usize> for Counts {
     }
 }
 
-impl Add for Counts {
-    type Output = Counts;
+impl Add for MirrorStats {
+    type Output = MirrorStats;
 
-    fn add(self, rhs: Counts) -> Self::Output {
-        Counts {
+    fn add(self, rhs: MirrorStats) -> Self::Output {
+        MirrorStats {
             total_count: self.total_count + rhs.total_count,
             mirrored_count: self.mirrored_count + rhs.mirrored_count,
             dropped_count: self.dropped_count + rhs.dropped_count,
@@ -56,9 +58,9 @@ impl Add for Counts {
     }
 }
 
-impl Sum for Counts {
+impl Sum for MirrorStats {
     fn sum<I: Iterator<Item = Self>>(iter: I) -> Self {
-        let mut result = Counts::default();
+        let mut result = MirrorStats::default();
         for next in iter {
             result = result + next;
         }
@@ -68,8 +70,9 @@ impl Sum for Counts {
 }
 
 #[derive(Debug, Clone, Default, Copy)]
-pub struct MirrorStats {
-    pub counts: Counts,
+pub struct ClusterStats {
+    pub mirrors: MirrorStats,
+    pub rewrite: RewriteStats,
 }
 
 #[cfg(test)]
@@ -78,24 +81,24 @@ mod tests {
 
     #[test]
     fn test_queue_length_default_is_zero() {
-        let stats = MirrorStats::default();
+        let stats = ClusterStats::default();
         assert_eq!(
-            stats.counts.queue_length, 0,
+            stats.mirrors.queue_length, 0,
             "queue_length should be 0 by default"
         );
     }
 
     #[test]
     fn test_queue_length_arithmetic_operations() {
-        let counts1 = Counts {
+        let counts1 = MirrorStats {
             total_count: 10,
             mirrored_count: 5,
             dropped_count: 3,
             error_count: 2,
             queue_length: 7,
         };
 
-        let counts2 = Counts {
+        let counts2 = MirrorStats {
             total_count: 5,
             mirrored_count: 3,
             dropped_count: 1,
@@ -127,15 +130,15 @@ mod tests {
 
     #[test]
     fn test_queue_length_saturating_sub() {
-        let counts1 = Counts {
+        let counts1 = MirrorStats {
             total_count: 10,
             mirrored_count: 5,
             dropped_count: 3,
             error_count: 2,
             queue_length: 3,
         };
 
-        let counts2 = Counts {
+        let counts2 = MirrorStats {
             total_count: 5,
             mirrored_count: 3,
             dropped_count: 1,