add DUCKDB_NO_THREADS check

lnkuiper · Nov 4, 2024 · c24736c · c24736c
1 parent 961af3a
commit c24736c
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/extension/jemalloc/include/malloc_ncpus.h b/extension/jemalloc/include/malloc_ncpus.h
@@ -0,0 +1,22 @@
+//===----------------------------------------------------------------------===//
+//                         DuckDB
+//
+// malloc_ncpus.h
+//
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef MALLOC_NCPUS_H
+#define MALLOC_NCPUS_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+unsigned duckdb_malloc_ncpus();
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif // MALLOC_NCPUS_H
diff --git a/extension/jemalloc/jemalloc_extension.cpp b/extension/jemalloc/jemalloc_extension.cpp
@@ -118,7 +118,11 @@ std::string JemallocExtension::Version() const {
 extern "C" {
 
 unsigned duckdb_malloc_ncpus() {
+#ifdef DUCKDB_NO_THREADS
+	return 1
+#else
 	return duckdb::NumericCast<unsigned>(std::thread::hardware_concurrency());
+#endif
 }
 
 DUCKDB_EXTENSION_API void jemalloc_init(duckdb::DatabaseInstance &db) {