Chapril/xmpp.chapril.org-ejabberd
25
1
Fork 0
mirror of https://github.com/processone/ejabberd.git synced 2024-10-31 15:21:38 +01:00
Code Releases Activity

put cleaning code to pubsub_debug

Browse Source
This commit is contained in:
Christophe Romain 2012-09-12 15:08:39 +02:00
parent e8d008c392
commit e3085a4d48
2 changed files with 27 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/mod_pubsub/pubsub_clean.erl
View File

@ -1,57 +0,0 @@
-module(pubsub_clean).
-define(TIMEOUT, 1000 * 600).
-export([start/0, loop/0, purge/0, offline/1]).
start() ->
Pid = spawn(?MODULE, loop, []),
register(pubsub_clean, Pid),
Pid.
loop() ->
receive purge -> purge() after ?TIMEOUT -> purge() end,
loop().
purge() ->
Sessions = lists:sum([mnesia:table_info(session, size)
| [rpc:call(N, mnesia, table_info, [session, size])
|| N <- nodes()]]),
Subscriptions = mnesia:table_info(pubsub_state, size),
if Subscriptions > Sessions + 500 ->
lists:foreach(fun (K) ->
[N] = mnesia:dirty_read({pubsub_node, K}),
I = element(3, N),
lists:foreach(fun (JID) ->
case
mnesia:dirty_read({pubsub_state,
{JID,
I}})
of
[{pubsub_state, K, _,
_, _,
[{subscribed,
S}]}] ->
mnesia:dirty_delete({pubsub_subscription,
S});
_ -> ok
end,
mnesia:dirty_delete({pubsub_state,
{JID,
I}})
end,
offline(pubsub_debug:subscribed(I)))
end,
mnesia:dirty_all_keys(pubsub_node));
true -> ok
end.
offline(Jids) ->
lists:filter(fun ({U, S, <<"">>}) ->
ejabberd_sm:get_user_resources(U, S) == [];
({U, S, R}) ->
not
lists:member(R, ejabberd_sm:get_user_resources(U, S))
end,
Jids).%%ejabberd_cluster:get_node({LUser, LServer})

27
src/mod_pubsub/pubsub_debug.erl
View File

@ -107,6 +107,16 @@ subscribed(NodeId) ->
|| S <- states(NodeId),
S#pubsub_state.subscriptions =/= []].
offline_subscribers(NodeId) ->
lists:filter(fun ({U, S, <<"">>}) ->
ejabberd_sm:get_user_resources(U, S) == [];
({U, S, R}) ->
not
lists:member(R, ejabberd_sm:get_user_resources(U, S))
end,
subscribed(NodeId)).
owners(NodeId) ->
[stateid(S)
|| S <- states(NodeId),
@ -254,3 +264,20 @@ pep_subscriptions(LUser, LServer, LResource) ->
end;
_ -> []
end.
purge_offline_subscriptions() ->
lists:foreach(fun (K) ->
[N] = mnesia:dirty_read({pubsub_node, K}),
I = element(3, N),
lists:foreach(fun (JID) ->
case mnesia:dirty_read({pubsub_state, {JID, I}}) of
[{pubsub_state, K, _, _, _, [{subscribed, S}]}] ->
mnesia:dirty_delete({pubsub_subscription, S});
_ ->
ok
end,
mnesia:dirty_delete({pubsub_state, {JID, I}})
end,
offline_subscribers(I))
end,
mnesia:dirty_all_keys(pubsub_node)).