internal: fix gunicorn not being restarted correctly

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2021-07-17 16:59:31 +02:00
parent be9ca48de0
commit 7cbe33d65d

View file

@ -11,30 +11,40 @@ import (
type GoUnicorn struct {
log *log.Entry
p *exec.Cmd
started bool
}
func NewGoUnicorn() *GoUnicorn {
logger := log.WithField("logger", "authentik.g.unicorn")
g := &GoUnicorn{
log: logger,
started: false,
}
g.initCmd()
return g
}
func (g *GoUnicorn) initCmd() {
command := "gunicorn"
args := []string{"-c", "./lifecycle/gunicorn.conf.py", "authentik.root.asgi:application"}
if config.G.Debug {
command = "python"
args = []string{"manage.py", "runserver", "localhost:8000"}
}
logger.WithField("args", args).WithField("cmd", command).Debug("Starting gunicorn")
p := exec.Command(command, args...)
p.Env = append(os.Environ(),
g.log.WithField("args", args).WithField("cmd", command).Debug("Starting gunicorn")
g.p = exec.Command(command, args...)
g.p.Env = append(os.Environ(),
"WORKERS=2",
)
p.Stdout = os.Stdout
p.Stderr = os.Stderr
return &GoUnicorn{
log: logger,
p: p,
}
g.p.Stdout = os.Stdout
g.p.Stderr = os.Stderr
}
func (g *GoUnicorn) Start() error {
if g.started {
g.initCmd()
}
g.started = true
return g.p.Run()
}