async-hooks-wrapper.h

Go to the documentation of this file.

// Copyright 2018 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
 
#ifndef V8_D8_ASYNC_HOOKS_WRAPPER_H_
#define V8_D8_ASYNC_HOOKS_WRAPPER_H_
 
#include <stack>
#include <vector>
 
#include "include/v8-function-callback.h"
#include "include/v8-local-handle.h"
#include "include/v8-promise.h"
 
namespace v8 {
 
class Function;
class Isolate;
class ObjectTemplate;
class Value;
 
using async_id_t = double;
 
struct AsyncContext {
  async_id_t execution_async_id;
  async_id_t trigger_async_id;
};
 
class AsyncHooksWrap {
 public:
  static constexpr internal::ExternalPointerTag kManagedTag =
      internal::kGenericManagedTag;
 
  explicit AsyncHooksWrap(Isolate* isolate)
      : isolate_(isolate), enabled_(false) {}
  void Enable();
  void Disable();
  bool IsEnabled() const { return enabled_; }
 
  inline v8::Local<v8::Function> init_function() const;
  inline void set_init_function(v8::Local<v8::Function> value);
  inline v8::Local<v8::Function> before_function() const;
  inline void set_before_function(v8::Local<v8::Function> value);
  inline v8::Local<v8::Function> after_function() const;
  inline void set_after_function(v8::Local<v8::Function> value);
  inline v8::Local<v8::Function> promiseResolve_function() const;
  inline void set_promiseResolve_function(v8::Local<v8::Function> value);
 
 private:
  Isolate* isolate_;
 
  Persistent<v8::Function> init_function_;
  Persistent<v8::Function> before_function_;
  Persistent<v8::Function> after_function_;
  Persistent<v8::Function> promiseResolve_function_;
 
  bool enabled_;
};
 
class AsyncHooks {
 public:
  explicit AsyncHooks(Isolate* isolate);
  ~AsyncHooks();
 
  async_id_t GetExecutionAsyncId() const;
  async_id_t GetTriggerAsyncId() const;
 
  Local<Object> CreateHook(const v8::FunctionCallbackInfo<v8::Value>& info);
 
  Persistent<FunctionTemplate> async_hook_ctor;
 
 private:
  std::vector<std::shared_ptr<AsyncHooksWrap>> async_wraps_;
  v8::Isolate* v8_isolate_;
  Persistent<ObjectTemplate> async_hooks_templ;
  Persistent<Private> async_id_symbol;
  Persistent<Private> trigger_id_symbol;
 
  static void ShellPromiseHook(PromiseHookType type, Local<Promise> promise,
                               Local<Value> parent);
  static void PromiseHookDispatch(PromiseHookType type, Local<Promise> promise,
                                  Local<Value> parent,
                                  const AsyncHooksWrap& wrap,
                                  AsyncHooks* hooks);
 
  std::stack<AsyncContext> asyncContexts;
  async_id_t current_async_id;
  // We might end up in an invalid state after skipping steps due to
  // terminations.
  bool skip_after_termination_ = false;
};
 
}  // namespace v8
 
#endif  // V8_D8_ASYNC_HOOKS_WRAPPER_H_