mirror of
https://github.com/dcarrillo/whatismyip.git
synced 2024-12-23 02:48:01 +00:00
133 lines
3.0 KiB
Go
133 lines
3.0 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"errors"
|
||
|
"log"
|
||
|
"net/http"
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
"syscall"
|
||
|
|
||
|
"github.com/dcarrillo/whatismyip/internal/httputils"
|
||
|
"github.com/dcarrillo/whatismyip/internal/setting"
|
||
|
"github.com/dcarrillo/whatismyip/models"
|
||
|
"github.com/dcarrillo/whatismyip/router"
|
||
|
|
||
|
"github.com/gin-gonic/gin"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
tcpServer *http.Server
|
||
|
tlsServer *http.Server
|
||
|
engine *gin.Engine
|
||
|
)
|
||
|
|
||
|
func main() {
|
||
|
setting.Setup()
|
||
|
models.Setup(setting.App.GeodbPath.City, setting.App.GeodbPath.ASN)
|
||
|
setupEngine()
|
||
|
router.SetupTemplate(engine)
|
||
|
router.Setup(engine)
|
||
|
|
||
|
if setting.App.BindAddress != "" {
|
||
|
runTCPServer()
|
||
|
}
|
||
|
|
||
|
if setting.App.TLSAddress != "" {
|
||
|
runTLSServer()
|
||
|
}
|
||
|
|
||
|
runHandler()
|
||
|
}
|
||
|
|
||
|
func runHandler() {
|
||
|
signalChan := make(chan os.Signal, 3)
|
||
|
signal.Notify(signalChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
|
||
|
ctx := context.Background()
|
||
|
var s os.Signal
|
||
|
|
||
|
for {
|
||
|
s = <-signalChan
|
||
|
|
||
|
if s == syscall.SIGHUP {
|
||
|
models.CloseDBs()
|
||
|
models.Setup(setting.App.GeodbPath.City, setting.App.GeodbPath.ASN)
|
||
|
router.SetupTemplate(engine)
|
||
|
|
||
|
if setting.App.BindAddress != "" {
|
||
|
if err := tcpServer.Shutdown(ctx); err != nil {
|
||
|
log.Printf("TCP server forced to shutdown: %s", err)
|
||
|
}
|
||
|
runTCPServer()
|
||
|
}
|
||
|
if setting.App.TLSAddress != "" {
|
||
|
if err := tlsServer.Shutdown(ctx); err != nil {
|
||
|
log.Printf("TLS server forced to shutdown: %s", err)
|
||
|
}
|
||
|
runTLSServer()
|
||
|
}
|
||
|
} else {
|
||
|
log.Printf("Shutting down...")
|
||
|
if setting.App.BindAddress != "" {
|
||
|
if err := tcpServer.Shutdown(ctx); err != nil {
|
||
|
log.Printf("TCP server forced to shutdown: %s", err)
|
||
|
}
|
||
|
}
|
||
|
if setting.App.TLSAddress != "" {
|
||
|
if err := tlsServer.Shutdown(ctx); err != nil {
|
||
|
log.Printf("TLS server forced to shutdown: %s", err)
|
||
|
}
|
||
|
}
|
||
|
models.CloseDBs()
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func runTCPServer() {
|
||
|
tcpServer = &http.Server{
|
||
|
Addr: setting.App.BindAddress,
|
||
|
Handler: engine,
|
||
|
ReadTimeout: setting.App.Server.ReadTimeout,
|
||
|
WriteTimeout: setting.App.Server.WriteTimeout,
|
||
|
}
|
||
|
|
||
|
go func() {
|
||
|
log.Printf("Starting TCP server listening on %s", setting.App.BindAddress)
|
||
|
if err := tcpServer.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
log.Printf("Stopping TCP server...")
|
||
|
}()
|
||
|
}
|
||
|
|
||
|
func runTLSServer() {
|
||
|
tlsServer = &http.Server{
|
||
|
Addr: setting.App.TLSAddress,
|
||
|
Handler: engine,
|
||
|
ReadTimeout: setting.App.Server.ReadTimeout,
|
||
|
WriteTimeout: setting.App.Server.WriteTimeout,
|
||
|
}
|
||
|
|
||
|
go func() {
|
||
|
log.Printf("Starting TLS server listening on %s", setting.App.TLSAddress)
|
||
|
if err := tlsServer.ListenAndServeTLS(setting.App.TLSCrtPath, setting.App.TLSKeyPath); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
log.Printf("Stopping TLS server...")
|
||
|
}()
|
||
|
}
|
||
|
|
||
|
func setupEngine() {
|
||
|
gin.DisableConsoleColor()
|
||
|
if os.Getenv(gin.EnvGinMode) == "" {
|
||
|
gin.SetMode(gin.ReleaseMode)
|
||
|
}
|
||
|
engine = gin.New()
|
||
|
engine.Use(gin.LoggerWithFormatter(httputils.GetLogFormatter))
|
||
|
engine.Use(gin.Recovery())
|
||
|
_ = engine.SetTrustedProxies(nil)
|
||
|
engine.TrustedPlatform = setting.App.TrustedHeader
|
||
|
}
|