hmq/cmd/broker/main.go

package main

import (
	"log"
	"os"
	"os/signal"

	"github.com/fhmq/hmq/broker"
)

func main() {
	b, err := broker.NewBroker(nil)
	if err != nil {
		log.Fatal("New Broker error: ", err)
	}
	b.Start()

	s := waitForSignal()
	log.Println("signal received, broker closed.", s)
}

func waitForSignal() os.Signal {
	signalChan := make(chan os.Signal, 1)
	defer close(signalChan)
	signal.Notify(signalChan, os.Kill, os.Interrupt)
	s := <-signalChan
	signal.Stop(signalChan)
	return s
}