Add support for std::future<void> with enqueue

Added support for `std::future<void>` return type with `dp::thread_pool::enqueue`. Fixes #37
DeveloperPaul123 · Jun 1, 2023 · ffd1318 · ffd1318
1 parent a442e8b
commit ffd1318
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/include/thread_pool/thread_pool.h b/include/thread_pool/thread_pool.h
@@ -117,7 +117,12 @@ namespace dp {
             auto task = [func = std::move(f), ... largs = std::move(args),
                          promise = std::move(promise)]() mutable {
                 try {
-                    promise.set_value(func(largs...));
+                    if constexpr (std::is_same_v<ReturnType, void>) {
+                        func(largs...);
+                        promise.set_value();
+                    } else {
+                        promise.set_value(func(largs...));
+                    }
                 } catch (...) {
                     promise.set_exception(std::current_exception());
                 }
@@ -139,7 +144,13 @@ namespace dp {
             auto task = [func = std::move(f), ... largs = std::move(args),
                          promise = shared_promise]() {
                 try {
-                    promise->set_value(func(largs...));
+                    if constexpr (std::is_same_v<ReturnType, void>) {
+                        func(largs...);
+                        promise->set_value();
+                    } else {
+                        promise->set_value(func(largs...));
+                    }
+
                 } catch (...) {
                     promise->set_exception(std::current_exception());
                 }