fix: systemd always restart service even though success running

Author: mkitsdts

deploy/systemd/mkblog.service.tpl

@@ -1,14 +1,16 @@
 [Unit]
 Description=__APP_NAME__ blog service
 After=network.target
+StartLimitIntervalSec=60
+StartLimitBurst=5
 
 [Service]
 Type=simple
 User=__USER__
 Group=__GROUP__
 WorkingDirectory=__ROOT_DIR__
 ExecStart=__BIN_PATH__
-Restart=always
+Restart=on-failure
 RestartSec=3
 
 [Install]

main.go

@@ -1,48 +1,54 @@
-package main
-
-import (
-	"flag"
-	"mkBlog/config"
-	"mkBlog/models"
-	"mkBlog/pkg/bloom"
-	"mkBlog/pkg/cache"
-	"mkBlog/pkg/log"
-	"mkBlog/pkg/middleware"
-	"mkBlog/pkg/router"
-	"mkBlog/service"
-	"os"
-)
-
-func Init(debugflag *bool) {
-	if err := os.MkdirAll(models.Default_Data_Path, 0755); err != nil {
-		return
-	}
-	if debugflag != nil {
-		log.Init(*debugflag)
-	} else {
-		log.Init(false)
-	}
-
-	config.Init()
-
-	bloom.Init()
-	cache.Init(models.Default_Static_File_Path)
-	middleware.Init()
-}
-
-func main() {
-	debug := flag.Bool("debug", false, "启用调试模式")
-
-	// 解析命令行参数
-	flag.Parse()
-	Init(debug)
-
-	if err := router.InitRouter(); err != nil {
-		panic("failed to create router: " + err.Error())
-	}
-
-	if s, err := service.NewBlogService(); err == nil {
-		s.Start()
-	}
-
-}
+package main
+
+import (
+	"flag"
+	"log/slog"
+	"mkBlog/config"
+	"mkBlog/models"
+	"mkBlog/pkg/bloom"
+	"mkBlog/pkg/cache"
+	applog "mkBlog/pkg/log"
+	"mkBlog/pkg/middleware"
+	"mkBlog/pkg/router"
+	"mkBlog/service"
+	"os"
+)
+
+func Init(debugflag *bool) {
+	if err := os.MkdirAll(models.Default_Data_Path, 0755); err != nil {
+		return
+	}
+	if debugflag != nil {
+		applog.Init(*debugflag)
+	} else {
+		applog.Init(false)
+	}
+
+	config.Init()
+
+	bloom.Init()
+	cache.Init(models.Default_Static_File_Path)
+	middleware.Init()
+}
+
+func main() {
+	debug := flag.Bool("debug", false, "启用调试模式")
+
+	// 解析命令行参数
+	flag.Parse()
+	Init(debug)
+
+	if err := router.InitRouter(); err != nil {
+		panic("failed to create router: " + err.Error())
+	}
+
+	s, err := service.NewBlogService()
+	if err != nil {
+		slog.Error("failed to initialize blog service", "error", err)
+		os.Exit(1)
+	}
+	if err := s.Start(); err != nil {
+		slog.Error("blog service stopped", "error", err)
+		os.Exit(1)
+	}
+}

service/service.go

@@ -2,6 +2,7 @@ package service
 
 import (
 	"crypto/tls"
+	"errors"
 	"fmt"
 	"log"
 	"log/slog"
@@ -10,6 +11,7 @@ import (
 	"mkBlog/pkg/router"
 	tlscert "mkBlog/pkg/tls_cert"
 	"mkBlog/service/api"
+	"net"
 	"net/http"
 	_ "net/http/pprof"
 	"os"
@@ -76,17 +78,22 @@ func NewBlogService() (*BlogService, error) {
 	return &service, nil
 }
 
-func (s *BlogService) Start() {
+func (s *BlogService) Start() error {
 	if config.Cfg.Server.Devmode {
 		go func() {
 			log.Println(http.ListenAndServe(":6060", nil))
 		}()
 	}
+	addr := ":" + fmt.Sprint(config.Cfg.Server.Port)
 	if config.Cfg.Server.HTTP3Enabled {
 		tlscert.LoadCert()
+		conn, err := net.ListenPacket("udp", addr)
+		if err != nil {
+			return fmt.Errorf("start HTTP3 server: %w", err)
+		}
 		srv := http3.Server{
 			Handler: router.GetRouter(),
-			Addr:    ":" + fmt.Sprint(config.Cfg.Server.Port),
+			Addr:    addr,
 			TLSConfig: http3.ConfigureTLSConfig(&tls.Config{
 				MinVersion:     tls.VersionTLS13,
 				GetCertificate: tlscert.GetCurrentCert,
@@ -95,7 +102,7 @@ func (s *BlogService) Start() {
 		}
 		slog.Info("starting HTTP3 server", "port", config.Cfg.Server.Port)
 		go func() {
-			if err := srv.ListenAndServe(); err != nil {
+			if err := srv.Serve(conn); err != nil && !errors.Is(err, http.ErrServerClosed) {
 				slog.Error("failed to start HTTP3 server", "error", err)
 			}
 		}()
@@ -108,20 +115,23 @@ func (s *BlogService) Start() {
 	if config.Cfg.TLS.Enabled {
 		tlscert.LoadCert()
 		srv := &http.Server{
-			Addr:    ":" + fmt.Sprint(config.Cfg.Server.Port),
+			Addr:    addr,
 			Handler: router.GetRouter(),
 			TLSConfig: &tls.Config{
 				MinVersion:     tls.VersionTLS12,
 				GetCertificate: tlscert.GetCurrentCert,
 			},
 		}
 		// Start HTTPS server
-		if err := srv.ListenAndServeTLS("", ""); err != nil {
+		if err := srv.ListenAndServeTLS("", ""); err != nil && !errors.Is(err, http.ErrServerClosed) {
 			slog.Error("failed to start HTTPS server", "error", err)
+			return fmt.Errorf("start HTTPS server: %w", err)
 		}
 	} else {
-		if err := router.GetRouter().Run(":" + fmt.Sprint(config.Cfg.Server.Port)); err != nil {
+		if err := router.GetRouter().Run(addr); err != nil {
 			slog.Error("failed to start HTTP server", "error", err)
+			return fmt.Errorf("start HTTP server: %w", err)
 		}
 	}
+	return nil
 }