forked from nodejs/node-addon-api
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathasync_worker.cc
More file actions
106 lines (91 loc) · 2.88 KB
/
async_worker.cc
File metadata and controls
106 lines (91 loc) · 2.88 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include "napi.h"
using namespace Napi;
class TestWorker : public AsyncWorker {
public:
static void DoWork(const CallbackInfo& info) {
bool succeed = info[0].As<Boolean>();
Object resource = info[1].As<Object>();
Function cb = info[2].As<Function>();
Value data = info[3];
TestWorker* worker = new TestWorker(cb, "TestResource", resource);
worker->Receiver().Set("data", data);
worker->_succeed = succeed;
worker->Queue();
}
protected:
void Execute() override {
if (!_succeed) {
SetError("test error");
}
}
private:
TestWorker(Function cb, const char* resource_name, const Object& resource)
: AsyncWorker(cb, resource_name, resource) {}
bool _succeed;
};
class TestWorkerWithResult : public AsyncWorker {
public:
static void DoWork(const CallbackInfo& info) {
bool succeed = info[0].As<Boolean>();
Object resource = info[1].As<Object>();
Function cb = info[2].As<Function>();
Value data = info[3];
TestWorkerWithResult* worker =
new TestWorkerWithResult(cb, "TestResource", resource);
worker->Receiver().Set("data", data);
worker->_succeed = succeed;
worker->Queue();
}
protected:
void Execute() override {
if (!_succeed) {
SetError("test error");
}
}
std::vector<napi_value> GetResult(Napi::Env env) override {
return {Boolean::New(env, _succeed),
String::New(env, _succeed ? "ok" : "error")};
}
private:
TestWorkerWithResult(Function cb,
const char* resource_name,
const Object& resource)
: AsyncWorker(cb, resource_name, resource) {}
bool _succeed;
};
class TestWorkerNoCallback : public AsyncWorker {
public:
static Value DoWork(const CallbackInfo& info) {
napi_env env = info.Env();
bool succeed = info[0].As<Boolean>();
Object resource = info[1].As<Object>();
TestWorkerNoCallback* worker =
new TestWorkerNoCallback(env, "TestResource", resource);
worker->_succeed = succeed;
worker->Queue();
return worker->_deferred.Promise();
}
protected:
void Execute() override {}
virtual void OnOK() override { _deferred.Resolve(Env().Undefined()); }
virtual void OnError(const Napi::Error& /* e */) override {
_deferred.Reject(Env().Undefined());
}
private:
TestWorkerNoCallback(napi_env env,
const char* resource_name,
const Object& resource)
: AsyncWorker(env, resource_name, resource),
_deferred(Napi::Promise::Deferred::New(env)) {}
Promise::Deferred _deferred;
bool _succeed;
};
Object InitAsyncWorker(Env env) {
Object exports = Object::New(env);
exports["doWork"] = Function::New(env, TestWorker::DoWork);
exports["doWorkNoCallback"] =
Function::New(env, TestWorkerNoCallback::DoWork);
exports["doWorkWithResult"] =
Function::New(env, TestWorkerWithResult::DoWork);
return exports;
}