Code cleanup and formatting
This commit is contained in:
57
cmd/app/handler/handler.go
Normal file
57
cmd/app/handler/handler.go
Normal file
@@ -0,0 +1,57 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/kjannette/go_http_server/internal/config"
|
||||
"github.com/labstack/echo/v4"
|
||||
"github.com/pkg/errors"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// Handler has services that are required by the handler.
|
||||
type Handler struct {
|
||||
logger *zap.Logger
|
||||
}
|
||||
|
||||
// New creates a new Handler instance.
|
||||
func New(logger *zap.Logger) (*echo.Echo, Handler) {
|
||||
h := Handler{
|
||||
logger: logger,
|
||||
}
|
||||
|
||||
e := echo.New()
|
||||
h.setupHealthCheckRoutes(e)
|
||||
|
||||
return e, h
|
||||
}
|
||||
|
||||
// RunServer runs http server and listens until context is canceled.
|
||||
func (h *Handler) RunServer(ctx context.Context, conf *config.Config, e *echo.Echo) error {
|
||||
server := &http.Server{
|
||||
Addr: fmt.Sprintf(":%d", conf.API.Port),
|
||||
ReadTimeout: time.Duration(conf.API.ReadTimeout) * time.Second,
|
||||
WriteTimeout: time.Duration(conf.API.WriteTimeout) * time.Second,
|
||||
IdleTimeout: time.Duration(conf.API.IdleTimeout) * time.Second,
|
||||
}
|
||||
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
|
||||
shutdownCtx, cancel := context.WithTimeout(context.Background(), time.Duration(conf.API.ReadTimeout)*time.Second)
|
||||
defer cancel()
|
||||
|
||||
if err := server.Shutdown(shutdownCtx); err != nil {
|
||||
h.logger.Warn("graceful shutdown failed", zap.Error(err))
|
||||
}
|
||||
}()
|
||||
|
||||
if err := e.StartServer(server); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||
return errors.Wrap(err, "server error")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user