mugit: internal/cli/serve.go (96821ba3e784d93a9effc172c698b5444f02eccc)

Oleksandr Smirnov goimports -local 'olexsmir.xyz' -w ., 1 month ago

package cli

import (
	"context"
	"log/slog"
	"net"
	"net/http"
	"os"
	"os/signal"
	"strconv"
	"syscall"

	"github.com/urfave/cli/v3"

	"olexsmir.xyz/mugit/internal/handlers"
	"olexsmir.xyz/mugit/internal/mirror"
)

func (c *Cli) serveAction(ctx context.Context, cmd *cli.Command) error {
	httpServer := &http.Server{
		Addr:    net.JoinHostPort(c.cfg.Server.Host, strconv.Itoa(c.cfg.Server.Port)),
		Handler: handlers.InitRoutes(c.cfg),
	}
	go func() {
		slog.Info("starting http server", "host", c.cfg.Server.Host, "port", c.cfg.Server.Port)
		if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			slog.Error("HTTP server error", "err", err)
		}
	}()

	if c.cfg.Mirror.Enable {
		mirrorer := mirror.NewWorker(c.cfg)
		go func() {
			slog.Info("starting mirroring worker")
			if err := mirrorer.Start(ctx); err != nil {
				slog.Error("failed to start mirrorer", "err", err)
			}
		}()
	}

	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

	sig := <-sigChan
	slog.Info("received signal, starting graceful shutdown", "signal", sig)

	if err := httpServer.Shutdown(ctx); err != nil {
		slog.Error("HTTP server shutdown error", "err", err)
	} else {
		slog.Info("HTTP server shutdown complete")
	}

	return nil
}

1	package cli
2
3	import (
4	"context"
5	"log/slog"
6	"net"
7	"net/http"
8	"os"
9	"os/signal"
10	"strconv"
11	"syscall"
12
13	"github.com/urfave/cli/v3"
14
15	"olexsmir.xyz/mugit/internal/handlers"
16	"olexsmir.xyz/mugit/internal/mirror"
17	)
18
19	func (c Cli) serveAction(ctx context.Context, cmd cli.Command) error {
20	httpServer := &http.Server{
21	Addr: net.JoinHostPort(c.cfg.Server.Host, strconv.Itoa(c.cfg.Server.Port)),
22	Handler: handlers.InitRoutes(c.cfg),
23	}
24	go func() {
25	slog.Info("starting http server", "host", c.cfg.Server.Host, "port", c.cfg.Server.Port)
26	if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
27	slog.Error("HTTP server error", "err", err)
28	}
29	}()
30
31	if c.cfg.Mirror.Enable {
32	mirrorer := mirror.NewWorker(c.cfg)
33	go func() {
34	slog.Info("starting mirroring worker")
35	if err := mirrorer.Start(ctx); err != nil {
36	slog.Error("failed to start mirrorer", "err", err)
37	}
38	}()
39	}
40
41	sigChan := make(chan os.Signal, 1)
42	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
43
44	sig := <-sigChan
45	slog.Info("received signal, starting graceful shutdown", "signal", sig)
46
47	if err := httpServer.Shutdown(ctx); err != nil {
48	slog.Error("HTTP server shutdown error", "err", err)
49	} else {
50	slog.Info("HTTP server shutdown complete")
51	}
52
53	return nil
54	}