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.
88 lines
1.5 KiB
88 lines
1.5 KiB
package base
|
|
|
|
import (
|
|
"base/component"
|
|
"base/config"
|
|
"base/etc"
|
|
"base/eventbus"
|
|
"base/log"
|
|
"base/task"
|
|
"base/utils/xfile"
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
"strconv"
|
|
"syscall"
|
|
)
|
|
|
|
type Container struct {
|
|
sig chan os.Signal
|
|
components []component.Component
|
|
}
|
|
|
|
// NewContainer 创建一个容器
|
|
func NewContainer() *Container {
|
|
return &Container{sig: make(chan os.Signal)}
|
|
}
|
|
|
|
// Add 添加组件
|
|
func (c *Container) Add(components ...component.Component) {
|
|
c.components = append(c.components, components...)
|
|
}
|
|
|
|
// Serve 启动容器
|
|
func (c *Container) Serve() {
|
|
log.Debugf("Welcome to the bzGame framework %s, Learn more at %s", Version, Website)
|
|
|
|
for _, comp := range c.components {
|
|
comp.Init()
|
|
}
|
|
|
|
for _, comp := range c.components {
|
|
comp.Start()
|
|
}
|
|
|
|
c.doSavePID()
|
|
|
|
switch runtime.GOOS {
|
|
case `windows`:
|
|
signal.Notify(c.sig, syscall.SIGINT, syscall.SIGTERM)
|
|
default:
|
|
signal.Notify(c.sig, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGABRT, syscall.SIGTERM)
|
|
}
|
|
|
|
sig := <-c.sig
|
|
|
|
log.Warnf("process got signal %v, container will close", sig)
|
|
|
|
signal.Stop(c.sig)
|
|
|
|
for _, comp := range c.components {
|
|
comp.Destroy()
|
|
}
|
|
|
|
if err := eventbus.Close(); err != nil {
|
|
log.Errorf("eventbus close failed: %v", err)
|
|
}
|
|
|
|
task.Release()
|
|
|
|
config.Close()
|
|
|
|
etc.Close()
|
|
|
|
log.Close()
|
|
|
|
}
|
|
|
|
func (c *Container) doSavePID() {
|
|
filename := etc.Get("etc.pid").String()
|
|
if filename == "" {
|
|
return
|
|
}
|
|
|
|
err := xfile.WriteFile(filename, []byte(strconv.Itoa(syscall.Getpid())))
|
|
if err != nil {
|
|
log.Fatalf("pid save failed: %v", err)
|
|
}
|
|
}
|
|
|