feat: travse module graph

Author: SyMind

bench.config.js

@@ -1,16 +1,17 @@
 export default {
 	jobs: [
-		"10000_development-mode",
-		"10000_development-mode_hmr",
+		// "10000_development-mode",
+		// "10000_development-mode_hmr",
 		"10000_production-mode",
-		"10000_big_production-mode_disable-minimize",
-		"arco-pro_development-mode",
-		"arco-pro_development-mode_hmr",
-		"arco-pro_production-mode",
-		"arco-pro_production-mode_generate-package-json-webpack-plugin",
-		"threejs_development-mode_10x",
-		"threejs_development-mode_10x_hmr",
-		"threejs_production-mode_10x"
+		"10000_production-mode_traverse-module-graph",
+		// "arco-pro_development-mode",
+		// "arco-pro_development-mode_hmr",
+		// "arco-pro_production-mode",
+		// "arco-pro_production-mode_generate-package-json-webpack-plugin",
+		// "threejs_development-mode_10x",
+		// "threejs_development-mode_10x_hmr",
+		// "threejs_production-mode_10x",
+		// "10000_big_production-mode",
 	],
 	rspackDirectory: process.env.RSPACK_DIR
 };

lib/addons/traverse-module-graph/index.js

@@ -0,0 +1,17 @@
+import path from "path";
+import { Addon } from "../common.js";
+
+export default class extends Addon {
+	options = {
+		times: 10
+	};
+	async afterSetup(ctx) {
+		ctx.config =
+			ctx.config +
+			`
+module.exports.plugins = module.exports.plugins || [];
+const TraverseModuleGraphPlugin = require("${path.join(__dirname, 'plugin.js')}");
+module.exports.plugins.push(new TraverseModuleGraphPlugin());
+`;
+	}
+}

lib/addons/traverse-module-graph/plugin.js

@@ -0,0 +1,33 @@
+const PLUGIN_NAME = "TraverseModuleGraphPlugin";
+
+class TraverseModuleGraphPlugin {
+	apply(compiler) {
+		compiler.hooks.compilation.tap(PLUGIN_NAME, compilation => {
+			compilation.hooks.finishModules.tap(PLUGIN_NAME, () => {
+				const entries = compilation.entries.values();
+
+				const visitedModules = new Set();
+
+				function traverse(dependency) {
+					const module = compilation.moduleGraph.getModule(dependency);
+					if (module) {
+						visitedModules.add(module);
+						for (const dep of module.dependencies) {
+							traverse(dep)
+						}
+					}
+				}
+
+				for (const entry of entries) {
+					for (const dependency of entry.dependencies) {
+						traverse(dependency)
+					}
+				}
+
+				console.log("Visited module number: ", visitedModules.size);
+			});
+		});
+	}
+}
+
+module.exports = TraverseModuleGraphPlugin;