Producer Consumer

producer_consumer.go

package main

import (
	"fmt"
	"time"
	"math/rand"
	"sync"
)

type Message struct {
	Title		string
	Content		int
}

func sleep() {
	randomTime := rand.Intn(5)
	time.Sleep(time.Duration(randomTime) * time.Second)
}

func produc(m chan Message, wg *sync.WaitGroup) {
	for i := 0; i < 10; i++ {
		m <- Message{ Title: "message", Content: i }
		fmt.Println("Produced message ", i)
		sleep()
	}
	fmt.Println("Finished producing messages")
	wg.Done()
}

func consum(m chan Message, wg *sync.WaitGroup) {
	for i := 0; i < 10; i++ {
		receivedValue := <- m
		sleep()
		fmt.Println("Message received ", receivedValue.Content)
	}
	fmt.Println("Finished consuming messages")
	wg.Done()
}

func main() {
	messageChannel := make(chan Message, 3)
	
	var wg sync.WaitGroup
	wg.Add(2)

	go consum(messageChannel, &wg)
	go produc(messageChannel, &wg)

	wg.Wait()
	fmt.Println("Finished producer and consumer")
}