Introduction to Generators in ES6

DamCosset
995 views

Open Source Your Knowledge, Become a Contributor

Technology knowledge has to be shared and made accessible for free. Join the movement.

Create Content

Introduction

With ES6, we were given a new tool: generators. In a normal function, there is only one entry point: the invocation of the function itself. A generator allows you to pause the execution of a function and resume it later. Generators are useful when dealing with iterators and can simplify the asynchronous nature of Javascript.

Syntax

So, how would we define a generator, compared to a normal function? You declare a generator function by using the * ( asterisk ) operator after the function keyword:

function* awesomeGenerator(){
  //code
}

To pause the execution inside the generator, we use the statement yield:

function* awesomeGenerator(){
  yield 'Hey friend' // We pause the execution here
  console.log('Back again') // When we resume, we are here
}

next() method

A generator gives you a next() method, which is used to start/resume the execution. This method returns an object with two keys:

{
  value: [ yielded value ],
  done: [ true if we reach the end]
}

Let's see a very simple generator in action:

1
2
3
4
5
6
7
8
9
10
11
12
13
function* groceriesGenerator(){
yield 'Eggs'
yield 'Tomatoes'
yield 'Milk'
return 'Paper Bag'
}
const groceries = groceriesGenerator()
console.log(groceries.next()) // [1]
console.log(groceries.next()) // [2]
console.log(groceries.next()) // [3]
console.log(groceries.next()) // [4]
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

At our first groceries.next() call, our groceriesGenerator reached our first yield statement and paused, returning the value Eggs to the caller. Our second next() resumed the execution at the second yield statement, pausing it again and returning Tomatoes...

Our last next() terminates the generator, returns Paper Bag and sets done to true.

As iterators

In the introduction, I said that generators could help to implement iterators. Let's see an example:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
function* iterationGenerator( arr ){
for( let i = 0; i < arr.length; i++ ){
yield arr[i]
}
}
const iterator = iterationGenerator( ['John', 'Sarah', 'Joe', 'Emily'])
let current = iterator.next()
while( !current.done ){
console.log( current.value )
current = iterator.next()
}
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

In this example, you can see how the state of the generator is maintained accross invocations. Whne we resume the execution by calling next(), the variables and loops are the same.

Pass values back to generators

Finally, you can also pass a value back to a generator. Here is a example:

1
2
3
4
5
6
7
8
9
10
11
12
13
function* twoWayGenerator(arr){
for( let i = 0; i < arr.length; i++ ) {
const symbol = yield
console.log(`${arr[i]} ${symbol}`)
}
}
const prices = twoWayGenerator([10, 23, 12])
prices.next()
prices.next('€')
prices.next('£')
prices.next('$')
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

Here, our first next() will not print anything because we pause the execution before the first console.log. The second next() is called with an argument. This argument is provided as the returned value of the yield statement. We store this value inside the symbol variable and use it next.

You can also force a generator to throw an error:

1
2
3
4
5
6
7
8
9
10
11
12
function* twoWayGenerator(arr){
for( let i = 0; i < arr.length; i++ ) {
const symbol = yield
console.log(`${arr[i]} ${symbol}`)
}
}
const prices = twoWayGenerator([10, 23, 12])
prices.next()
prices.throw( new Error('Invalid Amount'))
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

That's it for a quick and dirty introduction about generators. In the next article, we will go a bit deeper and explore how generators can help us control the asynchronous flow of Javascript.

Open Source Your Knowledge: become a Contributor and help others learn. Create New Content