Fix initial start of etcd only nodes (k3s-io#3748)

Signed-off-by: Brad Davidson <[email protected]> Signed-off-by: galal-hussein <[email protected]> (cherry picked from commit 2069cdf)
brandond · Apr 15, 2022 · bff7930 · bff7930
1 parent cda3cc3
commit bff7930
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/cli/server/server.go b/pkg/cli/server/server.go
@@ -418,6 +418,8 @@ func run(app *cli.Context, cfg *cmds.Server, leaderControllers server.CustomCont
 
 	logrus.Info("Starting " + version.Program + " " + app.App.Version)
 
+	notifySocket := os.Getenv("NOTIFY_SOCKET")
+
 	ctx := signals.SetupSignalHandler(context.Background())
 
 	if err := server.StartServer(ctx, &serverConfig, cfg); err != nil {
@@ -434,7 +436,8 @@ func run(app *cli.Context, cfg *cmds.Server, leaderControllers server.CustomCont
 		}
 
 		logrus.Info(version.Program + " is up and running")
-		if (cfg.DisableAgent || cfg.DisableAPIServer) && os.Getenv("NOTIFY_SOCKET") != "" {
+		if (cfg.DisableAgent || cfg.DisableAPIServer) && notifySocket != "" {
+			os.Setenv("NOTIFY_SOCKET", notifySocket)
 			systemd.SdNotify(true, "READY=1\n")
 		}
 	}()