From cf93360b9c174c19f148c015272efdfe6586e7e3 Mon Sep 17 00:00:00 2001 From: Pavan-Nambi Date: Fri, 24 Oct 2025 19:02:27 +0530 Subject: [PATCH] cleanup MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Mikaël Francoeur cleanup Co-authored-by: Mikaël Francoeur --- simulator/generation/query.rs | 4 +++- simulator/profiles/query.rs | 1 - sql_generation/model/query/pragma.rs | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/simulator/generation/query.rs b/simulator/generation/query.rs index 443a67231..02d82c17f 100644 --- a/simulator/generation/query.rs +++ b/simulator/generation/query.rs @@ -88,7 +88,7 @@ fn random_create_index( fn random_pragma(rng: &mut R, _conn_ctx: &impl GenerationContext) -> Query { const ALL_MODES: [VacuumMode; 2] = [ VacuumMode::None, - // VacuumMode::Incremental, not implemented yer + // VacuumMode::Incremental, not implemented yet VacuumMode::Full, ]; @@ -162,6 +162,8 @@ impl QueryDiscriminants { fn weight(&self, remaining: &Remaining) -> u32 { match self { QueryDiscriminants::Create => remaining.create, + // remaining.select / 3 is for the random_expr generation + // have a max of 1 so that we always generate at least a non zero weight for `QueryDistribution` QueryDiscriminants::Select => (remaining.select + remaining.select / 3).max(1), QueryDiscriminants::Insert => remaining.insert, QueryDiscriminants::Delete => remaining.delete, diff --git a/simulator/profiles/query.rs b/simulator/profiles/query.rs index 0992303d2..3f0de4bd9 100644 --- a/simulator/profiles/query.rs +++ b/simulator/profiles/query.rs @@ -71,5 +71,4 @@ pub enum QueryTypes { Update, Delete, DropTable, - pragma_weight, } diff --git a/sql_generation/model/query/pragma.rs b/sql_generation/model/query/pragma.rs index 351f36f9c..0bee9cbff 100644 --- a/sql_generation/model/query/pragma.rs +++ b/sql_generation/model/query/pragma.rs @@ -24,7 +24,7 @@ impl Display for Pragma { VacuumMode::Full => "full", }; - write!(f, "PRAGMA auto_vacuum={mode} ")?; + write!(f, "PRAGMA auto_vacuum={mode}")?; Ok(()) } }