move collectd_flush function to base class

1 files changed, 70 insertions, 0 deletions

diff --git a/type/Base.class.php b/type/Base.class.php
index 2851280..fe31814 100644
--- a/type/Base.class.php
+++ b/type/Base.class.php
@@ -27,6 +27,9 @@ class Type_Base {
         var $tinstances;
         var $identifiers;
 
+        var $flush_socket;
+        var $flush_type;
+
         function __construct($config, $_get) {
                 $this->datadir = $config['datadir'];
                 $this->rrdtool = $config['rrdtool'];
@@ -39,6 +42,8 @@ class Type_Base {
                 $this->graph_type = $config['graph_type'];
                 $this->negative_io = $config['negative_io'];
                 $this->graph_smooth = $config['graph_smooth'];
+                $this->flush_socket = $config['socket'];
+                $this->flush_type = $config['flush_type'];
         }
 
         function rainbow_colors() {
@@ -167,6 +172,8 @@ class Type_Base {
         }
 
         function rrd_graph($debug = false) {
+                $this->collectd_flush();
+
                 if (!$this->colors)
                         $this->rainbow_colors();
 
@@ -295,4 +302,67 @@ class Type_Base {
                         $this->ds_names[$index] = sprintf($format, $value);
                 }
         }
+
+        function socket_cmd($socket, $cmd) {
+                $r = fwrite($socket, $cmd, strlen($cmd));
+                if ($r === false || $r != strlen($cmd)) {
+                        error_log(sprintf('ERROR: Failed to write full command to unix-socket: %d out of %d written',
+                                $r === false ? -1 : $r, strlen($cmd)));
+                        return FALSE;
+                }
+
+                $resp = fgets($socket,128);
+                if ($resp === false) {
+                        error_log(sprintf('ERROR: Failed to read response from collectd for command: %s',
+                                trim($cmd)));
+                        return FALSE;
+                }
+
+                $n = (int)$resp;
+                while ($n-- > 0)
+                        fgets($socket,128);
+
+                return TRUE;
+        }
+
+        # tell collectd to FLUSH all data of the identifier(s)
+        function collectd_flush() {
+                $identifier = $this->identifiers;
+
+                if (!$this->flush_socket)
+                        return FALSE;
+
+                if (!$identifier || (is_array($identifier) && count($identifier) == 0) ||
+                                !(is_string($identifier) || is_array($identifier)))
+                        return FALSE;
+
+                if (!is_array($identifier))
+                        $identifier = array($identifier);
+
+                $u_errno  = 0;
+                $u_errmsg = '';
+                if (! $socket = @fsockopen($this->flush_socket, 0, $u_errno, $u_errmsg)) {
+                        error_log(sprintf('ERROR: Failed to open unix-socket to %s (%d: %s)',
+                                $this->flush_socket, $u_errno, $u_errmsg));
+                        return FALSE;
+                }
+
+                if ($this->flush_type == 'collectd'){
+                        $cmd = 'FLUSH';
+                        foreach ($identifier as $val)
+                                $cmd .= sprintf(' identifier="%s"', $val);
+                        $cmd .= "\n";
+                        $this->socket_cmd($socket, $cmd);
+                }
+                elseif ($this->flush_type == 'rrdcached') {
+                        foreach ($identifier as $val) {
+                                $cmd = sprintf("FLUSH %s.rrd\n", $val);
+                                $this->socket_cmd($socket, $cmd);
+                        }
+                }
+
+                fclose($socket);
+
+                return TRUE;
+        }
 }