X-Git-Url: https://git.octo.it/?p=collectd.git;a=blobdiff_plain;f=src%2Ffilecount.c;h=e2b002074fd9462fc0bc047e1bdc3b11b2428a1d;hp=f0713795f52100f909cb61a6fb52f313a19737d5;hb=576d966de9a98aacd1a2740abb349022cc32d913;hpb=e628f39838a67b40d52dfb8425b4d8474fbd0550 diff --git a/src/filecount.c b/src/filecount.c index f0713795..e2b00207 100644 --- a/src/filecount.c +++ b/src/filecount.c @@ -31,11 +31,15 @@ #include #include +#define FC_RECURSIVE 1 + struct fc_directory_conf_s { char *path; char *instance; + int options; + /* Data counters */ uint64_t files_num; uint64_t files_size; @@ -307,6 +311,25 @@ static int fc_config_add_dir_size (fc_directory_conf_t *dir, return (0); } /* int fc_config_add_dir_size */ +static int fc_config_add_dir_recursive (fc_directory_conf_t *dir, + oconfig_item_t *ci) +{ + if ((ci->values_num != 1) + || (ci->values[0].type != OCONFIG_TYPE_BOOLEAN)) + { + WARNING ("filecount plugin: The `Recursive' config options needs exactly " + "one boolean argument."); + return (-1); + } + + if (ci->values[0].value.boolean) + dir->options |= FC_RECURSIVE; + else + dir->options &= ~FC_RECURSIVE; + + return (0); +} /* int fc_config_add_dir_recursive */ + static int fc_config_add_dir (oconfig_item_t *ci) { fc_directory_conf_t *dir; @@ -338,6 +361,8 @@ static int fc_config_add_dir (oconfig_item_t *ci) fc_config_set_instance (dir, dir->path); + dir->options = FC_RECURSIVE; + dir->name = NULL; dir->mtime = 0; dir->size = 0; @@ -355,6 +380,8 @@ static int fc_config_add_dir (oconfig_item_t *ci) status = fc_config_add_dir_mtime (dir, option); else if (strcasecmp ("Size", option->key) == 0) status = fc_config_add_dir_size (dir, option); + else if (strcasecmp ("Recursive", option->key) == 0) + status = fc_config_add_dir_recursive (dir, option); else { WARNING ("filecount plugin: fc_config_add_dir: " @@ -447,7 +474,7 @@ static int fc_read_dir_callback (const char *dirname, const char *filename, return (-1); } - if (S_ISDIR (statbuf.st_mode)) + if (S_ISDIR (statbuf.st_mode) && (dir->options & FC_RECURSIVE)) { status = walk_directory (abs_path, fc_read_dir_callback, dir); return (status);