ProducerConsumer In Go

producer_consumer.go

package main

import (
	"fmt"
	"strconv"
	"sync"
	"time"
)

const MaxItems = 100

type Message struct {
	id   int
	desc string
}

var msgChan = make(chan Message)

func NewMessage(i int) Message {
	fmt.Println("Producing message :: " + strconv.Itoa(i))
	return Message{
		id:   i,
		desc: fmt.Sprintf("this is message no %d", i),
	}
}

func consumeMessage(m chan Message) {
	for {
		fmt.Println("Consuming :: " + (<-m).desc)
	}
}

func produceMessage(w *sync.WaitGroup, m chan Message) {
	defer w.Done()
	for i := 0; i < MaxItems; i++ {
		m <- NewMessage(i)
		time.Sleep(time.Millisecond)
	}
}

func main() {
	var wg sync.WaitGroup
	wg.Add(1)
	go produceMessage(&wg, msgChan)
	go consumeMessage(msgChan)
	wg.Wait()
	close(msgChan)
}