Go ping pong with timeout context

pingpong.go

package main

import (
	"context"
	"fmt"
	"sync"
	"time"
)

func pingPong(ctx context.Context, player string, wg *sync.WaitGroup, in <-chan struct{}, out chan<- struct{}) {
PingPongLoop:
	for {
		select {
		case <-in:
			fmt.Println(player)
			select {
			case out <- struct{}{}:
			default:
			}
		case <-ctx.Done():
			close(out)
			fmt.Printf("Terminated %q\n", player)
			break PingPongLoop
		}
	}
	wg.Done()
}

func main() {
	var wg sync.WaitGroup
	wg.Add(2)

	timeoutCtx, timeoutCancel := context.WithTimeout(context.Background(), time.Second)
	defer timeoutCancel()

	pingChan := make(chan struct{})
	pongChan := make(chan struct{})

	go pingPong(timeoutCtx, "ping", &wg, pingChan, pongChan)
	go pingPong(timeoutCtx, "pong", &wg, pongChan, pingChan)

	pingChan <- struct{}{}

	wg.Wait()
}