MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

extract generalized StubForwardIndexExternalizer

Browse Source 
GitOrigin-RevId: ff7ad119c9c571119d6b58cbdb8bd66439425c32
This commit is contained in:
Dmitry Batkovich
2019-06-12 14:15:06 +03:00
committed by intellij-monorepo-bot
parent af2ad448b7
commit 2ee67f4761
2 changed files with 90 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files
platform/lang-impl/src/com/intellij/psi/stubs/StubForwardIndexExternalizer.java
+75
View File
@@ -0,0 +1,75 @@
// Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.psi.stubs;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.util.indexing.ID;
import com.intellij.util.io.DataExternalizer;
import com.intellij.util.io.DataInputOutputUtil;
import gnu.trove.THashMap;
import org.jetbrains.annotations.NotNull;
import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Collections;
import java.util.Map;
import java.util.Set;
abstract class StubForwardIndexExternalizer<StubKeySerializationState> implements DataExternalizer<IndexedStubs> {
private volatile boolean myEnsuredStubElementTypesLoaded;
protected abstract StubKeySerializationState createStubIndexKeySerializationState(@NotNull DataOutput out, @NotNull Set<StubIndexKey> set) throws IOException;
protected abstract void writeStubIndexKey(@NotNull DataOutput out, @NotNull StubIndexKey key, StubKeySerializationState state) throws IOException;
protected abstract StubKeySerializationState createStubIndexKeySerializationState(@NotNull DataInput input, int stubIndexKeyCount) throws IOException;
protected abstract ID<?, ?> readStubIndexKey(@NotNull DataInput input, StubKeySerializationState stubKeySerializationState) throws IOException;
@Override
public void save(@NotNull DataOutput out, IndexedStubs indexedStubs) throws IOException {
byte[] hash = indexedStubs.getStubTreeHash();
DataInputOutputUtil.writeINT(out, hash.length);
out.write(hash);
Map<StubIndexKey, Map<Object, StubIdList>> stubIndicesValueMap = indexedStubs.getStubIndicesValueMap();
DataInputOutputUtil.writeINT(out, stubIndicesValueMap.size());
if (!stubIndicesValueMap.isEmpty()) {
StubKeySerializationState stubKeySerializationState = createStubIndexKeySerializationState(out, stubIndicesValueMap.keySet());
StubIndexImpl stubIndex = (StubIndexImpl)StubIndex.getInstance();
for (StubIndexKey stubIndexKey : stubIndicesValueMap.keySet()) {
writeStubIndexKey(out, stubIndexKey, stubKeySerializationState);
Map<Object, StubIdList> map = stubIndicesValueMap.get(stubIndexKey);
stubIndex.serializeIndexValue(out, stubIndexKey, map);
}
}
}
@Override
public IndexedStubs read(@NotNull DataInput in) throws IOException {
byte[] hash = new byte[DataInputOutputUtil.readINT(in)];
in.readFully(hash);
if (!myEnsuredStubElementTypesLoaded) {
ProgressManager.getInstance().executeNonCancelableSection(() -> {
SerializationManager.getInstance().initSerializers();
StubIndexImpl.initExtensions();
});
myEnsuredStubElementTypesLoaded = true;
}
int stubIndicesValueMapSize = DataInputOutputUtil.readINT(in);
if (stubIndicesValueMapSize > 0) {
THashMap<StubIndexKey, Map<Object, StubIdList>> stubIndicesValueMap = new THashMap<>(stubIndicesValueMapSize);
StubIndexImpl stubIndex = (StubIndexImpl)StubIndex.getInstance();
StubKeySerializationState stubKeySerializationState = createStubIndexKeySerializationState(in, stubIndicesValueMapSize);
for (int i = 0; i < stubIndicesValueMapSize; ++i) {
ID<Object, ?> indexKey = (ID<Object, ?>)readStubIndexKey(in, stubKeySerializationState);
if (indexKey instanceof StubIndexKey) { // indexKey can be ID in case of removed index
StubIndexKey<Object, ?> stubIndexKey = (StubIndexKey<Object, ?>)indexKey;
stubIndicesValueMap.put(stubIndexKey, stubIndex.deserializeIndexValue(in, stubIndexKey));
}
}
return new IndexedStubs(hash, stubIndicesValueMap);
}
return new IndexedStubs(hash, Collections.emptyMap());
}
}
platform/lang-impl/src/com/intellij/psi/stubs/StubUpdatingForwardIndexAccessor.java
+15 -47
View File
@@ -1,73 +1,41 @@
// Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.psi.stubs;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.indexing.ID;
import com.intellij.util.indexing.impl.InputData;
import com.intellij.util.indexing.impl.InputDataDiffBuilder;
import com.intellij.util.indexing.impl.forward.AbstractForwardIndexAccessor;
import com.intellij.util.io.DataExternalizer;
import com.intellij.util.io.DataInputOutputUtil;
import gnu.trove.THashMap;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Collections;
import java.util.Map;
import java.util.Set;
class StubUpdatingForwardIndexAccessor extends AbstractForwardIndexAccessor<Integer, SerializedStubTree, IndexedStubs> {
StubUpdatingForwardIndexAccessor() {super(new DataExternalizer<IndexedStubs>() {
private volatile boolean myEnsuredStubElementTypesLoaded;
StubUpdatingForwardIndexAccessor() {super(new StubForwardIndexExternalizer<Void>() {
@Override
public void save(@NotNull DataOutput out, IndexedStubs indexedStubs) throws IOException {
byte[] hash = indexedStubs.getStubTreeHash();
DataInputOutputUtil.writeINT(out, hash.length);
out.write(hash);
Map<StubIndexKey, Map<Object, StubIdList>> stubIndicesValueMap = indexedStubs.getStubIndicesValueMap();
DataInputOutputUtil.writeINT(out, stubIndicesValueMap.size());
if (!stubIndicesValueMap.isEmpty()) {
StubIndexImpl stubIndex = (StubIndexImpl)StubIndex.getInstance();
for (StubIndexKey stubIndexKey : stubIndicesValueMap.keySet()) {
DataInputOutputUtil.writeINT(out, stubIndexKey.getUniqueId());
Map<Object, StubIdList> map = stubIndicesValueMap.get(stubIndexKey);
stubIndex.serializeIndexValue(out, stubIndexKey, map);
}
}
protected void writeStubIndexKey(@NotNull DataOutput out, @NotNull StubIndexKey key, Void aVoid) throws IOException {
DataInputOutputUtil.writeINT(out, key.getUniqueId());
}
@Override
public IndexedStubs read(@NotNull DataInput in) throws IOException {
byte[] hash = new byte[DataInputOutputUtil.readINT(in)];
in.readFully(hash);
if (!myEnsuredStubElementTypesLoaded) {
ProgressManager.getInstance().executeNonCancelableSection(() -> {
SerializationManager.getInstance().initSerializers();
StubIndexImpl.initExtensions();
});
myEnsuredStubElementTypesLoaded = true;
}
int stubIndicesValueMapSize = DataInputOutputUtil.readINT(in);
if (stubIndicesValueMapSize > 0) {
THashMap<StubIndexKey, Map<Object, StubIdList>> stubIndicesValueMap = new THashMap<>(stubIndicesValueMapSize);
StubIndexImpl stubIndex = (StubIndexImpl)StubIndex.getInstance();
protected Void createStubIndexKeySerializationState(@NotNull DataOutput out, @NotNull Set<StubIndexKey> set) {
return null;
}
for (int i = 0; i < stubIndicesValueMapSize; ++i) {
int stubIndexId = DataInputOutputUtil.readINT(in);
ID<Object, ?> indexKey = (ID<Object, ?>)ID.findById(stubIndexId);
if (indexKey instanceof StubIndexKey) { // indexKey can be ID in case of removed index
StubIndexKey<Object, ?> stubIndexKey = (StubIndexKey<Object, ?>)indexKey;
stubIndicesValueMap.put(stubIndexKey, stubIndex.deserializeIndexValue(in, stubIndexKey));
}
}
return new IndexedStubs(hash, stubIndicesValueMap);
}
return new IndexedStubs(hash, Collections.emptyMap());
@Override
protected ID<?, ?> readStubIndexKey(@NotNull DataInput input, Void aVoid) throws IOException {
return ID.findById(DataInputOutputUtil.readINT(input));
}
@Override
protected Void createStubIndexKeySerializationState(@NotNull DataInput input, int stubIndexKeyCount) {
return null;
}
});}