From 74f0d7c8ea47b363136c86de5d7ea78a48c40570 Mon Sep 17 00:00:00 2001
From: Rickard Green <rickard@erlang.org>
Date: Wed, 17 Jun 2015 00:01:14 +0200
Subject: Save IO bytes in scheduler specific data

---
 erts/preloaded/ebin/erts_internal.beam | Bin 5380 -> 5964 bytes
 erts/preloaded/src/erts_internal.erl   |  19 ++++++++++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

(limited to 'erts/preloaded')

diff --git a/erts/preloaded/ebin/erts_internal.beam b/erts/preloaded/ebin/erts_internal.beam
index 0e0811af3f..dc8c711e1a 100644
Binary files a/erts/preloaded/ebin/erts_internal.beam and b/erts/preloaded/ebin/erts_internal.beam differ
diff --git a/erts/preloaded/src/erts_internal.erl b/erts/preloaded/src/erts_internal.erl
index 65a1f1ed3a..cf8edefd7d 100644
--- a/erts/preloaded/src/erts_internal.erl
+++ b/erts/preloaded/src/erts_internal.erl
@@ -40,7 +40,7 @@
 
 -export([flush_monitor_messages/3]).
 
--export([await_result/1]).
+-export([await_result/1, gather_io_bytes/2]).
 
 -export([time_unit/0]).
 
@@ -66,6 +66,23 @@ await_result(Ref) when is_reference(Ref) ->
 	    Result
     end.
 
+%%
+%% statistics(io) end up in gather_io_bytes/2
+%%
+
+gather_io_bytes(Ref, No) when is_reference(Ref),
+			      is_integer(No),
+			      No > 0 ->
+    gather_io_bytes(Ref, No, 0, 0).
+
+gather_io_bytes(_Ref, 0, InAcc, OutAcc) ->
+    {{input, InAcc}, {output, OutAcc}};
+gather_io_bytes(Ref, No, InAcc, OutAcc) ->
+    receive
+	{Ref, _SchedId, In, Out} ->
+	    gather_io_bytes(Ref, No-1, InAcc + In, OutAcc + Out)
+    end.
+
 %%
 %% Statically linked port NIFs
 %%
-- 
cgit v1.2.3