docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-22 17:06:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename a couple of variables (#839)

Browse Source
This commit is contained in:
Mostafa
2024-10-23 06:38:07 -05:00
committed by GitHub
parent 186e72298f
commit c27d801abf
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/query_router.rs
View File

@@ -386,14 +386,14 @@ impl QueryRouter {
}
}
/// Determines if a query is mutable or not.
fn query_is_mutable_statement(q: &sqlparser::ast::Query) -> bool {
/// Determines if a query is a mutation or not.
fn is_mutation_query(q: &sqlparser::ast::Query) -> bool {
use sqlparser::ast::*;
match q.body.as_ref() {
SetExpr::Insert(_) => true,
SetExpr::Update(_) => true,
SetExpr::Query(q) => Self::query_is_mutable_statement(q),
SetExpr::Query(q) => Self::is_mutation_query(q),
_ => false,
}
}
@@ -440,9 +440,9 @@ impl QueryRouter {
};
let has_locks = !query.locks.is_empty();
let is_mutable_statement = Self::query_is_mutable_statement(query);
let has_mutation = Self::is_mutation_query(query);
if has_locks || is_mutable_statement {
if has_locks || has_mutation {
self.active_role = Some(Role::Primary);
} else if !visited_write_statement {
// If we already visited a write statement, we should be going to the primary.