多进程打包利器：thread-loader 源码精讲（完结篇）

2023-11-24 19:56:31

多进程打包：thread-loader 源码（17） #

在小作文的最后一篇，让我们再见 thread-loader 源码的精彩之处。

了解了 thread-loader 的工作流程和关键步骤之后，我们来看看它的代码实现。由于篇幅原因，我将重点介绍一些关键部分。

// 引入所需的模块
const path = require('path');
const {
  getLoaders,
  LoaderOptionsPlugin,
  NormalModule,
} = require('webpack');

// 创建 thread-loader 类
class ThreadLoader {
  // 构造函数
  constructor(options) {
    // ... 省略其他代码 ...

    // 初始化 worker 进程
    this.workerPool = createWorkerPool(options.workers);
  }

  // apply 方法
  apply(compiler) {
    // ... 省略其他代码 ...

    // 注册 LoaderOptionsPlugin 插件
    compiler.options.plugins.push(
      new LoaderOptionsPlugin({
        // 在 thread-loader 选项中添加 workerPool
        threadLoader: {
          workerPool: this.workerPool,
        },
      })
    );

    // ... 省略其他代码 ...
  }

  // run 方法
  run(request, pitchLoaderData, callback) {
    // ... 省略其他代码 ...

    // 创建 NormalModule 实例
    const module = new NormalModule(
      {
        context: compiler.context,
        resource: request,
      },
      {
        // 将 compiler 和 thread-loader 选项传递给 NormalModule
        compiler,
        options: compiler.options,
        threadLoaderOptions: this.options,
      }
    );

    // ... 省略其他代码 ...
  }
}

// 创建 thread-loader 实例
const threadLoader = new ThreadLoader(options);

// 将 thread-loader 添加到 webpack 配置中
module.exports = threadLoader;