From f504781a136b1992a1cee26a7da841892d796d91 Mon Sep 17 00:00:00 2001
From: Sverker Eriksson <sverker@erlang.org>
Date: Wed, 21 Nov 2018 15:54:17 +0100
Subject: erts: Add counters:put/3

---
 erts/preloaded/ebin/counters.beam      | Bin 2808 -> 3140 bytes
 erts/preloaded/ebin/erts_internal.beam | Bin 17508 -> 17632 bytes
 erts/preloaded/src/counters.erl        |  14 ++++++++++++++
 erts/preloaded/src/erts_internal.erl   |   6 +++++-
 4 files changed, 19 insertions(+), 1 deletion(-)

(limited to 'erts/preloaded')

diff --git a/erts/preloaded/ebin/counters.beam b/erts/preloaded/ebin/counters.beam
index caaa6167e1..4e1a3566f7 100644
Binary files a/erts/preloaded/ebin/counters.beam and b/erts/preloaded/ebin/counters.beam differ
diff --git a/erts/preloaded/ebin/erts_internal.beam b/erts/preloaded/ebin/erts_internal.beam
index e174f71966..651d5e9d05 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/counters.erl b/erts/preloaded/src/counters.erl
index 67354f648d..a0e3035e0f 100644
--- a/erts/preloaded/src/counters.erl
+++ b/erts/preloaded/src/counters.erl
@@ -26,6 +26,7 @@
          get/2,
          add/3,
          sub/3,
+         put/3,
          info/1]).
 
 -export_type([counters_ref/0]).
@@ -76,6 +77,19 @@ add(_, _, _) ->
 sub(Ref, Ix, Decr) ->
     add(Ref, Ix, -Decr).
 
+
+-spec put(Ref, Ix, Value) -> ok when
+      Ref  :: counters_ref(),
+      Ix :: integer(),
+      Value :: integer().
+put({atomics, Ref}, Ix, Value) ->
+    atomics:put(Ref, Ix, Value);
+put({write_concurrency, Ref}, Ix, Value) ->
+    erts_internal:counters_put(Ref, Ix, Value);
+put(_, _, _) ->
+    erlang:error(badarg).
+
+
 -spec info(Ref) -> Info when
       Ref  :: counters_ref(),
       Info :: #{'size':=Size, 'memory':=Memory},
diff --git a/erts/preloaded/src/erts_internal.erl b/erts/preloaded/src/erts_internal.erl
index d491a505c6..8f29a569f2 100644
--- a/erts/preloaded/src/erts_internal.erl
+++ b/erts/preloaded/src/erts_internal.erl
@@ -95,7 +95,7 @@
 -export([atomics_new/2]).
 
 -export([counters_new/1, counters_get/2, counters_add/3,
-         counters_info/1]).
+         counters_put/3, counters_info/1]).
 
 %%
 %% Await result of send to port
@@ -719,6 +719,10 @@ counters_get(_Ref, _Ix) ->
 counters_add(_Ref, _Ix, _Incr) ->
     erlang:nif_error(undef).
 
+-spec counters_put(reference(), pos_integer(), integer()) -> ok.
+counters_put(_Ref, _Ix, _Value) ->
+    erlang:nif_error(undef).
+
 -spec counters_info(reference()) -> #{}.
 counters_info(_Ref) ->
     erlang:nif_error(undef).
-- 
cgit v1.2.3