a05daeda2e
* Basic setup to serve HTTPS * Correctly handle the case of missing TLS configs * Documenting TLS * Refactor TLS configuration setup * Add TLS Encryption section again to README * Extending TOC in README * Moving TLS settings to subsection of web settings * Adding tests for config/web * Add test for handling TLS * Rename some variables as suggested * Corrected error formatting * Update test module import * Polishing the readme file * Error handling for TLSConfig() --------- Co-authored-by: TwiN <twin@linux.com>
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/TwiN/gatus/v5/config"
|
|
"github.com/TwiN/gatus/v5/controller/handler"
|
|
)
|
|
|
|
var (
|
|
// server is the http.Server created by Handle.
|
|
// The only reason it exists is for testing purposes.
|
|
server *http.Server
|
|
)
|
|
|
|
// Handle creates the router and starts the server
|
|
func Handle(cfg *config.Config) {
|
|
var router http.Handler = handler.CreateRouter(cfg)
|
|
if os.Getenv("ENVIRONMENT") == "dev" {
|
|
router = handler.DevelopmentCORS(router)
|
|
}
|
|
tlsConfig, err := cfg.Web.TLSConfig()
|
|
if err != nil {
|
|
panic(err) // Should be unreachable, because the config is validated before
|
|
}
|
|
|
|
server = &http.Server{
|
|
Addr: fmt.Sprintf("%s:%d", cfg.Web.Address, cfg.Web.Port),
|
|
TLSConfig: tlsConfig,
|
|
Handler: router,
|
|
ReadTimeout: 15 * time.Second,
|
|
WriteTimeout: 15 * time.Second,
|
|
IdleTimeout: 15 * time.Second,
|
|
}
|
|
log.Println("[controller][Handle] Listening on " + cfg.Web.SocketAddress())
|
|
if os.Getenv("ROUTER_TEST") == "true" {
|
|
return
|
|
}
|
|
if tlsConfig != nil {
|
|
log.Println("[controller][Handle]", server.ListenAndServeTLS("", ""))
|
|
} else {
|
|
log.Println("[controller][Handle]", server.ListenAndServe())
|
|
}
|
|
}
|
|
|
|
// Shutdown stops the server
|
|
func Shutdown() {
|
|
if server != nil {
|
|
_ = server.Shutdown(context.TODO())
|
|
server = nil
|
|
}
|
|
}
|