unix stats plugin

Makefile.am

@@ -7,7 +7,8 @@ SUBDIRS = \
     . \
     src/evcollect \
     plugins/eventql \
-    plugins/hostname
+    plugins/hostname \
+    plugins/unix_stats
 
 EXTRA_DIST =                             \
   doc                                    \

configure.ac

@@ -64,5 +64,5 @@ AM_CONDITIONAL([HAVE_ZLIB], [test $HAVE_ZLIB = 1])
 ACX_PTHREAD
 AM_CONDITIONAL([HAVE_PTHREAD], [test "x$acx_pthread_ok" = "xyes"])
 
-AC_CONFIG_FILES([Makefile src/evcollect/Makefile plugins/hostname/Makefile plugins/eventql/Makefile])
+AC_CONFIG_FILES([Makefile src/evcollect/Makefile plugins/hostname/Makefile plugins/eventql/Makefile plugins/unix_stats/Makefile])
 AC_OUTPUT

plugins/unix_stats/Makefile.am

@@ -0,0 +1,32 @@
+MAINTAINERCLEANFILES = Makefile.in
+
+AM_CXXFLAGS = -std=c++0x -Wall -Wextra -Wdelete-non-virtual-dtor -g -fvisibility=hidden -I$(top_srcdir)/src
+AM_CFLAGS = -std=c11 -Wall -pedantic -g
+AM_LDFLAGS = -fvisibility=hidden -module -avoid-version -shared -export-dynamic -static-libstdc++ -static-libgcc -rpath $(libdir)
+
+noinst_LTLIBRARIES = plugin_unix_stats.la
+
+plugin_unix_stats_la_SOURCES = \
+    util/stringutil.cc \
+    util/stringutil.h \
+    util/stringutil_impl.h \
+    util/time.cc \
+    util/time.h \
+    util/time_impl.h \
+    disk_stats.h \
+    disk_stats.cc \
+    kernel_stats.h \
+    kernel_stats.cc \
+    process_stats.h \
+    process_stats.cc \
+    unix_stats_plugin.cc
+
+PLUGINDIR=$(DESTDIR)$(libdir)/evcollect/plugins
+
+install-data-hook: $(noinst_LTLIBRARIES)
+	@for soname in `echo | $(EGREP) "^dlname=" $^ | $(SED) -e "s|^dlname='\(.*\)'|\1|"`; do  \
+		mkdir -p ${PLUGINDIR};                                                                 \
+		echo Installing $$soname to ${PLUGINDIR};                                              \
+		cp $(abs_builddir)/.libs/$$soname ${PLUGINDIR};                                        \
+	done
+

plugins/unix_stats/disk_stats.cc

