Increased call stack size.

1 files changed, 34 insertions, 2 deletions

diff --git a/src/yue.cpp b/src/yue.cpp
index fc57767..2c3a469 100644
--- a/src/yue.cpp
+++ b/src/yue.cpp
@@ -30,6 +30,38 @@ using namespace std::chrono_literals;
 #include "ghc/fs_std.hpp"
 #include "linenoise.hpp"
 
+#if __has_include(<pthread.h>)
+#include <pthread.h>
+template<class R>
+std::future<R> async(const std::function<R()>& f) {
+        using Fn = std::packaged_task<R()>;
+        auto task = new Fn(f);
+        std::future<R> fut = task->get_future();
+
+        pthread_attr_t attr;
+        pthread_attr_init(&attr);
+        pthread_attr_setstacksize(&attr, 8 * 1024 * 1024);
+
+        pthread_t th;
+        pthread_create(&th, &attr,
+                [](void* p)->void* {
+                        std::unique_ptr<Fn> fn(static_cast<Fn*>(p));
+                        (*fn)();
+                        return nullptr;
+                },
+        task);
+        pthread_attr_destroy(&attr);
+        pthread_detach(th);
+        return fut;
+}
+#else
+template<class R>
+std::future<R> async(const std::function<R()>& f) {
+    // fallback: ignore stack size
+    return std::async(std::launch::async, f);
+}
+#endif
+
 #if not(defined YUE_NO_MACRO && defined YUE_COMPILER_ONLY)
 #define _DEFER(code, line) std::shared_ptr<void> _defer_##line(nullptr, [&](auto) { \
         code; \
@@ -699,7 +731,7 @@ int main(int narg, const char** args) {
                 }
                 std::list<std::future<std::string>> results;
                 for (const auto& file : files) {
-                        auto task = std::async(std::launch::async, [=]() {
+                        auto task = async<std::string>([=]() {
 #ifndef YUE_COMPILER_ONLY
                                 return compileFile(fs::absolute(file.first), config, fullWorkPath, fullTargetPath, minify, rewrite);
 #else
@@ -737,7 +769,7 @@ int main(int narg, const char** args) {
 #endif // YUE_NO_WATCHER
         std::list<std::future<std::tuple<int, std::string, std::string>>> results;
         for (const auto& file : files) {
-                auto task = std::async(std::launch::async, [=]() {
+                auto task = async<std::tuple<int, std::string, std::string>>([=]() {
                         std::ifstream input(file.first, std::ios::in);
                         if (input) {
                                 std::string s(