|
1443
|
1 -- Log common stats to statsd |
|
|
2 -- |
|
|
3 -- Copyright (C) 2014 Daurnimator |
|
|
4 -- |
|
|
5 -- This module is MIT/X11 licensed. |
|
|
6 |
|
|
7 local socket = require "socket" |
|
|
8 local iterators = require "util.iterators" |
|
|
9 local jid = require "util.jid" |
|
|
10 |
|
|
11 local options = module:get_option("statsd") or {} |
|
|
12 |
|
|
13 -- Create UDP socket to statsd server |
|
|
14 local sock = socket.udp() |
|
|
15 sock:setpeername(options.hostname or "127.0.0.1", options.port or 8125) |
|
|
16 |
|
|
17 -- Metrics are namespaced by ".", and seperated by newline |
|
|
18 function clean(s) return (s:gsub("[%.\n]", "_")) end |
|
|
19 |
|
|
20 -- A 'safer' send function to expose |
|
|
21 function send(s) return sock:send(s) end |
|
|
22 |
|
|
23 -- prefix should end in "." |
|
|
24 local prefix = (options.prefix or ("prosody." .. clean(module.host))) .. "." |
|
|
25 |
|
|
26 -- Track users as they bind/unbind |
|
|
27 -- count bare sessions every time, as we have no way to tell if it's a new bare session or not |
|
|
28 module:hook("resource-bind", function(event) |
|
|
29 send(prefix.."bare_sessions:"..iterators.count(bare_sessions).."|g") |
|
|
30 send(prefix.."full_sessions:+1|g") |
|
|
31 end, 1) |
|
|
32 module:hook("resource-unbind", function(event) |
|
|
33 send(prefix.."bare_sessions:"..iterators.count(bare_sessions).."|g") |
|
|
34 send(prefix.."full_sessions:-1|g") |
|
|
35 end, 1) |
|
|
36 |
|
|
37 -- Track MUC occupants as they join/leave |
|
|
38 module:hook("muc-occupant-joined", function(event) |
|
|
39 send(prefix.."n_occupants:+1|g") |
|
|
40 local room_node = jid.split(event.room.jid) |
|
|
41 send(prefix..clean(room_node)..".occupants:+1|g") |
|
|
42 end) |
|
|
43 module:hook("muc-occupant-left", function(event) |
|
|
44 send(prefix.."n_occupants:-1|g") |
|
|
45 local room_node = jid.split(event.room.jid) |
|
|
46 send(prefix..clean(room_node)..".occupants:-1|g") |
|
|
47 end) |
|
|
48 |
|
|
49 -- Misc other MUC |
|
|
50 module:hook("muc-broadcast-message", function(event) |
|
|
51 send(prefix.."broadcast-message:1|c") |
|
|
52 local room_node = jid.split(event.room.jid) |
|
|
53 send(prefix..clean(room_node)..".broadcast-message:1|c") |
|
|
54 end) |
|
|
55 module:hook("muc-invite", function(event) |
|
|
56 send(prefix.."invite:1|c") |
|
|
57 local room_node = jid.split(event.room.jid) |
|
|
58 send(prefix..clean(room_node)..".invite:1|c") |
|
|
59 local to_node, to_host, to_resource = jid.split(event.stanza.attr.to) |
|
|
60 send(prefix..clean(to_node)..".invites:1|c") |
|
|
61 end) |
