В Go, подход к Graceful shutdown зависит от того, какой сервер вы используете. Важными факторами являются то, насколько корректно обрабатываются активные соединения, закрываются слушающие сокеты и завершаются все запущенные задачи.
Если вы используете стандартную библиотеку net/http для создания HTTP-сервера, вам потребуется немного дополнительного кода, чтобы обрабатывать Graceful shutdown. Это связано с тем, что net/http не предоставляет встроенной поддержки Graceful shutdown.
Один из способов реализации Graceful shutdown с использованием net/http - использовать контексты. Контексты позволяют передавать сигналы отмены и управлять временем жизни запросов. Создайте корневой контекст и передайте его вашему HTTP-серверу при его запуске. Затем можно использовать этот контекст для сигналов отмены при Graceful shutdown.
package main
import (
"context"
"log"
"net/http"
"os"
"os/signal"
"time"
)
func main() {
server := &http.Server{
Addr: ":8080",
Handler: myHandler(),
}
// Создаем канал для получения сигналов от OS
stop := make(chan os.Signal, 1)
signal.Notify(stop, os.Interrupt)
// Создаем контекст с таймаутом
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
// Запускаем сервер в отдельной горутине
go func() {
if err := server.ListenAndServe(); err != http.ErrServerClosed {
log.Fatalf("HTTP server ListenAndServe: %v", err)
}
}()
// Ожидаем сигнал остановки
<-stop
// Останавливаем сервер с использованием контекста
if err := server.Shutdown(ctx); err != nil {
log.Fatalf("HTTP server Shutdown: %v", err)
}
log.Println("Server gracefully stopped")
}
func myHandler() http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// Обработка запросов
})
}
В этом примере мы создаем HTTP-сервер с помощью net/http, затем создаем контекст с тайм-аутом и запускаем сервер в отдельной горутине. Затем мы настраиваем обработку сигналов от операционной системы (SIGINT) и используем <-stop, чтобы ожидать получения этого сигнала, после получения этого сигнала мы вызываем server.Shutdown() с использованием контекста, чтобы корректно остановить сервер. Важно отметить, что сервер будет продолжать обслуживать активные соединения в течение 5 секунд (вы можете изменить время тайм-аута в функции context.WithTimeout), а затем грациозно завершит свою работу. Если у вас есть дополнительные действия, которые должны быть выполнены при завершении сервера, вы можете добавить их после server.Shutdown(). Общая идея заключается в том, чтобы использовать контексты для управления временем жизни запросов и корректного завершения сервера при получении сигнала от операционной системы. Это позволяет обрабатывать активные соединения и завершать работу сервера без потери запросов или нарушения целостности данных.