mugit/internal/cli/serve.go(view raw)
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 |
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"
"olexsmir.xyz/mugit/internal/ssh"
)
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.SSH.Enable {
sshServer := ssh.NewServer(c.cfg)
go func() {
slog.Info("starting ssh server", "port", c.cfg.SSH.Port)
if err := sshServer.Start(); err != nil {
slog.Error("ssh server error", "err", err)
}
}()
}
if c.cfg.Mirror.Enable {
mirrorer := mirror.NewWorker(c.cfg)
go func() {
slog.Info("starting mirroring worker")
mirrorer.Start(context.TODO())
}()
}
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(context.TODO()); err != nil {
slog.Error("HTTP server shutdown error", "err", err)
} else {
slog.Info("HTTP server shutdown complete")
}
return nil
}
|