service: Add pprof option (#153)

Author: sunshineplan

service/go.mod

@@ -7,7 +7,7 @@ require (
 	github.com/sunshineplan/httpproxy v0.0.0-00010101000000-000000000000
 	github.com/sunshineplan/limiter v1.0.0
 	github.com/sunshineplan/service v1.0.22
-	github.com/sunshineplan/utils v0.1.77
+	github.com/sunshineplan/utils v0.1.78
 	golang.org/x/net v0.42.0
 	golang.org/x/time v0.12.0
 )

service/go.sum

@@ -4,8 +4,8 @@ github.com/sunshineplan/limiter v1.0.0 h1:wx3q5eS5J+ggXlAxzg9k6UbDyJYrysNmHyxt5c
 github.com/sunshineplan/limiter v1.0.0/go.mod h1:+Pjd5Pu7i5YclrnFz+MBFxGB9+MZ2cytQeV+S9kXOxY=
 github.com/sunshineplan/service v1.0.22 h1:MFXKIbcd3POiJb9KRI8oeybc2qN0p79vv9f+OL88EDU=
 github.com/sunshineplan/service v1.0.22/go.mod h1:JR7cTSNJVlajjXfXobA3Rr46z88M+6VK6coxq7JmyrY=
-github.com/sunshineplan/utils v0.1.77 h1:hDJtJahnkzvJ39L0GMc7mGFcZPjwYIbRcSwX3wPNVbM=
-github.com/sunshineplan/utils v0.1.77/go.mod h1:R7MInPRKnExzzNGJ9qY1W0+11P/3jc6IO5ICuGDPA1w=
+github.com/sunshineplan/utils v0.1.78 h1:4aBDOTqWUozGBKe3wI1jjxz03yjUdn2GvPMs7oKILuw=
+github.com/sunshineplan/utils v0.1.78/go.mod h1:R7MInPRKnExzzNGJ9qY1W0+11P/3jc6IO5ICuGDPA1w=
 golang.org/x/net v0.42.0 h1:jzkYrhi3YQWD6MLBJcsklgQsoAcw89EcZbJw8Z614hs=
 golang.org/x/net v0.42.0/go.mod h1:FF1RA5d3u7nAYA4z2TkclSCKh68eSXtiFwcWQpPXdt8=
 golang.org/x/sys v0.34.0 h1:H5Y5sJ2L2JRdyv7ROF1he/lPdvFsd0mJHFw2ThKHxLA=

service/main.go

@@ -22,6 +22,7 @@ var (
 	status    = flag.String("status", "", "Path to status file")
 	keep      = flag.Int("keep", 100, "Count of status files")
 	debug     = flag.Bool("debug", false, "debug")
+	pprof     = flag.String("pprof", "", "pprof port")
 )
 
 const commonFlag = `

service/misc.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"io"
+	"net"
 	"net/url"
 	"path/filepath"
 	"time"
@@ -58,7 +59,7 @@ func watchFile(file string, fnChange, fnRemove func()) error {
 	return nil
 }
 
-func transfer(dst io.WriteCloser, src io.ReadCloser, user user, lim *limiter.Limiter) {
+func transfer(dst, src net.Conn, user user, lim *limiter.Limiter) {
 	defer dst.Close()
 	defer src.Close()
 	if lim == nil {

service/run.go

@@ -2,6 +2,8 @@ package main
 
 import (
 	"net"
+	"net/http"
+	_ "net/http/pprof"
 	"net/url"
 	"strconv"
 
@@ -19,6 +21,11 @@ type Runner interface {
 }
 
 func run() error {
+	if addr := *pprof; addr != "" {
+		go func() {
+			svc.Print(http.ListenAndServe(addr, nil))
+		}()
+	}
 	base := NewBase(*host, *port)
 	base.ErrorLog = errorLogger.Logger
 	servers := []*httpsvr.Server{base.Server}