Skip to content

Commit 26f416a

Browse files
committed
chore: fix codestyle
1 parent b59ab93 commit 26f416a

4 files changed

Lines changed: 57 additions & 37 deletions

File tree

src/ast/mod.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,8 @@ pub use self::ddl::{
8484
pub use self::dml::{
8585
Delete, Insert, Merge, MergeAction, MergeClause, MergeClauseKind, MergeInsertExpr,
8686
MergeInsertKind, MergeUpdateExpr, MultiTableInsertIntoClause, MultiTableInsertType,
87-
MultiTableInsertValue, MultiTableInsertValues, MultiTableInsertWhenClause, OutputClause, Update,
87+
MultiTableInsertValue, MultiTableInsertValues, MultiTableInsertWhenClause, OutputClause,
88+
Update,
8889
};
8990
pub use self::operator::{BinaryOperator, UnaryOperator};
9091
pub use self::query::{

src/ast/spans.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1312,8 +1312,8 @@ impl Spanned for Insert {
13121312
priority: _, // todo, mysql specific
13131313
insert_alias: _, // todo, mysql specific
13141314
assignments,
1315-
settings: _, // todo, clickhouse specific
1316-
format_clause: _, // todo, clickhouse specific
1315+
settings: _, // todo, clickhouse specific
1316+
format_clause: _, // todo, clickhouse specific
13171317
multi_table_insert_type: _, // snowflake multi-table insert
13181318
multi_table_into_clauses: _, // snowflake multi-table insert
13191319
multi_table_when_clauses: _, // snowflake multi-table insert

src/dialect/snowflake.rs

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,11 @@ use crate::ast::{
3636
RefreshModeKind, RowAccessPolicy, ShowObjects, SqlOption, Statement,
3737
StorageSerializationPolicy, TableObject, TagsColumnOption, Value, WrappedCollection,
3838
};
39-
use crate::tokenizer::TokenWithSpan;
4039
use crate::dialect::{Dialect, Precedence};
4140
use crate::keywords::Keyword;
4241
use crate::parser::{IsOptional, Parser, ParserError};
4342
use crate::tokenizer::Token;
43+
use crate::tokenizer::TokenWithSpan;
4444
#[cfg(not(feature = "std"))]
4545
use alloc::boxed::Box;
4646
#[cfg(not(feature = "std"))]
@@ -1824,7 +1824,9 @@ fn parse_multi_table_insert_into_clause(
18241824
}
18251825

18261826
/// Parse a single value in a multi-table INSERT VALUES clause.
1827-
fn parse_multi_table_insert_value(parser: &mut Parser) -> Result<MultiTableInsertValue, ParserError> {
1827+
fn parse_multi_table_insert_value(
1828+
parser: &mut Parser,
1829+
) -> Result<MultiTableInsertValue, ParserError> {
18281830
if parser.parse_keyword(Keyword::DEFAULT) {
18291831
Ok(MultiTableInsertValue::Default)
18301832
} else {
@@ -1835,8 +1837,13 @@ fn parse_multi_table_insert_value(parser: &mut Parser) -> Result<MultiTableInser
18351837
/// Parse WHEN clauses for conditional multi-table INSERT.
18361838
fn parse_multi_table_insert_when_clauses(
18371839
parser: &mut Parser,
1838-
) -> Result<(Vec<MultiTableInsertWhenClause>, Option<Vec<MultiTableInsertIntoClause>>), ParserError>
1839-
{
1840+
) -> Result<
1841+
(
1842+
Vec<MultiTableInsertWhenClause>,
1843+
Option<Vec<MultiTableInsertIntoClause>>,
1844+
),
1845+
ParserError,
1846+
> {
18401847
let mut when_clauses = vec![];
18411848
let mut else_clause = None;
18421849

tests/sqlparser_snowflake.rs

Lines changed: 42 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -3434,8 +3434,9 @@ fn test_multi_table_insert_unconditional() {
34343434
snowflake().verified_stmt("INSERT ALL INTO t1 (c1, c2, c3) SELECT n1, n2, n3 FROM src");
34353435

34363436
// With VALUES clause
3437-
snowflake()
3438-
.verified_stmt("INSERT ALL INTO t1 (c1, c2, c3) VALUES (n2, n1, DEFAULT) SELECT n1, n2, n3 FROM src");
3437+
snowflake().verified_stmt(
3438+
"INSERT ALL INTO t1 (c1, c2, c3) VALUES (n2, n1, DEFAULT) SELECT n1, n2, n3 FROM src",
3439+
);
34393440

34403441
// Complex example from Snowflake docs
34413442
snowflake().verified_stmt(
@@ -3450,24 +3451,19 @@ fn test_multi_table_insert_unconditional() {
34503451
fn test_multi_table_insert_conditional() {
34513452
// Basic conditional multi-table insert with WHEN clause
34523453
// See: https://docs.snowflake.com/en/sql-reference/sql/insert-multi-table
3453-
snowflake().verified_stmt(
3454-
"INSERT ALL WHEN n1 > 100 THEN INTO t1 SELECT n1 FROM src"
3455-
);
3454+
snowflake().verified_stmt("INSERT ALL WHEN n1 > 100 THEN INTO t1 SELECT n1 FROM src");
34563455

34573456
// Multiple WHEN clauses
34583457
snowflake().verified_stmt(
3459-
"INSERT ALL WHEN n1 > 100 THEN INTO t1 WHEN n1 > 10 THEN INTO t2 SELECT n1 FROM src"
3458+
"INSERT ALL WHEN n1 > 100 THEN INTO t1 WHEN n1 > 10 THEN INTO t2 SELECT n1 FROM src",
34603459
);
34613460

34623461
// WHEN with multiple INTO clauses
3463-
snowflake().verified_stmt(
3464-
"INSERT ALL WHEN n1 > 10 THEN INTO t1 INTO t2 SELECT n1 FROM src"
3465-
);
3462+
snowflake().verified_stmt("INSERT ALL WHEN n1 > 10 THEN INTO t1 INTO t2 SELECT n1 FROM src");
34663463

34673464
// With ELSE clause
3468-
snowflake().verified_stmt(
3469-
"INSERT ALL WHEN n1 > 100 THEN INTO t1 ELSE INTO t2 SELECT n1 FROM src"
3470-
);
3465+
snowflake()
3466+
.verified_stmt("INSERT ALL WHEN n1 > 100 THEN INTO t1 ELSE INTO t2 SELECT n1 FROM src");
34713467

34723468
// Complex conditional insert from Snowflake docs
34733469
snowflake().verified_stmt(
@@ -3481,36 +3477,30 @@ fn test_multi_table_insert_conditional() {
34813477

34823478
// With OVERWRITE
34833479
snowflake().verified_stmt(
3484-
"INSERT OVERWRITE ALL WHEN n1 > 100 THEN INTO t1 ELSE INTO t2 SELECT n1 FROM src"
3480+
"INSERT OVERWRITE ALL WHEN n1 > 100 THEN INTO t1 ELSE INTO t2 SELECT n1 FROM src",
34853481
);
34863482

34873483
// WHEN with always-true condition
3488-
snowflake().verified_stmt(
3489-
"INSERT ALL WHEN 1 = 1 THEN INTO t1 SELECT n1 FROM src"
3490-
);
3484+
snowflake().verified_stmt("INSERT ALL WHEN 1 = 1 THEN INTO t1 SELECT n1 FROM src");
34913485
}
34923486

34933487
#[test]
34943488
fn test_multi_table_insert_with_values() {
34953489
// INTO clause with VALUES using column references
3496-
snowflake().verified_stmt(
3497-
"INSERT ALL INTO t1 VALUES (n1, n2) SELECT n1, n2 FROM src"
3498-
);
3490+
snowflake().verified_stmt("INSERT ALL INTO t1 VALUES (n1, n2) SELECT n1, n2 FROM src");
34993491

35003492
// INTO clause with VALUES using DEFAULT
35013493
snowflake().verified_stmt(
3502-
"INSERT ALL INTO t1 (c1, c2, c3) VALUES (n1, n2, DEFAULT) SELECT n1, n2 FROM src"
3494+
"INSERT ALL INTO t1 (c1, c2, c3) VALUES (n1, n2, DEFAULT) SELECT n1, n2 FROM src",
35033495
);
35043496

35053497
// INTO clause with VALUES using NULL
35063498
snowflake().verified_stmt(
3507-
"INSERT ALL INTO t1 (c1, c2, c3) VALUES (n1, NULL, n2) SELECT n1, n2 FROM src"
3499+
"INSERT ALL INTO t1 (c1, c2, c3) VALUES (n1, NULL, n2) SELECT n1, n2 FROM src",
35083500
);
35093501

35103502
// Positional alias in VALUES
3511-
snowflake().verified_stmt(
3512-
"INSERT ALL INTO t1 VALUES ($1, $2) SELECT 1, 50 AS an_alias"
3513-
);
3503+
snowflake().verified_stmt("INSERT ALL INTO t1 VALUES ($1, $2) SELECT 1, 50 AS an_alias");
35143504
}
35153505

35163506
/// Unit tests for multi-table INSERT AST structure validation
@@ -3631,15 +3621,33 @@ fn test_multi_table_insert_ast_conditional() {
36313621
assert_eq!(multi_table_when_clauses.len(), 2);
36323622

36333623
// First WHEN clause: WHEN n1 > 100 THEN INTO t1
3634-
assert_eq!(multi_table_when_clauses[0].condition.to_string(), "n1 > 100");
3624+
assert_eq!(
3625+
multi_table_when_clauses[0].condition.to_string(),
3626+
"n1 > 100"
3627+
);
36353628
assert_eq!(multi_table_when_clauses[0].into_clauses.len(), 1);
3636-
assert_eq!(multi_table_when_clauses[0].into_clauses[0].table_name.to_string(), "t1");
3629+
assert_eq!(
3630+
multi_table_when_clauses[0].into_clauses[0]
3631+
.table_name
3632+
.to_string(),
3633+
"t1"
3634+
);
36373635

36383636
// Second WHEN clause: WHEN n1 > 10 THEN INTO t2 INTO t3
36393637
assert_eq!(multi_table_when_clauses[1].condition.to_string(), "n1 > 10");
36403638
assert_eq!(multi_table_when_clauses[1].into_clauses.len(), 2);
3641-
assert_eq!(multi_table_when_clauses[1].into_clauses[0].table_name.to_string(), "t2");
3642-
assert_eq!(multi_table_when_clauses[1].into_clauses[1].table_name.to_string(), "t3");
3639+
assert_eq!(
3640+
multi_table_when_clauses[1].into_clauses[0]
3641+
.table_name
3642+
.to_string(),
3643+
"t2"
3644+
);
3645+
assert_eq!(
3646+
multi_table_when_clauses[1].into_clauses[1]
3647+
.table_name
3648+
.to_string(),
3649+
"t3"
3650+
);
36433651

36443652
// ELSE clause: ELSE INTO t4
36453653
let else_clause = multi_table_else_clause.expect("Expected ELSE clause");
@@ -3653,7 +3661,8 @@ fn test_multi_table_insert_ast_conditional() {
36533661
#[test]
36543662
fn test_multi_table_insert_ast_first() {
36553663
// Test INSERT FIRST vs INSERT ALL
3656-
let sql = "INSERT FIRST WHEN n1 > 100 THEN INTO t1 WHEN n1 > 10 THEN INTO t2 SELECT n1 FROM src";
3664+
let sql =
3665+
"INSERT FIRST WHEN n1 > 100 THEN INTO t1 WHEN n1 > 10 THEN INTO t2 SELECT n1 FROM src";
36573666
let stmt = snowflake().verified_stmt(sql);
36583667

36593668
match stmt {
@@ -3704,7 +3713,10 @@ fn test_multi_table_insert_ast_complex_values() {
37043713
}) => {
37053714
assert_eq!(multi_table_into_clauses.len(), 1);
37063715

3707-
let values = multi_table_into_clauses[0].values.as_ref().expect("Expected VALUES");
3716+
let values = multi_table_into_clauses[0]
3717+
.values
3718+
.as_ref()
3719+
.expect("Expected VALUES");
37083720
assert_eq!(values.values.len(), 3);
37093721

37103722
// First value: n1 + n2 (binary expression)

0 commit comments

Comments
 (0)