mugit: internal/cli/serve.go (a940f43f0e7d974f75672988227623e2e1fa9a08)

Oleksandr Smirnov ssh: write only ssh logs to file (othewise they are shown to user..., 22 days 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	"olexsmir.xyz/mugit/internal/handlers"
15	"olexsmir.xyz/mugit/internal/mirror"
16	)
17
18	func (c Cli) serveAction(ctx context.Context, cmd cli.Command) error {
19	httpServer := &http.Server{
20	Addr: net.JoinHostPort(c.cfg.Server.Host, strconv.Itoa(c.cfg.Server.Port)),
21	Handler: handlers.InitRoutes(c.cfg),
22	}
23	go func() {
24	slog.Info("starting http server", "host", c.cfg.Server.Host, "port", c.cfg.Server.Port)
25	if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
26	slog.Error("HTTP server error", "err", err)
27	}
28	}()
29
30	if c.cfg.Mirror.Enable {
31	mirrorer := mirror.NewWorker(c.cfg)
32	go func() {
33	slog.Info("starting mirroring worker")
34	if err := mirrorer.Start(ctx); err != nil {
35	slog.Error("failed to start mirrorer", "err", err)
36	}
37	}()
38	}
39
40	sigChan := make(chan os.Signal, 1)
41	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
42
43	sig := <-sigChan
44	slog.Info("received signal, starting graceful shutdown", "signal", sig)
45
46	if err := httpServer.Shutdown(ctx); err != nil {
47	slog.Error("HTTP server shutdown error", "err", err)
48	} else {
49	slog.Info("HTTP server shutdown complete")
50	}
51
52	return nil
53	}