nodejs---require() 源碼解讀

原文鏈接:https://blog.csdn.net/a460550542/article/details/82694332

2009年,Node.js 項目誕生,所有模塊一律爲 CommonJS 格式。

時至今日,Node.js 的模塊倉庫 npmjs.com ,已經存放了15萬個模塊,其中絕大部分都是 CommonJS 格式。

這種格式的核心就是 require 語句,模塊通過它加載。學習 Node.js ,必學如何使用 require 語句。本文通過源碼分析,詳細介紹 require 語句的內部運行機制,幫你理解 Node.js 的模塊機制。

 

一、require() 的基本用法

分析源碼之前,先介紹 require 語句的內部邏輯。如果你只想瞭解 require 的用法,只看這一段就夠了。

下面的內容翻譯自《Node使用手冊》

當 Node 遇到 require(X) 時,按下面的順序處理。

(1)如果 X 是內置模塊(比如 require('http')) 
  a. 返回該模塊。 
  b. 不再繼續執行。

(2)如果 X 以 "./" 或者 "/" 或者 "../" 開頭 
  a. 根據 X 所在的父模塊,確定 X 的絕對路徑。 
  b. 將 X 當成文件,依次查找下面文件,只要其中有一個存在,就返回該文件,不再繼續執行。

  • X
  • X.js
  • X.json
  • X.node

  c. 將 X 當成目錄,依次查找下面文件,只要其中有一個存在,就返回該文件,不再繼續執行。

  • X/package.json(main字段)
  • X/index.js
  • X/index.json
  • X/index.node

(3)如果 X 不帶路徑 
  a. 根據 X 所在的父模塊,確定 X 可能的安裝目錄。 
  b. 依次在每個目錄中,將 X 當成文件名或目錄名加載。

(4) 拋出 "not found"

請看一個例子。

當前腳本文件 /home/ry/projects/foo.js 執行了 require('bar') ,這屬於上面的第三種情況。Node 內部運行過程如下。

首先,確定 x 的絕對路徑可能是下面這些位置,依次搜索每一個目錄。


 
  1.  
  2. /home/ry/projects/node_modules/bar

  3. /home/ry/node_modules/bar

  4. /home/node_modules/bar

  5. /node_modules/bar

搜索時,Node 先將 bar 當成文件名,依次嘗試加載下面這些文件,只要有一個成功就返回。


 
  1.  
  2. bar

  3. bar.js

  4. bar.json

  5. bar.node

如果都不成功,說明 bar 可能是目錄名,於是依次嘗試加載下面這些文件。


 
  1.  
  2. bar/package.json(main字段)

  3. bar/index.js

  4. bar/index.json

  5. bar/index.node

如果在所有目錄中,都無法找到 bar 對應的文件或目錄,就拋出一個錯誤。

二、Module 構造函數

瞭解內部邏輯以後,下面就來看源碼。

require 的源碼在 Node 的 lib/module.js 文件。爲了便於理解,本文引用的源碼是簡化過的,並且刪除了原作者的註釋。


 
  1.  
  2. function Module(id, parent) {

  3. this.id = id;

  4. this.exports = {};

  5. this.parent = parent;

  6. this.filename = null;

  7. this.loaded = false;

  8. this.children = [];

  9. }

  10.  
  11. module.exports = Module;

  12.  
  13. var module = new Module(filename, parent);

上面代碼中,Node 定義了一個構造函數 Module,所有的模塊都是 Module 的實例。可以看到,當前模塊(module.js)也是 Module 的一個實例。

每個實例都有自己的屬性。下面通過一個例子,看看這些屬性的值是什麼。新建一個腳本文件 a.js 。


 
  1.  
  2. // a.js

  3.  
  4. console.log('module.id: ', module.id);

  5. console.log('module.exports: ', module.exports);

  6. console.log('module.parent: ', module.parent);

  7. console.log('module.filename: ', module.filename);

  8. console.log('module.loaded: ', module.loaded);

  9. console.log('module.children: ', module.children);

  10. console.log('module.paths: ', module.paths);

運行這個腳本。


 
  1.  
  2. $ node a.js

  3.  
  4. module.id: .

  5. module.exports: {}

  6. module.parent: null

  7. module.filename: /home/ruanyf/tmp/a.js

  8. module.loaded: false

  9. module.children: []

  10. module.paths: [ '/home/ruanyf/tmp/node_modules',

  11. '/home/ruanyf/node_modules',

  12. '/home/node_modules',

  13. '/node_modules' ]

