onasty: mailer/main.go (cd94b5c1238f63c87e4906cc492be0c70ae0cec7)

1

package main

2

3

import (

4

	"errors"

5

	"fmt"

6

	"log/slog"

7

	"net/http"

8

	"os"

9

	"os/signal"

10

	"strings"

11

	"syscall"

12

13

	"github.com/nats-io/nats.go"

14

	"github.com/nats-io/nats.go/micro"

15

	"github.com/olexsmir/onasty/internal/logger"

16

	"github.com/olexsmir/onasty/internal/transport/http/httpserver"

17

18

	_ "embed"

19

20

21

//go:embed version

22

var _version string

23

24

var version = strings.Trim(_version, "\n")

25

26

func main() {

27

	if err := run(); err != nil {

28

		fmt.Fprintf(os.Stderr, "error: %v\n", err)

29

		os.Exit(1)

30

31

32

33

func run() error {

34

	cfg := NewConfig()

35

	nc, err := nats.Connect(cfg.NatsURL)

36

	if err != nil {

37

		return err

38

39

40

	logger, err := logger.NewCustomLogger(cfg.LogLevel, cfg.LogFormat, cfg.LogShowLine)

41

	if err != nil {

42

		return err

43

44

45

	slog.SetDefault(logger)

46

47

	//nolint:exhaustruct

48

	svc, err := micro.AddService(nc, micro.Config{

49

		Name:    "mailer",

50

		Version: version,

51

})

52

	if err != nil {

53

		return err

54

55

56

	mg := NewMailgun(cfg.MailgunFrom, cfg.MailgunDomain, cfg.MailgunAPIKey)

57

	service := NewService(cfg.AppURL, mg)

58

	handlers := NewHandlers(service)

59

60

	if err := handlers.RegisterAll(svc); err != nil {

61

		return err

62

63

64

	if cfg.MetricsEnabled {

65

		srv := httpserver.NewServer(cfg.MetricsPort, MetricsHandler())

66

		go func() {

67

			slog.Info("starting metrics server", "port", cfg.MetricsPort)

68

			if err := srv.Start(); !errors.Is(err, http.ErrServerClosed) {

69

				slog.Error("failed to start metrics server", "error", err)

70

71

}()

72

73

74

	slog.Info("the service is listening")

75

76

	// graceful shutdown

77

	quitCh := make(chan os.Signal, 1)

78

	signal.Notify(quitCh, syscall.SIGINT, syscall.SIGTERM)

79

	<-quitCh

80

81

	slog.Info("stopping the service")

82

83

	if err := svc.Stop(); err != nil {

84

		return err

85

86

87

	if err := nc.Drain(); err != nil {

88

		return err

89

90

91

	return nil

92

1	package main
2
3	import (
4	"errors"
5	"fmt"
6	"log/slog"
7	"net/http"
8	"os"
9	"os/signal"
10	"strings"
11	"syscall"
12
13	"github.com/nats-io/nats.go"
14	"github.com/nats-io/nats.go/micro"
15	"github.com/olexsmir/onasty/internal/logger"
16	"github.com/olexsmir/onasty/internal/transport/http/httpserver"
17
18	_ "embed"
19	)
20
21	//go:embed version
22	var _version string
23
24	var version = strings.Trim(_version, "\n")
25
26	func main() {
27	if err := run(); err != nil {
28	fmt.Fprintf(os.Stderr, "error: %v\n", err)
29	os.Exit(1)
30	}
31	}
32
33	func run() error {
34	cfg := NewConfig()
35	nc, err := nats.Connect(cfg.NatsURL)
36	if err != nil {
37	return err
38	}
39
40	logger, err := logger.NewCustomLogger(cfg.LogLevel, cfg.LogFormat, cfg.LogShowLine)
41	if err != nil {
42	return err
43	}
44
45	slog.SetDefault(logger)
46
47	//nolint:exhaustruct
48	svc, err := micro.AddService(nc, micro.Config{
49	Name: "mailer",
50	Version: version,
51	})
52	if err != nil {
53	return err
54	}
55
56	mg := NewMailgun(cfg.MailgunFrom, cfg.MailgunDomain, cfg.MailgunAPIKey)
57	service := NewService(cfg.AppURL, mg)
58	handlers := NewHandlers(service)
59
60	if err := handlers.RegisterAll(svc); err != nil {
61	return err
62	}
63
64	if cfg.MetricsEnabled {
65	srv := httpserver.NewServer(cfg.MetricsPort, MetricsHandler())
66	go func() {
67	slog.Info("starting metrics server", "port", cfg.MetricsPort)
68	if err := srv.Start(); !errors.Is(err, http.ErrServerClosed) {
69	slog.Error("failed to start metrics server", "error", err)
70	}
71	}()
72	}
73
74	slog.Info("the service is listening")
75
76	// graceful shutdown
77	quitCh := make(chan os.Signal, 1)
78	signal.Notify(quitCh, syscall.SIGINT, syscall.SIGTERM)
79	<-quitCh
80
81	slog.Info("stopping the service")
82
83	if err := svc.Stop(); err != nil {
84	return err
85	}
86
87	if err := nc.Drain(); err != nil {
88	return err
89	}
90
91	return nil
92	}