honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Internal: make sure that multi_percolate request hands over its context and headers to its corresponding shard requests 

Closes #7371
This commit is contained in:
javanna 2014-08-21 13:31:32 +02:00 committed by Luca Cavanna
parent 9dd3597f1f
commit b6cdaff30c
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/org/elasticsearch/action/percolate/TransportMultiPercolateAction.java
View File

@ -122,7 +122,7 @@ public class TransportMultiPercolateAction extends HandledTransportAction<MultiP
percolateRequests.set(slot, itemResponse.getFailure());
}
}
new ASyncAction(percolateRequests, listener, clusterState).run();
new ASyncAction(request, percolateRequests, listener, clusterState).run();
}
@Override
@ -131,7 +131,7 @@ public class TransportMultiPercolateAction extends HandledTransportAction<MultiP
}
});
} else {
new ASyncAction(percolateRequests, listener, clusterState).run();
new ASyncAction(request, percolateRequests, listener, clusterState).run();
}
}
@ -140,6 +140,7 @@ public class TransportMultiPercolateAction extends HandledTransportAction<MultiP
final ActionListener<MultiPercolateResponse> finalListener;
final Map<ShardId, TransportShardMultiPercolateAction.Request> requestsByShard;
final MultiPercolateRequest multiPercolateRequest;
final List<Object> percolateRequests;
final Map<ShardId, IntArrayList> shardToSlots;
@ -148,8 +149,9 @@ public class TransportMultiPercolateAction extends HandledTransportAction<MultiP
final AtomicReferenceArray<AtomicInteger> expectedOperationsPerItem;
final AtomicReferenceArray<AtomicReferenceArray> responsesByItemAndShard;
ASyncAction(List<Object> percolateRequests, ActionListener<MultiPercolateResponse> finalListener, ClusterState clusterState) {
ASyncAction(MultiPercolateRequest multiPercolateRequest, List<Object> percolateRequests, ActionListener<MultiPercolateResponse> finalListener, ClusterState clusterState) {
this.finalListener = finalListener;
this.multiPercolateRequest = multiPercolateRequest;
this.percolateRequests = percolateRequests;
responsesByItemAndShard = new AtomicReferenceArray<>(percolateRequests.size());
expectedOperationsPerItem = new AtomicReferenceArray<>(percolateRequests.size());
@ -192,7 +194,7 @@ public class TransportMultiPercolateAction extends HandledTransportAction<MultiP
ShardId shardId = shard.shardId();
TransportShardMultiPercolateAction.Request requests = requestsByShard.get(shardId);
if (requests == null) {
requestsByShard.put(shardId, requests = new TransportShardMultiPercolateAction.Request(shardId.getIndex(), shardId.getId(), percolateRequest.preference()));
requestsByShard.put(shardId, requests = new TransportShardMultiPercolateAction.Request(multiPercolateRequest, shardId.getIndex(), shardId.getId(), percolateRequest.preference()));
}
logger.trace("Adding shard[{}] percolate request for item[{}]", shardId, slot);
requests.add(new TransportShardMultiPercolateAction.Request.Item(slot, new PercolateShardRequest(shardId, percolateRequest)));

4
src/main/java/org/elasticsearch/action/percolate/TransportShardMultiPercolateAction.java
View File

@ -126,8 +126,8 @@ public class TransportShardMultiPercolateAction extends TransportShardSingleOper
Request() {
}
Request(String concreteIndex, int shardId, String preference) {
this.index = concreteIndex;
Request(MultiPercolateRequest multiPercolateRequest, String concreteIndex, int shardId, String preference) {
super(multiPercolateRequest, concreteIndex);
this.shardId = shardId;
this.preference = preference;
this.items = new ArrayList<>();