2024年前端最全手写generator核心原理，再也不怕面试官问我generator原理(1)，2024年最新前端面试题

最后

喜欢的话别忘了关注、点赞哦~

开源分享：【大厂前端面试题解析+核心总结学习笔记+真实项目实战+最新讲解视频】

for (let v of foo()) {

console.log(v);

}

其中foo()是迭代器对象，可以把它赋值给变量，然后遍历这个变量。

function* foo() {

yield 1;

yield 2;

yield 3;

yield 4;

yield 5;

return 6;

}

let a = foo();

for (let v of a) {

console.log(v);

}

// 1 2 3 4 5

上面代码使用for…of循环，依次显示5个yield语句的值。这里需要注意，一旦next方法的返回对象的done属性为true，for…of循环就会中止，且不包含该返回对象，所以上面代码的return语句返回的6，不包括在for…of循环之中。

下面是一个利用Generator函数和for…of循环，实现斐波那契数列的例子。

斐波那契数列是什么？它指的是这样一个数列 0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144…

这个数列前两项是0和1，从第3项开始，每一项都等于前两项之和。

function* fibonacci() {

let [prev, curr] = [0, 1];

for (;😉 { // 这里请思考：为什么这个循环不设定结束条件？

[prev, curr] = [curr, prev + curr];

yield curr;

}

}

for (let n of fibonacci()) {

if (n > 1000) {

break;

}

console.log(n);

}

2.手写generator核心原理

---

我们从一个简单的例子开始，一步步探究Generator的实现原理：

function* foo() {

yield ‘result1’

yield ‘result2’

yield ‘result3’

}

const gen = foo()

console.log(gen.next()) //{value: “result1”, done: false}

console.log(gen.next()) //{value: “result2”, done: false}

console.log(gen.next()) //{value: “result3”, done: false}

console.log(gen.next()) //{value: undefined, done: true}

看到这种整齐的结构，我想起了switch case，也是这么地整齐，所以这两种之间应该存在一种关系。

我们尝试写一个用switch/case来实现下：

function gen$(nextStep) {

while (1) {

switch (nextStep) {

case 0:

return ‘result1’;

case 2:

return ‘result2’;

case 4:

return ‘result3’;

case 6:

return undefined;

}

}

}

如代码所示，我们每次调用gen$然后传对应的参数，就能返回对应的值（也就是原本函数yield后面的值）

但是nextStep应该是一个自动增加的函数，应该不是我们传进去的。所以这里应该用一个闭包来实现

function gen$() {

var nextStep = 0

return function () {

while (1) {

switch (nextStep) {

case 0:

nextStep = 2;

return ‘result1’;

case 2:

nextStep = 4;

return ‘result2’;

case 4:

nextStep = 6;

return ‘result3’;

case 6:

return undefined

}

}

}

}

现在我们可以通过

var a = gen$()

获得内函数。

这样每次执行

a()

nextStep就会改成下一次执行a()应该对应的值，并且返回相应的result了。

但是generator的底层原理不是用闭包的。而是用一个全局变量，因为这样为了后面的实现方便很多，为了遵循原理，我们改成用全局变量来实现。

先定义一个全局变量

context = {

prev:0,

next:0

}

function gen$(context) {

while (1) {

switch (context.prev = context.next) {

case 0:

context.next = 2;

return ‘result1’;

case 2:

context.next = 4;

return ‘result2’;

case 4:

context.next = 6;

return ‘result3’;

case 6:

return undefined

}

}

}

第一次执行gen$(context),swtich判断的时候，是用prev来判断这一次应该执行那个case，执行case时再改变next的值，next表示下次应该执行哪个case。第二次执行gen$(context)的时候，将next的值赋给prev。

但是直接返回这么一个值是不对的。我们看前面的例子是返回一个对象。那该怎么实现呢？

再把例子搬下来：

function* foo() {

yield ‘result1’

yield ‘result2’

yield ‘result3’

}

const gen = foo()

console.log(gen.next()) //{value: “result1”, done: false}

console.log(gen.next()) //{value: “result2”, done: false}

console.log(gen.next()) //{value: “result3”, done: false}

console.log(gen.next()) //{value: undefined, done: true}

我们发现 gen 有next这个方法。所以可以判断出 执行foo返回的应该是一个对象，这个对象有next这个方法。所以我们初步实现foo的转化后的函数。

let foo = function () {

return {

next: function () {

}

}

}

而每次执行next，就会返回拥有value和done的对象，

所以，可以完善返回值

let foo = function () {

return {

next: function () {

return {

value,

done

}

}

}

}

但是我们这里还没定义这value和done啊，该怎么定义呢？

我们先看value的实现。我们在上面实现gen 的 时 候 ， 就 发 现 它 返 回 的 是 v a l u e 了 。 所 以 可 以 在 这 里 获 取 ‘ 的时候，就发现它返回的是value了。所以可以在这里获取` 的时候，就发现它返回的是value了。所以可以在这里获取‘gen`的返回值作为value。

let foo = function () {

return {

next: function () {

value = gen$(context)

return {

value,

done

}

}

}

}

那done怎么定义呢？

其实done作为一个全局状态表示generator是否执行结束，因此，我们可以在

context里定义，默认值为false。

var context = {

next:0,

prev: 0,

done: false,

}

所以，每次返回，直接返回context.done就可以了

let foo = function () {

return {

next: function () {

value = gen$(context);

done = context.done

return {

value,

done

}

}

}

}

那done是怎么改变为true的。我们知道，generator执行到后面，就会返回done:true。我们可以看例子的第四个执行结果

function* foo() {

yield ‘result1’

yield ‘result2’

yield ‘result3’

}

const gen = foo()

console.log(gen.next()) //{value: “result1”, done: false}

console.log(gen.next()) //{value: “result2”, done: false}

console.log(gen.next()) //{value: “result3”, done: false}

console.log(gen.next()) //{value: undefined, done: true}

因此，我们需要在最后一次执行gen$的时候改变context.done的值。

思路，给context添加一个stop方法。用来改变自身的done为true。在执行$gen的时时候让context执行stop就好

var context = {

next:0,

prev: 0,

done: false,

新增代码

stop: function stop () {

this.done = true

}

}

function gen$(context) {

while (1) {

switch (context.prev = context.next) {

case 0:

context.next = 2;

return ‘result1’;

case 2:

context.next = 4;

return ‘result2’;

case 4:

context.next = 6;

return ‘result3’;

case 6:

新增代码

context.stop();

return undefined

}

}

}

let foo = function () {

return {

next: function () {

value = gen$(context);

done = context.done

return {

value,

done

}

}

}

}

这样执行到case为6的时候就会改变done的值了。

实际上这就是generator的大致原理

并不难理解，我们分析一下流程：

我们定义的function*生成器函数被转化为以上代码

转化后的代码分为三大块：

gen$(_context)由yield分割生成器函数代码而来

context对象用于储存函数执行上下文

迭代器法定义next()，用于执行gen$(_context)来跳到下一步

从中我们可以看出，「Generator实现的核心在于上下文的保存，函数并没有真的被挂起，每一次yield，其实都执行了一遍传入的生成器函数，只是在这个过程中间用了一个context对象储存上下文，使得每次执行生成器函数的时候，都可以从上一个执行结果开始执行，看起来就像函数被挂起了一样」

3.参照源码实现Context类

---

不过，我们这里的context是个全局对象啊？我们都知道如果是下面这种情况：

function* g() {

var o = 1;

yield o++;

yield o++;

yield o++;

}

var gen = g();

console.log(gen.next()); // 1

var xxx = g();

console.log(gen.next()); // 2

console.log(xxx.next()); // 1

console.log(gen.next()); // 3

我们发现 每个迭代器之间互不干扰，作用域独立。

也就是说每个迭代器的context是独立的。但是与我们目前实现的一个全局context不一致，这个我是百思不得其解，所以看下源码。

利用babel将下面代码转化一下

function* foo() {

yield ‘result1’

yield ‘result2’

yield ‘result3’

}

我们可以在babel官网上在线转化这段代码，看看ES5环境下是如何实现Generator的：

“use strict”;

var _marked =

/#PURE/

regeneratorRuntime.mark(foo);

function foo() {

return regeneratorRuntime.wrap(function foo$(_context) {

while (1) {

switch (_context.prev = _context.next) {

case 0:

_context.next = 2;

return ‘result1’;

case 2:

_context.next = 4;

return ‘result2’;

case 4:

_context.next = 6;

return ‘result3’;

case 6:

case “end”:

return _context.stop();

}

}

}, _marked);

}

看源码，你可能觉得跟我们实现的有点不一样，实际上结构是基本一样的，基本都是分成那三部分

发现源码是将我们的gen$(context)方法传入了wrap中。

我们看下wrap方法

function wrap(innerFn, outerFn, self) {

var generator = Object.create(outerFn.prototype);

var context = new Context([]);

generator._invoke = makeInvokeMethod(innerFn, self, context);

return generator;

}

发现它是每生foo()执行一次 ，就会执行一次wrap方法，而在wrap方法里就会new 一个Context对象。这就说明了每个迭代器的context是独立的。

Soga原来如此~~~

也就是说如果我们要实现独立context还是 把context改成一个类。

在执行var gen = g();的时候再生成context实例即可：

class Context {

constructor() {

this.next = 0

this.prev = 0

this.done = false

}

top() {

this.done = true

}

}

let foo = function () {

var context = new Context() 新增代码

return {

next: function () {

value = gen$(context);

done = context.done

return {

value,

done

}

}

}

}

4.参照源码实现参数值的保存

---

好了，这个独立context问题解决。但是发现哈有一个问题：

function* foo() {

var a = yield ‘result1’

console.log(a);

yield ‘result2’

yield ‘result3’

}

const gen = foo()

console.log(gen.next().value)

console.log(gen.next(222).value)

console.log(gen.next().value)

我们发现这里用var a 来接收传入的参数。

当我们第一次执行gen.next()，foo内部会执行到yield这里。还没给a赋值

当我们第二次执行gen.next()，foo内部会再第一个yield这里执行。把传入的参数222赋值给a

。

那原理是怎么实现的呢？我依旧百思不得其解，不得不再看下源码。

最后

开源分享：【大厂前端面试题解析+核心总结学习笔记+真实项目实战+最新讲解视频】

uctor() {

this.next = 0

this.prev = 0

this.done = false

}

top() {

this.done = true

}

}

let foo = function () {

var context = new Context() 新增代码

return {

next: function () {

value = gen$(context);

done = context.done

return {

value,

done

}

}

}

}

4.参照源码实现参数值的保存

---

好了，这个独立context问题解决。但是发现哈有一个问题：

function* foo() {

var a = yield ‘result1’

console.log(a);

yield ‘result2’

yield ‘result3’

}

const gen = foo()

console.log(gen.next().value)

console.log(gen.next(222).value)

console.log(gen.next().value)

我们发现这里用var a 来接收传入的参数。

当我们第一次执行gen.next()，foo内部会执行到yield这里。还没给a赋值

当我们第二次执行gen.next()，foo内部会再第一个yield这里执行。把传入的参数222赋值给a

。

那原理是怎么实现的呢？我依旧百思不得其解，不得不再看下源码。

最后

开源分享：【大厂前端面试题解析+核心总结学习笔记+真实项目实战+最新讲解视频】

[外链图片转存中…(img-PY3txVg3-1715534795868)]