62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"golang.org/x/sys/unix"
|
|
|
|
"github.com/containerd/containerd/log"
|
|
"github.com/containerd/containerd/reaper"
|
|
"github.com/containerd/containerd/sys"
|
|
"github.com/urfave/cli"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const (
|
|
defaultConfigPath = "/etc/containerd/config.toml"
|
|
)
|
|
|
|
var (
|
|
handledSignals = []os.Signal{unix.SIGTERM, unix.SIGINT, unix.SIGUSR1, unix.SIGCHLD}
|
|
)
|
|
|
|
func platformInit(context *cli.Context) error {
|
|
if conf.Subreaper {
|
|
log.G(global).Info("setting subreaper...")
|
|
if err := sys.SetSubreaper(1); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if conf.OOMScore != 0 {
|
|
log.G(global).Infof("changing OOM score to %d", conf.OOMScore)
|
|
if err := sys.SetOOMScore(os.Getpid(), conf.OOMScore); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if err := os.MkdirAll(conf.State, 0750); err != nil {
|
|
return err
|
|
}
|
|
if err := os.Chown(conf.State, conf.GRPC.Uid, conf.GRPC.Gid); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func handleSignals(signals chan os.Signal, server *grpc.Server) error {
|
|
for s := range signals {
|
|
log.G(global).WithField("signal", s).Debug("received signal")
|
|
switch s {
|
|
case unix.SIGCHLD:
|
|
if err := reaper.Reap(); err != nil {
|
|
log.G(global).WithError(err).Error("reap containerd processes")
|
|
}
|
|
case unix.SIGUSR1:
|
|
dumpStacks()
|
|
default:
|
|
server.Stop()
|
|
return nil
|
|
}
|
|
}
|
|
return nil
|
|
}
|