Go言語でPHPのarray_chunkのようなことをする

例えば、

[]string{"a", "b", "c", "d", "e", "f", "g", "h"}

のような1次元のスライスを

[][]string{
	[]string{"a", "b", "c"}, 
	[]string{"d", "e", "f"},
	[]string{"g", "h"},
}

のように3個ずつのスライスに分割した2次元スライスにしたい。

実装してみた

main.go

package main

import (
	"log"
)

func main() {
	slice := []string{"a", "b", "c", "d", "e", "f", "g", "h"}
	chunkedSlice := chunkStringSlice(slice, 3)
	log.Printf("%#v", chunkedSlice)
}

func chunkStringSlice(slice []string, size int) [][]string {
	var chunks [][]string
	
	sliceSize :=  len(slice)
	
	for i := 0; i < sliceSize; i += size {
		end := i+size
		if sliceSize < end {
			end = sliceSize
		}
		chunks = append(chunks, slice[i:end])
	}
	
	return chunks
}

実行結果

2013/12/12 17:20:23 [][]string{[]string{"a", "b", "c"}, []string{"d", "e", "f"}, []string{"g", "h"}}

課題

文字列型のスライスにのみ使える関数になってしまった。スライスに対する操作しかないので、もう少し汎用度高めたい。