@@ -0,0 +1,135 @@
+/**
+ * Copyright (c) 2016 DeepCortex GmbH <legal@eventql.io>
+ * Authors:
+ *   - Paul Asmuth <paul@eventql.io>
+ *   - Laura Schlimmer <laura@eventql.io>
+ *
+ * This program is free software: you can redistribute it and/or modify it under
+ * the terms of the GNU Affero General Public License ("the license") as
+ * published by the Free Software Foundation, either version 3 of the License,
+ * or any later version.
+ *
+ * In accordance with Section 7(e) of the license, the licensing of the Program
+ * under the license does not imply a trademark license. Therefore any rights,
+ * title and interest in our trademarks remain entirely with us.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the license for more details.
+ *
+ * You can be released from the requirements of the license by purchasing a
+ * commercial license. Buying such a license is mandatory as soon as you develop
+ * commercial activities involving this program without disclosing the source
+ * code of your own applications
+ */
+
+#include <sys/statvfs.h>
+#include <unistd.h>
+#include <stdio.h>
+#include "util/stringutil.h"
+#include "disk_stats.h"
+
+#if __linux__
+#include <mntent.h>
+#endif
+
+#if __APPLE__
+#include <sys/mount.h>
+#endif
+
+namespace evcollect {
+namespace plugin_unix_stats {
+
+std::vector<MountInfo> getMountInfo() {
+  std::vector<MountInfo> mount_info;
+
+#if __linux__
+
+  auto file = setmntent(MOUNTED, "r");
+  while (auto mntent = getmntent(file)) {
+    MountInfo mn_info = {
+      .device = mntent->mnt_fsname,
+      .mount_point = mntent->mnt_dir
+    };
+
+    mount_info.emplace_back(mn_info);
+  }
+  fclose(file);
+
+#elif __APPLE__
+  struct statfs* mntbuf;
+  auto mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
+  for (auto i = 0; i < mntsize; ++i) {
+    MountInfo mn_info = {
+      .device = mntbuf[i].f_mntfromname,
+      .mount_point = mntbuf[i].f_mntonname
+    };
+
+    mount_info.emplace_back(mn_info);
+  }
+
+#endif
+
+  return mount_info;
+}
+
+
+bool getDiskInfo(std::vector<DiskInfo> disk_info) {
+  auto mount_info = getMountInfo();
+
+  for (size_t i = 0; i < mount_info.size(); ++i) {
+    struct statvfs buf;
+    if (statvfs(mount_info[i].mount_point.c_str(), &buf) == -1) {
+      continue;
+    }
+
+    DiskInfo di;
+    di.total = buf.f_blocks * buf.f_frsize;
+    di.available = buf.f_bavail * buf.f_frsize;
+    di.used = di.total - di.available;
+    di.capacity = di.total > 0 ? (di.used / di.total) * 100 : 100;
+    di.ifree = buf.f_favail;
+    di.iused = buf.f_files - di.ifree;
+
+    disk_info.emplace_back(di);
+  }
+
+  return true;
+}
+
+std::string toJSON(std::vector<DiskInfo> disk_info) {
+  std::string json;
+  json.append(R"({disk: )");
+
+  for (size_t i = 0; i < disk_info.size(); ++i) {
+    if (i > 0) {
+      json.append(",");
+    }
+
+    json.append(StringUtil::format(R"([
+        "filesystem": $0,
+        "mount_point": $1,
+        "total": $2,
+        "available": $3,
+        "used": $4,
+        "capacity": $5,
+        "ifree": $6,
+        "iused": $7
+      ])",
+      disk_info[i].filesystem,
+      disk_info[i].mount_point,
+      disk_info[i].total,
+      disk_info[i].available,
+      disk_info[i].used,
+      disk_info[i].capacity,
+      disk_info[i].ifree,
+      disk_info[i].used
+    ));
+  }
+
+  json.append("}");
+  return json;
+}
+
+} //namespace plugin_unix_stats
+} //namespace evcollect

plugins/unix_stats/disk_stats.h

@@ -0,0 +1,53 @@
+/**
+ * Copyright (c) 2016 DeepCortex GmbH <legal@eventql.io>
+ * Authors:
+ *   - Paul Asmuth <paul@eventql.io>
+ *   - Laura Schlimmer <laura@eventql.io>
+ *
+ * This program is free software: you can redistribute it and/or modify it under
+ * the terms of the GNU Affero General Public License ("the license") as
+ * published by the Free Software Foundation, either version 3 of the License,
+ * or any later version.
+ *
+ * In accordance with Section 7(e) of the license, the licensing of the Program
+ * under the license does not imply a trademark license. Therefore any rights,
+ * title and interest in our trademarks remain entirely with us.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the license for more details.
+ *
+ * You can be released from the requirements of the license by purchasing a
+ * commercial license. Buying such a license is mandatory as soon as you develop
+ * commercial activities involving this program without disclosing the source
+ * code of your own applications
+ */
+
+#include <vector>
+
+namespace evcollect {
+namespace plugin_unix_stats {
+
+struct MountInfo {
+  std::string device;
+  std::string mount_point;
+};
+
+struct DiskInfo {
+  std::string filesystem;
+  std::string mount_point;
+  uint64_t total;
+  uint64_t used;
+  uint64_t available;
+  uint8_t capacity;
+  uint64_t iused;
+  uint64_t ifree;
+};
+
+
+bool getDiskInfo(std::vector<DiskInfo> disk_info);
+
+std::string toJSON(std::vector<DiskInfo> disk_info);
+
+} //namespace plugin_unix_stats
+} //namespace evcollect