use goroutine to record cpu usage

main.go

@@ -1,71 +1,72 @@
 package main
 
 import (
-    "context"
-    "flag"
-    "github.com/0xJacky/Nginx-UI/server/model"
-    "github.com/0xJacky/Nginx-UI/server/router"
-    "github.com/0xJacky/Nginx-UI/server/settings"
-    tool2 "github.com/0xJacky/Nginx-UI/server/tool"
-    "github.com/gin-gonic/gin"
-    "log"
-    "mime"
-    "net/http"
-    "os/signal"
-    "syscall"
-    "time"
+	"context"
+	"flag"
+	"github.com/0xJacky/Nginx-UI/server/model"
+	"github.com/0xJacky/Nginx-UI/server/router"
+	"github.com/0xJacky/Nginx-UI/server/settings"
+	"github.com/0xJacky/Nginx-UI/server/tool"
+	"github.com/gin-gonic/gin"
+	"log"
+	"mime"
+	"net/http"
+	"os/signal"
+	"syscall"
+	"time"
 )
 
 func main() {
-    // Create context that listens for the interrupt signal from the OS.
-    ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
-    defer stop()
+	// Create context that listens for the interrupt signal from the OS.
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
 
-    // Hack: fix wrong Content Type of .js file on some OS platforms
-    // See https://github.com/golang/go/issues/32350
-    _ = mime.AddExtensionType(".js", "text/javascript; charset=utf-8")
+	// Hack: fix wrong Content Type of .js file on some OS platforms
+	// See https://github.com/golang/go/issues/32350
+	_ = mime.AddExtensionType(".js", "text/javascript; charset=utf-8")
 
-    var confPath string
-    flag.StringVar(&confPath, "config", "app.ini", "Specify the configuration file")
-    flag.Parse()
+	var confPath string
+	flag.StringVar(&confPath, "config", "app.ini", "Specify the configuration file")
+	flag.Parse()
 
-    gin.SetMode(settings.ServerSettings.RunMode)
-    
-    settings.Init(confPath)
-    log.Printf("nginx config dir path: %s", tool2.GetNginxConfPath(""))
-    if "" != settings.ServerSettings.JwtSecret {
-        model.Init()
-        go tool2.AutoCert()
-    }
+	gin.SetMode(settings.ServerSettings.RunMode)
 
-    srv := &http.Server{
-        Addr:    ":" + settings.ServerSettings.HttpPort,
-        Handler: router.InitRouter(),
-    }
+	settings.Init(confPath)
+	log.Printf("nginx config dir path: %s", tool.GetNginxConfPath(""))
+	if "" != settings.ServerSettings.JwtSecret {
+		model.Init()
+		go tool.AutoCert()
+		go tool.RecordCpuUsage()
+	}
 
-    // Initializing the server in a goroutine so that
-    // it won't block the graceful shutdown handling below
-    go func() {
-        if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
-            log.Fatalf("listen: %s\n", err)
-        }
-    }()
+	srv := &http.Server{
+		Addr:    ":" + settings.ServerSettings.HttpPort,
+		Handler: router.InitRouter(),
+	}
 
-    // Listen for the interrupt signal.
-    <-ctx.Done()
+	// Initializing the server in a goroutine so that
+	// it won't block the graceful shutdown handling below
+	go func() {
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			log.Fatalf("listen: %s\n", err)
+		}
+	}()
 
-    // Restore default behavior on the interrupt signal and notify user of shutdown.
-    stop()
-    log.Println("shutting down gracefully, press Ctrl+C again to force")
+	// Listen for the interrupt signal.
+	<-ctx.Done()
 
-    // The context is used to inform the server it has 5 seconds to finish
-    // the request it is currently handling
-    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-    defer cancel()
-    if err := srv.Shutdown(ctx); err != nil {
-        log.Fatal("Server forced to shutdown: ", err)
-    }
+	// Restore default behavior on the interrupt signal and notify user of shutdown.
+	stop()
+	log.Println("shutting down gracefully, press Ctrl+C again to force")
 
-    log.Println("Server exiting")
+	// The context is used to inform the server it has 5 seconds to finish
+	// the request it is currently handling
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(ctx); err != nil {
+		log.Fatal("Server forced to shutdown: ", err)
+	}
+
+	log.Println("Server exiting")
 
 }