Skip to content

Commit 60b4203

Browse files
xteraoxterao
committed
Support for EXISTS clause format
1 parent a661681 commit 60b4203

File tree

7 files changed

+157
-1
lines changed

7 files changed

+157
-1
lines changed

‎src/main/kotlin/org/domaframework/doma/intellij/formatter/block/group/keyword/option/SqlExistsGroupBlock.kt‎

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
/*
2+
* Copyright Doma Tools Authors
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.domaframework.doma.intellij.formatter.block.group.keyword.option
17+
18+
import com.intellij.lang.ASTNode
19+
import org.domaframework.doma.intellij.formatter.block.SqlBlock
20+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlKeywordGroupBlock
21+
import org.domaframework.doma.intellij.formatter.block.group.keyword.create.SqlCreateTableColumnDefinitionRawGroupBlock
22+
import org.domaframework.doma.intellij.formatter.util.IndentType
23+
import org.domaframework.doma.intellij.formatter.util.SqlBlockFormattingContext
24+
25+
class SqlExistsGroupBlock(
26+
node: ASTNode,
27+
context: SqlBlockFormattingContext,
28+
) : SqlKeywordGroupBlock(node, IndentType.OPTIONS, context) {
29+
override fun setParentGroupBlock(lastGroup: SqlBlock?) {
30+
super.setParentGroupBlock(lastGroup)
31+
indent.indentLen = createBlockIndentLen()
32+
indent.groupIndentLen = createGroupIndentLen()
33+
}
34+
35+
override fun createBlockIndentLen(): Int = parentBlock?.indent?.groupIndentLen?.plus(1) ?: 1
36+
37+
override fun createGroupIndentLen(): Int {
38+
parentBlock?.let { parent ->
39+
return parent.indent.groupIndentLen.plus(topKeywordBlocks.sumOf { it.getNodeText().length.plus(1) })
40+
}
41+
return topKeywordBlocks.sumOf { it.getNodeText().length.plus(1) }.minus(1)
42+
}
43+
44+
override fun isSaveSpace(lastGroup: SqlBlock?): Boolean {
45+
if (lastGroup is SqlCreateTableColumnDefinitionRawGroupBlock) {
46+
return false
47+
}
48+
return super.isSaveSpace(lastGroup)
49+
}
50+
}

‎src/main/kotlin/org/domaframework/doma/intellij/formatter/builder/SqlBlockRelationBuilder.kt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlKeywordG
2828
import org.domaframework.doma.intellij.formatter.block.group.keyword.create.SqlCreateViewGroupBlock
2929
import org.domaframework.doma.intellij.formatter.block.group.keyword.inline.SqlInlineGroupBlock
3030
import org.domaframework.doma.intellij.formatter.block.group.keyword.inline.SqlInlineSecondGroupBlock
31+
import org.domaframework.doma.intellij.formatter.block.group.keyword.option.SqlExistsGroupBlock
3132
import org.domaframework.doma.intellij.formatter.block.group.keyword.option.SqlInGroupBlock
3233
import org.domaframework.doma.intellij.formatter.block.group.keyword.option.SqlLateralGroupBlock
3334
import org.domaframework.doma.intellij.formatter.block.group.keyword.second.SqlReturningGroupBlock
@@ -64,6 +65,7 @@ class SqlBlockRelationBuilder(
6465
SqlColumnDefinitionRawGroupBlock::class,
6566
SqlLateralGroupBlock::class,
6667
SqlInGroupBlock::class,
68+
SqlExistsGroupBlock::class,
6769
)
6870

6971
private val TOP_LEVEL_EXPECTED_TYPES =

‎src/main/kotlin/org/domaframework/doma/intellij/formatter/handler/JoinClauseHandler.kt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ object JoinClauseHandler {
3030
child: ASTNode,
3131
sqlBlockFormattingCtx: SqlBlockFormattingContext,
3232
): SqlBlock =
33-
if (SqlKeywordUtil.Companion.isJoinKeyword(keywordText)) {
33+
if (SqlKeywordUtil.isJoinKeyword(keywordText)) {
3434
SqlJoinGroupBlock(
3535
child,
3636
sqlBlockFormattingCtx,

‎src/main/kotlin/org/domaframework/doma/intellij/formatter/util/SqlKeywordUtil.kt‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,15 @@ class SqlKeywordUtil {
292292

293293
fun isWithOptionKeyword(keyword: String): Boolean = WITH_OPTION_KEYWORDS.contains(keyword.lowercase())
294294

295+
private val EXISTS_KEYWORDS =
296+
setOf(
297+
"if",
298+
"exists",
299+
"not",
300+
)
301+
302+
fun isExistsKeyword(keyword: String): Boolean = EXISTS_KEYWORDS.contains(keyword.lowercase())
303+
295304
private val SET_LINE_KEYWORDS =
296305
mapOf(
297306
"into" to setOf("insert"),
@@ -307,6 +316,7 @@ class SqlKeywordUtil {
307316
"by" to setOf("group", "order", "first"),
308317
"and" to setOf("between"),
309318
"if" to setOf("table", "create"),
319+
"exists" to setOf("if", "where"),
310320
"conflict" to setOf("on"),
311321
"nothing" to setOf("do"),
312322
"constraint" to setOf("on"),

‎src/test/kotlin/org/domaframework/doma/intellij/formatter/SqlFormatterTest.kt‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,6 +250,10 @@ class SqlFormatterTest : BasePlatformTestCase() {
250250
formatSqlFile("ConditionalSubquery.sql", "ConditionalSubquery$formatDataPrefix.sql")
251251
}
252252

253+
fun testConditionalExistsFormatter() {
254+
formatSqlFile("ConditionalExists.sql", "ConditionalExists$formatDataPrefix.sql")
255+
}
256+
253257
private fun formatSqlFile(
254258
beforeFile: String,
255259
afterFile: String,

‎src/test/testData/sql/formatter/ConditionalExists.sql‎

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,46 @@
1+
SELECT e.id
2+
, e.name
3+
, d.department_name
4+
FROM employees e
5+
WHERE
6+
/*%if isNot*/
7+
NOT
8+
/*%end*/
9+
/*%if filterByHighPerformers */
10+
EXISTS ( SELECT 1
11+
FROM performance_reviews pr
12+
WHERE pr.employee_id = e.id
13+
/*%if reviewPeriod != null */
14+
AND pr.review_date BETWEEN /* reviewPeriod.startDate */'2023-01-01' AND /* reviewPeriod.endDate */'2023-12-31'
15+
/*%end*/
16+
/*%if minScore != null */
17+
AND pr.performance_score >= /* minScore */4.0
18+
/*%end*/ )
19+
/*%elseif filterByLowPerformers */
20+
/*%if isNot*/
21+
NOT
22+
/*%end*/
23+
EXISTS ( SELECT 1
24+
FROM performance_reviews pr
25+
WHERE pr.employee_id = e.id
26+
/*%if reviewPeriod != null */
27+
AND pr.review_date BETWEEN /* reviewPeriod.startDate */'2023-01-01' AND /* reviewPeriod.endDate */'2023-12-31'
28+
/*%end*/
29+
/*%if minScore != null */
30+
AND pr.performance_score >= /* minScore */4.0
31+
/*%end*/ )
32+
/*%elseif filterByMediumPerformers */
33+
NOT
34+
EXISTS
35+
( SELECT 1
36+
FROM performance_reviews pr
37+
WHERE pr.employee_id = e.id
38+
/*%if reviewPeriod != null */
39+
AND pr.review_date BETWEEN /* reviewPeriod.startDate */'2023-01-01' AND /* reviewPeriod.endDate */'2023-12-31'
40+
/*%end*/
41+
/*%if minScore != null */
42+
AND pr.performance_score >= /* minScore */4.0
43+
/*%end*/ )
44+
/*%else*/
45+
e.is_active = true
46+
/*%end*/

