feat: add automaxprocs in the template

caddyserver · Jan 16, 2024 · bc13c89 · bc13c89
1 parent a36cef4
commit bc13c89
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/environment.go b/environment.go
@@ -353,8 +353,12 @@ type goModTemplateContext struct {
 const mainModuleTemplate = `package main
 
 import (
+	"go.uber.org/automaxprocs/maxprocs"
+	"go.uber.org/zap"
+
 	caddycmd "{{.CaddyModule}}/cmd"
 
+	"{{.CaddyModule}}"
 	// plug in Caddy modules here
 	_ "{{.CaddyModule}}/modules/standard"
 	{{- range .Plugins}}
@@ -363,6 +367,12 @@ import (
 )
 
 func main() {
+	undo, err := maxprocs.Set()
+	defer undo()
+	if err != nil {
+		caddy.Log().Warn("failed to set GOMAXPROCS", zap.Error(err))
+	}
+
 	caddycmd.Main()
 }
 `