JavaScript 中的闭包

121 阅读5分钟

闭包 (closure) 是程序语言的一种特性,在 JavaScript 中也扮演相当重要的角色,被广泛应用在 JavaScript 程序库中。许多被开发者大量使用的重要功能,也都看得到闭包的身影,举例来说最热门的 JavaScript 函数库 React 中的 useState 就是通过闭包来实现。

以面试的角度来说,不仅仅要了解什么是闭包,同时也要知道闭包会有什么应用的情景。假如你目前仍不熟闭包的概念,或不确定可以怎么应用,千万要在面试前弄熟。

什么是闭包?

MDN 文档中,闭包被定义为函数以及该函数被声明时所在的作用域环境 (lexical environment) 的组合。白话一点说,闭包就是内部函数能够取得函数外部的变量,并且记住这个变量。因为能够记住这个外部变量,闭包很常被用来做状态保存。

以下是一个最简单的例子,在下方代码中的 inner 函数,能拿到外部函数 outer 的 a 变量,并将其保存在内存中。当我们调用 inner 时,之所以不是每次都返回 1 ,而是返回 123 不断加上去,正是因为之前的 a 的状态被记住了。

function outer() {
  let a = 0;
  function inner() {
    a += 1;
    console.log(a);
  }
  return inner;
}

const inner = outer();

inner(); // 1
inner(); // 2
inner(); // 3

我们可以理解成:闭包这种特性,可以让我们在一个内层函数中 (这边的 inner),访问到外部函数的作用域 (这边的 outer),并且会记住外部函数的变量 (这边的 a)。

在了解闭包是什么后,接着我们来看看闭包的实际用途。

闭包的应用 1 — 状态保存

在写程序时,我们很常会需要记住某个状态,JavaScript 的热门函数库 React 就有提供一个 useState 让开发者来管理状态。以下我们模拟一个简化版的 useState ,可以在下方的代码看到, getStatesetState 作为内部函数,可以取得外部函数当中的 state,在实际调用后,如果这个 state 有改变, getState 可以持续取得最新改变的值。

// 因为闭包的关系,getState 与 setState 可以取得与记得 state
function useState(initialState) {
  let state = initialState;

  function getState() {
    return state;
  }

  function setState(updatedState) {
    state = updatedState;
  }
  return [getState, setState];
}

const [count, setCount] = useState(0);

count(); // 0
setCount(1);
count(); // 1
setCount(500);
count(); // 500

又或者先前 React 核心团队成员 Sebastian Markbåge 分享的一段代码,在说 React 的 Server Actions 中,可以运用闭包来做版本检查。下面这段代码 verifiedVersion 记住的是在首次渲染时拿到的版本,因为闭包的关系,内部的函数 publish 函数,能取得 verifiedVersion

这时,如果要做版本检查,可以在 publish 里面再调用一次 await getVersion,拿到当下的版本。这时就可以比较首次渲染时的版本,以及当下的版本,并当版本不同时,可以做处理 (这边是返回一个错误信息)

image

闭包的应用 2:缓存机制

因为闭包可以让内部函数记住外部的变量,我们可以依照这个特性,通过闭包来实现缓存机制。以下面的例子来说,因为闭包原理, cache 变量可以被返回的箭头函数取得与记得,所以我们能够重复用 cache 来放想要缓存的东西。

function cached(fn) {
  const cache = {};

  // 被返回的箭头函数,可以取得外面的 cache 变量,同时记住这个变量
  // 因此可以把这个 cache 拿来存已经计算的结果
  return (...args) => {
    // 把输入字符串化并当成 key
    const key = JSON.stringify(args);

    // 如果 key 已经在 cache 中,则不用重复算,直接返回之前存的运算结果
    // 如果 key 还不在,则运算完后,把 key 与运算结果放到 cache 中,未来可以避免重复运算
    if (key in cache) {
      return cache[key];
    } else {
      const val = fn(...args);
      cache[key] = val;
      return val;
    }
  };
}

闭包的应用 3:模拟私有变量

许多程序语言有声明私有方法的语法,这些私有变量对于外部来讲是隐藏的,这是一项很重要的特性,因为有时候我们在开发的代码内部细节,并不想让外部来获取。JavaScript 并不支持私有变量,但我们可以通过闭包做出类似的功能。如下方代码示例:

// privateCounter 没被法被外部修改,
// 因为闭包的关系 increment 与 decrement 可以存取到 privateCounter
// 因此 privateCounter 只能够通过 increment 与 decrement 来改,这能有效避免被误触到
var counter = (function () {
  var privateCounter = 0;
  function changeBy(val) {
    privateCounter += val;
  }
  return {
    increment: function () {
      changeBy(1);
    },
    decrement: function () {
      changeBy(-1);
    },
    value: function () {
      return privateCounter;
    },
  };
})();

console.log(counter.value()); // logs 0
counter.increment();
counter.increment();
console.log(counter.value()); // logs 2
counter.decrement();
console.log(counter.value()); // logs 1

闭包缺点 — 内存泄漏

虽然说闭包很好用,但也不是没有缺点。从内存的角度来看,闭包的缺点是显而易见的,由于闭包会让内部函数记得外部的变量,这可能会造成变量常驻在内存当中,如果使用过多可能会造成内存泄露 (memory leak),需要小心使用。

以下面的例子来说, longArray 没有被使用到,但是因为闭包的原因会一直被 addNumbers 记住。假如今天 longArray 有被使用,那就没问题,但因为它没有被用到但仍存在于内存中没被清除,这种情况就是典型的内存泄漏。

function outer() {
  const longArray = [];
  return function inner(num) {
    longArray.push(num);
  };
}
const addNumbers = outer();

for (let i = 0; i < 100000000; i++) {
  addNumbers(i);
}