From 6dcc3374feb84adcebf20996f8f3487ae106a0ad Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Thu, 25 Oct 2018 17:04:07 +0300
Subject: [PATCH] Get string entries from stubs (KT-23738)

---
 .../org/jetbrains/kotlin/psi/KtStringTemplateEntry.java  | 7 ++++++-
 .../jetbrains/kotlin/psi/KtStringTemplateExpression.java | 9 ++++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateEntry.java b/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateEntry.java
index 9a0ee00e659..42d6c74cef9 100644
--- a/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateEntry.java
+++ b/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateEntry.java
@@ -23,11 +23,16 @@ import org.jetbrains.annotations.Nullable;
 import org.jetbrains.kotlin.psi.stubs.KotlinPlaceHolderWithTextStub;
 
 public abstract class KtStringTemplateEntry extends KtElementImplStub<KotlinPlaceHolderWithTextStub<? extends KtStringTemplateEntry>> {
+    public static final KtStringTemplateEntry[] EMPTY_ARRAY = new KtStringTemplateEntry[0];
+
     public KtStringTemplateEntry(@NotNull ASTNode node) {
         super(node);
     }
 
-    public KtStringTemplateEntry(@NotNull KotlinPlaceHolderWithTextStub<? extends KtStringTemplateEntry> stub, @NotNull IStubElementType elementType) {
+    public KtStringTemplateEntry(
+            @NotNull KotlinPlaceHolderWithTextStub<? extends KtStringTemplateEntry> stub,
+            @NotNull IStubElementType elementType
+    ) {
         super(stub, elementType);
     }
 
diff --git a/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateExpression.java b/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateExpression.java
index d49a8552a08..b9c64061aa2 100644
--- a/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateExpression.java
+++ b/compiler/psi/src/org/jetbrains/kotlin/psi/KtStringTemplateExpression.java
@@ -49,9 +49,16 @@ public class KtStringTemplateExpression extends KtElementImplStub<KotlinPlaceHol
         return visitor.visitStringTemplateExpression(this, data);
     }
 
+    private static final TokenSet STRING_ENTRIES_TYPES = TokenSet.create(
+            KtStubElementTypes.LONG_STRING_TEMPLATE_ENTRY,
+            KtStubElementTypes.SHORT_STRING_TEMPLATE_ENTRY,
+            KtStubElementTypes.LITERAL_STRING_TEMPLATE_ENTRY,
+            KtStubElementTypes.ESCAPE_STRING_TEMPLATE_ENTRY
+    );
+
     @NotNull
     public KtStringTemplateEntry[] getEntries() {
-        return findChildrenByClass(KtStringTemplateEntry.class);
+        return getStubOrPsiChildren(STRING_ENTRIES_TYPES, KtStringTemplateEntry.EMPTY_ARRAY);
     }
 
     @Override