nixos/module: add metrics options

marijanp · Sep 19, 2024 · 8f0138e · 8f0138e
1 parent 170ddcc
commit 8f0138e
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 5 deletions.
diff --git a/nixos/default.nix b/nixos/default.nix
@@ -12,6 +12,7 @@
               self.nixosModules.server
             ];
             services.server.enable = true;
+            services.server.metrics.enable = true;
           };
           verifyServices = [ "server.service" ];
         };

diff --git a/nixos/modules/server.nix b/nixos/modules/server.nix
@@ -4,9 +4,9 @@ let
     types
     mkOption
     mkIf
-    mkMerge
     mkEnableOption
     escapeShellArgs
+    optionals
     ;
   cfg = config.services.server;
 in
@@ -37,6 +37,27 @@ in
       '';
     };
 
+    metrics = {
+      enable = lib.mkEnableOption "Prometheus metrics server";
+
+      address = mkOption {
+        type = types.str;
+        default = "0.0.0.0";
+        example = "0.0.0.0";
+        description = ''
+          Listen address of the metrics server.
+        '';
+      };
+
+      port = mkOption {
+        type = types.port;
+        default = 8081;
+        description = ''
+          Listen port of the metrics service.
+        '';
+      };
+    };
+
     logLevel = mkOption {
       type = types.str;
       default = "info";
@@ -50,10 +71,16 @@ in
 
     systemd.services.server =
       let
-        args = escapeShellArgs [
-          "--listen-address"
-          "${cfg.address}:${toString cfg.port}"
-        ];
+        args = escapeShellArgs (
+          [
+            "--listen-address"
+            "${cfg.address}:${toString cfg.port}"
+          ]
+          ++ optionals cfg.metrics.enable [
+            "--metrics-listen-address"
+            "${cfg.metrics.address}:${toString cfg.metrics.port}"
+          ]
+        );
       in
       {
         description = "server";