init

Signed-off-by: jason yang <jasonyangshadow@gmail.com>

internal/pkg/metric/pushgateway.go

@@ -0,0 +1,29 @@
+package metric
+
+import (
+	"net"
+	"path/filepath"
+
+	"github.com/apptainer/apptainer/pkg/syfs"
+	"github.com/apptainer/apptainer/pkg/util/apptainerconf"
+)
+
+const protocol = "unix"
+
+type Pushgateway struct {
+	net.Conn
+}
+
+func New() (*Pushgateway, error) {
+	var socketPath string
+	socketPath = apptainerconf.GetCurrentConfig().PushgatewaySocketPath
+	if socketPath == "" {
+		socketPath = filepath.Join(syfs.ConfigDir(), "pushgateway.socket")
+	}
+	conn, err := net.Dial(protocol, socketPath)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Pushgateway{conn}, nil
+}

pkg/util/apptainerconf/config.go

@@ -128,6 +128,8 @@ type File struct {
 	DownloadPartSize    uint   `default:"5242880" directive:"download part size"`
 	DownloadBufferSize  uint   `default:"32768" directive:"download buffer size"`
 	SystemdCgroups      bool   `default:"yes" authorized:"yes,no" directive:"systemd cgroups"`
+	// pushgateway unix socket
+	PushgatewaySocketPath string `directive:"pushgateway communication socket path"`
 }
 
 // NOTE: if you think that we may want to change the default for any
@@ -530,4 +532,10 @@ download buffer size = {{ .DownloadBufferSize }}
 # Whether to use systemd to manage container cgroups. Required for rootless cgroups
 # functionality. 'no' will manage cgroups directly via cgroupfs.
 systemd cgroups = {{ if eq .SystemdCgroups true }}yes{{ else }}no{{ end }}
+
+# PUSHGATEWAY SOCKET PATH: [STRING]
+# DEFAULT: Undefined
+# Defines pushgateway socket path
+#pushgateway socket path=
+{{ if ne .PushgatewaySocketPath "" }}pushgateway socket path = {{ .PushgatewaySocketPath}}{{ end }}
 `