private void repopulateAllIndexes() throws IOException {
final IndexRule[] rules = getIndexesNeedingPopulation();
final IndexPopulator[] populators = new IndexPopulator[rules.length];
// the store is uncontended at this point, so creating a local LockService is safe.
LockService locks = new ReentrantLockService();
IndexStoreView storeView = new NeoStoreIndexStoreView(locks, neoStore);
final int[] labelIds = new int[rules.length];
final int[] propertyKeyIds = new int[rules.length];
for (int i = 0; i < labelIds.length; i++) {
IndexRule rule = rules[i];
labelIds[i] = rule.getLabel();
propertyKeyIds[i] = rule.getPropertyKey();
populators[i] =
schemaIndexProviders
.apply(rule.getProviderDescriptor())
.getPopulator(rule.getId(), new IndexConfiguration(rule.isConstraintIndex()));
populators[i].create();
}
Visitor<NodePropertyUpdate, IOException> propertyUpdateVisitor =
new Visitor<NodePropertyUpdate, IOException>() {
@Override
public boolean visit(NodePropertyUpdate update) throws IOException {
// Do a lookup from which property has changed to a list of indexes worried about that
// property.
int propertyKeyInQuestion = update.getPropertyKeyId();
for (int i = 0; i < propertyKeyIds.length; i++) {
if (propertyKeyIds[i] == propertyKeyInQuestion) {
if (update.forLabel(labelIds[i])) {
try {
populators[i].add(update.getNodeId(), update.getValueAfter());
} catch (IndexEntryConflictException conflict) {
throw conflict.notAllowed(rules[i].getLabel(), rules[i].getPropertyKey());
}
}
}
}
return true;
}
};
NodeLabelUpdateVisitor labelUpdateVisitor = new NodeLabelUpdateVisitor();
StoreScan<IOException> storeScan =
storeView.visitNodes(labelIds, propertyKeyIds, propertyUpdateVisitor, labelUpdateVisitor);
storeScan.run();
for (IndexPopulator populator : populators) {
populator.close(true);
}
labelUpdateVisitor.close();
}
private IndexRule[] getIndexesNeedingPopulation() {
List<IndexRule> indexesNeedingPopulation = new ArrayList<>();
for (SchemaRule rule : schemaCache.schemaRules()) {
if (rule.getKind().isIndex()) {
IndexRule indexRule = (IndexRule) rule;
SchemaIndexProvider provider =
schemaIndexProviders.apply(indexRule.getProviderDescriptor());
if (provider.getInitialState(indexRule.getId()) != InternalIndexState.FAILED) {
indexesNeedingPopulation.add(indexRule);
}
}
}
return indexesNeedingPopulation.toArray(new IndexRule[indexesNeedingPopulation.size()]);
}
