authenticationService/main.go at main · SafetyLink/authenticationService · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main

import (
	"context"
	"fmt"
	"github.com/SafetyLink/authenticationService/internal"
	"github.com/SafetyLink/authenticationService/internal/controller"
	"github.com/SafetyLink/authenticationService/internal/domain/authenticationService"
	"github.com/SafetyLink/authenticationService/internal/infra/adapter/sql"
	"github.com/SafetyLink/authenticationService/internal/infra/bcryptRepository"
	"github.com/SafetyLink/authenticationService/internal/infra/userRepository"
	"github.com/SafetyLink/commons/config"
	"github.com/SafetyLink/commons/grpc"
	"github.com/SafetyLink/commons/logger"
	"github.com/SafetyLink/commons/otel"
	"github.com/joho/godotenv"
	"go.uber.org/fx"
	"go.uber.org/zap"
	grpc2 "google.golang.org/grpc"
	"net"
)

func init() {
	//load env file
	err := godotenv.Load(".env")
	if err != nil {
		panic(err)
	}
}

func main() {
	fx.New(
		fx.Provide(logger.InitLogger),
		fx.Provide(config.ReadConfig[internal.Config]),
		fx.Provide(otel.InitTracer),

		fx.Provide(sql.NewPostgresProvider),

		fx.Provide(userRepository.NewUserRepository),
		fx.Provide(bcryptRepository.NewBcryptRepository),

		fx.Provide(authenticationService.NewAuthenticationService),

		fx.Provide(SetupAuthenticationServiceGrpcServer),

		fx.Invoke(controller.RegisterAuthenticationService),

		fx.Invoke(StartGrpcServer),
	).Run()
}

func StartGrpcServer(lc fx.Lifecycle, s *grpc2.Server, lis net.Listener, logger *zap.Logger, config *internal.Config) {
	lc.Append(fx.Hook{
		OnStart: func(ctx context.Context) error {
			logger.Info(fmt.Sprintf("Starting %s Grpc Server on port %s!", config.Grpc.ServiceName, config.Grpc.Port))

			go func() {
				if err := s.Serve(lis); err != nil {
					logger.Info(fmt.Sprintf("Failed to start %s Grpc Server on port %s!", config.Grpc.ServiceName, config.Grpc.Port))
				}
			}()
			return nil
		},
		OnStop: func(ctx context.Context) error {
			return nil
		},
	})
}

func SetupAuthenticationServiceGrpcServer(logger *zap.Logger, config *internal.Config) (net.Listener, *grpc2.Server) {
	s := grpc.NewServer()
	lis, err := net.Listen("tcp", config.Grpc.Port)
	if err != nil {
		logger.Fatal(fmt.Sprintf("Failed to start %s grpc server", config.Grpc.ServiceName), zap.Error(err))
	}

	return lis, s
}