Skip to content

Commit 1df51e7

Browse files
committed
[SOLR-17337] Show proper distributed stage id
- Shows a proper distributed stage id instead of an empty string
1 parent 554c0de commit 1df51e7

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

solr/core/src/java/org/apache/solr/handler/component/DebugComponent.java

+13-2
Original file line numberDiff line numberDiff line change
@@ -181,16 +181,27 @@ public void modifyRequest(ResponseBuilder rb, SearchComponent who, ShardRequest
181181
}
182182
}
183183

184+
private String getDistributedStageName(int stage) {
185+
String stageName = stages.get(stage);
186+
187+
if (stageName == null) {
188+
stageName = "STAGE_" + Integer.toString(stage);
189+
}
190+
191+
return stageName;
192+
}
193+
184194
@Override
185195
public void handleResponses(ResponseBuilder rb, ShardRequest sreq) {
186196
if (rb.isDebugTrack() && rb.isDistrib && !rb.finished.isEmpty()) {
187197
@SuppressWarnings("unchecked")
188198
NamedList<Object> stageList =
189199
(NamedList<Object>)
190-
((NamedList<Object>) rb.getDebugInfo().get("track")).get(stages.get(rb.stage));
200+
((NamedList<Object>) rb.getDebugInfo().get("track"))
201+
.get(getDistributedStageName(rb.stage));
191202
if (stageList == null) {
192203
stageList = new SimpleOrderedMap<>();
193-
rb.addDebug(stageList, "track", stages.get(rb.stage));
204+
rb.addDebug(stageList, "track", getDistributedStageName(rb.stage));
194205
}
195206
for (ShardResponse response : sreq.responses) {
196207
stageList.add(response.getShard(), getTrackResponse(response));

0 commit comments

Comments
 (0)