--- /dev/null
+package org.argeo.slc.systemd.dbus;
+
+import java.util.Map;
+
+import org.freedesktop.dbus.connections.impl.DBusConnection;
+import org.freedesktop.dbus.interfaces.DBusInterface;
+import org.freedesktop.dbus.interfaces.Introspectable;
+import org.freedesktop.dbus.interfaces.Properties;
+import org.freedesktop.dbus.types.Variant;
+
+public class PlainDBusTest {
+ final static String SYSTEMD_SERVICE = "org.freedesktop.systemd1.Service";
+
+ public static void main(String[] args) throws Exception {
+// try (DBusConnection dBusConnection = DBusConnectionBuilder.forSystemBus().build()) {
+
+ try (DBusConnection dBusConnection = DBusConnection.getConnection(DBusConnection.DBusBusType.SYSTEM)) {
+
+ String source = "org.freedesktop.systemd1";
+ String objectPath = "/org/freedesktop/systemd1/unit/ipsec_2eservice";
+// String objectPath = "/org/freedesktop/systemd1";
+ DBusInterface object = dBusConnection.getExportedObject(source, objectPath);
+ System.out.println(object);
+//
+// Introspectable introspectable = dBusConnection.getExportedObject(source, objectPath, Introspectable.class);
+// System.out.println(introspectable.Introspect());
+//
+ Properties props = dBusConnection.getExportedObject(source, objectPath, Properties.class);
+// System.out.println(props);
+
+ System.out.println(props.Get(SYSTEMD_SERVICE, "CPUUsageNSec").toString());
+
+ Map<String, Variant<?>> values = props.GetAll(SYSTEMD_SERVICE);
+ for (String key : values.keySet()) {
+ Variant<?> value = values.get(key);
+ System.out.println(key + "=" + value.getValue() + " (" + value.getType() + ") " + value.getSig());
+ }
+ }
+ }
+
+}
--- /dev/null
+package org.argeo.slc.systemd.dbus;
+
+import static java.lang.System.Logger.Level.DEBUG;
+import static java.lang.System.Logger.Level.INFO;
+import static java.lang.System.Logger.Level.WARNING;
+
+import java.io.IOException;
+import java.io.Writer;
+import java.lang.System.Logger;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardOpenOption;
+import java.time.Instant;
+import java.time.ZoneOffset;
+import java.time.format.DateTimeFormatter;
+
+import org.argeo.cms.util.CsvWriter;
+import org.freedesktop.dbus.exceptions.DBusException;
+
+import de.thjom.java.systemd.Manager;
+import de.thjom.java.systemd.Service;
+import de.thjom.java.systemd.Systemd;
+import de.thjom.java.systemd.types.UnitType;
+
+/** Gathers statistics about the runnign process, if it is a systemd unit. */
+public class ServiceStatistics {
+ private final static Logger logger = System.getLogger(ServiceStatistics.class.getName());
+
+ private Manager manager;
+
+ private String unitName;
+ private Service service;
+
+ private long frequency = 60 * 1000;
+
+ public void start() {
+ final long pid = ProcessHandle.current().pid();
+ try {
+ manager = Systemd.get().getManager();
+ // find own service
+ for (UnitType unitType : manager.listUnits()) {
+ if (unitType.isService()) {
+ Service s = manager.getService(unitType.getUnitName());
+ if (s.getMainPID() == pid) {
+ service = s;
+ unitName = unitType.getUnitName();
+ logger.log(INFO, "Systemd service found for pid " + pid + ": " + unitName);
+ }
+ }
+ }
+
+ } catch (DBusException e) {
+ logger.log(WARNING, "Cannot connect to systemd", e);
+ }
+
+ if (service == null) {
+ logger.log(DEBUG, () -> "No systemd service found for pid " + pid + ", disconnecting from DBus...");
+ manager = null;
+ Systemd.disconnect();
+ } else {
+ // start collecting
+ collectStatistics();
+ }
+ }
+
+ public void stop() {
+ if (manager != null) {
+ Systemd.disconnect();
+ manager = null;
+ notifyAll();
+ }
+ }
+
+ protected void collectStatistics() {
+ // standard systemd property
+ String logsDirectory = System.getenv("LOGS_DIRECTORY");
+ if (logsDirectory == null) {
+ logsDirectory = System.getProperty("user.dir");
+ }
+ // MainPID,CPUUsageNSec,MemoryCurrent,IPIngressBytes,IPEgressBytes,IOReadBytes,IOWriteBytes,TasksCurrent
+ Path basePath = Paths.get(logsDirectory);
+
+ logger.log(DEBUG, () -> "Writing statistics for " + unitName + " to " + basePath);
+ try {
+ while (manager != null) {
+ String dateSuffix = Instant.now().atOffset(ZoneOffset.UTC).format(DateTimeFormatter.ISO_LOCAL_DATE);
+
+ Path csvPath = basePath.resolve("statistics-" + unitName + "-" + dateSuffix + ".csv");
+ boolean writeHeader = !Files.exists(csvPath);
+ try (Writer writer = Files.newBufferedWriter(csvPath, StandardCharsets.UTF_8,
+ StandardOpenOption.APPEND)) {
+ CsvWriter csvWriter = new CsvWriter(writer);
+
+ if (writeHeader)// header
+ csvWriter.writeLine("CurrentTimeMillis", "CPUUsageNSec", "MemoryCurrent", "IPIngressBytes",
+ "IPEgressBytes", "IOReadBytes", "IOWriteBytes", "TasksCurrent");
+
+ // TODO better synchronise with stop
+ csvWriter.writeLine(System.currentTimeMillis(), service.getCPUUsageNSec(),
+ service.getMemoryCurrent(), service.getIPIngressBytes(), service.getIPEgressBytes(),
+ service.getIOReadBytes(), service.getIOWriteBytes(), service.getTasksCurrent());
+ }
+ Thread.sleep(frequency);
+ }
+ } catch (IOException e) {
+ throw new IllegalStateException("Cannot collect statistics", e);
+ } catch (InterruptedException e) {
+ logger.log(WARNING, "Statistics collection interrupted for " + unitName);
+ }
+
+ }
+
+ public static void main(String[] args) throws Exception {
+ try {
+ Systemd systemd = Systemd.get();
+ Service service = systemd.getManager().getService("ipsec.service");
+ System.out.println(service.getCPUUsageNSec());
+
+ for (UnitType unitType : systemd.getManager().listUnits()) {
+ if (unitType.isService()) {
+ System.out.println(unitType.getUnitName());
+ }
+ }
+
+ } finally {
+ Systemd.disconnect();
+ }
+ }
+
+}
projects / gpl / argeo-slc.git / commitdiff