Make SemiFuture::via throw on nullptr executor.

Summary: Make SemiFuture throw if no executor provided to via because in that situation the deferred work will never run.

Reviewed By: yfeldblum

Differential Revision: D6233233

fbshipit-source-id: 43b642c46cc0be80b1f13c03bdaf20b8038acec2

diff --git a/folly/futures/Future-inl.h b/folly/futures/Future-inl.h
index 20cf87bc647a449fe462ca9b67195e3d1eecb39d..dea9e863095638bf62e82ca77b02734c5c0fa8ca 100644 (file)
--- a/folly/futures/Future-inl.h
+++ b/folly/futures/Future-inl.h
@@ -483,6 +483,9 @@ void SemiFuture<T>::boost_() {
 template <class T>
 inline Future<T> SemiFuture<T>::via(Executor* executor, int8_t priority) && {
   throwIfInvalid();
+  if (!executor) {
+    throwNoExecutor();
+  }
 
   // If current executor is deferred, boost block to ensure that work
   // progresses and is run on the new executor.

diff --git a/folly/futures/FutureException.cpp b/folly/futures/FutureException.cpp
index ecbf03da3c5f4bda7233e67dcdf5d65cbf39e517..d21d6f9425b2abb8c0ae827113960e466a2cb25e 100644 (file)
--- a/folly/futures/FutureException.cpp
+++ b/folly/futures/FutureException.cpp
@@ -42,7 +42,9 @@ namespace folly {
   throw PredicateDoesNotObtain();
 }
 
-[[noreturn]] void throwNoFutureInSplitter() {
-  throw NoFutureInSplitter();
+[[noreturn]] void throwNoFutureInSplitter() { throw NoFutureInSplitter(); }
+
+    [[noreturn]] void throwNoExecutor() {
+  throw NoExecutor();
 }
 } // namespace folly

diff --git a/folly/futures/FutureException.h b/folly/futures/FutureException.h
index 9d3bd77dd20c1fbf08024c2dabcd2ca1b7731fa8..0df8d8b37fc6a727bdc4460180b8f9d8c9edee7c 100644 (file)
--- a/folly/futures/FutureException.h
+++ b/folly/futures/FutureException.h
@@ -94,4 +94,11 @@ class FOLLY_EXPORT NoTimekeeper : public FutureException {
  public:
   NoTimekeeper() : FutureException("No timekeeper available") {}
 };
+
+[[noreturn]] void throwNoExecutor();
+
+class FOLLY_EXPORT NoExecutor : public FutureException {
+ public:
+  NoExecutor() : FutureException("No executor provided to via") {}
+};
 } // namespace folly

diff --git a/folly/futures/test/SemiFutureTest.cpp b/folly/futures/test/SemiFutureTest.cpp
index 953148324bdeaccb630bf634251878faa84cfda5..39ea108b31a6d68437fd9588792857ebb835ab35 100644 (file)
--- a/folly/futures/test/SemiFutureTest.cpp
+++ b/folly/futures/test/SemiFutureTest.cpp
@@ -143,6 +143,10 @@ TEST(SemiFuture, makeSemiFutureNoThrow) {
   makeSemiFuture().value();
 }
 
+TEST(SemiFuture, ViaThrowOnNull) {
+  EXPECT_THROW(makeSemiFuture().via(nullptr), NoExecutor);
+}
+
 TEST(SemiFuture, ConstructSemiFutureFromEmptyFuture) {
   auto f = SemiFuture<int>{Future<int>::makeEmpty()};
   EXPECT_THROW(f.isReady(), NoState);