Go で Iterator を書く

golangでのイテレーターの実装方法を
ググって見つけたのがこちらの記事
http://ewencp.org/blog/golang-iterators/
こちらにある4種類の方法が主な方法かと思いますが、
気に入ったのは、
チャンネルを使ったパターンとコールバックを使ったパターンの2つ。

チャンネルパターン

コードの書き方としては、
このパターンが一番しっくりくるのですが、、

チャンネルパターン

func IntChannelIterator() <-chan int {
    ch := make(chan int)
    go func() {
        for _, val := range int_data {
            ch <- val
        }
        close(ch) // Remember to close or the loop never ends!
    }()
    return ch
}

チャンネルパターン使用例

var sum int = 0
for val := range IntChannelIterator() {
    sum += val
}

他のパターンと比べると致命的に遅いのが欠点です。
以下は、上記ページにあったコードをgo ver1.4.1を使って
MacBook Pro (Retina, Mid 2012) で実測結果です。

$ go version
go version go1.4.1 darwin/amd64
$ go test -bench .
testing: warning: no tests to run
PASS
BenchmarkIntsCallbackIterator	     500	   2733180 ns/op
BenchmarkDataCallbackIterator	     500	   3142514 ns/op
BenchmarkIntsChannelIterator	       5	 269527633 ns/op
BenchmarkDataChannelIterator	       5	 266374326 ns/op
BenchmarkIntsBufferedChannelIterator	      20	  94554939 ns/op
BenchmarkDataBufferedChannelIterator	      20	  93447309 ns/op
BenchmarkIntsClosureIterator	     300	   5030297 ns/op
BenchmarkDataClosureIterator	     300	   4855251 ns/op
BenchmarkIntStatefulIterator	    1000	   2333693 ns/op
BenchmarkDataStatefulIterator	     500	   2490458 ns/op
BenchmarkIntStatefulIteratorInterface	     200	   6054515 ns/op
BenchmarkDataStatefulIteratorInterface	     200	   6797890 ns/op
ok  	github.com/masahide/golang-iterators-benchmark	26.820s

ここまで速度差が出てしまうと、繰り返し処理としては致命的かもしれません。。
チャンネルバッファを使ってもまだまだ・・・。

コールバックパターン

そこで、速さと、コードの書きやすさを両立したコールバックパターンです。

コールバックパターン

func IntCallbackIterator(cb func(int)) {
    for _, val := range int_data {
        cb(val)
    }
}

コールバックパターンの使用例

var sum int = 0
IntCallbackIterator(func(val int) {
    sum += val
})

その他のパターン

なんというか中途半端感があって興味が出ません。。
スピードにこだわるならStatefulパターンですね。でもコールバックとそれほど違わないので、うーん。