[feature] #19 增加pprof

EOEFANS · Feb 25, 2023 · ba0680e · ba0680e
1 parent 6cd959b
commit ba0680e
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/cmd/api/main.go b/cmd/api/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"go.uber.org/zap"
 
 	"git.vtb.link/eoefans/internal/app"
 	"git.vtb.link/eoefans/internal/app/api/service"
@@ -24,7 +25,12 @@ func newAPI() fx.Option {
 	)
 }
 
-func lc(lifecycle fx.Lifecycle, ginServer *httpserver.Server, randPicsCache *service.RandomPicsCache) {
+func lc(
+	lifecycle fx.Lifecycle,
+	ginServer *httpserver.Server,
+	randPicsCache *service.RandomPicsCache,
+	logger *zap.Logger,
+) {
 	lifecycle.Append(fx.Hook{
 		OnStart: func(ctx context.Context) error {
 			return ginServer.Start()
@@ -33,6 +39,11 @@ func lc(lifecycle fx.Lifecycle, ginServer *httpserver.Server, randPicsCache *ser
 			return ginServer.Stop()
 		},
 	})
+	lifecycle.Append(fx.Hook{
+		OnStart: func(ctx context.Context) error {
+			return httpserver.RunPprofHttpServer(logger)
+		},
+	})
 	lifecycle.Append(fx.Hook{
 		OnStart: func(ctx context.Context) error {
 			return randPicsCache.Run()

diff --git a/internal/pkg/httpserver/pprof.go b/internal/pkg/httpserver/pprof.go
@@ -0,0 +1,18 @@
+package httpserver
+
+import (
+	"go.uber.org/zap"
+	"net/http"
+	_ "net/http/pprof"
+)
+
+func RunPprofHttpServer(logger *zap.Logger) error {
+	go func() {
+		err := http.ListenAndServe(":9651", nil)
+		if err != nil {
+			logger.Error("RunPprofHttpServer failed: ", zap.Error(err))
+		}
+	}()
+
+	return nil
+}