azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

KT-61689 [LL API] Improve readability of withInterceptor function

Browse Source
This commit is contained in:
Roman Golyshev
2023-09-04 16:06:13 +02:00
parent 90ed90c9a7
commit 83049dc25c
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt
+8 -3
View File
@@ -156,7 +156,7 @@ private class ContextCollectorVisitor(
private val holder: SessionHolder, private val holder: SessionHolder,
private val shouldCollectBodyContext: Boolean, private val shouldCollectBodyContext: Boolean,
private val filter: (PsiElement) -> FilterResponse, private val filter: (PsiElement) -> FilterResponse,
private val interceptor: () -> FirElement? private val designationPathInterceptor: () -> FirElement?
) : FirDefaultVisitorVoid() { ) : FirDefaultVisitorVoid() {
private data class ContextKey(val element: PsiElement, val kind: ContextKind) private data class ContextKey(val element: PsiElement, val kind: ContextKind)
@@ -595,8 +595,13 @@ private class ContextCollectorVisitor(
} }
} }
private inline fun withInterceptor(block: () -> Unit) { /**
val target = interceptor() * Ensures that the visitor is going through the path specified by the initial [FirDesignation].
*
* If the designation is over, then allows the [block] code to take control.
*/
private fun withInterceptor(block: () -> Unit) {
val target = designationPathInterceptor()
if (target != null) { if (target != null) {
target.accept(this) target.accept(this)
} else { } else {