Allow store nullable qualified names in object stubs

This commit is contained in:
Nikolay Krasko
2012-07-02 17:16:09 +04:00
parent de4e302578
commit 72271cbdd4
4 changed files with 16 additions and 11 deletions
@@ -17,7 +17,7 @@
package org.jetbrains.jet.lang.psi.stubs; package org.jetbrains.jet.lang.psi.stubs;
import com.intellij.psi.stubs.NamedStub; import com.intellij.psi.stubs.NamedStub;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.lang.psi.JetObjectDeclaration; import org.jetbrains.jet.lang.psi.JetObjectDeclaration;
import org.jetbrains.jet.lang.resolve.name.FqName; import org.jetbrains.jet.lang.resolve.name.FqName;
@@ -25,6 +25,6 @@ import org.jetbrains.jet.lang.resolve.name.FqName;
* @author Nikolay Krasko * @author Nikolay Krasko
*/ */
public interface PsiJetObjectStub extends NamedStub<JetObjectDeclaration> { public interface PsiJetObjectStub extends NamedStub<JetObjectDeclaration> {
@NotNull @Nullable
FqName getFQName(); FqName getFQName();
} }
@@ -69,22 +69,21 @@ public class JetObjectElementType extends JetStubElementType<PsiJetObjectStub, J
assert name != null; assert name != null;
FqName fqName = psi.getFqName(); FqName fqName = psi.getFqName();
assert fqName != null;
return new PsiJetObjectStubImpl(JetStubElementTypes.OBJECT_DECLARATION, parentStub, name, fqName); return new PsiJetObjectStubImpl(JetStubElementTypes.OBJECT_DECLARATION, parentStub, name, fqName);
} }
@Override @Override
public void serialize(PsiJetObjectStub stub, StubOutputStream dataStream) throws IOException { public void serialize(PsiJetObjectStub stub, StubOutputStream dataStream) throws IOException {
dataStream.writeName(stub.getName()); dataStream.writeName(stub.getName());
dataStream.writeName(stub.getFQName().toString()); FqName fqName = stub.getFQName();
dataStream.writeName(fqName != null ? fqName.toString() : null);
} }
@Override @Override
public PsiJetObjectStub deserialize(StubInputStream dataStream, StubElement parentStub) throws IOException { public PsiJetObjectStub deserialize(StubInputStream dataStream, StubElement parentStub) throws IOException {
StringRef name = dataStream.readName(); StringRef name = dataStream.readName();
StringRef fqNameStr = dataStream.readName(); StringRef fqNameStr = dataStream.readName();
FqName fqName = new FqName(fqNameStr.toString()); FqName fqName = fqNameStr != null ? new FqName(fqNameStr.toString()) : null;
return new PsiJetObjectStubImpl(JetStubElementTypes.OBJECT_DECLARATION, parentStub, name, fqName); return new PsiJetObjectStubImpl(JetStubElementTypes.OBJECT_DECLARATION, parentStub, name, fqName);
} }
@@ -21,6 +21,7 @@ import com.intellij.psi.stubs.StubBase;
import com.intellij.psi.stubs.StubElement; import com.intellij.psi.stubs.StubElement;
import com.intellij.util.io.StringRef; import com.intellij.util.io.StringRef;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.lang.psi.JetObjectDeclaration; import org.jetbrains.jet.lang.psi.JetObjectDeclaration;
import org.jetbrains.jet.lang.psi.stubs.PsiJetObjectStub; import org.jetbrains.jet.lang.psi.stubs.PsiJetObjectStub;
import org.jetbrains.jet.lang.resolve.name.FqName; import org.jetbrains.jet.lang.resolve.name.FqName;
@@ -36,7 +37,7 @@ public class PsiJetObjectStubImpl extends StubBase<JetObjectDeclaration> impleme
@NotNull IStubElementType elementType, @NotNull IStubElementType elementType,
@NotNull StubElement parent, @NotNull StubElement parent,
@NotNull String name, @NotNull String name,
@NotNull FqName fqName) { @Nullable FqName fqName) {
this(elementType, parent, StringRef.fromString(name), fqName); this(elementType, parent, StringRef.fromString(name), fqName);
} }
@@ -44,7 +45,7 @@ public class PsiJetObjectStubImpl extends StubBase<JetObjectDeclaration> impleme
@NotNull IStubElementType elementType, @NotNull IStubElementType elementType,
@NotNull StubElement parent, @NotNull StubElement parent,
@NotNull StringRef name, @NotNull StringRef name,
@NotNull FqName fqName) { @Nullable FqName fqName) {
super(parent, elementType); super(parent, elementType);
this.name = name; this.name = name;
@@ -56,7 +57,7 @@ public class PsiJetObjectStubImpl extends StubBase<JetObjectDeclaration> impleme
return StringRef.toString(name); return StringRef.toString(name);
} }
@NotNull @Nullable
@Override @Override
public FqName getFQName() { public FqName getFQName() {
return fqName; return fqName;
@@ -68,7 +69,8 @@ public class PsiJetObjectStubImpl extends StubBase<JetObjectDeclaration> impleme
builder.append("PsiJetObjectStubImpl["); builder.append("PsiJetObjectStubImpl[");
builder.append("name=").append(getName()); builder.append("name=").append(getName());
builder.append(" fqName=").append(getFQName().toString());
builder.append(" fqName=").append(fqName != null ? fqName.toString() : "null");
builder.append("]"); builder.append("]");
return builder.toString(); return builder.toString();
@@ -52,7 +52,11 @@ public class StubIndexServiceImpl implements StubIndexService {
assert name != null; assert name != null;
sink.occurrence(JetIndexKeys.SHORT_NAME_KEY, name); sink.occurrence(JetIndexKeys.SHORT_NAME_KEY, name);
sink.occurrence(JetIndexKeys.FQN_KEY, stub.getFQName().toString());
FqName fqName = stub.getFQName();
if (fqName != null) {
sink.occurrence(JetIndexKeys.FQN_KEY, fqName.toString());
}
} }
@Override @Override