Add LIBDISPATCH_DEFAULT_THREAD_POOL_SIZE env variable

Author: Anton Pogonets

src/internal.h

@@ -977,4 +977,11 @@ extern int _dispatch_kevent_workqueue_enabled;
 #include "inline_internal.h"
 #include "firehose/firehose_internal.h"
 
+#if !defined(HAVE_PTHREAD_WORKQUEUE_KEVENT)
+// copied from https://opensource.apple.com/source/libpthread/libpthread-301.50.1/kern/workqueue_internal.h.auto.html
+#define WORKQUEUE_MAXTHREADS 512
+#define WORKQUEUE_CONSTRAINED_MAXTHREADS (WORKQUEUE_MAXTHREADS >> 3)
+#define WORKQUEUE_CONSTRAINED_FACTOR 5
+#endif
+
 #endif /* __DISPATCH_INTERNAL__ */

src/queue.c

@@ -755,8 +755,16 @@ _dispatch_root_queue_init_pthread_pool(dispatch_root_queue_context_t qc,
 		int32_t pool_size, bool overcommit)
 {
 	dispatch_pthread_root_queue_context_t pqc = qc->dgq_ctxt;
-	int32_t thread_pool_size = overcommit ? DISPATCH_WORKQ_MAX_PTHREAD_COUNT :
-			(int32_t)dispatch_hw_config(active_cpus);
+	int32_t default_pool_size = 0;
+	char* default_pool_size_env = getenv("LIBDISPATCH_DEFAULT_THREAD_POOL_SIZE");
+	if (default_pool_size_env) {
+		default_pool_size = (int32_t) atoi(default_pool_size_env);
+	}
+	if (!default_pool_size) {
+		default_pool_size = (int32_t) MAX(dispatch_hw_config(active_cpus) * 5, WORKQUEUE_CONSTRAINED_MAXTHREADS);
+	}
+	int32_t thread_pool_size = overcommit ? DISPATCH_WORKQ_MAX_PTHREAD_COUNT : default_pool_size;
+
 	if (slowpath(pool_size) && pool_size < thread_pool_size) {
 		thread_pool_size = pool_size;
 	}