Make dcgm-exporter listen address configurable

snap/hooks/configure

@@ -3,4 +3,10 @@
 # Register config options if unset,
 # so users can see available options by running
 # `sudo snap get dcgm`.
-[ -z "$(snapctl get nv-hostengine-port)" ] && snapctl set nv-hostengine-port=
+if [ -z "$(snapctl get nv-hostengine-port)" ]; then
+    snapctl set nv-hostengine-port=""
+fi
+
+if [ -z "$(snapctl get dcgm_exporter_listen)" ]; then
+    snapctl set dcgm_exporter_listen=""
+fi

snap/local/run_dcgm_exporter.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+set -euo pipefail
+
+# Build the argument list for the dcgm-exporter command
+args=()
+
+# Add the dcgm-exporter-port option if it is set. Default: “:9400”
+dcgm_exporter_listen="$(snapctl get dcgm_exporter_listen)"
+
+if [ -n "$dcgm_exporter_listen" ]; then
+    args+=("-a" "$dcgm_exporter_listen")
+fi
+
+exec "$SNAP/bin/dcgm-exporter" "${args[@]}"

snap/snapcraft.yaml

@@ -25,14 +25,13 @@ package-repositories:
 
 apps:
   dcgm-exporter:
-    command: bin/dcgm-exporter
+    command: run_dcgm_exporter.sh
     plugs:
       - network-bind
       - opengl
     daemon: simple
     install-mode: disable
     restart-condition: on-failure
-    restart-delay: 2s
   dcgmi:
     command: usr/bin/dcgmi
     plugs:
@@ -61,6 +60,7 @@ parts:
     override-build: |
       craftctl default
       chmod +x run_nv_hostengine.sh
+      chmod +x run_dcgm_exporter.sh
   dcgm-exporter:
     plugin: go
     stage-packages: [datacenter-gpu-manager]