Add profiler

.golangci.yml

@@ -1,5 +1,5 @@
 service:
-  golangci-lint-version: 1.16.0
+  golangci-lint-version: 1.20.0
 run:
   deadline: 5m
   issues-exit-code: 1
@@ -13,6 +13,8 @@ linters-settings:
     locale: US
   gocyclo:
     min-complexity: 20
+  wsl:
+    allow-trailing-comment: true
 
 issues:
   exclude-rules:
@@ -26,10 +28,11 @@ issues:
         - funlen
     - linters:
         - stylecheck
-      source: "package (cert_manager|domain_checker)"
+      source: "package (acme_client_manager|cert_manager|domain_checker)"
 
 linters:
   enable-all: true
   disable:
     - gochecknoglobals
+    - wsl
 

cmd/config.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/gobuffalo/packr"
 
+	"github.com/rekby/lets-proxy2/internal/profiler"
 	"github.com/rekby/lets-proxy2/internal/tlslistener"
 
 	"github.com/rekby/lets-proxy2/internal/proxy"
@@ -40,6 +41,7 @@ type configType struct {
 	Proxy        proxy.Config
 	CheckDomains domain_checker.Config
 	Listen       tlslistener.Config
+	Profiler     profiler.Config
 }
 
 //nolint:maligned
@@ -135,6 +137,7 @@ func mergeConfigByFilepath(ctx context.Context, c *configType, filename string)
 			log.DebugFatal(logger, err, "Restore workdir to", zap.String("dir", dir))
 		}()
 	}
+
 	mergeConfigBytes(ctx, c, content, filename)
 }
 

cmd/config_test.go

@@ -55,6 +55,7 @@ StorageDir = "storage2"
 `), 0600)
 
 	var config configType
+
 	mergeConfigBytes(ctx, &config, defaultConfig(ctx), "")
 	mergeConfigByFilepath(ctx, &config, filepath.Join(tmpDir, "config.toml"))
 	td.CmpDeeply(config.General.IssueTimeout, 1)

cmd/log.go

@@ -47,6 +47,7 @@ func initLogger(config logConfig) *zap.Logger {
 		if err != nil {
 			panic("Can't open stderr to log")
 		}
+
 		writers = append(writers, writer)
 	}
 
@@ -77,7 +78,7 @@ func initLogger(config logConfig) *zap.Logger {
 func parseLogLevel(logLevelS string) (zapcore.Level, error) {
 	logLevelS = strings.TrimSpace(logLevelS)
 	logLevelS = strings.ToLower(logLevelS)
-	switch logLevelS {
+	switch logLevelS { //nolint:wsl
 	case "debug":
 		return zapcore.DebugLevel, nil
 	case "info":
@@ -91,7 +92,6 @@ func parseLogLevel(logLevelS string) (zapcore.Level, error) {
 	default:
 		return zapcore.InfoLevel, errors.New("undefined log level")
 	}
-
 }
 
 func getLogOptions(config logConfig) (res []zap.Option) {

cmd/main.go

@@ -10,7 +10,10 @@ import (
 	"runtime"
 	"time"
 
+	"go.uber.org/zap/zapcore"
+
 	"github.com/rekby/lets-proxy2/internal/cert_manager"
+	"github.com/rekby/lets-proxy2/internal/profiler"
 
 	_ "github.com/kardianos/minwinsvc"
 	"github.com/rekby/lets-proxy2/internal/acme_client_manager"
@@ -57,6 +60,8 @@ func startProgram(config *configType) {
 
 	logger.Info("StartAutoRenew program version", zap.String("version", version()))
 
+	startProfiler(ctx, config.Profiler)
+
 	err := os.MkdirAll(config.General.StorageDir, defaultDirMode)
 	log.InfoFatal(logger, err, "Create storage dir", zap.String("dir", config.General.StorageDir))
 
@@ -98,3 +103,29 @@ func startProgram(config *configType) {
 	}
 	log.DebugErrorCtx(ctx, effectiveError, "Handle request stopped")
 }
+
+func startProfiler(ctx context.Context, config profiler.Config) {
+	logger := zc.L(ctx)
+
+	if !config.Enable {
+		logger.Info("Profiler disabled")
+		return
+	}
+
+	go func() {
+		httpServer := http.Server{
+			Addr:    config.BindAddress,
+			Handler: profiler.New(logger.Named("profiler"), config),
+		}
+
+		logger.Info("Start profiler", zap.String("bind_address", httpServer.Addr))
+		err := httpServer.ListenAndServe()
+		var logLevel zapcore.Level
+		if err == http.ErrServerClosed {
+			logLevel = zapcore.InfoLevel
+		} else {
+			logLevel = zapcore.ErrorLevel
+		}
+		log.LevelParam(logger, logLevel, "Profiler stopped")
+	}()
+}

cmd/static/default-config.toml

@@ -122,3 +122,13 @@ TLSAddresses = [":443"]
 
 # Bind addresses without TLS secure (for HTTP reverse proxy and http-01 validation without redirect to https)
 TCPAddresses = []
+
+[Profiler]
+Enable = false
+
+# IP networks for allow to use profiler.
+# Default - deny from all.
+# Example:
+# [ "1.2.3.4/32", "192.168.0.0/24", "::1/128" ]
+AllowedNetworks = []
+BindAddress = "localhost:31344"