Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix retry logic

This commit is contained in:
fjy 2014-09-12 15:47:41 -07:00
parent 469ccbbe5e
commit 257fb3f18e
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
processing/src/main/java/io/druid/query/RetryQueryRunner.java
View File

@ -27,6 +27,7 @@ import com.metamx.common.guava.Sequence;
import com.metamx.common.guava.Yielder; import com.metamx.common.guava.Yielder;
import com.metamx.common.guava.YieldingAccumulator; import com.metamx.common.guava.YieldingAccumulator;
import com.metamx.common.guava.YieldingSequenceBase; import com.metamx.common.guava.YieldingSequenceBase;
import io.druid.query.spec.MultipleSpecificSegmentSpec;
import io.druid.segment.SegmentMissingException; import io.druid.segment.SegmentMissingException;
import java.util.List; import java.util.List;
@ -73,13 +74,18 @@ public class RetryQueryRunner<T> implements QueryRunner<T>
for (int i = 0; i < config.numTries(); i++) { for (int i = 0; i < config.numTries(); i++) {
context.put(MISSING_SEGMENTS_KEY, Lists.newArrayList()); context.put(MISSING_SEGMENTS_KEY, Lists.newArrayList());
yielder = baseRunner.run(query, context).toYielder(initValue, accumulator); final Query<T> retryQuery = query.withQuerySegmentSpec(
new MultipleSpecificSegmentSpec(
missingSegments
)
);
yielder = baseRunner.run(retryQuery, context).toYielder(initValue, accumulator);
if (getMissingSegments(context).isEmpty()) { if (getMissingSegments(context).isEmpty()) {
break; break;
} }
} }
final List<SegmentDescriptor> finalMissingSegs= getMissingSegments(context); final List<SegmentDescriptor> finalMissingSegs = getMissingSegments(context);
if (!config.returnPartialResults() && !finalMissingSegs.isEmpty()) { if (!config.returnPartialResults() && !finalMissingSegs.isEmpty()) {
throw new SegmentMissingException("No results found for segments[%s]", finalMissingSegs); throw new SegmentMissingException("No results found for segments[%s]", finalMissingSegs);
} }