‎src/test/testData/sql/formatter/ConditionalExists_format.sql‎

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
SELECT e.id
2+
, e.name
3+
, d.department_name
4+
FROM employees e
5+
WHERE
6+
/*%if isNot*/
7+
NOT
8+
/*%end*/
9+
/*%if filterByHighPerformers */
10+
EXISTS ( SELECT 1
11+
FROM performance_reviews pr
12+
WHERE pr.employee_id = e.id
13+
/*%if reviewPeriod != null */
14+
AND pr.review_date BETWEEN /* reviewPeriod.startDate */'2023-01-01' AND /* reviewPeriod.endDate */'2023-12-31'
15+
/*%end*/
16+
/*%if minScore != null */
17+
AND pr.performance_score >= /* minScore */4.0
18+
/*%end*/ )
19+
/*%elseif filterByLowPerformers */
20+
/*%if isNot*/
21+
NOT
22+
/*%end*/
23+
EXISTS ( SELECT 1
24+
FROM performance_reviews pr
25+
WHERE pr.employee_id = e.id
26+
/*%if reviewPeriod != null */
27+
AND pr.review_date BETWEEN /* reviewPeriod.startDate */'2023-01-01' AND /* reviewPeriod.endDate */'2023-12-31'
28+
/*%end*/
29+
/*%if minScore != null */
30+
AND pr.performance_score >= /* minScore */4.0
31+
/*%end*/ )
32+
/*%elseif filterByMediumPerformers */
33+
NOT EXISTS ( SELECT 1
34+
FROM performance_reviews pr
35+
WHERE pr.employee_id = e.id
36+
/*%if reviewPeriod != null */
37+
AND pr.review_date BETWEEN /* reviewPeriod.startDate */'2023-01-01' AND /* reviewPeriod.endDate */'2023-12-31'
38+
/*%end*/
39+
/*%if minScore != null */
40+
AND pr.performance_score >= /* minScore */4.0
41+
/*%end*/ )
42+
/*%else*/
43+
e.is_active = true
44+
/*%end*/

0 commit comments

Comments
 (0)