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

Add support for ASTs without parents to suppress cache

Browse Source
This commit is contained in:
Ilya Chernikov
2021-10-15 12:40:25 +02:00
parent 7ef806b2ed
commit bfe31c97ac
3 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/frontend.common/src/org/jetbrains/kotlin/diagnostics/AbstractKotlinSuppressCache.kt
+10 -7
View File
@@ -12,18 +12,18 @@ abstract class AbstractKotlinSuppressCache<Element> {
// The cache is weak: we're OK with losing it
protected val suppressors = ContainerUtil.createConcurrentWeakValueMap<Element, Suppressor<Element>>()
fun isSuppressed(element: Element, suppressionKey: String, severity: Severity) =
isSuppressed(StringSuppressRequest(element, severity, suppressionKey.lowercase()))
fun isSuppressed(element: Element, rootElement: Element, suppressionKey: String, severity: Severity) =
isSuppressed(StringSuppressRequest(element, rootElement, severity, suppressionKey.lowercase()))
protected open fun isSuppressed(request: SuppressRequest<Element>): Boolean {
val annotated = getParentAnnotatedElement(request.element, false) ?: return false
val annotated = getClosestAnnotatedAncestorElement(request.element, request.rootElement, false) ?: return false
return isSuppressedByAnnotated(request.suppressKey, request.severity, annotated, 0)
return isSuppressedByAnnotated(request.suppressKey, request.severity, annotated, request.rootElement, 0)
}
protected abstract fun getParentAnnotatedElement(element: Element, strict: Boolean): Element?
protected abstract fun getClosestAnnotatedAncestorElement(element: Element, rootElement: Element, excludeSelf: Boolean): Element?
/*
The cache is optimized for the case where no warnings are suppressed (most frequent one)
@@ -58,14 +58,15 @@ abstract class AbstractKotlinSuppressCache<Element> {
suppressionKey: String,
severity: Severity,
annotated: Element,
rootElement: Element,
debugDepth: Int
): Boolean {
val suppressor = getOrCreateSuppressor(annotated)
if (suppressor.isSuppressed(suppressionKey, severity)) return true
val annotatedAbove = getParentAnnotatedElement(suppressor.annotatedElement, true) ?: return false
val annotatedAbove = getClosestAnnotatedAncestorElement(suppressor.annotatedElement, rootElement, true) ?: return false
val suppressed = isSuppressedByAnnotated(suppressionKey, severity, annotatedAbove, debugDepth + 1)
val suppressed = isSuppressedByAnnotated(suppressionKey, severity, annotatedAbove, rootElement, debugDepth + 1)
val suppressorAbove = suppressors[annotatedAbove]
if (suppressorAbove != null && suppressorAbove.dominates(suppressor)) {
suppressors[annotated] = suppressorAbove
@@ -126,12 +127,14 @@ abstract class AbstractKotlinSuppressCache<Element> {
protected interface SuppressRequest<Element> {
val element: Element
val rootElement: Element
val severity: Severity
val suppressKey: String
}
private class StringSuppressRequest<Element>(
override val element: Element,
override val rootElement: Element,
override val severity: Severity,
override val suppressKey: String
) : SuppressRequest<Element>
compiler/frontend.common/src/org/jetbrains/kotlin/diagnostics/DiagnosticReporter.kt
+1 -1
View File
@@ -35,7 +35,7 @@ open class KtDiagnosticReporterWithContext(
) : DiagnosticReporter() {
override fun report(diagnostic: KtDiagnostic?, context: DiagnosticContext) = diagnosticReporter.report(diagnostic, context)
fun at(sourceElement: AbstractKtSourceElement?, containingFilePath: String): DiagnosticContextImpl =
open fun at(sourceElement: AbstractKtSourceElement?, containingFilePath: String): DiagnosticContextImpl =
DiagnosticContextImpl(sourceElement, containingFilePath)
open inner class DiagnosticContextImpl(
compiler/frontend/src/org/jetbrains/kotlin/resolve/diagnostics/KotlinSuppressCache.kt
+3 -2
View File
@@ -99,11 +99,12 @@ abstract class KotlinSuppressCache : AbstractKotlinSuppressCache<PsiElement>() {
return super.isSuppressed(request)
}
override fun getParentAnnotatedElement(element: PsiElement, strict: Boolean) =
KtStubbedPsiUtil.getPsiOrStubParent(element, KtAnnotated::class.java, strict)
override fun getClosestAnnotatedAncestorElement(element: PsiElement, rootElement: PsiElement, excludeSelf: Boolean): PsiElement? =
KtStubbedPsiUtil.getPsiOrStubParent(element, KtAnnotated::class.java, excludeSelf)
protected class DiagnosticSuppressRequest(val diagnostic: Diagnostic) : SuppressRequest<PsiElement> {
override val element: PsiElement get() = diagnostic.psiElement
override val rootElement: PsiElement get() = element.containingFile
override val severity: Severity get() = diagnostic.severity
override val suppressKey: String get() = getDiagnosticSuppressKey(diagnostic)
}