1
0
Fork 0

refactor: extract server from main.go to it's own package

This commit is contained in:
Vojtěch Mareš 2024-05-04 12:22:42 +02:00
parent 1413fde9f0
commit 24e99a3b08
Signed by: vojtech.mares
GPG key ID: C6827B976F17240D
2 changed files with 72 additions and 44 deletions

67
internal/server/server.go Normal file
View file

@ -0,0 +1,67 @@
package server
import (
"context"
"fmt"
"time"
fiberzap "github.com/gofiber/contrib/fiberzap/v2"
"github.com/gofiber/fiber/v2"
"go.uber.org/zap"
)
type Server struct {
port int
logger *zap.Logger
}
func NewServer(port int, logger *zap.Logger) *Server {
return &Server{
port: port,
logger: logger,
}
}
func (s *Server) Run(ctx context.Context) {
sugared := s.logger.Sugar()
config := fiber.Config{
DisableStartupMessage: true,
IdleTimeout: 10 * time.Second,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
}
app := fiber.New(config)
app.Use(fiberzap.New(fiberzap.Config{
Logger: s.logger,
}))
app.Get("/", func(c *fiber.Ctx) error {
return c.SendString("Hello, World!")
})
go func() {
err := app.Listen(fmt.Sprintf(":%d", s.port))
if err != nil {
panic(err)
}
}()
sugared.Infow("HTTP server is now running", "port", s.port)
<-ctx.Done()
shutdownBegan := time.Now()
sugared.Infoln("Shutting HTTP server down gracefully...")
err := app.ShutdownWithTimeout(10 * time.Second)
if err != nil {
panic(err)
}
// Cleanup code here
sugared.Infow("HTTP server shut down gracefully.", "took", time.Since(shutdownBegan))
}