main.go

package main

import (
	"fmt"

	"github.com/ragul28/rabbitmq-benchmark/queue"
	"github.com/ragul28/rabbitmq-benchmark/utils"
)

func main() {

	cfg := utils.LoadFlags()

	fmt.Println("Runing as", cfg.Role)

	switch cfg.Role {
	case "consumer":
		// Start consumer worker threads using goroutine
		for w := 1; w <= cfg.NumWorker; w++ {
			fmt.Printf("Consumer Worker %d started..\n", w)
			go queue.ConsumerMQ(cfg)
		}

	case "producer":
		// Start publisher worker threads using goroutine
		for w := 1; w <= cfg.NumWorker; w++ {
			fmt.Printf("Publisher Worker %d started..\n", w)
			go queue.PublishMQ(cfg)
		}

	default:
		fmt.Printf("Not a valid role, Please choose consumer / publisher")
	}

	utils.CloserHandler()

	// Block main thread to allow running goroutines forever.
	select {}
}