From 8a5519cafad355f740bc3c4698e0c4335ab7d303 Mon Sep 17 00:00:00 2001
From: terry-xiaoyu <506895667@qq.com>
Date: Thu, 30 Aug 2018 14:29:07 +0800
Subject: [PATCH] attrs for ws_connection

---
 src/emqx_ws_connection.erl | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/emqx_ws_connection.erl b/src/emqx_ws_connection.erl
index 026d160b3..ed1532565 100644
--- a/src/emqx_ws_connection.erl
+++ b/src/emqx_ws_connection.erl
@@ -17,7 +17,7 @@
 -include("emqx.hrl").
 -include("emqx_mqtt.hrl").
 
--export([info/1]).
+-export([info/1, attrs/1]).
 -export([stats/1]).
 -export([kick/1]).
 -export([session/1]).
@@ -53,9 +53,14 @@
 %% API
 %%------------------------------------------------------------------------------
 
+%% for debug
 info(WSPid) ->
     call(WSPid, info).
 
+%% for dashboard
+attrs(CPid) when is_pid(CPid) ->
+    call(CPid, attrs).
+
 stats(WSPid) ->
     call(WSPid, stats).
 
@@ -170,6 +175,15 @@ websocket_info({call, From, info}, State = #state{peername    = Peername,
     gen_server:reply(From, lists:append([ConnInfo, ProtoInfo])),
     {ok, State};
 
+websocket_info({call, From, attrs}, State = #state{peername    = Peername,
+                                                   sockname    = Sockname,
+                                                   proto_state = ProtoState}) ->
+    SockAttrs = [{peername, Peername},
+                 {sockname, Sockname}],
+    ProtoAttrs = emqx_protocol:attrs(ProtoState),
+    gen_server:reply(From, lists:usort(lists:append(SockAttrs, ProtoAttrs))),
+    {ok, State};
+
 websocket_info({call, From, stats}, State = #state{proto_state = ProtoState}) ->
     Stats = lists:append([wsock_stats(), emqx_misc:proc_stats(), emqx_protocol:stats(ProtoState)]),
     gen_server:reply(From, Stats),
@@ -262,4 +276,3 @@ shutdown(Reason, State) ->
 
 wsock_stats() ->
     [{Key, get(Key)} || Key <- ?SOCK_STATS].
-