You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.1 KiB
59 lines
1.1 KiB
package main |
|
|
|
import ( |
|
"flag" |
|
"os" |
|
"os/signal" |
|
"syscall" |
|
|
|
"go-common/app/job/main/reply/conf" |
|
"go-common/app/job/main/reply/http" |
|
"go-common/app/job/main/reply/service" |
|
"go-common/library/exp/feature" |
|
"go-common/library/log" |
|
"go-common/library/net/trace" |
|
) |
|
|
|
var ( |
|
s *service.Service |
|
) |
|
|
|
func main() { |
|
feature.DefaultGate.AddFlag(flag.CommandLine) |
|
flag.Parse() |
|
if err := conf.Init(); err != nil { |
|
log.Error("conf.Init() error(%v)", err) |
|
panic(err) |
|
} |
|
log.Init(conf.Conf.XLog) |
|
trace.Init(conf.Conf.Tracer) |
|
defer trace.Close() |
|
defer log.Close() |
|
log.Info("reply_consumer start") |
|
s = service.New(conf.Conf) |
|
http.Init(conf.Conf, s) |
|
signalHandler() |
|
} |
|
|
|
func signalHandler() { |
|
var ( |
|
err error |
|
ch = make(chan os.Signal, 1) |
|
) |
|
signal.Notify(ch, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT) |
|
for { |
|
si := <-ch |
|
switch si { |
|
case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT: |
|
log.Info("get a signal %s, stop the consume process", si.String()) |
|
if err = s.Close(); err != nil { |
|
log.Error("close consumer error(%v)", err) |
|
} |
|
s.Wait() |
|
return |
|
case syscall.SIGHUP: |
|
default: |
|
return |
|
} |
|
} |
|
}
|
|
|