DEV Community

Soft Heart Engineer
Soft Heart Engineer

Posted on

Tree Shaking in JS

Understanding Tree Shaking in JavaScript: A Complete Guide

In the world of modern web development, optimizing your application's performance is crucial. One powerful technique to achieve smaller and faster JavaScript bundles is Tree Shaking. If you’ve worked with build tools like Webpack or Rollup, you’ve likely heard this term. But what exactly is tree shaking, and how does it work?

In this in-depth guide, we’ll explore tree shaking in JavaScript, understand its importance, see practical examples, and learn how to implement it effectively. Whether you’re a beginner or an experienced developer, this article is tailored to help you grasp the core concepts and apply them to your projects.


What Is Tree Shaking?

Tree shaking is a process of eliminating unused or dead code from your JavaScript bundles during the build process. The term originates from the idea of "shaking the tree" of code and removing the branches (code) that aren’t being used.

Tree shaking primarily works with ES6 (ES2015) module syntax, which provides static structure for imports and exports. This static nature allows bundlers to determine which parts of the code are used and which can be safely removed.


Why Is Tree Shaking Important?

Tree shaking is a critical optimization technique that can significantly improve the performance of your web application. Here are some key benefits:

  1. Reduced Bundle Size: By removing unused code, your JavaScript bundles are smaller, which reduces load times.
  2. Improved Performance: Smaller bundles mean less code to parse, compile, and execute in the browser.
  3. Better Developer Experience: Cleaner, optimized codebase and easier debugging.
  4. Lower Bandwidth Costs: Smaller files reduce bandwidth consumption, which is particularly beneficial for mobile users.

How Tree Shaking Works

Tree shaking relies on static analysis of ES6 module imports and exports. It identifies code that is not used and eliminates it during the build process. To enable tree shaking, the following requirements must be met:

  1. ES6 Module Syntax:
    Tree shaking only works with import and export. CommonJS (require/module.exports) is dynamic and cannot be statically analyzed.

  2. Bundlers with Tree Shaking Support:
    Tools like Webpack, Rollup, or Parcel support tree shaking.

  3. Proper Configuration:
    Ensure your bundler is configured correctly to remove unused code.


Example: Basic Tree Shaking in Action

Let’s walk through an example of tree shaking.

Step 1: Write a Module with Multiple Exports

Create a module file, math.js:

// math.js
export const add = (a, b) => a + b;
export const subtract = (a, b) => a - b;
export const multiply = (a, b) => a * b;
export const divide = (a, b) => a / b;
Enter fullscreen mode Exit fullscreen mode

Step 2: Import Only What You Need

In another file, main.js, use only the add function:

// main.js
import { add } from './math.js';

console.log(add(5, 3)); // Output: 8
Enter fullscreen mode Exit fullscreen mode

Step 3: Bundle Your Code

Use a bundler like Webpack or Rollup. With tree shaking enabled, the unused functions (subtract, multiply, divide) will be excluded from the final bundle.


Tree Shaking with Webpack

Step 1: Set Up Webpack

Install Webpack and Babel:

npm install --save-dev webpack webpack-cli babel-loader @babel/core @babel/preset-env
Enter fullscreen mode Exit fullscreen mode

Step 2: Configure Webpack

Create a webpack.config.js file:

const path = require('path');

module.exports = {
  mode: 'production', // Enables optimizations like tree shaking
  entry: './
Enter fullscreen mode Exit fullscreen mode
src/main.js', // Your main JavaScript file
  output: {
    filename: 'bundle.js',
    path: path.resolve(__dirname, 'dist'),
  },
  module: {
    rules: [
      {
        test: /\.js$/,
        exclude: /node_modules/,
        use: {
          loader: 'babel-loader',
          options: {
            presets: ['@babel/preset-env'],
          },
        },
      },
    ],
  },
  optimization: {
    usedExports: true, // Tells Webpack to identify unused exports
  },
};
Enter fullscreen mode Exit fullscreen mode

Step 3: Run Webpack

Run the build command:

npx webpack
Enter fullscreen mode Exit fullscreen mode

Webpack will automatically remove unused exports like subtract, multiply, and divide from the final bundle.


Tree Shaking with Rollup

Rollup is another popular bundler known for its excellent tree-shaking capabilities.

Step 1: Install Rollup

Install Rollup and Babel:

npm install --save-dev rollup @rollup/plugin-babel @rollup/plugin-node-resolve
Enter fullscreen mode Exit fullscreen mode

Step 2: Configure Rollup

Create a rollup.config.js file:

import babel from '@rollup/plugin-babel';
import resolve from '@rollup/plugin-node-resolve';

export default {
  input: 'src/main.js',
  output: {
    file: 'dist/bundle.js',
    format: 'es', // Keep the ES module format
  },
  plugins: [
    resolve(),
    babel({
      babelHelpers: 'bundled',
      presets: ['@babel/preset-env'],
    }),
  ],
};
Enter fullscreen mode Exit fullscreen mode

Step 3: Run Rollup

Run the Rollup build:

npx rollup -c
Enter fullscreen mode Exit fullscreen mode

The final bundle will exclude unused exports, just like Webpack.


Common Issues with Tree Shaking

Tree shaking can sometimes fail due to specific coding practices or misconfigurations. Here are some common issues:

  1. Dynamic Imports:
    If you use dynamic imports (e.g., require() or dynamic import()), tree shaking won’t work as bundlers can’t statically analyze them.

  2. Side Effects:
    Some modules perform actions just by being imported.
    For example:

    // module.js
    console.log('This is a side effect');
    export const example = () => {};
    

    Bundlers won’t remove such imports by default unless you explicitly mark them as “side-effect-free” in package.json:

    {
    "sideEffects": false
    }
    
  3. Incorrect Module Format:
    Ensure your dependencies support ES6 module syntax. If they use CommonJS, tree shaking won’t work.

  4. Dead Code Not Removed:
    Check if your bundler’s optimization settings (usedExports in Webpack or proper plugins in Rollup) are enabled.


Best Practices for Effective Tree Shaking

  • Write Modular Code: Use ES6 modules (import and export) instead of CommonJS.

  • Mark Side Effects: Declare side-effect-free modules in your package.json file.

  • Choose the Right Tools: Use bundlers like Rollup for library development and Webpack for applications.

  • Keep Dependencies in Check: Minimize the number of third-party libraries, as they may include unused code.

  • Enable Production Mode: Tree shaking is often only performed in production mode. Ensure your bundler is set to production.


Conclusion

Tree shaking is an essential technique for modern JavaScript development. By removing unused code, it helps optimize your application’s performance and reduces bundle sizes. With tools like Webpack and Rollup, implementing tree shaking is straightforward as long as you follow best practices and use ES6 modules.

Understanding how tree shaking works and addressing its limitations will ensure you’re building efficient and performant web applications. Start shaking your code today and enjoy faster, leaner builds!


Further Reading

Top comments (0)