package httpserver import ( "context" "io/ioutil" "net/http" "strings" "time" "github.com/labstack/echo" "github.com/labstack/echo/middleware" "github.com/rs/zerolog" "go.dev.pztrn.name/giredore/internal/configuration" "go.dev.pztrn.name/giredore/internal/logger" ) var ( log zerolog.Logger Srv *echo.Echo ) func Initialize() { log = logger.Logger.With().Str("type", "internal").Str("package", "httpserver").Logger() log.Info().Msg("Initializing...") Srv = echo.New() Srv.Use(middleware.Recover()) Srv.Use(requestLogger()) Srv.Use(checkAllowedIPs()) Srv.DisableHTTP2 = true Srv.HideBanner = true Srv.HidePort = true Srv.Binder = echo.Binder(&StrictJSONBinder{}) Srv.GET("/_internal/waitForOnline", waitForHTTPServerToBeUpHandler) } // Shutdown stops HTTP server. Returns true on success and false on failure. func Shutdown() { log.Info().Msg("Shutting down HTTP server...") err := Srv.Shutdown(context.Background()) if err != nil { log.Fatal().Err(err).Msg("Failed to stop HTTP server") } log.Info().Msg("HTTP server shutted down") } // Start starts HTTP server and checks that server is ready to process // requests. Returns true on success and false on failure. func Start() { log.Info().Str("address", configuration.Cfg.HTTP.Listen).Msg("Starting HTTP server...") go func() { err := Srv.Start(configuration.Cfg.HTTP.Listen) if !strings.Contains(err.Error(), "Server closed") { log.Fatal().Err(err).Msg("HTTP server critical error occurred") } }() // Check that HTTP server was started. // nolint:exhaustruct httpc := &http.Client{Timeout: time.Second * 1} checks := 0 for { checks++ if checks >= configuration.Cfg.HTTP.WaitForSeconds { log.Fatal().Int("seconds passed", checks).Msg("HTTP server isn't up") } time.Sleep(time.Second * 1) localCtx, cancelFunc := context.WithTimeout(context.Background(), time.Second*1) req, err := http.NewRequestWithContext(localCtx, "GET", "http://"+configuration.Cfg.HTTP.Listen+"/_internal/waitForOnline", nil) if err != nil { log.Panic().Err(err).Msg("Failed to create HTTP request!") } resp, err := httpc.Do(req) if err != nil { log.Debug().Err(err).Msg("HTTP error occurred, HTTP server isn't ready, waiting...") continue } response, err := ioutil.ReadAll(resp.Body) resp.Body.Close() if err != nil { log.Debug().Err(err).Msg("Failed to read response body, HTTP server isn't ready, waiting...") continue } log.Debug().Str("status", resp.Status).Int("body length", len(response)).Msg("HTTP response received") if resp.StatusCode == http.StatusOK { if len(response) == 0 { log.Debug().Msg("Response is empty, HTTP server isn't ready, waiting...") continue } log.Debug().Int("status code", resp.StatusCode).Msgf("Response: %+v", string(response)) if len(response) == 17 { // This is useless context cancel function call. Thanks to lostcancel linter. cancelFunc() break } } } log.Info().Msg("HTTP server is ready to process requests") } func waitForHTTPServerToBeUpHandler(ectx echo.Context) error { response := map[string]string{ "error": "None", } // nolint:wrapcheck return ectx.JSON(200, response) }