Quellcode durchsuchen

move systemd ready notification to server.go

bergquist vor 8 Jahren
Ursprung
Commit
7a497fd617
2 geänderte Dateien mit 28 neuen und 36 gelöschten Zeilen
  1. 28 2
      pkg/cmd/grafana-server/server.go
  2. 0 34
      pkg/util/sdnotify.go

+ 28 - 2
pkg/cmd/grafana-server/server.go

@@ -3,7 +3,9 @@ package main
 import (
 import (
 	"context"
 	"context"
 	"flag"
 	"flag"
+	"fmt"
 	"io/ioutil"
 	"io/ioutil"
+	"net"
 	"os"
 	"os"
 	"path/filepath"
 	"path/filepath"
 	"strconv"
 	"strconv"
@@ -29,7 +31,6 @@ import (
 
 
 	"github.com/grafana/grafana/pkg/social"
 	"github.com/grafana/grafana/pkg/social"
 	"github.com/grafana/grafana/pkg/tracing"
 	"github.com/grafana/grafana/pkg/tracing"
-	"github.com/grafana/grafana/pkg/util"
 )
 )
 
 
 func NewGrafanaServer() models.GrafanaServer {
 func NewGrafanaServer() models.GrafanaServer {
@@ -97,7 +98,7 @@ func (g *GrafanaServerImpl) Start() {
 		return
 		return
 	}
 	}
 
 
-	util.SdNotify("READY=1")
+	SendSystemdReady("READY=1")
 	g.startHttpServer()
 	g.startHttpServer()
 }
 }
 
 
@@ -171,3 +172,28 @@ func (g *GrafanaServerImpl) writePIDFile() {
 
 
 	g.log.Info("Writing PID file", "path", *pidFile, "pid", pid)
 	g.log.Info("Writing PID file", "path", *pidFile, "pid", pid)
 }
 }
+
+func SendSystemdReady(state string) error {
+	notifySocket := os.Getenv("NOTIFY_SOCKET")
+
+	if notifySocket == "" {
+		return fmt.Errorf("NOTIFY_SOCKET environment variable empty or unset.")
+	}
+
+	socketAddr := &net.UnixAddr{
+		Name: notifySocket,
+		Net:  "unixgram",
+	}
+
+	conn, err := net.DialUnix(socketAddr.Net, nil, socketAddr)
+
+	if err != nil {
+		return err
+	}
+
+	_, err = conn.Write([]byte(state))
+
+	conn.Close()
+
+	return err
+}

+ 0 - 34
pkg/util/sdnotify.go

@@ -1,34 +0,0 @@
-package util
-
-import (
-	"errors"
-	"net"
-	"os"
-)
-
-var NoNotifySocket = errors.New("NOTIFY_SOCKET environment variable empty or unset.")
-
-func SdNotify(state string) error {
-	notifySocket := os.Getenv("NOTIFY_SOCKET")
-
-	if notifySocket == "" {
-		return NoNotifySocket
-	}
-
-	socketAddr := &net.UnixAddr{
-		Name: notifySocket,
-		Net:  "unixgram",
-	}
-
-	conn, err := net.DialUnix(socketAddr.Net, nil, socketAddr)
-
-	if err != nil {
-		return err
-	}
-
-	_, err = conn.Write([]byte(state))
-
-	conn.Close()
-
-	return err
-}