Golang优雅停机：信号处理应该放在main文件还是HTTP服务器文件中？

Golang优雅停机：信号处理应该放在main文件还是HTTP服务器文件中？ 我正在尝试找出两种当前实现相同功能（例如当按下 Ctrl+C 时，优雅地关闭应用程序/服务器）的实现之间是否存在任何差异。两者都运行良好，并且基于文档。

我的一位朋友提到，示例2在应用程序级别处理关闭，这会关闭整个应用程序中的所有上下文。然而，示例1在HTTP服务器级别处理，这不一定关闭整个应用程序中的所有上下文。由于我是初学者，我无法反驳，因此需要您对此提供意见。

示例1

信号在 http.go 文件中处理，因此整个优雅关闭过程在单个文件中完成。

cmd/app/main.go

package main

import (
	"context"
	"internal/http"
	"os"
	"os/signal"
	"syscall"
)

func main() {
	// 引导配置、日志记录器等

	http.Start()
}

internal/http/server.go

package http

import (
	"context"
	"github.com/prometheus/common/log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"
)

func Start() {
	log.Infof("starting HTTP server")

	srv := &http.Server{Addr: ":8080", Handler: nil}

	idle := make(chan struct{})

	go shutdown(srv, idle)

	if err := srv.ListenAndServe(); err != http.ErrServerClosed {
		log.Fatalf("failed to start/close HTTP server [%v]", err)
	}

	<-idle

	log.Info("gracefully shutdown HTTP server")
}

func shutdown(srv *http.Server, idle chan<- struct{}) {
	sig := make(chan os.Signal, 1)

	signal.Notify(sig, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)

	<-sig

	ctx, cancel := context.WithTimeout(context.Background(), time.Duration(10)*time.Second)
	defer cancel()

	if err := srv.Shutdown(ctx); err != nil {
		log.Fatalf("shutdown HTTP server by interrupting idle connections [%v]", err)
	}

	close(idle)
}

示例2

信号在应用程序的 main.go 文件中处理，因此整个优雅关闭过程被拆分到两个文件中。此示例唯一的额外之处是使用了 WithCancel 上下文。

cmd/app/main.go

package main

import (
	"context"
	"internal/http"
	"os"
	"os/signal"
	"syscall"
)

func main() {
	// 引导配置、日志记录器等

	backgroundCtx := context.Background()
	withCancelCtx, cancel := context.WithCancel(backgroundCtx)
	go shutdown(cancel)

	http.Start(withCancelCtx)
}

func shutdown(cancel context.CancelFunc) {
	sig := make(chan os.Signal, 1)

	signal.Notify(sig, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)

	<-sig

	cancel()
}

internal/http/server.go

package http

import (
	"context"
	"github.com/prometheus/common/log"
	"net/http"
	"time"
)

func Start(ctx context.Context) {
	log.Infof("starting HTTP server")

	srv := &http.Server{Addr: ":8080", Handler: nil}

	idle := make(chan struct{})

	go shutdown(ctx, srv, idle)

	if err := srv.ListenAndServe(); err != http.ErrServerClosed {
		log.Fatalf("failed to start/close HTTP server [%v]", err)
	}

	<-idle

	log.Info("gracefully shutdown HTTP server")
}

func shutdown(ctx context.Context, srv *http.Server, idle chan struct{}) {
	<-ctx.Done()

	ctx, cancel := context.WithTimeout(context.Background(), time.Duration(10)*time.Second)
	defer cancel()

	if err := srv.Shutdown(ctx); err != nil {
		log.Fatalf("shutdown HTTP server by interrupting idle connections [%v]", err)
	}

	close(idle)
}