remove go worker stuff

Signed-off-by: Marc 'risson' Schmitt <marc.schmitt@risson.space>
This commit is contained in:
Marc 'risson' Schmitt
2025-06-19 19:10:38 +02:00
parent d18a54e9e6
commit 92a33a408f
3 changed files with 0 additions and 131 deletions

View File

@ -4,7 +4,6 @@ import (
"fmt"
"net/http"
"net/url"
"syscall"
"time"
"github.com/getsentry/sentry-go"
@ -20,7 +19,6 @@ import (
sentryutils "goauthentik.io/internal/utils/sentry"
webutils "goauthentik.io/internal/utils/web"
"goauthentik.io/internal/web"
"goauthentik.io/internal/worker"
)
var rootCmd = &cobra.Command{
@ -58,14 +56,6 @@ var rootCmd = &cobra.Command{
panic(err)
}
worker := worker.New()
if config.Get().Worker.Embedded {
err = worker.Start()
if err != nil {
panic(err)
}
}
ws := web.NewWebServer()
ws.Core().AddHealthyCallback(func() {
if config.Get().Outposts.DisableEmbeddedOutpost {
@ -77,7 +67,6 @@ var rootCmd = &cobra.Command{
<-ex
l.Info("shutting down webserver")
go ws.Shutdown()
go worker.Kill(syscall.SIGTERM)
},
}

View File

@ -7,7 +7,6 @@ type Config struct {
ErrorReporting ErrorReportingConfig `yaml:"error_reporting" env:", prefix=AUTHENTIK_ERROR_REPORTING__"`
Redis RedisConfig `yaml:"redis" env:", prefix=AUTHENTIK_REDIS__"`
Outposts OutpostConfig `yaml:"outposts" env:", prefix=AUTHENTIK_OUTPOSTS__"`
Worker WorkerConfig `yaml:"worker" env:", prefix=AUTHENTIK_WORKER__"`
// Config for core and embedded outpost
SecretKey string `yaml:"secret_key" env:"AUTHENTIK_SECRET_KEY, overwrite"`
@ -78,7 +77,3 @@ type OutpostConfig struct {
type WebConfig struct {
Path string `yaml:"path" env:"PATH, overwrite"`
}
type WorkerConfig struct {
Embedded bool `yaml:"embedded" env:"EMBEDDED, overwrite"`
}

View File

@ -1,115 +0,0 @@
package worker
import (
"fmt"
"os"
"os/exec"
"os/signal"
"runtime"
"syscall"
log "github.com/sirupsen/logrus"
"goauthentik.io/internal/config"
)
type Worker struct {
Healthcheck func() bool
HealthyCallback func()
log *log.Entry
p *exec.Cmd
pidFile string
started bool
killed bool
}
func New() *Worker {
logger := log.WithField("logger", "authentik.router.worker")
w := &Worker{
log: logger,
started: false,
killed: false,
}
w.initCmd()
c := make(chan os.Signal, 1)
signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
go func() {
for sig := range c {
switch sig {
case syscall.SIGHUP:
w.log.Info("SIGHUP received, forwarding to dramatiq")
w.Reload()
case syscall.SIGINT:
w.log.Info("SIGINT received, stopping dramatiq")
w.Kill(syscall.SIGINT)
case syscall.SIGTERM:
w.log.Info("SIGTERM received, stopping dramatiq")
w.Kill(syscall.SIGTERM)
}
}
}()
return w
}
func (w *Worker) initCmd() {
command := "./manage.py"
args := []string{"worker"}
if config.Get().Debug {
args = append(args, "--reload")
}
pidFile, err := os.CreateTemp("", "authentik-dramatiq.pid")
if err != nil {
panic(fmt.Errorf("failed to create temporary pid file: %v", err))
}
w.pidFile = pidFile.Name()
args = append(args, "--pid-file", w.pidFile)
w.log.WithField("args", args).WithField("cmd", command).Debug("Starting dramatiq")
w.p = exec.Command(command, args...)
w.p.Env = os.Environ()
w.p.Stdout = os.Stdout
w.p.Stderr = os.Stderr
}
func (w *Worker) Start() error {
if !w.started {
w.initCmd()
}
w.killed = false
w.started = true
return w.p.Run()
}
func (w *Worker) Reload() {
w.log.WithField("method", "reload").Info("reloading dramatiq")
err := w.p.Process.Signal(syscall.SIGHUP)
if err != nil {
w.log.WithError(err).Warning("failed to reload dramatiq")
}
}
func (w *Worker) Kill(sig syscall.Signal) {
if !w.started {
return
}
var err error
if runtime.GOOS == "darwin" {
w.log.WithField("method", "processKill").Warning("stopping dramatiq")
err = w.p.Process.Kill()
} else {
w.log.WithField("method", "syscallKill").Warning("stopping dramatiq")
err = syscall.Kill(w.p.Process.Pid, sig)
}
if err != nil {
w.log.WithError(err).Warning("failed to stop dramatiq")
}
if w.pidFile != "" {
err := os.Remove(w.pidFile)
if err != nil {
w.log.WithError(err).Warning("failed to remove pidfile")
}
}
w.killed = true
}