mirror of
https://github.com/fhmq/hmq.git
synced 2026-05-04 07:08:32 +00:00
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
/* Copyright (c) 2018, joy.zhou <chowyu08@gmail.com>
|
|
|
|
Permission to use, copy, modify, and/or distribute this software for any
|
|
purpose with or without fee is hereby granted, provided that the above
|
|
copyright notice and this permission notice appear in all copies.
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/fhmq/hmq/broker"
|
|
"github.com/fhmq/hmq/logger"
|
|
)
|
|
|
|
var (
|
|
log = logger.Get().Named("main")
|
|
)
|
|
|
|
func main() {
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
config, err := broker.ConfigureConfig(os.Args[1:])
|
|
if err != nil {
|
|
log.Fatal("configure broker config error: ", zap.Error(err))
|
|
}
|
|
|
|
b, err := broker.NewBroker(config)
|
|
if err != nil {
|
|
log.Fatal("New Broker error: ", zap.Error(err))
|
|
}
|
|
b.Start()
|
|
|
|
s := waitForSignal()
|
|
log.Info("signal received, broker closed.", zap.String("singal", s.String()))
|
|
}
|
|
|
|
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
|
|
}
|