diff --git a/cmd/main.go b/cmd/main.go index e91d1bbd..3b32ad80 100755 --- a/cmd/main.go +++ b/cmd/main.go @@ -38,7 +38,7 @@ func main() { select { case <-done: return - case <-time.After(time.Second * 10): + case <-time.After(common.InitTimeout): log.Fatal().Msgf("timeout waiting for initialization to complete, exiting...") } }() diff --git a/internal/common/env.go b/internal/common/env.go index f181d0c3..172f45df 100644 --- a/internal/common/env.go +++ b/internal/common/env.go @@ -13,6 +13,8 @@ var ( IsDebug = env.GetEnvBool("DEBUG", IsTest) IsTrace = env.GetEnvBool("TRACE", false) && IsDebug + InitTimeout = env.GetEnvDuation("INIT_TIMEOUT", 1*time.Minute) + ShortLinkPrefix = env.GetEnvString("SHORTLINK_PREFIX", "go") ProxyHTTPAddr,