Improve folly::RequestContext onSet and onUnset efficiency
[folly.git] / folly / io / async / test / RequestContextTest.cpp
index 750ff7fac9bd0c3c2ad7c2f4e737aaf303d53e37..6758a9a9ef6815339b65619d03413126df2846c2 100644 (file)
@@ -27,12 +27,19 @@ class TestData : public RequestData {
  public:
   explicit TestData(int data) : data_(data) {}
   ~TestData() override {}
+
+  bool hasCallback() override {
+    return true;
+  }
+
   void onSet() override {
     set_++;
   }
+
   void onUnset() override {
     unset_++;
   }
+
   int set_ = 0, unset_ = 0;
   int data_;
 };
@@ -136,9 +143,9 @@ TEST(RequestContext, deadlockTest) {
           val_, std::make_unique<TestData>(1));
     }
 
-    void onSet() override {}
-
-    void onUnset() override {}
+    bool hasCallback() override {
+      return false;
+    }
 
     std::string val_;
   };