From: Scott Sanders <scott@jssjr.com>
Date: Sun, 9 Oct 2011 23:02:06 +0000 (-0400)
Subject: Allow the user to append a string to the hostname
X-Git-Tag: collectd-5.1.0~35^2~25^2~5
X-Git-Url: https://git.verplant.org/?a=commitdiff_plain;h=677ce0421a6c77570eba780197310f8d7be43ea1;p=collectd.git

Allow the user to append a string to the hostname

The user should be able to append a string to the hostname to provide
better organization if they wish. The directive used to set this option
is Postfix.
---

diff --git a/src/write_graphite.c b/src/write_graphite.c
index 0de419f2..4fc38fb3 100644
--- a/src/write_graphite.c
+++ b/src/write_graphite.c
@@ -54,7 +54,8 @@
 #define WG_FORMAT_NAME(ret, ret_len, vl, cb, name) \
         wg_format_name (ret, ret_len, (vl)->host, (vl)->plugin, \
                          (vl)->plugin_instance, (vl)->type, \
-                         (vl)->type_instance, (cb)->prefix, name, (cb)->dotchar)
+                         (vl)->type_instance, (cb)->prefix, (cb)->postfix, \
+                         name, (cb)->dotchar)
 #endif
 
 #ifndef WG_SEND_BUF_SIZE
@@ -72,6 +73,7 @@ struct wg_callback
     char    *host;
     int      port;
     char    *prefix;
+    char    *postfix;
     char     dotchar;
 
     char     send_buf[WG_SEND_BUF_SIZE];
@@ -211,6 +213,7 @@ static void wg_callback_free (void *data)
     close(cb->sock_fd);
     sfree(cb->host);
     sfree(cb->prefix);
+    sfree(cb->postfix);
 
     sfree(cb);
 }
@@ -335,7 +338,8 @@ static int wg_format_name (char *ret, int ret_len,
         const char *hostname,
         const char *plugin, const char *plugin_instance,
         const char *type, const char *type_instance,
-        const char *prefix, const char *ds_name, const char dotchar)
+        const char *prefix, const char *postfix,
+        const char *ds_name, const char dotchar)
 {
     int  status;
     char *n_hostname = 0;
@@ -374,21 +378,21 @@ static int wg_format_name (char *ret, int ret_len,
         if ((n_type_instance == NULL) || (n_type_instance[0] == '\0'))
         {
             if ((ds_name == NULL) || (ds_name[0] == '\0'))
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s",
-                        prefix, n_hostname, plugin, type);
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s",
+                        prefix, n_hostname, postfix, plugin, type);
             else
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s.%s",
-                        prefix, n_hostname, plugin, type, ds_name);
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s.%s",
+                        prefix, n_hostname, postfix, plugin, type, ds_name);
         }
         else
         {
             if ((ds_name == NULL) || (ds_name[0] == '\0'))
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s-%s",
-                        prefix, n_hostname, plugin, type,
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s-%s",
+                        prefix, n_hostname, postfix, plugin, type,
                         n_type_instance);
             else
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s-%s.%s",
-                        prefix, n_hostname, plugin, type,
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s-%s.%s",
+                        prefix, n_hostname, postfix, plugin, type,
                         n_type_instance, ds_name);
         }
     }
@@ -397,23 +401,23 @@ static int wg_format_name (char *ret, int ret_len,
         if ((n_type_instance == NULL) || (n_type_instance[0] == '\0'))
         {
             if ((ds_name == NULL) || (ds_name[0] == '\0'))
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s.%s",
-                        prefix, n_hostname, plugin,
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s.%s",
+                        prefix, n_hostname, postfix, plugin,
                         plugin_instance, type);
             else
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s.%s.%s",
-                        prefix, n_hostname, plugin,
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s.%s.%s",
+                        prefix, n_hostname, postfix, plugin,
                         plugin_instance, type, ds_name);
         }
         else
         {
             if ((ds_name == NULL) || (ds_name[0] == '\0'))
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s.%s-%s",
-                        prefix, n_hostname, plugin,
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s.%s-%s",
+                        prefix, n_hostname, postfix, plugin,
                         plugin_instance, type, n_type_instance);
             else
-                status = ssnprintf (ret, ret_len, "%s.%s.%s.%s.%s-%s.%s",
-                        prefix, n_hostname, plugin,
+                status = ssnprintf (ret, ret_len, "%s.%s%s.%s.%s.%s-%s.%s",
+                        prefix, n_hostname, postfix, plugin,
                         plugin_instance, type, n_type_instance, ds_name);
         }
     }
@@ -661,6 +665,7 @@ static int wg_config_carbon (oconfig_item_t *ci)
     cb->host = NULL;
     cb->port = 2003;
     cb->prefix = NULL;
+    cb->postfix = NULL;
     cb->server = NULL;
     cb->dotchar = '_';
 
@@ -676,6 +681,8 @@ static int wg_config_carbon (oconfig_item_t *ci)
             config_set_number (&cb->port, child);
         else if (strcasecmp ("Prefix", child->key) == 0)
             config_set_string (&cb->prefix, child);
+        else if (strcasecmp ("Postfix", child->key) == 0)
+            config_set_string (&cb->postfix, child);
         else if (strcasecmp ("DotCharacter", child->key) == 0)
             config_set_char (&cb->dotchar, child);
         else
@@ -685,6 +692,15 @@ static int wg_config_carbon (oconfig_item_t *ci)
         }
     }
 
+    if (cb->postfix == NULL) {
+        if ((cb->postfix = malloc((int)sizeof(char))) == NULL)
+        {
+            ERROR ("Unable to allocate memory for normalized hostname buffer");
+            return (-1);
+        }
+        cb->postfix[0] = '\0';
+    }
+
     DEBUG ("write_graphite: Registering write callback to carbon agent "
             "%s:%d", cb->host, cb->port);