override fun applyFix()

in src/main/kotlin/com/netflix/dgs/plugin/hints/DgsFieldSimplifyingInspector.kt [78:93]

15 lines of code
1 McCabe index (conditional complexity)


        override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
            val newAnnotation : PsiElement = when (descriptor.psiElement) {
                is PsiAnnotation -> {
                    val newAnnotationText = replaceField(descriptor.psiElement.text)
                    PsiElementFactory.getInstance(project).createAnnotationFromText(newAnnotationText, null)
                }
                is KtAnnotationEntry -> {
                    val newAnnotationText = replaceField(descriptor.psiElement.text)
                    KtPsiFactory(project).createAnnotationEntry(newAnnotationText)
                }
                else -> return
            }

            descriptor.psiElement.replace(newAnnotation)
            project.getService(DgsService::class.java).clearCache()
        }