customer/internal/app/app.go

package app

import (
	"context"
	"fmt"
	"time"

	"go.uber.org/zap"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/initialize"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/models"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/server"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/swagger"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/utils"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/pkg/closer"
	"penahub.gitlab.yandexcloud.net/pena-services/customer/pkg/mongo"
)

const (
	shutdownTimeout = 5 * time.Second
)

func Run(ctx context.Context, config *models.Config, logger *zap.Logger) error {
	closer := closer.New()

	mongoDB, err := mongo.Connect(ctx, &mongo.ConnectDeps{
		Configuration: &config.Database,
		Timeout:       10 * time.Second,
	})
	if err != nil {
		return fmt.Errorf("failed connection to db: %w", err)
	}

	clients := initialize.NewClients(&initialize.ClientsDeps{
		Logger:                       logger,
		AuthURL:                      &config.Service.AuthMicroservice.URL,
		HubadminURL:                  &config.Service.HubadminMicroservice.URL,
		CurrencyURL:                  &config.Service.CurrencyMicroservice.URL,
		DiscountServiceConfiguration: &config.Service.DiscountMicroservice,
		PaymentServiceConfiguration:  &config.Service.PaymentMicroservice,
	})

	repositories := initialize.NewRepositories(&initialize.RepositoriesDeps{
		Logger:  logger,
		MongoDB: mongoDB,
	})

	services := initialize.NewServices(&initialize.ServicesDeps{
		Logger:       logger,
		Repositories: repositories,
		Clients:      clients,
	})

	controllers := initialize.NewControllers(&initialize.ControllersDeps{
		Logger:   logger,
		Services: services,
	})

	openapi, err := swagger.GetSwagger()
	if err != nil {
		return fmt.Errorf("failed to loading openapi spec: %w", err)
	}

	api := initialize.NewAPI(controllers)

	serverHTTP, httpErr := server.NewHTTP(server.DepsHTTP{
		Logger:             logger,
		Swagger:            openapi,
		AuthenticationFunc: utils.NewAuthenticator(utils.NewJWT(&config.Service.JWT)),
	})
	if httpErr != nil {
		return httpErr.Wrap("failed to init http server")
	}

	serverGRPC, grpcErr := server.NewGRPC(server.DepsGRPC{Logger: logger})
	if grpcErr != nil {
		return httpErr.Wrap("failed to init grpc server")
	}

	serverHTTP.Register(api)
	serverGRPC.Register(controllers)

	go serverHTTP.Run(&config.HTTP)
	go serverGRPC.Run(&config.GRPC)

	closer.Add(mongoDB.Client().Disconnect)
	closer.Add(serverHTTP.Stop)
	closer.Add(serverGRPC.Stop)

	<-ctx.Done()

	logger.Info("shutting down app gracefully")

	shutdownCtx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)

	defer cancel()

	if err := closer.Close(shutdownCtx); err != nil {
		return fmt.Errorf("closer: %w", err)
	}

	return nil
}