可以看到,如果沒有父模塊,直接調用當前模塊,parent 屬性就是 null,id 屬性就是一個點。filename 屬性是模塊的絕對路徑,path 屬性是一個數組,包含了模塊可能的位置。另外,輸出這些內容時,模塊還沒有全部加載,所以 loaded 屬性爲 false 。

新建另一個腳本文件 b.js,讓其調用 a.js 。


 
  1.  
  2. // b.js

  3.  
  4. var a = require('./a.js');

運行 b.js 。


 
  1.  
  2. $ node b.js

  3.  
  4. module.id: /home/ruanyf/tmp/a.js

  5. module.exports: {}

  6. module.parent: { object }

  7. module.filename: /home/ruanyf/tmp/a.js

  8. module.loaded: false

  9. module.children: []

  10. module.paths: [ '/home/ruanyf/tmp/node_modules',

  11. '/home/ruanyf/node_modules',

  12. '/home/node_modules',

  13. '/node_modules' ]

上面代碼中,由於 a.js 被 b.js 調用,所以 parent 屬性指向 b.js 模塊,id 屬性和 filename 屬性一致,都是模塊的絕對路徑。

三、模塊實例的 require 方法

每個模塊實例都有一個 require 方法。


 
  1.  
  2. Module.prototype.require = function(path) {

  3. return Module._load(path, this);

  4. };

由此可知,require 並不是全局性命令,而是每個模塊提供的一個內部方法,也就是說,只有在模塊內部才能使用 require 命令(唯一的例外是 REPL 環境)。另外,require 其實內部調用 Module._load 方法。

