feat(observability): 为 API 和 worker 增加独立 pprof 诊断端口

backend/internal/config/loadconfig.go

@@ -12,6 +12,7 @@ type Config struct {
 	Database DatabaseConfig `yaml:"database"`
 	Redis    RedisConfig    `yaml:"redis"`
 	RabbitMQ RabbitMQConfig `yaml:"rabbitmq"`
+	ObservabilityConfig ObservabilityConfig `yaml:"observability"`
 }
 
 type ServerConfig struct {
@@ -40,6 +41,14 @@ type RabbitMQConfig struct {
 	Password string `yaml:"password"`
 }
 
+type ObservabilityConfig struct {
+	Pprof PprofConfig `yaml:"pprof"`
+}
+type PprofConfig struct {
+	Enabled bool `yaml:"enabled"`
+	ApiAddr string `yaml:"api_addr"`
+	WorkerAddr string `yaml:"worker_addr"`
+}
 func Load(filename string) (Config, error) {
 	data, err := os.ReadFile(filename)
 	if err != nil {
@@ -90,5 +99,12 @@ func DefaultLocalConfig() Config {
 			Username: "admin",
 			Password: "password123",
 		},
+		ObservabilityConfig: ObservabilityConfig{
+			Pprof: PprofConfig{
+				Enabled: true,
+				ApiAddr: "localhost:6060",
+				WorkerAddr: "localhost:6061",
+			},
+		},
 	}
 }

backend/internal/observability/pprof.go

@@ -0,0 +1,51 @@
+package observability
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"net/http"
+	"net/http/pprof"
+	"time"
+	"context"
+)
+func NewPprofMux() *http.ServeMux {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/debug/pprof/", pprof.Index)
+	mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+	mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
+	mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+	mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
+
+	return mux
+}
+
+func StartPprofServer(name string, enabled bool, addr string) (*http.Server, error) {
+	if !enabled || addr == "" {
+		return nil, nil
+	}
+	ln, err := net.Listen("tcp", addr)
+	if err != nil {
+		return nil, fmt.Errorf("failed to start %s pprof server on %s: %w", name, addr, err)
+	}
+	server := &http.Server{
+		Addr: addr,
+		Handler: NewPprofMux(),
+		ReadHeaderTimeout: 5 * time.Second,
+	}
+	go func() {
+		log.Printf("%s pprof listening on %s", name, addr)
+		if err := server.Serve(ln); err != nil && !errors.Is(err, http.ErrServerClosed) {
+			log.Printf("%s pprof server error: %v", name, err)
+		}
+	}()
+	return server, nil
+}
+
+func Shutdown(ctx context.Context, srv *http.Server) error{
+	if srv == nil {
+		return nil
+	}
+	return srv.Shutdown(ctx)
+}

backend/internal/observability/pprof_test.go

@@ -0,0 +1,31 @@
+package observability
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+func TestNewPprofMux(t *testing.T) {
+	t.Parallel()
+
+	req := httptest.NewRequest(http.MethodGet, "/debug/pprof/", nil)
+	rr := httptest.NewRecorder()
+
+	NewPprofMux().ServeHTTP(rr, req)
+
+	if rr.Code != http.StatusOK {
+		t.Errorf("Expected status code 200, got %d", rr.Code)
+	}
+}
+	
+func TestStartPprofServerWithDisabled(t *testing.T) {
+	t.Parallel()
+
+	server, err := StartPprofServer("api", false, "localhost:6060")
+	if err != nil {
+		t.Fatalf("Failed to start pprof server: %v", err)
+	}
+	if server != nil {
+		t.Fatalf("Expected nil server when pprof is disabled, got non-nil")
+	}
+}