add the actual carbon file

diff --git a/pdns/dnsdist-carbon.cc b/pdns/dnsdist-carbon.cc
new file mode 100644 (file)
index 0000000..f94cd38
--- /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;
+}