summaryrefslogtreecommitdiff
path: root/files/munin/ejabberd_
blob: 84f52bd0c3206f8cb3abd48cc5786a51600eef20 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
#!/bin/bash
# 
# Munin plugin for ejabberd.
#
# Written by Lasse Karstensen <lkarsten@hyse.org> 2007-05-27.
# Based on ejabberd-plugin by Christian Dröge <Christian@draugr.de>
#
# As connected users, registered users and server-connections have somewhat
# different scales, this plugin uses munins suggest feature to create three
# different graphs.
# 
# ejabberd_connections
# ejabberd_users
# ejabberd_registrations 	
#
# use ln -s ejabberd ejabberd_(connections|users|registrations) 
# to activate.
#
# If the autodetect-feature for vhosts breaks, you can set  
# """
# [ejabberd*]
# env.vhosts foo.com bar.com
# """
# in plugin-conf.d/munin-node to override it.
# (user root may also be smart/not so smart depending on your setup.)
#
#%# family=auto
#%# capabilities=autoconf suggest

EJCTL=$ejctl
if [ "$EJCTL" == "" ]; then
    EJCTL=`which ejabberdctl||echo "/usr/sbin/ejabberdctl"`
fi
if [ "$1" == "autoconf" ]; then
        if [ -x $EJCTL > /dev/null ]; then
                echo yes
                exit 0
        fi
        echo "no (ejabberdctl not found)"
        exit 1
fi

if [ "$1" == "suggest" ]; then
    echo "connections"
    echo "users"
    echo "registrations"
    exit 0
fi

# trying to autodetect running vhosts.
if [ -z "$vhosts" ]; then
    for CFGPATH in /etc/ejabberd /usr/local/ejabberd/etc; do
	if [ -f "$CFGPATH/ejabberd.cfg" ]; then
	    EJCFG=$CFGPATH/ejabberd.cfg;
	fi
    done
    if [ -z "$EJCFG" ]; then
        echo "Unable to find ejabberd.cfg. Exiting." > /dev/stderr
        exit -1
    fi

    # this is pretty trivial, and may break if you have many vhosts.
    vhosts=$(egrep -A5 '^\{hosts' $EJCFG | \
       egrep -v "^#" | \
    # make it one whole line.
       tr '\n' ' ' | \
    # eat what is before [ and after ] on the line.
       sed -e 's/.*\[//' -e 's/\].*//' | \
    # remove the "s
       sed -e 's/\"//g' | \
    # clean up whitespaces
       sed -e 's/ //g' | \
       sed -e 's/,/ /g' | \
    # hmm, wonder if this should be here. 
       sed -e 's/localhost//' )
    # all set!
    # okay, not so trivial. hope it works. if you get funky results, 
    # try adjusting the "egrep -Ax" above. 
fi

if [ -z "$vhosts" ]; then
    echo "No vhosts to sample." > /dev/stderr
    echo "Please set env.vhosts in plugins-conf.d/munin-node." > /dev/stderr
    exit -1
fi

MODE=`basename $0 | sed 's/^ejabberd_//g'`

if ! [ "$MODE" == "connections" -o "$MODE" == "users" \
  -o "$MODE" == "registrations" ]; then
    echo "ERROR: Unknown mode \"$MODE\". Exiting." > /dev/stderr
    exit -1
fi

if [ "$1" = "config" ]; then
    echo 'graph_args --base 1000 -l 0'
    echo 'graph_scale no'
    echo 'graph_category ejabberd'
    echo 'graph_info This graph shows a statistic of ejabberd '

    if [ "$MODE" == "connections" ]; then
        echo 's2s_connections_out.label incoming s2s connections'
        echo 's2s_connections_out.info Number of outgoing server to server connections'

        echo 's2s_connections_in.label outgoing s2s connections'
        echo 's2s_connections_in.info Number of incoming server to server connections'
    elif [ "$MODE" == "users" ]; then
	echo 'graph_title ejabberd connected users'
	echo 'graph_vlabel users'
#        echo 'connected_users.info Number of logged in users'
#        echo 'connected_users.label connected users'
	for host in $vhosts; do
	    formathost=$(echo $host | tr '.-' '_')
	    echo "connected_users_$formathost.label $host connected users";
	    echo "connected_unique_users_$formathost.label $host unique connected users";
	done;
    elif [ "$MODE" == "registrations" ]; then
	echo 'graph_title ejabberd registrations'
	echo 'graph_vlabel users'
	for host in $vhosts; do
	    formathost=$(echo $host | tr '.-' '_')
	    echo "registered_$formathost.label $host registered users"; 
	    echo "registered_$formathost.info Registered users for vhost $host"
	done;
    fi
    exit 0
fi


if [ "$MODE" == "connections" ]; then
    echo -n "s2s_connections_out.value "
    $EJCTL outgoing-s2s-number
    echo -n "s2s_connections_in.value "
    $EJCTL incoming-s2s-number
    exit 0
fi

if [ "$MODE" == "users" ]; then
#    echo -n "connected_unique_users.value "
#    $EJCTL connected-users | awk -F/ '{print $1}' | sort | uniq | wc -l

    for host in $vhosts; do
	formathost=$(echo $host | tr '.-' '_')
	echo -n "connected_users_$formathost.value "; 
	$EJCTL connected-users | grep "@$host" | wc -l
	echo -n "connected_unique_users_$formathost.value "; 
	$EJCTL connected-users | grep "@$host" | awk -F/ '{print $1}' | sort | uniq | wc -l
    done
    exit 0
fi

if [ "$MODE" == "registrations" ]; then
    for host in $vhosts; do
	formathost=$(echo $host | tr '.-' '_')
	echo -n "registered_$formathost.value "; 
	num=$($EJCTL registered-users $host)
        if [ "$?" != 0 ]; then
            echo "U"
        else
            echo $num | wc -w
        fi
    done
    exit 0
fi