From: bert hubert <bert.hubert@netherlabs.nl>
Date: Fri, 8 May 2015 19:55:03 +0000 (+0200)
Subject: add the actual carbon file
X-Git-Tag: dnsdist-1.0.0-alpha1~248^2~28^2~52^2~1
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=68ee58b65c120ea875d1732d70517c22d9921579;p=thirdparty%2Fpdns.git

add the actual carbon file
---

diff --git a/pdns/dnsdist-carbon.cc b/pdns/dnsdist-carbon.cc
new file mode 100644
index 0000000000..f94cd3839e
--- /dev/null
+++ b/pdns/dnsdist-carbon.cc
@@ -0,0 +1,81 @@
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+#include "iputils.hh"
+#include "dolog.hh"
+#include "sstuff.hh"
+#include <boost/foreach.hpp>
+#include "namespaces.hh"
+#undef L
+#include "dnsdist.hh"
+
+GlobalStateHolder<CarbonConfig> g_carbon;
+
+void* carbonDumpThread()
+try
+{
+  auto localCarbon = g_carbon.getLocal();
+  for(int numloops=0;;++numloops) {
+    if(localCarbon->server == ComboAddress("0.0.0.0", 0)) {
+      sleep(1);
+      continue;
+    }
+    if(numloops) 
+      sleep(localCarbon->interval);
+
+    try {
+      Socket s(localCarbon->server.sin4.sin_family, SOCK_STREAM);
+      
+      s.setNonBlocking();
+      s.connect(localCarbon->server);  // we do the connect so the attempt happens while we gather stats
+      
+      ostringstream str;
+      time_t now=time(0);
+      string hostname=localCarbon->ourname;
+      if(hostname.empty()) {
+	char tmp[80];
+	memset(tmp, 0, sizeof(tmp));
+	gethostname(tmp, sizeof(tmp));
+	char *p = strchr(tmp, '.');
+	if(p) *p=0;
+	hostname=tmp;
+	boost::replace_all(hostname, ".", "_");
+      }
+      for(const auto& e : g_stats.entries) {
+	str<<"dnsdist."<<hostname<<".main."<<e.first<<' '<<e.second->load()<<' '<<now<<"\r\n";
+      }
+      const string msg = str.str();
+      
+      int ret = waitForRWData(s.getHandle(), false, 1 , 0); 
+      if(ret <= 0 ) {
+	infolog("Unable to write data to carbon server on %s: %s", localCarbon->server.toStringWithPort(), (ret<0 ? strerror(errno) : "Timeout"));
+	continue;
+      }
+      s.setBlocking();
+      ret=writen2(s.getHandle(), msg.c_str(), msg.size());
+      if(ret < 0)
+	warnlog("Error writing carbon data to %s: %s", localCarbon->server.toStringWithPort(), strerror(errno));
+      if(ret==0)
+	warnlog("EOF writing carbon data to %s", localCarbon->server.toStringWithPort());
+    }
+    catch(std::exception& e) {
+      warnlog("Problem sending carbon data: %s", e.what());
+    }
+  }
+  return 0;
+}
+catch(std::exception& e)
+{
+  errlog("Carbon thread died: %s", e.what());
+  return 0;
+}
+catch(PDNSException& e)
+{
+  errlog("Carbon thread died, PDNSException: %s", e.reason);
+  return 0;
+}
+catch(...)
+{
+  errlog("Carbon thread died");
+  return 0;
+}