JetClass#getBody() by stub

2014-03-28 21:34:01 +04:00
parent 40369f1e3b
commit 19cf463831
1 changed files with 2 additions and 2 deletions
@@ -49,7 +49,7 @@ public class JetClass extends JetTypeParameterListOwnerStub<PsiJetClassStub> imp
    @NotNull
    @Override
    public List<JetDeclaration> getDeclarations() {
-        JetClassBody body = getStubOrPsiChild(JetStubElementTypes.CLASS_BODY);
+        JetClassBody body = getBody();
        if (body == null) return Collections.emptyList();

        return body.getDeclarations();
@@ -111,7 +111,7 @@ public class JetClass extends JetTypeParameterListOwnerStub<PsiJetClassStub> imp

    @Override
    public JetClassBody getBody() {
-        return (JetClassBody) findChildByType(JetNodeTypes.CLASS_BODY);
+        return getStubOrPsiChild(JetStubElementTypes.CLASS_BODY);
    }

    @Nullable