use folly::Function for callback in async IO interface.

Summary:
folly::Function is non-copyable. This means it can be used with lambdas that capture
a unique_ptr, whereas std::function can not.

See https://www.internalfb.com/intern/wiki/Folly/Function/

Differential Revision: D23883275

fbshipit-source-id: a26cda0d3833e3ea822beef99c6a28a6dd90fa83

folly/experimental/io/AsyncBase.cpp

@@ -219,13 +219,14 @@ void AsyncBaseQueue::maybeDequeue() {
     queue_.pop_front();
 
     // Interpose our completion callback
-    auto& nextCb = op->notificationCallback();
-    op->setNotificationCallback([this, nextCb](AsyncBaseOp* op2) {
-      this->onCompleted(op2);
-      if (nextCb) {
-        nextCb(op2);
-      }
-    });
+    auto nextCb = op->getNotificationCallback();
+    op->setNotificationCallback(
+        [this, nextCb{std::move(nextCb)}](AsyncBaseOp* op2) mutable {
+          this->onCompleted(op2);
+          if (nextCb) {
+            nextCb(op2);
+          }
+        });
 
     asyncBase_->submit(op);
   }

folly/experimental/io/AsyncBase.h

@@ -27,6 +27,7 @@
 #include <utility>
 #include <vector>
 
+#include <folly/Function.h>
 #include <folly/Portability.h>
 #include <folly/Range.h>
 #include <folly/portability/SysUio.h>
@@ -44,7 +45,7 @@ class AsyncBaseOp {
   friend class AsyncBase;
 
  public:
-  using NotificationCallback = std::function<void(AsyncBaseOp*)>;
+  using NotificationCallback = folly::Function<void(AsyncBaseOp*)>;
 
   explicit AsyncBaseOp(NotificationCallback cb = NotificationCallback());
   AsyncBaseOp(const AsyncBaseOp&) = delete;
@@ -119,8 +120,16 @@ class AsyncBaseOp {
   void setNotificationCallback(NotificationCallback cb) {
     cb_ = std::move(cb);
   }
-  const NotificationCallback& notificationCallback() const {
-    return cb_;
+
+  /**
+   * Get the notification callback from the op.
+   *
+   * Note that this moves the callback out, leaving the callback in an
+   * uninitialized state! You must call setNotificationCallback before
+   * submitting the operation!
+   */
+  NotificationCallback getNotificationCallback() {
+    return std::move(cb_);
   }
 
   /**