给大家变个Node.js的小魔术

本文转载自微信公众号「神光的编程秘籍」，作者神说要有光。转载本文请联系神光的编程秘籍公众号。

网站建设哪家好，找成都创新互联公司！专注于网页设计、网站建设、微信开发、小程序定制开发、集团企业网站建设等服务项目。为回馈新老客户创新互联还提供了莲池免费建站欢迎大家使用！

魔术演出

我们准备一个 Node.js 的模块 input.js：

 
 
 
 
  
  
  
  // input.js   
  
  
  function func() {   
  
  
      return '卡颂'   
  
  
  }   
  
  
  module.exports = func();

这个模块返回的值是啥?

东东：是“卡颂”。

那我在另一个模块 test.js 中引入这个 input.js，然后打印一下：

 
 
 
 
  
  
  
  // test.js   
  
  
  const data = require('./input.js');   
  
  
  console.log(data);

之后我在 entry.js 里面引入 test.js:

 
 
 
 
  
  
  
  require('./test.js');

执行之后打印的是啥?

东东：是“卡颂”。

真的么?那我们跑一下：

打印的是啥：

东东：是 “卡帅”，哇，好神奇，怎么做到的。

我：想不想学?

东东：想。

我：那接下来就进入魔术揭秘时间。

魔术揭秘

Node.js 加载模块的流程是这样的：

模块加载会调用 load 方法， load 会调用对应后缀名的 _extensions 的方法来处理，其中会调用 _compile 来编译并把结果放入 cache，之后返回。

所以呢?我们想改变 js 模块的返回值，只需要改造下 Module._extensions['.js'] 就可以了。

 
 
 
 
  
  
  
  const Module = require('module');   
  
  
  const fs = require('fs');   
  
  
     
  
  
  Module._extensions['.js'] = function (module, filename) {   
  
  
      let content = fs.readFileSync(filename, 'utf8');   
  
  
      if (filename.includes('input')) {   
  
  
          content = content.replace('卡颂', '卡帅');   
  
  
      }   
  
  
      module._compile(content, filename);   
  
  
  };

我们对 filename 为 input 的文件，读取内容之后进行了替换，之后再调用 module._compile 来编译，后续流程不变。

模块引入方式不变，但是模块内容已经悄悄的被修改了，这个魔术的名字叫做 require hook。

东东：原来是你藏了一段代码没展示。

我：魔术都是这样的啊。而且你别小看了这个 require hook，它能做到很多强大的功能呢。

东东：哦?比如说

我：比如说 ts-node，它是怎么做到直接 require ts 模块的?就是通过 require hook 偷偷做了编译，其实你执行的是编译后的 js。

比如说 babel-register 它是怎么做到直接执行带有 esnext 新特性的代码的?也是通过 require hook 偷偷做了编译。

还有覆盖率测试，其实是通过函数插桩做到的，也就是你每执行一条语句都会计数。怎么插桩呢?跑单测的时候也没手动插桩啊，就是因为工具内部偷偷通过 require hook 做了插桩，才能得到覆盖率数据。

东东：这个魔术还挺有用的嘛。学会了～

总结

Node.js 的 js 模块加载的流程是 load -> _extensions['.js'] -> _compile，可以通过修改 _extensions['.js'] 来达到 hook 的目的，比如在 _compile 之前做一些代码转换。

这种 hook 在 babel-register、ts-node 还有单测的覆盖率测试中都有应用，能够达到透明的修改代码的目的。

因为开发者不知道代码什么时候被修改的，所以看起来比较神奇。

网页名称：给大家变个Node.js的小魔术
文章位置：http://www.csdahua.cn/qtweb/news14/539614.html

网站建设、网络推广公司-快上网，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源：快上网

成都快上网为您推荐相关内容