From b7cf191e2a731f2916649d7dd67450cf4f85de27 Mon Sep 17 00:00:00 2001
From: XieHan <XieHan@sogou-inc.com>
Date: Tue, 23 Feb 2021 17:33:20 +0800
Subject: [PATCH] add WFTimerTask::set_callback

---
 src/factory/WFTask.h | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/factory/WFTask.h b/src/factory/WFTask.h
index 06f3c280..f4b17956 100644
--- a/src/factory/WFTask.h
+++ b/src/factory/WFTask.h
@@ -346,6 +346,12 @@ public:
 	int get_state() const { return this->state; }
 	int get_error() const { return this->error; }
 
+public:
+	void set_callback(std::function<void (WFTimerTask *)> cb)
+	{
+		this->callback = std::move(cb);
+	}
+
 protected:
 	virtual SubTask *done()
 	{