✅ Graceful Shutdown в Go: полная защита с отклонением новых запросов

server.Shutdown() ждёт текущие запросы, но не закрывает входящий трафик мгновенно. Между сигналом и фактической остановкой сервер продолжает принимать новые соединения. При высокой нагрузке это окно создаёт проблемы.

Финальный паттерн закрывает и эту щель.

📎 Добавляем middleware для отклонения новых запросов

Используем атомарный флаг и промежуточный обработчик:

package main

import (
    "context"
    "fmt"
    "net/http"
    "os"
    "os/signal"
    "sync/atomic"
    "syscall"
    "time"
)

var isShuttingDown int32

func rejectMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if atomic.LoadInt32(&isShuttingDown) == 1 {
            http.Error(w, "server is shutting down", http.StatusServiceUnavailable)
            return
        }
        next.ServeHTTP(w, r)
    })
}

func handler(w http.ResponseWriter, r *http.Request) {
    id := r.URL.Query().Get("id")
    fmt.Printf("START %s\n", id)
    time.Sleep(5 * time.Second)
    fmt.Printf("DONE %s\n", id)
    w.Write([]byte("COMPLETED!"))
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/", handler)

    server := &http.Server{
        Addr:    ":8080",
        Handler: rejectMiddleware(mux),
    }

    go func() {
        fmt.Println("server started at :8080")
        if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            panic(err)
        }
    }()

    ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
    defer stop()

    <-ctx.Done()
    fmt.Println("shutdown signal received")

    atomic.StoreInt32(&isShuttingDown, 1)

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

    server.Shutdown(shutdownCtx)

    fmt.Println("graceful shutdown complete")
}

❔ Что здесь происходит

После получения сигнала сначала ставится флаг isShuttingDown = 1. rejectMiddleware проверяет этот флаг перед каждым запросом. Если флаг поднят, новый клиент получает 503 Service Unavailable. Уже выполняющиеся обработчики продолжают работу и завершаются штатно.

atomic.LoadInt32 и atomic.StoreInt32 нужны потому, что флаг читается и пишется из разных горутин. Без атомарных операций здесь была бы гонка данных.

Итоговая картина

Путь от первого паттерна до последнего выглядит так. Сначала сервер просто падает. Потом учится слышать сигналы. Потом начинает ждать текущие запросы. Наконец, закрывает вход для новых до того, как начнёт завершаться.

Каждый уровень добавляет одну гарантию. Финальный вариант даёт полный контроль над состоянием сервера во время остановки.

📍 Навигация: Вакансии • Задачи • Собесы

🐸 Библиотека Go-разработчика

#GoToProduction