From f348a4461d5c3b5e887eeb550117f08508adf2d6 Mon Sep 17 00:00:00 2001
From: Anders Svensson <anders@erlang.org>
Date: Thu, 5 Oct 2017 14:16:14 +0200
Subject: Update appup and code_change for ERIERL-83

---
 lib/diameter/src/base/diameter_reg.erl | 5 +++++
 lib/diameter/src/diameter.appup.src    | 6 ++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

(limited to 'lib')

diff --git a/lib/diameter/src/base/diameter_reg.erl b/lib/diameter/src/base/diameter_reg.erl
index e0c41a2834..5b7cfab31a 100644
--- a/lib/diameter/src/base/diameter_reg.erl
+++ b/lib/diameter/src/base/diameter_reg.erl
@@ -298,6 +298,11 @@ terminate(_Reason, _State)->
 %% # code_change/3
 %% ----------------------------------------------------------
 
+code_change(_, State, "2.1") ->
+    {ok, lists:foldl(fun add_monitor/2,
+                     State,
+                     ets:select(?TABLE, [{{'_', '$1'}, [], ['$1']}]))};
+
 code_change(_OldVsn, State, _Extra) ->
     {ok, State}.
 
diff --git a/lib/diameter/src/diameter.appup.src b/lib/diameter/src/diameter.appup.src
index 7566cf25c3..c2198de9ea 100644
--- a/lib/diameter/src/diameter.appup.src
+++ b/lib/diameter/src/diameter.appup.src
@@ -53,7 +53,8 @@
   {"1.12",    [{restart_application, diameter}]},  %% 19.0
   {"1.12.1",  [{restart_application, diameter}]},  %% 19.1
   {"1.12.2",  [{restart_application, diameter}]},  %% 19.3
-  {"2.0",     [{restart_application, diameter}]}   %% 20.0
+  {"2.0",     [{restart_application, diameter}]},  %% 20.0
+  {"2.1",     [{update, diameter_reg, {advanced, "2.1"}}]} %% 20.1
  ],
  [
   {"0.9",     [{restart_application, diameter}]},
@@ -88,6 +89,7 @@
   {"1.12",    [{restart_application, diameter}]},
   {"1.12.1",  [{restart_application, diameter}]},
   {"1.12.2",  [{restart_application, diameter}]},
-  {"2.0",     [{restart_application, diameter}]}
+  {"2.0",     [{restart_application, diameter}]},
+  {"2.1",     [{restart_application, diameter}]}
  ]
 }.
-- 
cgit v1.2.3