Add df plugin

[collecty.git] / src / collecty / plugins / latency.py

#!/usr/bin/python3
###############################################################################
#                                                                             #
# collecty - A system statistics collection daemon for IPFire                 #
# Copyright (C) 2012 IPFire development team                                  #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

import socket

import collecty._collecty
from . import base

from ..i18n import _

PING_HOSTS = [
	"ping.ipfire.org",
]

class GraphTemplateLatency(base.GraphTemplate):
	name = "latency"

	lower_limit = 0

	@property
	def rrd_graph(self):
		_ = self.locale.translate

		return [
			"DEF:latency6=%(file)s:latency6:AVERAGE",
			"DEF:loss6=%(file)s:loss6:AVERAGE",
			"DEF:stddev6=%(file)s:stddev6:AVERAGE",

			"DEF:latency4=%(file)s:latency4:AVERAGE",
			"DEF:loss4=%(file)s:loss4:AVERAGE",
			"DEF:stddev4=%(file)s:stddev4:AVERAGE",

			# Compute the biggest loss and convert into percentage
			"CDEF:ploss=loss6,loss4,MAX,100,*",

			# Compute standard deviation
			"CDEF:stddevarea6=stddev6,2,*",
			"CDEF:spacer6=latency6,stddev6,-",
			"CDEF:stddevarea4=stddev4,2,*",
			"CDEF:spacer4=latency4,stddev4,-",

			"CDEF:l005=ploss,0,5,LIMIT,UN,UNKN,INF,IF",
			"CDEF:l010=ploss,5,10,LIMIT,UN,UNKN,INF,IF",
			"CDEF:l025=ploss,10,25,LIMIT,UN,UNKN,INF,IF",
			"CDEF:l050=ploss,25,50,LIMIT,UN,UNKN,INF,IF",
			"CDEF:l099=ploss,50,99,LIMIT,UN,UNKN,INF,IF",

			"VDEF:latency6min=latency6,MINIMUM",
			"VDEF:latency6max=latency6,MAXIMUM",
			"VDEF:latency6avg=latency6,AVERAGE",
			"VDEF:latency4min=latency4,MINIMUM",
			"VDEF:latency4max=latency4,MAXIMUM",
			"VDEF:latency4avg=latency4,AVERAGE",

			"LINE1:latency6avg#00ff0066:%s" % _("Average latency (IPv6)"),
			"LINE1:latency4avg#ff000066:%s\\r" % _("Average latency (IPv4)"),

			"COMMENT:%s" % _("Packet Loss"),
			"AREA:l005#ffffff:%s" % _("0-5%%"),
			"AREA:l010#cccccc:%s" % _("5-10%%"),
			"AREA:l025#999999:%s" % _("10-25%%"),
			"AREA:l050#666666:%s" % _("25-50%%"),
			"AREA:l099#333333:%s" % _("50-99%%") + "\\r",

			"COMMENT: \\n", # empty line

			"AREA:spacer4",
			"AREA:stddevarea4#ff000033:STACK",
			"LINE2:latency4#ff0000:%s" % _("Latency (IPv4)"),
			"GPRINT:latency4max:%12s\:" % _("Maximum") + " %6.2lf",
			"GPRINT:latency4min:%12s\:" % _("Minimum") + " %6.2lf",
			"GPRINT:latency4avg:%12s\:" % _("Average") + " %6.2lf\\n",

			"AREA:spacer6",
			"AREA:stddevarea6#00ff0033:STACK",
			"LINE2:latency6#00ff00:%s" % _("Latency (IPv6)"),
			"GPRINT:latency6max:%12s\:" % _("Maximum") + " %6.2lf",
			"GPRINT:latency6min:%12s\:" % _("Minimum") + " %6.2lf",
			"GPRINT:latency6avg:%12s\:" % _("Average") + " %6.2lf\\n",
		]

	@property
	def graph_title(self):
		_ = self.locale.translate
		return _("Latency to %s") % self.object.hostname

	@property
	def graph_vertical_label(self):
		_ = self.locale.translate
		return _("Milliseconds")

	@property
	def rrd_graph_args(self):
		return [
			"--legend-direction=bottomup",
		]


class LatencyObject(base.Object):
	rrd_schema = [
		"DS:latency6:GAUGE:0:U",
		"DS:stddev6:GAUGE:0:U",
		"DS:loss6:GAUGE:0:100",
		"DS:latency4:GAUGE:0:U",
		"DS:stddev4:GAUGE:0:U",
		"DS:loss4:GAUGE:0:100",
	]

	def __repr__(self):
		return "<%s %s>" % (self.__class__.__name__, self.hostname)

	def init(self, hostname):
		self.hostname = hostname

	@property
	def id(self):
		return self.hostname

	def collect(self):
		result = []

		for family in (socket.AF_INET6, socket.AF_INET):
			try:
				p = collecty._collecty.Ping(self.hostname, family=family)
				p.ping(count=5, deadline=10)

				result += (p.average, p.stddev, p.loss)

			except collecty._collecty.PingAddHostError as e:
				self.log.debug(_("Could not add host %(host)s for family %(family)s") \
					% { "host" : self.hostname, "family" : family })

				# No data available
				result += (None, None, None)
				continue

			except collecty._collecty.PingError as e:
				self.log.warning(_("Could not run latency check for %(host)s: %(msg)s") \
					% { "host" : self.hostname, "msg" : e })

				# A hundred percent loss
				result += (None, None, 1)

		return result


class LatencyPlugin(base.Plugin):
	name = "latency"
	description = "Latency (ICMP ping) Plugin"

	templates = [GraphTemplateLatency]

	@property
	def objects(self):
		for hostname in PING_HOSTS:
			yield LatencyObject(self, hostname)