Create a generator in go - similar to yield in python

generator.go

package main
import "fmt"

func countGenerator() func() int{
	// state
	c := make(chan int)
	var i int

	// generator
	go func(){
		for {
			c <- i
			i++
		}
	}()

	// reader
	return func() int{
		return <-c
	}
}

func main() {
	c := countGenerator()
	c2 := countGenerator()
	fmt.Println(c(), c(), c())  // 0 1 2
	fmt.Println(c2(), c2(), c())  // 0 1 3
}