mirror of
https://github.com/yusing/godoxy.git
synced 2026-02-16 15:37:42 +01:00
45f856b8d1
Replaced hardcoded 10-second initialization timeout with a configurable `INIT_TIMEOUT` environment variable. The new default is 1 minute, allowing operators to adjust startup behavior based on their infrastructure requirements.
112 lines
2.7 KiB
Go
Executable File
112 lines
2.7 KiB
Go
Executable File
package main
|
|
|
|
import (
|
|
"os"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/yusing/godoxy/internal/api"
|
|
"github.com/yusing/godoxy/internal/auth"
|
|
"github.com/yusing/godoxy/internal/common"
|
|
"github.com/yusing/godoxy/internal/config"
|
|
"github.com/yusing/godoxy/internal/dnsproviders"
|
|
iconlist "github.com/yusing/godoxy/internal/homepage/icons/list"
|
|
"github.com/yusing/godoxy/internal/logging"
|
|
"github.com/yusing/godoxy/internal/logging/memlogger"
|
|
"github.com/yusing/godoxy/internal/metrics/systeminfo"
|
|
"github.com/yusing/godoxy/internal/metrics/uptime"
|
|
"github.com/yusing/godoxy/internal/net/gphttp/middleware"
|
|
"github.com/yusing/godoxy/internal/route/rules"
|
|
gperr "github.com/yusing/goutils/errs"
|
|
"github.com/yusing/goutils/server"
|
|
"github.com/yusing/goutils/task"
|
|
"github.com/yusing/goutils/version"
|
|
)
|
|
|
|
func parallel(fns ...func()) {
|
|
var wg sync.WaitGroup
|
|
for _, fn := range fns {
|
|
wg.Go(fn)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func main() {
|
|
done := make(chan struct{}, 1)
|
|
go func() {
|
|
select {
|
|
case <-done:
|
|
return
|
|
case <-time.After(common.InitTimeout):
|
|
log.Fatal().Msgf("timeout waiting for initialization to complete, exiting...")
|
|
}
|
|
}()
|
|
|
|
initProfiling()
|
|
|
|
logging.InitLogger(os.Stderr, memlogger.GetMemLogger())
|
|
log.Info().Msgf("GoDoxy version %s", version.Get())
|
|
log.Trace().Msg("trace enabled")
|
|
parallel(
|
|
dnsproviders.InitProviders,
|
|
iconlist.InitCache,
|
|
systeminfo.Poller.Start,
|
|
middleware.LoadComposeFiles,
|
|
)
|
|
|
|
if common.APIJWTSecret == nil {
|
|
log.Warn().Msg("API_JWT_SECRET is not set, using random key")
|
|
common.APIJWTSecret = common.RandomJWTKey()
|
|
}
|
|
|
|
for _, dir := range common.RequiredDirectories {
|
|
prepareDirectory(dir)
|
|
}
|
|
|
|
err := config.Load()
|
|
if err != nil {
|
|
gperr.LogWarn("errors in config", err)
|
|
}
|
|
|
|
config.StartProxyServers()
|
|
|
|
if err := auth.Initialize(); err != nil {
|
|
log.Fatal().Err(err).Msg("failed to initialize authentication")
|
|
}
|
|
rules.InitAuthHandler(auth.AuthOrProceed)
|
|
|
|
// API Handler needs to start after auth is initialized.
|
|
server.StartServer(task.RootTask("api_server", false), server.Options{
|
|
Name: "api",
|
|
HTTPAddr: common.APIHTTPAddr,
|
|
Handler: api.NewHandler(true),
|
|
})
|
|
|
|
// Local API Handler is used for unauthenticated access.
|
|
if common.LocalAPIHTTPAddr != "" {
|
|
server.StartServer(task.RootTask("local_api_server", false), server.Options{
|
|
Name: "local_api",
|
|
HTTPAddr: common.LocalAPIHTTPAddr,
|
|
Handler: api.NewHandler(false),
|
|
})
|
|
}
|
|
|
|
listenDebugServer()
|
|
|
|
uptime.Poller.Start()
|
|
config.WatchChanges()
|
|
|
|
close(done)
|
|
|
|
task.WaitExit(config.Value().TimeoutShutdown)
|
|
}
|
|
|
|
func prepareDirectory(dir string) {
|
|
if _, err := os.Stat(dir); os.IsNotExist(err) {
|
|
if err = os.MkdirAll(dir, 0o755); err != nil {
|
|
log.Fatal().Msgf("failed to create directory %s: %v", dir, err)
|
|
}
|
|
}
|
|
}
|