Github user danny0405 commented on a diff in the pull request: https://github.com/apache/storm/pull/2704#discussion_r193437371 --- Diff: storm-client/src/jvm/org/apache/storm/utils/TransferDrainer.java --- @@ -40,94 +41,39 @@ public void add(TaskMessage taskMsg) { } public void send(Map<Integer, NodeInfo> taskToNode, Map<NodeInfo, IConnection> connections) { - HashMap<NodeInfo, ArrayList<ArrayList<TaskMessage>>> bundleMapByDestination = groupBundleByDestination(taskToNode); - - for (Map.Entry<NodeInfo, ArrayList<ArrayList<TaskMessage>>> entry : bundleMapByDestination.entrySet()) { + HashMap<NodeInfo, Stream<TaskMessage>> bundleMapByDestination = groupBundleByDestination(taskToNode); + + for (Map.Entry<NodeInfo, Stream<TaskMessage>> entry : bundleMapByDestination.entrySet()) { --- End diff -- nit: whitespace
---