Load Balancing Channel Example

loadbalancing-channel.go

package main

import (
	"fmt"
	"sync"
)

func publisher() <-chan int {
	c := make(chan int)

	go func() {
		for i := 1; i <= 1000; i++ {
			c <- i
		}

		close(c)
	}()

	return c
}

func consumer(c <-chan int, name string, wg *sync.WaitGroup) {
	counter := 0

	for value := range c {
		fmt.Printf("Consumer %s is doing task %d\n", name, value)
		counter++
	}

	fmt.Printf("Consumer %s has finished %d task(s)\n", name, counter)
	wg.Done()
}

func main() {
	myChan := publisher()
	maxConsumer := 5
	var wg sync.WaitGroup

	wg.Add(maxConsumer)
	for i := 1; i <= maxConsumer; i++ {
		go consumer(myChan, fmt.Sprintf("%d", i), &wg)
	}
	wg.Wait()
}