{"id":6572,"date":"2024-09-19T19:32:32","date_gmt":"2024-09-19T11:32:32","guid":{"rendered":"https:\/\/t.n-years.com\/?p=6572"},"modified":"2024-09-19T19:32:32","modified_gmt":"2024-09-19T11:32:32","slug":"srv-shutdownctx-%e5%85%b3%e9%94%ae%e4%bb%a3%e7%a0%81","status":"publish","type":"post","link":"https:\/\/t.n-years.com\/?p=6572","title":{"rendered":"srv.Shutdown(ctx) \u5173\u952e\u4ee3\u7801"},"content":{"rendered":"

srv.Shutdown(ctx)<\/code> \u662f Go \u6807\u51c6\u5e93 net\/http<\/code> \u5305\u4e2d\u7684 http.Server<\/code> \u7c7b\u578b\u7684\u65b9\u6cd5\u3002\u5b83\u7528\u4e8e\u4f18\u96c5\u5730\u5173\u95ed HTTP \u670d\u52a1\u5668\uff0c\u786e\u4fdd\u6240\u6709\u6b63\u5728\u5904\u7406\u7684\u8bf7\u6c42\u5b8c\u6210\u540e\u518d\u9000\u51fa\u3002<\/p>\n

\u4ee5\u4e0b\u662f srv.Shutdown(ctx)<\/code> \u7684\u6838\u5fc3\u4ee3\u7801\u793a\u4f8b\uff0c\u5c55\u793a\u4e86\u5982\u4f55\u5728\u670d\u52a1\u5668\u5173\u95ed\u8fc7\u7a0b\u4e2d\u7b49\u5f85\u6240\u6709\u8bf7\u6c42\u5b8c\u6210\u3002\u8bf7\u6ce8\u610f\uff0c\u5b9e\u9645\u6e90\u7801\u4f1a\u6709\u6240\u4e0d\u540c\uff0c\u4ee5\u4e0b\u4ee3\u7801\u4ec5\u4e3a\u6838\u5fc3\u903b\u8f91\u7684\u7b80\u5316\u7248\u672c\uff1a<\/p>\n

\/\/ http\/server.go\n\n\/\/ Shutdown initiates a graceful shutdown of the server, waiting for\n\/\/ currently active connections to finish.\nfunc (srv *Server) Shutdown(ctx context.Context) error {\n    \/\/ Signal the server to stop accepting new connections.\n    srv.setClosing()\n\n    \/\/ Close the listener so no new connections will be accepted.\n    if err := srv.Listener.Close(); err != nil && err != ErrServerClosed {\n        return err\n    }\n\n    \/\/ Create a channel to signal when all active connections have finished.\n    done := make(chan struct{})\n\n    \/\/ Start a goroutine to wait for active connections to close.\n    go func() {\n        srv.wg.Wait()\n        close(done)\n    }()\n\n    \/\/ Wait for either the connections to close or the context to timeout.\n    select {\n    case <-ctx.Done():\n        return ctx.Err() \/\/ Return context error if context is canceled or timed out.\n    case <-done:\n        return nil \/\/ Successfully closed after all connections are done.\n    }\n}\n\n\/\/ setClosing sets the server to a "closing" state to stop accepting new connections.\nfunc (srv *Server) setClosing() {\n    srv.mu.Lock()\n    defer srv.mu.Unlock()\n    srv.closing = true\n}<\/code><\/pre>\n
\u5173\u952e\u90e8\u5206\u89e3\u6790<\/h3>\n
    \n
  1. \n
    \u505c\u6b62\u63a5\u53d7\u65b0\u8fde\u63a5<\/strong>\uff1a<\/p>\n