下面來看 Module._load 的源碼。


 
  1.  
  2. Module._load = function(request, parent, isMain) {

  3.  
  4. // 計算絕對路徑

  5. var filename = Module._resolveFilename(request, parent);

  6.  
  7. // 第一步:如果有緩存,取出緩存

  8. var cachedModule = Module._cache[filename];

  9. if (cachedModule) {

  10. return cachedModule.exports;

  11.  
  12. // 第二步:是否爲內置模塊

  13. if (NativeModule.exists(filename)) {

  14. return NativeModule.require(filename);

  15. }

  16.  
  17. // 第三步:生成模塊實例,存入緩存

  18. var module = new Module(filename, parent);

  19. Module._cache[filename] = module;

  20.  
  21. // 第四步:加載模塊

  22. try {

  23. module.load(filename);

  24. hadException = false;

  25. } finally {

  26. if (hadException) {

  27. delete Module._cache[filename];

  28. }

  29. }

  30.  
  31. // 第五步:輸出模塊的exports屬性

  32. return module.exports;

  33. };

上面代碼中,首先解析出模塊的絕對路徑(filename),以它作爲模塊的識別符。然後,如果模塊已經在緩存中,就從緩存取出;如果不在緩存中,就加載模塊。

因此,Module._load 的關鍵步驟是兩個。

  • Module._resolveFilename() :確定模塊的絕對路徑
  • module.load():加載模塊

四、模塊的絕對路徑

下面是 Module._resolveFilename 方法的源碼。


 
  1.  
  2. Module._resolveFilename = function(request, parent) {

  3.  
  4. // 第一步:如果是內置模塊,不含路徑返回

  5. if (NativeModule.exists(request)) {

  6. return request;

  7. }

  8.  
  9. // 第二步:確定所有可能的路徑

  10. var resolvedModule = Module._resolveLookupPaths(request, parent);

  11. var id = resolvedModule[0];

  12. var paths = resolvedModule[1];

  13.  
  14. // 第三步:確定哪一個路徑爲真

  15. var filename = Module._findPath(request, paths);

  16. if (!filename) {

  17. var err = new Error("Cannot find module '" + request + "'");

  18. err.code = 'MODULE_NOT_FOUND';

  19. throw err;

  20. }

  21. return filename;

  22. };

上面代碼中,在 Module.resolveFilename 方法內部,又調用了兩個方法 Module.resolveLookupPaths() 和 Module._findPath() ,前者用來列出可能的路徑,後者用來確認哪一個路徑爲真。

爲了簡潔起見,這裏只給出 Module._resolveLookupPaths() 的運行結果。


 
  1.  
  2. [ '/home/ruanyf/tmp/node_modules',

  3. '/home/ruanyf/node_modules',

  4. '/home/node_modules',

  5. '/node_modules'

  6. '/home/ruanyf/.node_modules',

  7. '/home/ruanyf/.node_libraries',

  8. '$Prefix/lib/node' ]

上面的數組,就是模塊所有可能的路徑。基本上是,從當前路徑開始一級級向上尋找 node_modules 子目錄。最後那三個路徑,主要是爲了歷史原因保持兼容,實際上已經很少用了。

有了可能的路徑以後,下面就是 Module._findPath() 的源碼,用來確定到底哪一個是正確路徑。


 
  1.  
  2. Module._findPath = function(request, paths) {

  3.  
  4. // 列出所有可能的後綴名:.js,.json, .node

  5. var exts = Object.keys(Module._extensions);

  6.  
  7. // 如果是絕對路徑,就不再搜索

  8. if (request.charAt(0) === '/') {

  9. paths = [''];

  10. }

  11.  
  12. // 是否有後綴的目錄斜槓

  13. var trailingSlash = (request.slice(-1) === '/');

  14.  
  15. // 第一步:如果當前路徑已在緩存中,就直接返回緩存

  16. var cacheKey = JSON.stringify({request: request, paths: paths});

  17. if (Module._pathCache[cacheKey]) {

  18. return Module._pathCache[cacheKey];

  19. }

  20.  
  21. // 第二步:依次遍歷所有路徑

  22. for (var i = 0, PL = paths.length; i < PL; i++) {

  23. var basePath = path.resolve(paths[i], request);

  24. var filename;

  25.  
  26. if (!trailingSlash) {

  27. // 第三步:是否存在該模塊文件

  28. filename = tryFile(basePath);

  29.  
  30. if (!filename && !trailingSlash) {

  31. // 第四步:該模塊文件加上後綴名,是否存在

  32. filename = tryExtensions(basePath, exts);

  33. }

  34. }

  35.  
  36. // 第五步:目錄中是否存在 package.json

  37. if (!filename) {

  38. filename = tryPackage(basePath, exts);

  39. }

  40.  
  41. if (!filename) {

  42. // 第六步:是否存在目錄名 + index + 後綴名

  43. filename = tryExtensions(path.resolve(basePath, 'index'), exts);

  44. }

  45.  
  46. // 第七步:將找到的文件路徑存入返回緩存,然後返回

  47. if (filename) {

  48. Module._pathCache[cacheKey] = filename;

  49. return filename;

  50. }

  51. }

  52.  
  53. // 第八步:沒有找到文件,返回false

  54. return false;

  55. };

經過上面代碼,就可以找到模塊的絕對路徑了。

有時在項目代碼中,需要調用模塊的絕對路徑,那麼除了 module.filename ,Node 還提供一個 require.resolve 方法,供外部調用,用於從模塊名取到絕對路徑。


 
  1.  
  2. require.resolve = function(request) {

  3. return Module._resolveFilename(request, self);

  4. };

  5.  
  6. // 用法

  7. require.resolve('a.js')

  8. // 返回 /home/ruanyf/tmp/a.js

五、加載模塊

有了模塊的絕對路徑,就可以加載該模塊了。下面是 module.load 方法的源碼。


 
  1.  
  2. Module.prototype.load = function(filename) {

  3. var extension = path.extname(filename) || '.js';

  4. if (!Module._extensions[extension]) extension = '.js';

  5. Module._extensions[extension](this, filename);

  6. this.loaded = true;

  7. };

上面代碼中,首先確定模塊的後綴名,不同的後綴名對應不同的加載方法。下面是 .js 和 .json 後綴名對應的處理方法。


 
  1.  
  2. Module._extensions['.js'] = function(module, filename) {

  3. var content = fs.readFileSync(filename, 'utf8');

  4. module._compile(stripBOM(content), filename);

  5. };

  6.  
  7. Module._extensions['.json'] = function(module, filename) {

  8. var content = fs.readFileSync(filename, 'utf8');

  9. try {

  10. module.exports = JSON.parse(stripBOM(content));

  11. } catch (err) {

  12. err.message = filename + ': ' + err.message;

  13. throw err;

  14. }

  15. };

這裏只討論 js 文件的加載。首先,將模塊文件讀取成字符串,然後剝離 utf8 編碼特有的BOM文件頭,最後編譯該模塊。

module._compile 方法用於模塊的編譯。


 
  1.  
  2. Module.prototype._compile = function(content, filename) {

  3. var self = this;

  4. var args = [self.exports, require, self, filename, dirname];

  5. return compiledWrapper.apply(self.exports, args);

  6. };

上面的代碼基本等同於下面的形式。


 
  1.  
  2. (function (exports, require, module, __filename, __dirname) {

  3. // 模塊源碼

  4. });

也就是說,模塊的加載實質上就是,注入exports、require、module三個全局變量,然後執行模塊的源碼,然後將模塊的 exports 變量的值輸出。

(完)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章