Redmage/server/server.go

package server

import (
	"context"
	"errors"
	"net/http"

	"github.com/go-chi/chi/v5"
	"github.com/tigorlazuardi/redmage/config"
	"github.com/tigorlazuardi/redmage/pkg/caller"
	"github.com/tigorlazuardi/redmage/pkg/errs"
	"github.com/tigorlazuardi/redmage/pkg/log"
	"github.com/tigorlazuardi/redmage/server/routes/api"
	"github.com/tigorlazuardi/redmage/server/routes/www"
)

type Server struct {
	server *http.Server
	config *config.Config
}

func (srv *Server) Start(exit <-chan struct{}) error {
	errch := make(chan error, 1)
	caller := caller.New(3)
	go func() {
		log.New(context.Background()).Caller(caller).Info("starting http server", "address", "http://"+srv.server.Addr)
		errch <- srv.server.ListenAndServe()
	}()

	select {
	case <-exit:
		log.New(context.Background()).Caller(caller).Info("received exit signal. shutting down server")
		ctx, cancel := context.WithTimeout(context.Background(), srv.config.Duration("http.shutdown_timeout"))
		defer cancel()
		return srv.server.Shutdown(ctx)
	case err := <-errch:
		if errors.Is(err, http.ErrServerClosed) {
			return nil
		}
		return errs.Wrap(err)
	}
}

func New(cfg *config.Config, api *api.API, www *www.WWW) *Server {
	router := chi.NewRouter()

	router.Route("/api", api.Register)
	router.Route("/", www.Register)

	server := &http.Server{
		Handler: router,
		Addr:    cfg.String("http.host") + ":" + cfg.String("http.port"),
	}

	return &Server{server: server, config: cfg}
}