1 files changed, 57 insertions, 0 deletions
diff --git a/deps/rabbitmq_top/src/rabbit_top_wm_ets_tables.erl b/deps/rabbitmq_top/src/rabbit_top_wm_ets_tables.erl
new file mode 100644
index 0000000000..5ae17b4f97
--- /dev/null
+++ b/deps/rabbitmq_top/src/rabbit_top_wm_ets_tables.erl
@@ -0,0 +1,57 @@
+%% This Source Code Form is subject to the terms of the Mozilla Public
+%% License, v. 2.0. If a copy of the MPL was not distributed with this
+%% file, You can obtain one at https://mozilla.org/MPL/2.0/.
+%%
+%% Copyright (c) 2007-2020 VMware, Inc. or its affiliates.  All rights reserved.
+%%
+
+-module(rabbit_top_wm_ets_tables).
+
+-export([init/2, to_json/2, content_types_provided/2, is_authorized/2]).
+
+-include_lib("rabbitmq_management_agent/include/rabbit_mgmt_records.hrl").
+-include_lib("amqp_client/include/amqp_client.hrl").
+
+%%--------------------------------------------------------------------
+
+init(Req, _State) ->
+    {cowboy_rest, rabbit_mgmt_cors:set_headers(Req, ?MODULE), #context{}}.
+
+
+content_types_provided(ReqData, Context) ->
+   {[{<<"application/json">>, to_json}], ReqData, Context}.
+
+to_json(ReqData, Context) ->
+    Sort     = rabbit_top_util:sort_by_param(ReqData, memory),
+    Node     = rabbit_data_coercion:to_atom(rabbit_mgmt_util:id(node, ReqData)),
+    Order    = rabbit_top_util:sort_order_param(ReqData),
+    RowCount = rabbit_top_util:row_count_param(ReqData, 20),
+
+    rabbit_mgmt_util:reply([{node,       Node},
+                            {row_count,  RowCount},
+                            {ets_tables, ets_tables(Node, Sort, Order, RowCount)}],
+                           ReqData, Context).
+
+is_authorized(ReqData, Context) ->
+    rabbit_mgmt_util:is_authorized_admin(ReqData, Context).
+
+%%--------------------------------------------------------------------
+
+ets_tables(Node, Sort, Order, RowCount) ->
+    try
+        [fmt(P) || P <- rabbit_top_worker:ets_tables(Node, Sort, Order, RowCount)]
+    catch
+        exit:{noproc, _} ->
+            []
+    end.
+
+fmt(Info) ->
+    {owner, OPid} = lists:keyfind(owner, 1, Info),
+    {heir, HPid} = lists:keyfind(heir, 1, Info),
+    %% OTP 21 introduced the 'id' element that contains a reference.
+    %% These cannot be serialised and must be removed from the proplist
+    Info1 = lists:keydelete(owner, 1,
+                            lists:keydelete(id, 1, Info)),
+    Info2 = lists:keydelete(heir, 1, Info1),
+    [{owner,  rabbit_top_util:fmt(OPid)},
+     {heir, rabbit_top_util:fmt(HPid)} | Info2].