steam chunk for golang

steam_chunk.go

package main

import (
	"encoding/json"
	"os"
)

func main() {
	stream := make(chan int, 1)

	go func() {
		defer close(stream)
		for i := 0; i < 6; i++ {
			stream <- i
		}
	}()

	const maxLength = 5
	const minLength = 1

	result := make([][]int, 0, 10)
	buffer := make([]int, 0, maxLength)

	for i := range stream {
		if len(buffer) == maxLength-1 {
			result = append(result, append(buffer, i))
			buffer = make([]int, 0, maxLength)
			continue
		}
		buffer = append(buffer, i)
	}

	if len(buffer) >= minLength {
		result = append(result, buffer)
	}

	_ = json.NewEncoder(os.Stdout).Encode(result)
}