Iterator&Generator - ChoDragon9/posts GitHub Wiki

Iterator

내부 표현부를 노출하지 않고 어떤 객체 집합에 속한 원소들을 순차적으로 접근할 수 있는 방법을 제공하는 패턴이다.

Iterable Protocol

순차적으로 접근할 수 있는 Iterator를 접근하기 위한 프로토콜

Java에서 Iterator를 구현할 때는 hasNext/next 인터페이스를 사용하지만 Javascript에서는 next만 사용된다.

죵료된 것은 done property가 true일 때로 확인 가능하다.

const str = "hi"
const iterator = str[Symbol.iterator]()

console.log(iterator.next()) //{ value: 'h', done: false }
console.log(iterator.next()) //{ value: 'i', done: false }
console.log(iterator.next()) //{ value: undefined, done: true }

Iterator Protocol

데이터를 순차적으로 접근할 때 실행되는 프로토콜

function makeIterator(array){
  var nextIndex = 0;

  return {
    next: function(){ //next 메소드를 가져야 함
      return nextIndex < array.length ?
        {value: array[nextIndex++], done: false} :
        {done: true};
         //done : iterator의 반복 작업 종료 유무. boolean 사용
         //value : iterator로 부터 반환되는 값, done이 true일 때 생략 가능
      }
  };
}

var it = makeIterator(['yo', 'ya']);

console.log(it.next().value); // 'yo'
console.log(it.next().value); // 'ya'
console.log(it.next().done); // true

Customize Iterables

Object는 Iterable 하지 않지만 Iterator/Iterable Protocol를 통해 Iterable 하게 만들 수 있다.

const obj = {
  name: 'Smith',
  age: 20,
  weight: 60
}

console.log([...obj]) //TypeError

obj[Symbol.iterator] = () => {
  const arr = Object.keys(obj)
  return {
    next () {
      return arr.length > 0 ?
        { value: obj[arr.pop()], done: false } :
        { done: true }
      }
    }
}

console.log([...obj]) // [60, 20, 'Smith']

Using Generator

const obj = {
  name: 'Smith',
  age: 20,
  weight: 60
}

obj[Symbol.iterator] = function* () {
  const arr = Object.keys(obj)
  yield obj[arr.pop()]
  yield obj[arr.pop()]
  yield obj[arr.pop()]
}

console.log([...obj]) // [60, 20, 'Smith']

Iterable data sources

Arrays, Maps, DOM queies, Strings, Sets, Arguments

Generator

yield로 Iterator를 쉽게 정의하여 Iterable를 쉽게 만들 수 있음. 코루틴을 지원함.

function*

함수 실행 시 Generator Object를 반환한다. Generator Object의 인터페이스는 next, return, throw가 있다.

next를 호출하게 되면 yield의 값이 반환되며, yield는 Generator를 일시정지 시킨다.

function* genFour () {
  yield 1
  yield 2
  yield 3
  return 4
}
let four = getFour()

Generators are both an iterator and an iterable.

four.next() // Object { value: 1, done: false)
four.next() // Object { value: 2, done: false)
four.next() // Object { value: 3, done: false)
four.next() // Object { value: 4, done: true)

[....getFour()] // Array [1, 2, 3, 4]

yield* yields every yield inside a called generator

function* flatten(arr){
  for (let x of arr) {
    if (x instanceof Array){
      yield* flatten(x)
    } else {
      yield x
    }
  }
}
let t = flatten([1, 2, [3, 4]])

cursor 예시

const log = txt => {
  console.log(`[cursor] ${txt}`)
}

function* getFour () {
  log(`getFour: 1`)
  yield 1
  log(`getFour: 2`)
  yield 2
  log(`getFour: 3`)
  yield 3
  log(`getFour: 4`)
  yield 4
}

for (const num of getFour()) {
  log(`num: ${num}`)
}

// "[cursor] getFour: 1"
// "[cursor] num: 1"
// "[cursor] getFour: 2"
// "[cursor] num: 2"
// "[cursor] getFour: 3"
// "[cursor] num: 3"
// "[cursor] getFour: 4"
// "[cursor] num: 4"