Skip to content

Commit 48ab871

Browse files
committed
Refactor for directive suggestions into a separate method
1 parent f15ccee commit 48ab871

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

src/main/kotlin/org/domaframework/doma/intellij/contributor/sql/provider/SqlParameterCompletionProvider.kt

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -366,11 +366,7 @@ class SqlParameterCompletionProvider : CompletionProvider<CompletionParameters>(
366366
}
367367
// Add ForDirective Items
368368
val forDirectives = ForDirectiveUtil.getForDirectiveBlocks(position)
369-
forDirectives.forEach {
370-
result.addElement(LookupElementBuilder.create(it.item.text))
371-
result.addElement(LookupElementBuilder.create("${it.item.text}_has_next"))
372-
result.addElement(LookupElementBuilder.create("${it.item.text}_index"))
373-
}
369+
addForDirectiveSuggestions(forDirectives, result)
374370
return null
375371
}
376372
if (findParam == null) {
@@ -380,6 +376,17 @@ class SqlParameterCompletionProvider : CompletionProvider<CompletionParameters>(
380376
return PsiClassTypeUtil.convertOptionalType(immediate, originalFile.project)
381377
}
382378

379+
private fun addForDirectiveSuggestions(
380+
forDirectives: List<ForDirectiveUtil.BlockToken>,
381+
result: CompletionResultSet,
382+
) {
383+
forDirectives.forEach {
384+
result.addElement(LookupElementBuilder.create(it.item.text))
385+
result.addElement(LookupElementBuilder.create("${it.item.text}_has_next"))
386+
result.addElement(LookupElementBuilder.create("${it.item.text}_index"))
387+
}
388+
}
389+
383390
private fun getRefClazz(
384391
top: PsiElement,
385392
fqdnGetter: () -> String,

0 commit comments

Comments
 (0)