Golang batch get from buffered channel with a maximum number.

testbatch_test.go

package testbatch

import (
	"testing"
	"time"
)

func TestBatch(t *testing.T) {
	bufCh := make(chan int, 20)
	go func() {
		for {
			var items []int
			items = append(items, <-bufCh)

			// As we need to batch get maximum 10 items, 9 remains to get.
			remains := 9

		Remaining:
			for i := 0; i < remains; i++ {
				select {
				case item := <-bufCh:
					items = append(items, item)
				default:
					break Remaining
				}
			}

			// The batch processing. Here we just log output.
			t.Log("Items:", items)
		}
	}()

	for i := 0; i < 50; i++ {
		bufCh <- i
		t.Log("Push:", i)
	}

	time.Sleep(time.Second)
}