Add missing FileModificationService call (EA-107395)

This commit is contained in:
Dmitry Jemerov
2017-10-06 19:21:01 +02:00
parent c2efdfe243
commit 6c7f558d36
@@ -16,6 +16,7 @@
package org.jetbrains.kotlin.idea.inspections package org.jetbrains.kotlin.idea.inspections
import com.intellij.codeInsight.FileModificationService
import com.intellij.codeInspection.* import com.intellij.codeInspection.*
import com.intellij.openapi.project.Project import com.intellij.openapi.project.Project
import com.intellij.psi.PsiElementVisitor import com.intellij.psi.PsiElementVisitor
@@ -58,6 +59,7 @@ class RedundantSamConstructorInspection : AbstractKotlinInspection() {
override fun getName() = "Remove redundant SAM-constructor" override fun getName() = "Remove redundant SAM-constructor"
override fun getFamilyName() = name override fun getFamilyName() = name
override fun applyFix(project: Project, descriptor: ProblemDescriptor) { override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
if (!FileModificationService.getInstance().preparePsiElementForWrite(expression)) return
replaceSamConstructorCall(expression) replaceSamConstructorCall(expression)
} }
} }
@@ -69,6 +71,7 @@ class RedundantSamConstructorInspection : AbstractKotlinInspection() {
override fun getFamilyName() = name override fun getFamilyName() = name
override fun applyFix(project: Project, descriptor: ProblemDescriptor) { override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
for (callExpression in expressions) { for (callExpression in expressions) {
if (!FileModificationService.getInstance().preparePsiElementForWrite(callExpression)) return
replaceSamConstructorCall(callExpression) replaceSamConstructorCall(callExpression)
} }
} }