src: add trace_event macros to AsyncWrap

joshgav · Mar 29, 2017 · fda406f · fda406f
1 parent 3d6d61c
commit fda406f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/async-wrap.cc b/src/async-wrap.cc
@@ -259,9 +259,18 @@ AsyncWrap::AsyncWrap(Environment* env,
   CHECK_NE(provider, PROVIDER_NONE);
   CHECK_GE(object->InternalFieldCount(), 1);
 
+ switch (provider_type()) {
+#define V(PROVIDER)                                                           \
+    case PROVIDER_ ## PROVIDER:                                               \
+      TRACE_EVENT_NESTABLE_ASYNC_BEGIN0("node", #PROVIDER, get_uid());        \
+      break;
+    NODE_ASYNC_PROVIDER_TYPES(V)
+#undef V
+  }
   // Shift provider value over to prevent id collision.
   persistent().SetWrapperClassId(NODE_ASYNC_ID_OFFSET + provider);
 
+
   Local<Function> init_fn = env->async_hooks_init_function();
 
   // No init callback exists, no reason to go on.
@@ -303,6 +312,14 @@ AsyncWrap::AsyncWrap(Environment* env,
 
 
 AsyncWrap::~AsyncWrap() {
+  switch (provider_type()) {
+#define V(PROVIDER)                                                           \
+    case PROVIDER_ ## PROVIDER:                                               \
+      TRACE_EVENT_NESTABLE_ASYNC_END0("node", #PROVIDER, get_uid());          \
+      break;
+    NODE_ASYNC_PROVIDER_TYPES(V)
+#undef V
+  }
   if (!ran_init_callback())
     return;