merch-api/internal/app/handler.go

package app

import (
	"context"
	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"merch-api/config"
	"merch-api/internal/merch"
	"merch-api/pkg/dbase"
	"merch-api/pkg/router"
	"merch-api/pkg/utils"
	"time"
)

const pkgLogHeader string = "Application |"

type App struct {
	cfg     config.Config
	router  *router.Router
	modules []Module
}

func New(cfg config.Config) *App {
	//providers
	r := router.NewRouter(router.Deps{
		Host:    cfg.Http.Host,
		Port:    cfg.Http.Port,
		Prefix:  cfg.Http.Prefix,
		GinMode: cfg.Http.GinMode,
	})

	db, err := dbase.Connect(dbase.Deps{
		Host:     cfg.DBase.Host,
		Port:     cfg.DBase.Port,
		Username: cfg.DBase.Username,
		Password: cfg.DBase.Password,
		DBName:   cfg.DBase.DBName,
	})

	u := utils.New()

	if err != nil {
		log.WithError(err).Fatalf("%v failed to connect database", pkgLogHeader)
	}

	//modules
	var modules []Module

	m := merch.New(merch.Deps{
		DB:    db,
		Utils: u,
	})
	modules = append(modules, m)

	return &App{
		cfg:     cfg,
		router:  r,
		modules: modules,
	}
}

func (app *App) Run(ctx context.Context) error {
	log.Infof("%v starting...", pkgLogHeader)

	baseGroup := app.router.BaseGroup()
	app.collectRoutes(baseGroup)

	errCh := make(chan error, 10)

	go func() {
		if err := app.router.Run(); err != nil {
			errCh <- err
		}
	}()

	select {
	case <-ctx.Done():
		app.shutdown(ctx)

	case err := <-errCh:
		return err
	}
	return nil
}

func (app *App) shutdown(ctx context.Context) {
	log.Infof("%v shutting down...", pkgLogHeader)
	shutdownCtx, shutdownCancel := context.WithTimeout(ctx, 15*time.Second)
	defer shutdownCancel()

	if err := app.router.Shutdown(shutdownCtx); err != nil {
		log.WithError(err).Warnf("%v error shutting down application", pkgLogHeader)
	}

	log.Infof("%v shutdown complete", pkgLogHeader)
}

func (app *App) collectRoutes(group *gin.RouterGroup) {
	for _, m := range app.modules {
		m.RegisterRoutes(group)
	}
	log.Infof("%v routes registered", pkgLogHeader)
}