refactor

projectdiscovery · Nov 26, 2024 · 6bcddac · 6bcddac
1 parent a864c49
commit 6bcddac
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 21 deletions.
diff --git a/global/global.go b/global/global.go
diff --git a/global/max_threads.go b/global/max_threads.go
@@ -0,0 +1,24 @@
+package global
+
+import (
+	"os"
+	"strconv"
+
+	"github.com/projectdiscovery/utils/sysutil"
+)
+
+const OS_MAX_THREADS_ENV = "OS_MAX_THREADS"
+
+func init() {
+	handleOSMaxThreads()
+}
+
+func handleOSMaxThreads() {
+	osMaxThreads := os.Getenv(OS_MAX_THREADS_ENV)
+	if osMaxThreads == "" {
+		return
+	}
+	if value, err := strconv.Atoi(osMaxThreads); err == nil && value > 0 {
+		_ = sysutil.SetMaxThreads(value)
+	}
+}