From 976e3ae10ba7e81c3ead1d2bd1fc852b0f9e3004 Mon Sep 17 00:00:00 2001
From: Michael Klishin <michael@clojurewerkz.org>
Date: Fri, 12 Feb 2016 06:20:01 +0300
Subject: [PATCH] Switch to ets:select_count/2

---
 src/rabbit_connection_tracking.erl | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/src/rabbit_connection_tracking.erl b/src/rabbit_connection_tracking.erl
index 23b7e4c3bb95..e450ab23ef54 100644
--- a/src/rabbit_connection_tracking.erl
+++ b/src/rabbit_connection_tracking.erl
@@ -66,14 +66,9 @@ is_over_connection_limit(VirtualHost) ->
     end.
 
 count_connections_in(VirtualHost) ->
-    %% TODO: optimize
-    Xs = rabbit_misc:execute_mnesia_transaction(
-        fun() ->
-            mnesia:index_read(
-                rabbit_tracked_connection, VirtualHost,
-                #tracked_connection.vhost)
-        end),
-    length(Xs).
+    ets:select_count(?TABLE, [{#tracked_connection{vhost = '$1', _ = '_'},
+                              [{'=:=','$1', VirtualHost}],
+                              [true]}]).
 
 %% Returns a #tracked_connection from connection_created
 %% event details.