[metrics] Add metrics-specific varlink sockets in /run/systemd/metrics

Author: YapingLi04

src/core/manager.h

@@ -463,6 +463,8 @@ typedef struct Manager {
          * systemd-oomd to report changes in ManagedOOM settings (systemd client - oomd server). */
         sd_varlink *managed_oom_varlink;
 
+        sd_varlink_server *metrics_varlink_server;
+
         /* Reference to RestrictFileSystems= BPF program */
         struct restrict_fs_bpf *restrict_fs;
 

src/core/varlink.c

@@ -6,6 +6,8 @@
 #include "errno-util.h"
 #include "json-util.h"
 #include "manager.h"
+#include "mkdir-label.h"
+#include "metrics.h"
 #include "path-util.h"
 #include "pidref.h"
 #include "string-util.h"
@@ -402,6 +404,22 @@ int manager_setup_varlink_server(Manager *m) {
         return 1;
 }
 
+static int vl_method_list(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
+        Manager *m = ASSERT_PTR(userdata);
+        int r;
+
+        assert(link);
+        assert(m->units);
+
+        r = sd_varlink_dispatch(link, parameters, /* dispatch_table= */ NULL, /* userdata= */ NULL);
+        if (r != 0)
+                return r;
+
+        return sd_varlink_replybo(
+                link,
+                SD_JSON_BUILD_PAIR("units.active_units", SD_JSON_BUILD_INTEGER(hashmap_size(m->units))));
+}
+
 static int manager_varlink_init_system(Manager *m) {
         int r;
 
@@ -430,6 +448,11 @@ static int manager_varlink_init_system(Manager *m) {
                 }
         }
 
+        (void) mkdir_label("/run/systemd/metrics", 0755);
+        r = metrics_setup_varlink_server(&m->metrics_varlink_server, m->event, vl_method_list, m, "/run/systemd/metrics/io.systemd.Manager");
+        if (r < 0)
+                return log_error_errno(r, "Failed to set up metrics varlink server: %m");
+
         return 1;
 }
 
@@ -464,6 +487,7 @@ static int manager_varlink_init_user(Manager *m) {
                         return log_error_errno(r, "Failed to bind to varlink socket '%s': %m", address);
         }
 
+        // TODO: Make metrics server work for user manager
         return manager_varlink_managed_oom_connect(m);
 }
 
@@ -482,4 +506,5 @@ void manager_varlink_done(Manager *m) {
 
         m->varlink_server = sd_varlink_server_unref(m->varlink_server);
         m->managed_oom_varlink = sd_varlink_close_unref(m->managed_oom_varlink);
+        m->metrics_varlink_server = sd_varlink_server_unref(m->metrics_varlink_server);
 }

src/shared/meson.build

@@ -125,6 +125,7 @@ shared_sources = files(
         'machine-pool.c',
         'macvlan-util.c',
         'main-func.c',
+        'metrics.c',
         'mkdir-label.c',
         'mkfs-util.c',
         'module-util.c',

src/shared/metrics.c

@@ -0,0 +1,33 @@
+#include "log.h"
+#include "metrics.h"
+#include "varlink-io.systemd.Metrics.h"
+#include "varlink-util.h"
+
+int metrics_setup_varlink_server(sd_varlink_server **m, sd_event *event, sd_varlink_method_t vl_method_list_cb, void *userdata, const char *socket) {
+        _cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
+        int r;
+
+        r = varlink_server_new(&s, SD_VARLINK_SERVER_INHERIT_USERDATA, userdata);
+        if (r < 0)
+                return log_error_errno(r, "Failed to allocate varlink server object: %m");
+
+        r = sd_varlink_server_add_interface(s, &vl_interface_io_systemd_Metrics);
+        if (r < 0)
+                return log_error_errno(r, "Failed to add Metrics interface to varlink server: %m");
+
+        r = sd_varlink_server_bind_method(
+                        s, "io.systemd.Metrics.List", vl_method_list_cb);
+        if (r < 0)
+                return r;
+
+        r = sd_varlink_server_listen_address(s, socket, 0600);
+        if (r < 0)
+                return r;
+
+        r = sd_varlink_server_attach_event(s, event, SD_EVENT_PRIORITY_NORMAL);
+        if (r < 0)
+                return r;
+
+        *m = TAKE_PTR(s);
+        return 0;
+}

src/shared/metrics.h

@@ -0,0 +1,5 @@
+#pragma once
+
+#include "sd-varlink.h"
+
+int metrics_setup_varlink_server(sd_varlink_server **m, sd_event *event, sd_varlink_method_t vl_method_list_cb, void *userdata, const char *socket);