summaryrefslogtreecommitdiff
path: root/src/async-wrap-inl.h
blob: 5841c58faf18d509ec53008d6c559a22ffebc164 (plain)
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
#ifndef SRC_ASYNC_WRAP_INL_H_
#define SRC_ASYNC_WRAP_INL_H_

#include "async-wrap.h"
#include "base-object.h"
#include "base-object-inl.h"
#include "env.h"
#include "env-inl.h"
#include "node_internals.h"
#include "util.h"
#include "util-inl.h"
#include "v8.h"

namespace node {

inline AsyncWrap::AsyncWrap(Environment* env,
                            v8::Handle<v8::Object> object,
                            ProviderType provider,
                            AsyncWrap* parent)
    : BaseObject(env, object), bits_(static_cast<uint32_t>(provider) << 1) {
  // Only set wrapper class id if object will be Wrap'd.
  if (object->InternalFieldCount() > 0)
    // Shift provider value over to prevent id collision.
    persistent().SetWrapperClassId(NODE_ASYNC_ID_OFFSET + provider);

  // Check user controlled flag to see if the init callback should run.
  if (!env->using_asyncwrap())
    return;

  // If callback hooks have not been enabled, and there is no parent, return.
  if (!env->async_wrap_callbacks_enabled() && parent == nullptr)
    return;

  // If callback hooks have not been enabled and parent has no queue, return.
  if (!env->async_wrap_callbacks_enabled() && !parent->has_async_queue())
    return;

  v8::HandleScope scope(env->isolate());
  v8::TryCatch try_catch;

  v8::Local<v8::Value> n = v8::Int32::New(env->isolate(), provider);
  env->async_hooks_init_function()->Call(object, 1, &n);

  if (try_catch.HasCaught())
    FatalError("node::AsyncWrap::AsyncWrap", "init hook threw");

  bits_ |= 1;  // has_async_queue() is true now.
}


inline bool AsyncWrap::has_async_queue() const {
  return static_cast<bool>(bits_ & 1);
}


inline AsyncWrap::ProviderType AsyncWrap::provider_type() const {
  return static_cast<ProviderType>(bits_ >> 1);
}


inline v8::Handle<v8::Value> AsyncWrap::MakeCallback(
    const v8::Handle<v8::String> symbol,
    int argc,
    v8::Handle<v8::Value>* argv) {
  v8::Local<v8::Value> cb_v = object()->Get(symbol);
  CHECK(cb_v->IsFunction());
  return MakeCallback(cb_v.As<v8::Function>(), argc, argv);
}


inline v8::Handle<v8::Value> AsyncWrap::MakeCallback(
    uint32_t index,
    int argc,
    v8::Handle<v8::Value>* argv) {
  v8::Local<v8::Value> cb_v = object()->Get(index);
  CHECK(cb_v->IsFunction());
  return MakeCallback(cb_v.As<v8::Function>(), argc, argv);
}

}  // namespace node

#endif  // SRC_ASYNC_WRAP_INL_H_