1 files changed, 66 insertions, 33 deletions

diff --git a/inc/collectd.inc.php b/inc/collectd.inc.php
index 7269e68..e364180 100644
--- a/inc/collectd.inc.php
+++ b/inc/collectd.inc.php
@@ -19,6 +19,26 @@ function collectd_hosts() {
         return($dir);
 }
 
+
+# return files in directory. this will recurse into subdirs
+# infinite loop may occur
+function get_host_rrd_files($dir) {
+        $files = array();
+
+        $objects = new RegexIterator(
+                new RecursiveIteratorIterator(
+                new RecursiveDirectoryIterator($dir),
+                RecursiveIteratorIterator::SELF_FIRST),
+                '/\.rrd$/');
+
+        foreach($objects as $object) {
+                $files[] = str_replace($dir.'/', '', $object->getPathname());
+        }
+
+        return $files;
+}
+
+
 # returns an array of plugins/pinstances/types/tinstances
 function collectd_plugindata($host, $plugin=NULL) {
         global $CONFIG;
@@ -26,8 +46,8 @@ function collectd_plugindata($host, $plugin=NULL) {
         if (!is_dir($CONFIG['datadir'].'/'.$host))
                 return false;
 
-        chdir($CONFIG['datadir'].'/'.$host);
-        $files = glob("*/*.rrd");
+        $hostdir = $CONFIG['datadir'].'/'.$host;
+        $files = get_host_rrd_files($hostdir);
         if (!$files)
                 return false;
 
@@ -215,6 +235,28 @@ function build_url($base, $items, $s=NULL) {
         return $base;
 }
 
+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) {
         global $CONFIG;
@@ -226,43 +268,34 @@ function collectd_flush($identifier) {
                         !(is_string($identifier) || is_array($identifier)))
                 return FALSE;
 
+        if (!is_array($identifier))
+                $identifier = array($identifier);
+
         $u_errno  = 0;
         $u_errmsg = '';
-        if ($socket = @fsockopen($CONFIG['socket'], 0, $u_errno, $u_errmsg)) {
-                $cmd = 'FLUSH plugin=rrdtool';
-                if (is_array($identifier)) {
-                        foreach ($identifier as $val)
-                                $cmd .= sprintf(' identifier="%s"', $val);
-                } else
-                        $cmd .= sprintf(' identifier="%s"', $identifier);
-                $cmd .= "\n";
-
-                $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;
-                }
+        if (! $socket = @fsockopen($CONFIG['socket'], 0, $u_errno, $u_errmsg)) {
+                error_log(sprintf('ERROR: Failed to open unix-socket to %s (%d: %s)',
+                        $CONFIG['socket'], $u_errno, $u_errmsg));
+                return FALSE;
+        }
 
-                $resp = fgets($socket);
-                if ($resp === false) {
-                        error_log(sprintf('ERROR: Failed to read response from collectd for command: %s',
-                                trim($cmd)));
-                        return FALSE;
+        if ($CONFIG['flush_type'] == 'collectd'){
+                $cmd = 'FLUSH';
+                foreach ($identifier as $val)
+                        $cmd .= sprintf(' identifier="%s"', $val);
+                $cmd .= "\n";
+                socket_cmd($socket, $cmd);
+        }
+        elseif ($CONFIG['flush_type'] == 'rrdcached') {
+                foreach ($identifier as $val) {
+                        $cmd = sprintf("FLUSH %s.rrd\n", $val);
+                        socket_cmd($socket, $cmd);
                 }
+        }
 
-                $n = (int)$resp;
-                while ($n-- > 0)
-                        fgets($socket);
-
-                fclose($socket);
+        fclose($socket);
 
-                return TRUE;
-        } else {
-                error_log(sprintf('ERROR: Failed to open unix-socket to collectd: %d: %s',
-                        $u_errno, $u_errmsg));
-                return FALSE;
-        }
+        return TRUE;
 }
 
 ?>