From 7443e5c1c98b63ba53f0f7dbaee319b4ca948c11 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=8A=B2=E9=B9=8F?= <jingpeng.wang@outlook.com>
Date: Thu, 19 Dec 2019 20:24:44 +0800
Subject: [PATCH] callback test

---
 doric/async/async_result.h       | 15 +++++++++++++--
 doric/doric.pro                  |  1 +
 doric/main.cpp                   | 14 ++++++++++----
 doric/template/custom_callback.h | 23 +++++++++++++++++++++++
 4 files changed, 47 insertions(+), 6 deletions(-)
 create mode 100644 doric/template/custom_callback.h

diff --git a/doric/async/async_result.h b/doric/async/async_result.h
index ffe14871..2b12b77b 100644
--- a/doric/async/async_result.h
+++ b/doric/async/async_result.h
@@ -11,7 +11,7 @@ class AsyncResult {
 
 private:
     QVariant result;
-    Callback<R> *callback;
+    Callback<R> *callback = nullptr;
 
 public:
     AsyncResult() {}
@@ -28,7 +28,7 @@ public:
         }
     }
 
-    void setError(QException *exception) {
+    void setError(QException exception) {
         this->result->setValue(exception);
         if (callback != nullptr) {
             this->callback->onError(exception);
@@ -44,6 +44,17 @@ public:
     R *getResult() {
         return static_cast<R*>(result.data());
     }
+
+    void setCallback(Callback<R> *callback) {
+        this->callback = callback;
+        if (QException *exception = static_cast<QException*>(result.data())) {
+            this->callback->onError(*exception);
+            this->callback->onFinish();
+        } else if (hasResult()) {
+            this->callback->onResult(*getResult());
+            this->callback->onFinish();
+        }
+    }
 };
 
 #endif // ASYNC_RESULT_H
diff --git a/doric/doric.pro b/doric/doric.pro
index 8f61c351..1d33eaf7 100644
--- a/doric/doric.pro
+++ b/doric/doric.pro
@@ -58,5 +58,6 @@ HEADERS += \
     plugin/shader_plugin.h \
     registry.h \
     shader/layer.h \
+    template/custom_callback.h \
     template/singleton.h \
     utility/utility.h
diff --git a/doric/main.cpp b/doric/main.cpp
index 994626b7..e633bd07 100644
--- a/doric/main.cpp
+++ b/doric/main.cpp
@@ -5,6 +5,7 @@
 
 #include "context_manager.h"
 #include "async/async_result.h"
+#include "template/custom_callback.h"
 
 int main(int argc, char *argv[])
 {
@@ -31,10 +32,15 @@ int main(int argc, char *argv[])
         delete source;
     }
 
-    QJsonValue *a = new QJsonValue();
-    AsyncResult<QJsonValue> *result = new AsyncResult<QJsonValue>(*a);
-    qDebug() << result->hasResult();
-    qDebug() << result->getResult();
+    {
+        // code for test
+        QJsonValue *a = new QJsonValue();
+        AsyncResult<QJsonValue> *result = new AsyncResult<QJsonValue>(*a);
+        CustomCallback<QJsonValue> *callback = new CustomCallback<QJsonValue>();
+        result->setCallback(callback);
+        qDebug() << result->hasResult();
+        qDebug() << result->getResult();
+    }
 
     return app.exec();
 }
diff --git a/doric/template/custom_callback.h b/doric/template/custom_callback.h
new file mode 100644
index 00000000..d0c20667
--- /dev/null
+++ b/doric/template/custom_callback.h
@@ -0,0 +1,23 @@
+#ifndef CUSTOM_CALLBACK_H
+#define CUSTOM_CALLBACK_H
+
+#include "async/callback.h"
+
+template <class R>
+class CustomCallback : public Callback<R> {
+
+public:
+    void onResult(R result) override {
+
+    }
+
+    void onError(QException exception) override {
+
+    }
+
+    void onFinish() override {
+
+    }
+};
+
+#endif // CUSTOM_CALLBACK_H