Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add multi-query support to label step
Fixes #3813 Signed-off-by: Oleksandr Porunov <alexandr.porunov@gmail.com>
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,114 @@ | ||
| // Copyright 2023 JanusGraph Authors | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| package org.janusgraph.graphdb.tinkerpop.optimize.step; | ||
|
|
||
| import org.apache.tinkerpop.gremlin.process.traversal.Traverser; | ||
| import org.apache.tinkerpop.gremlin.process.traversal.step.Profiling; | ||
| import org.apache.tinkerpop.gremlin.process.traversal.step.map.LabelStep; | ||
| import org.apache.tinkerpop.gremlin.process.traversal.util.MutableMetrics; | ||
| import org.apache.tinkerpop.gremlin.structure.Element; | ||
| import org.apache.tinkerpop.gremlin.structure.Vertex; | ||
| import org.janusgraph.core.BaseVertexQuery; | ||
| import org.janusgraph.graphdb.query.profile.QueryProfiler; | ||
| import org.janusgraph.graphdb.query.vertex.BasicVertexCentricQueryBuilder; | ||
| import org.janusgraph.graphdb.query.vertex.BasicVertexCentricQueryUtil; | ||
| import org.janusgraph.graphdb.tinkerpop.optimize.step.fetcher.LabelStepBatchFetcher; | ||
| import org.janusgraph.graphdb.tinkerpop.profile.TP3ProfileWrapper; | ||
| import org.janusgraph.graphdb.util.CopyStepUtil; | ||
| import org.janusgraph.graphdb.util.JanusGraphTraverserUtil; | ||
|
|
||
| /** | ||
| * This class extends the default TinkerPop's {@link LabelStep} and adds vertices multi-query optimization to this step. | ||
| * <p> | ||
| * Before this step is evaluated it usually receives multiple future vertices which might be processed next with this step. | ||
| * This step stores all these vertices which might be needed later for evaluation and whenever this step receives the | ||
| * vertex for evaluation which wasn't preFetched previously it sends multi-query for a batch of vertices to fetch their | ||
| * labels. | ||
| * <p> | ||
| * This step optimizes only access to Vertex properties and skips optimization for any other Element. | ||
| */ | ||
| public class JanusGraphLabelStep<S extends Element> extends LabelStep<S> implements Profiling, MultiQueriable<S,String> { | ||
|
|
||
| private boolean useMultiQuery = false; | ||
| private QueryProfiler queryProfiler = QueryProfiler.NO_OP; | ||
| private int batchSize = Integer.MAX_VALUE; | ||
| private LabelStepBatchFetcher labelStepBatchFetcher; | ||
|
|
||
| public JanusGraphLabelStep(LabelStep<S> originalStep){ | ||
| super(originalStep.getTraversal()); | ||
| CopyStepUtil.copyAbstractStepModifiableFields(originalStep, this); | ||
|
|
||
| if (originalStep instanceof JanusGraphLabelStep) { | ||
| JanusGraphLabelStep originalJanusGraphLabelStep = (JanusGraphLabelStep) originalStep; | ||
| setBatchSize(originalJanusGraphLabelStep.batchSize); | ||
| setUseMultiQuery(originalJanusGraphLabelStep.useMultiQuery); | ||
|
Check warning on line 56 in janusgraph-core/src/main/java/org/janusgraph/graphdb/tinkerpop/optimize/step/JanusGraphLabelStep.java
|
||
| } | ||
| } | ||
|
|
||
| @Override | ||
| protected String map(final Traverser.Admin<S> traverser) { | ||
| if (useMultiQuery && traverser.get() instanceof Vertex) { | ||
| return labelStepBatchFetcher.fetchData(getTraversal(), (Vertex) traverser.get(), JanusGraphTraverserUtil.getLoops(traverser)); | ||
| } | ||
| return super.map(traverser); | ||
| } | ||
|
|
||
| @Override | ||
| public void setUseMultiQuery(boolean useMultiQuery) { | ||
| this.useMultiQuery = useMultiQuery; | ||
| if(this.useMultiQuery && labelStepBatchFetcher == null){ | ||
| labelStepBatchFetcher = new LabelStepBatchFetcher(this::makeLabelsQuery, batchSize); | ||
| } | ||
| } | ||
|
|
||
| private <Q extends BaseVertexQuery> Q makeLabelsQuery(Q query) { | ||
| return (Q) BasicVertexCentricQueryUtil.withLabelVertices((BasicVertexCentricQueryBuilder) query) | ||
| .profiler(queryProfiler); | ||
| } | ||
|
|
||
| @Override | ||
| public void setBatchSize(int batchSize) { | ||
| this.batchSize = batchSize; | ||
| if(labelStepBatchFetcher != null){ | ||
| labelStepBatchFetcher.setBatchSize(batchSize); | ||
|
Check warning on line 85 in janusgraph-core/src/main/java/org/janusgraph/graphdb/tinkerpop/optimize/step/JanusGraphLabelStep.java
|
||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void registerFirstNewLoopFutureVertexForPrefetching(Vertex futureVertex, int futureVertexTraverserLoop) { | ||
| if(useMultiQuery){ | ||
| labelStepBatchFetcher.registerFirstNewLoopFutureVertexForPrefetching(futureVertex); | ||
|
Check warning on line 92 in janusgraph-core/src/main/java/org/janusgraph/graphdb/tinkerpop/optimize/step/JanusGraphLabelStep.java
|
||
| } | ||
| } | ||
|
Check warning on line 94 in janusgraph-core/src/main/java/org/janusgraph/graphdb/tinkerpop/optimize/step/JanusGraphLabelStep.java
|
||
|
|
||
| @Override | ||
| public void registerSameLoopFutureVertexForPrefetching(Vertex futureVertex, int futureVertexTraverserLoop) { | ||
| if(useMultiQuery){ | ||
| labelStepBatchFetcher.registerCurrentLoopFutureVertexForPrefetching(futureVertex, futureVertexTraverserLoop); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void registerNextLoopFutureVertexForPrefetching(Vertex futureVertex, int futureVertexTraverserLoop) { | ||
| if(useMultiQuery){ | ||
| labelStepBatchFetcher.registerNextLoopFutureVertexForPrefetching(futureVertex, futureVertexTraverserLoop); | ||
|
Check warning on line 106 in janusgraph-core/src/main/java/org/janusgraph/graphdb/tinkerpop/optimize/step/JanusGraphLabelStep.java
|
||
| } | ||
| } | ||
|
Check warning on line 108 in janusgraph-core/src/main/java/org/janusgraph/graphdb/tinkerpop/optimize/step/JanusGraphLabelStep.java
|
||
|
|
||
| @Override | ||
| public void setMetrics(MutableMetrics metrics) { | ||
| queryProfiler = new TP3ProfileWrapper(metrics); | ||
| } | ||
| } | ||
Oops, something went wrong.