From: Martin Martin <mcm@fb.com>
Date: Fri, 29 Apr 2016 18:54:34 +0000 (-0700)
Subject: Make addTask[Remote]Future() work for functions returning void.
X-Git-Tag: 2016.07.26~299
X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=b31eb722e444ab0293a73fe9de3f94e657ca6de9;p=folly.git

Make addTask[Remote]Future() work for functions returning void.

Summary:
There's no Future<void>, you're supposed to use Future<Unit>
instead.  Unit has a "Lift" template to do the conversion.

Reviewed By: andriigrynenko

Differential Revision: D3241498

fb-gh-sync-id: db12d4f343685bc613b701e023c860c2c903ed4c
fbshipit-source-id: db12d4f343685bc613b701e023c860c2c903ed4c
---

diff --git a/folly/experimental/fibers/FiberManager-inl.h b/folly/experimental/fibers/FiberManager-inl.h
index 6bad3618..8f31ede1 100644
--- a/folly/experimental/fibers/FiberManager-inl.h
+++ b/folly/experimental/fibers/FiberManager-inl.h
@@ -281,9 +281,10 @@ void FiberManager::addTask(F&& func) {
 }
 
 template <typename F>
-auto FiberManager::addTaskFuture(F&& func)
-    -> folly::Future<typename std::result_of<F()>::type> {
-  using T = typename std::result_of<F()>::type;
+auto FiberManager::addTaskFuture(F&& func) -> folly::Future<
+    typename folly::Unit::Lift<typename std::result_of<F()>::type>::type> {
+  using T =
+      typename folly::Unit::Lift<typename std::result_of<F()>::type>::type;
   folly::Promise<T> p;
   auto f = p.getFuture();
   addTaskFinally(
@@ -312,9 +313,11 @@ void FiberManager::addTaskRemote(F&& func) {
 }
 
 template <typename F>
-auto FiberManager::addTaskRemoteFuture(F&& func)
-    -> folly::Future<typename std::result_of<F()>::type> {
-  folly::Promise<typename std::result_of<F()>::type> p;
+auto FiberManager::addTaskRemoteFuture(F&& func) -> folly::Future<
+    typename folly::Unit::Lift<typename std::result_of<F()>::type>::type> {
+  folly::Promise<
+      typename folly::Unit::Lift<typename std::result_of<F()>::type>::type>
+      p;
   auto f = p.getFuture();
   addTaskRemote(
       [ p = std::move(p), func = std::forward<F>(func), this ]() mutable {
@@ -395,7 +398,8 @@ struct FiberManager::AddTaskFinallyHelper {
 
 template <typename F, typename G>
 void FiberManager::addTaskFinally(F&& func, G&& finally) {
-  typedef typename std::result_of<F()>::type Result;
+  typedef typename folly::Unit::Lift<typename std::result_of<F()>::type>::type
+      Result;
 
   static_assert(
       IsRvalueRefTry<typename FirstArgOf<G>::type>::value,
diff --git a/folly/experimental/fibers/FiberManager.h b/folly/experimental/fibers/FiberManager.h
index 73a243ce..559f70e7 100644
--- a/folly/experimental/fibers/FiberManager.h
+++ b/folly/experimental/fibers/FiberManager.h
@@ -190,8 +190,8 @@ class FiberManager : public ::folly::Executor {
    *             The object will be destroyed once task execution is complete.
    */
   template <typename F>
-  auto addTaskFuture(F&& func)
-      -> folly::Future<typename std::result_of<F()>::type>;
+  auto addTaskFuture(F&& func) -> folly::Future<
+      typename folly::Unit::Lift<typename std::result_of<F()>::type>::type>;
   /**
    * Add a new task to be executed. Safe to call from other threads.
    *
@@ -209,8 +209,8 @@ class FiberManager : public ::folly::Executor {
    *             The object will be destroyed once task execution is complete.
    */
   template <typename F>
-  auto addTaskRemoteFuture(F&& func)
-      -> folly::Future<typename std::result_of<F()>::type>;
+  auto addTaskRemoteFuture(F&& func) -> folly::Future<
+      typename folly::Unit::Lift<typename std::result_of<F()>::type>::type>;
 
   // Executor interface calls addTaskRemote
   void add(folly::Func f) override {
diff --git a/folly/experimental/fibers/test/FibersTest.cpp b/folly/experimental/fibers/test/FibersTest.cpp
index ac66df6e..08d182ff 100644
--- a/folly/experimental/fibers/test/FibersTest.cpp
+++ b/folly/experimental/fibers/test/FibersTest.cpp
@@ -1406,6 +1406,29 @@ TEST(FiberManager, remoteFutureTest) {
   EXPECT_EQ(v2, testValue2);
 }
 
+// Test that a void function produes a Future<Unit>.
+TEST(FiberManager, remoteFutureVoidUnitTest) {
+  FiberManager fiberManager(folly::make_unique<SimpleLoopController>());
+  auto& loopController =
+      dynamic_cast<SimpleLoopController&>(fiberManager.loopController());
+
+  bool ranLocal = false;
+  folly::Future<folly::Unit> futureLocal =
+      fiberManager.addTaskFuture([&]() { ranLocal = true; });
+
+  bool ranRemote = false;
+  folly::Future<folly::Unit> futureRemote =
+      fiberManager.addTaskRemoteFuture([&]() { ranRemote = true; });
+
+  loopController.loop([&]() { loopController.stop(); });
+
+  futureLocal.wait();
+  ASSERT_TRUE(ranLocal);
+
+  futureRemote.wait();
+  ASSERT_TRUE(ranRemote);
+}
+
 TEST(FiberManager, nestedFiberManagers) {
   folly::EventBase outerEvb;
   folly::EventBase innerEvb;