Chore: Convert background service registry to dskit module (#64062)

* Chore: Add initial support for deployment modes * revert CLI changes and start modules independently * add modules to codeowners * additional comments * add Engine and Manager interface to fix test issues * convert background service registry to dskit module * remove extra context from serviceListener logger Co-authored-by: Will Browne <wbrowne@users.noreply.github.com> * Remove whitespace * fix import * undo ide changes * only register All by default * with registry * add test * add comments * re-add debug log * fix import * reorganize arg * undo kind changes * add provide service test * fix import * rejig systemd calls * update codeowners --------- Co-authored-by: Todd Treece <todd.treece@grafana.com> Co-authored-by: Todd Treece <360020+toddtreece@users.noreply.github.com>
2023-07-06 14:45:47 +02:00
parent 76974009d0
commit 4818568c65
15 changed files with 287 additions and 190 deletions
@@ -0,0 +1,43 @@
+package systemd
+
+import (
+	"net"
+	"os"
+
+	"github.com/grafana/grafana/pkg/infra/log"
+)
+
+// NotifyReady sends READY state notifications to systemd.
+func NotifyReady(log log.Logger) {
+	notify(log, "READY=1")
+}
+
+// notify sends state notifications to systemd.
+func notify(log log.Logger, state string) {
+	notifySocket := os.Getenv("NOTIFY_SOCKET")
+	if notifySocket == "" {
+		log.Debug(
+			"NOTIFY_SOCKET environment variable empty or unset, can't send systemd notification")
+		return
+	}
+
+	socketAddr := &net.UnixAddr{
+		Name: notifySocket,
+		Net:  "unixgram",
+	}
+	conn, err := net.DialUnix(socketAddr.Net, nil, socketAddr)
+	if err != nil {
+		log.Warn("Failed to connect to systemd", "err", err, "socket", notifySocket)
+		return
+	}
+	defer func() {
+		if err = conn.Close(); err != nil {
+			log.Warn("Failed to close connection", "err", err)
+		}
+	}()
+
+	_, err = conn.Write([]byte(state))
+	if err != nil {
+		log.Warn("Failed to write notification to systemd", "err", err)
+	}
+}