工作流程
- 实例化 hook 注册事件监听
- 通过 hook 触发事件监听
- 运行懒编译的代码
Hook 分类
根据执行机制,hook 分为同步(Sync)和异步(Async)。异步 hook 根据执行方式又可以分为串行执行(Series)和并行执行(Parallel)。
根据执行特点,hook 又可以分为以下四类:
- Hook : 普通钩子,监听器之间互相不干扰。
- BailHook : 熔断钩子,某个监听返回非 undefined 时后续不执行。
- WaterfallHook : 瀑布钩子,上一个监听的返回值可以传递给下一个。
- LoopHook : 循环钩子,如果当前未返回 false 则一直执行。
同步钩子工作流程概述
- 导入钩子
- 导入钩子时,会自动创建一个对应的代码工厂 HookCodeFactory。
- 实例化钩子实例
- 创建一个 Hook 实例,传入形参,挂载args、taps、_x属性,重写构造函数,tap 方法,compile 方法,返回 hook 实例。
- 使用 tap 注册函数
- 把函数名、函数体合并成一个对象,将对象放入 taps 数组中。
- 触发事件
- 调用 call 方法,调用重写过的 compile 方法,
- 从 taps 数组中取出所有回调函数体,放入 _x 数组中。
- 从 _x 数组中取出事件函数,拼接成一段可执行代码。
- 传入实参到生成的可执行代码的自调用函数。
同步钩子源码实现
Hook类:
class Hook {
constructor(args = []) {
this.args = args
this.taps = [] // 将来用于存放组装好的 {}
this._x = undefined // 将来在代码工厂函数中会给 _x = [f1, f2, ...]
}
tap(options, fn) {
if (typeof options === 'string') {
options = { name: options }
}
options = Object.assign({fn}, options) // { fn: func, name: fn1 }
// 调用一下方法将组装好的 options 添加至 []
this._insert(options)
}
_insert(options) {
this.taps[this.taps.length] = options
}
call(...args) {
// 01 创建将来要具体执行的函数代码结构
let callFn = this._createCall()
// 02 调用上述的函数(args传入进去)
return callFn.apply(this, args)
}
_createCall() {
return this.compile({
taps: this.taps,
args: this.args
})
}
}
module.exports = Hook
SyncHook类和代码工厂类:
const Hook = require('./Hook')
class HookCodeFactory {
args() {
return this.options.args.join(',') // ['name', 'age'] -> 'name,age'
}
head() {
return `var _x = this._x;`
}
content() {
let code = ''
for (let i = 0; i < this.options.taps.length; i++) {
// code += `var _fn0 = _x[0];_fn0(name, age)`
code += `var _fn${i} = _x[${i}];_fn${i}(${this.args()});`
}
return code
}
setup(instance, options) { // 先准备后续需要使用的数据
this.options = options // 这里的操作在源码中是通过 init 方法实现,目前只是直接挂载在 this 上
instance._x = options.taps.map(t => t.fn) // this._x = [f1, f2,...]
}
create() { // 核心是创建一段可执行的代码体然后返回
let fn
fn = new Function(this.args(), this.head() + this.content())
return fn
}
}
let factory = new HookCodeFactory()
class SyncHook extends Hook {
constructor(args) {
super(args)
}
compile(options) { // options: { taps: [{}, {}...], args: [name, age] }
factory.setup(this, options)
return factory.create(options,)
}
}
module.exports = SyncHook
使用自实现 SyncHook :
const SyncHook = require('./SyncHook')
let hook = new SyncHook(['name', 'age'])
hook.tap('fn1', function (name, age) {
console.log('fn1--->', name, age)
})
hook.tap('fn2', function (name, age) {
console.log('fn2--->', name, age)
})
hook.call('zce', 18)
异步钩子工作流程概述
异步钩子工作流程大致与同步钩子相同,主要有两个区别:
- 拼接代码时,代码工厂类的 args 方法传入了 after 参数,值为 ‘_callback’,这样一来生成的代码的自调用函数就会多一个形参 _callback。
- 生成的自调用函数有一个计数器 _counter,一个最后执行的 done 方法,done 方法调用 _callback 参数传进来的函数。
异步钩子源码实现
Hook 类:
class Hook {
constructor(args = []) {
this.args = args
this.taps = [] // 将来用于存放组装好的 {}
this._x = undefined // 将来在代码工厂函数中会给 _x = [f1, f2, ...]
}
tapAsync(options, fn) {
if (typeof options === 'string') {
options = { name: options }
}
options = Object.assign({fn}, options) // { fn: func, name: fn1 }
// 调用一下方法将组装好的 options 添加至 []
this._insert(options)
}
_insert(options) {
this.taps[this.taps.length] = options
}
callAsync(...args) {
// 01 创建将来要具体执行的函数代码结构
let callFn = this._createCall()
// 02 调用上述的函数(args传入进去)
return callFn.apply(this, args)
}
_createCall() {
return this.compile({
taps: this.taps,
args: this.args
})
}
}
module.exports = Hook
AsyncParallelHook 和 工厂类:
const Hook = require('./Hook')
class HookCodeFactory {
args({ before, after } = {}) {
let allArgs = this.options.args
if (before) allArgs = [before].concat(allArgs)
if (after) allArgs = allArgs.concat(after)
return allArgs.join(',')
}
head() {
return `"use strict";var _x = this._x;var _content;`
}
content() {
let code = `var _counter = ${this.options.taps.length}; var _done = (function () { _callback(); });`
for (let i = 0; i < this.options.taps.length; i++) {
code += `var _fn${i} = _x[${i}]; _fn${i}(${this.args()}, (function () { if (--_counter === 0) _done(); }));`
}
return code
}
setup(instance, options) { // 先准备后续需要使用的数据
this.options = options // 这里的操作在源码中是通过 init 方法实现,目前只是直接挂载在 this 上
instance._x = options.taps.map(t => t.fn) // this._x = [f1, f2,...]
}
create() { // 核心是创建一段可执行的代码体然后返回
let fn
fn = new Function(this.args({ after: '_callback' }), this.head() + this.content())
return fn
}
}
let factory = new HookCodeFactory()
class AsyncParallelHook extends Hook {
constructor(args) {
super(args)
}
compile(options) { // options: { taps: [{}, {}...], args: [name, age] }
factory.setup(this, options)
return factory.create(options)
}
}
module.exports = AsyncParallelHook
使用自实现 AsyncParallelHook:
const AsyncParallelHook = require('./AsyncParallelHook')
let hook = new AsyncParallelHook(['name', 'age'])
hook.tapAsync('fn1', function (name, age, callback) {
console.log('fn1--->', name, age)
callback()
})
hook.tapAsync('fn2', function (name, age, callback) {
console.log('fn2--->', name, age)
callback()
})
hook.callAsync('zce', 18, function () {
console.log('end')
})