From 965a9c51c73bdc0b90ba6350107fad41b605c974 Mon Sep 17 00:00:00 2001
From: Michael Tremer <michael.tremer@ipfire.org>
Date: Sat, 25 Aug 2012 15:40:30 +0000
Subject: [PATCH] plugins: Automatically create RRAs.

---
 collecty/plugins/base.py | 61 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 60 insertions(+), 1 deletion(-)

diff --git a/collecty/plugins/base.py b/collecty/plugins/base.py
index e06df93..2554a89 100644
--- a/collecty/plugins/base.py
+++ b/collecty/plugins/base.py
@@ -19,7 +19,10 @@
 #                                                                             #
 ###############################################################################
 
+from __future__ import division
+
 import logging
+import math
 import os
 import rrdtool
 import threading
@@ -66,6 +69,11 @@ class Plugin(threading.Thread):
 	# The schema of the RRD database.
 	rrd_schema = None
 
+	# RRA properties.
+	rra_types = ["AVERAGE", "MIN", "MAX"]
+	rra_timespans = [3600, 86400, 604800, 2678400, 31622400]
+	rra_rows = 2880
+
 	# Instructions how to create the graph.
 	rrd_graph = None
 
@@ -128,6 +136,10 @@ class Plugin(threading.Thread):
 		# Otherwise return the default.
 		return self.default_interval
 
+	@property
+	def stepsize(self):
+		return self.interval
+
 	@property
 	def file(self):
 		"""
@@ -147,10 +159,57 @@ class Plugin(threading.Thread):
 		if not os.path.exists(dirname):
 			os.makedirs(dirname)
 
-		rrdtool.create(self.file, *self.rrd_schema)
+		# Create argument list.
+		args = [
+			"--step", "%s" % self.default_interval,
+		] + self.get_rrd_schema()
+
+		rrdtool.create(self.file, *args)
 
 		self.log.debug(_("Created RRD file %s.") % self.file)
 
+	def get_rrd_schema(self):
+		schema = [
+			"--step", "%s" % self.stepsize,
+		]
+		for line in self.rrd_schema:
+			if line.startswith("DS:"):
+				try:
+					(prefix, name, type, lower_limit, upper_limit) = line.split(":")
+
+					line = ":".join((
+						prefix,
+						name,
+						type,
+						"%s" % self.stepsize,
+						lower_limit,
+						upper_limit
+					))
+				except ValueError:
+					pass
+
+			schema.append(line)
+
+		xff = 0.1
+
+		cdp_length = 0
+		for rra_timespan in self.rra_timespans:
+			if (rra_timespan / self.stepsize) < self.rra_rows:
+				rra_timespan = self.stepsize * self.rra_rows
+
+			if cdp_length == 0:
+				cdp_length = 1
+			else:
+				cdp_length = rra_timespan // (self.rra_rows * self.stepsize)
+
+			cdp_number = math.ceil(rra_timespan / (cdp_length * self.stepsize))
+
+			for rra_type in self.rra_types:
+				schema.append("RRA:%s:%.10f:%d:%d" % \
+					(rra_type, xff, cdp_length, cdp_number))
+
+		return schema
+
 	def info(self):
 		return rrdtool.info(self.file)
 
-- 
2.47.2