Merge pull request #572 from projectdiscovery/add_max_thread_env_var

add global `OS_MAX_THREADS` env variable
projectdiscovery · Nov 26, 2024 · 70cf636 · 70cf636
2 parents af7d763 + 6bcddac
commit 70cf636
Showing 1 changed file with 24 additions and 0 deletions.
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)
+	}
+}