如何在 Go 中停止監聽伺服器

在Go語言中，如何優雅地停止監聽伺服器是一個常見的問題。當我們需要停止伺服器時，我們希望能夠正常關閉所有連接，並且不影響正在處理的請求。本文將介紹幾種常見的方法，幫助你解決這個問題。 首先，我們可以使用一個信號量來控制伺服器的關閉。 Go語言中的os套件提供了一個Signal方法，可以用來捕捉作業系統發送的訊號。我們可以透過監聽os.Interrupt訊號，即Ctrl C，來觸發伺服器的關閉操作。在訊號處理函數中，我們可以執行一些清理工作，例如關閉資料庫連線、儲存資料等。然後，呼叫伺服器的Shutdown方法，它會等待現有的請求處理完畢後關閉伺服器。這樣，我們就可以優雅地停止監聽伺服器了。值得注意的是，如果伺服器在一定時間內無法關閉，我們可以使用context套件中的WithTimeout方法來設定逾時時間，超過這個時間後，伺服器會強制關閉。使用這種方法，我們可以確保伺服器在任何情況下都能夠正常關閉。 除了使用信號量外，我們還可以使用一個bool類型的變數來控制伺服器的關閉。在伺服器啟動時，我們可以定義一個全域的bool變量，例如isShutdown。然後，在處理請求的函數中，我們可以檢查這個變數的值，如果為true，則立即返回，停止處理當前請求。在關閉伺服器時，我們將isShutdown設為true，然後等待所有請求處理完畢後退出程式。這種方法相對簡單，但需要在處理請求的函數中加入額外的邏輯判斷，可能會稍微影響效能。 總結起來，無論是使用信號量還是使用bool變量，我們都可以實現在Go語言中優雅地停止監聽伺服器的功能。選擇哪種方法，取決於你的特定需求和專案的複雜程度

問題內容

我一直在嘗試找到一種方法來優雅地停止 go 中的偵聽伺服器。因為 listen.accept 阻止，所以有必要關閉偵聽套接字以發出結束訊號，但我無法區分該錯誤和任何其他錯誤，因為相關錯誤未匯出。

我能做得更好嗎？請參閱fixme中以下程式碼中的serve()

package main

import (
    "io"
    "log"
    "net"
    "time"
)

// echo server struct
type echoserver struct {
    listen net.listener
    done   chan bool
}

// respond to incoming connection
//
// write the address connected to then echo
func (es *echoserver) respond(remote *net.tcpconn) {
    defer remote.close()
    _, err := io.copy(remote, remote)
    if err != nil {
        log.printf("error: %s", err)
    }
}

// listen for incoming connections
func (es *echoserver) serve() {
    for {
        conn, err := es.listen.accept()
        // fixme i'd like to detect "use of closed network connection" here
        // fixme but it isn't exported from net
        if err != nil {
            log.printf("accept failed: %v", err)
            break
        }
        go es.respond(conn.(*net.tcpconn))
    }
    es.done <- true
}

// stop the server by closing the listening listen
func (es *echoserver) stop() {
    es.listen.close()
    <-es.done
}

// make a new echo server
func newechoserver(address string) *echoserver {
    listen, err := net.listen("tcp", address)
    if err != nil {
        log.fatalf("failed to open listening socket: %s", err)
    }
    es := &echoserver{
        listen: listen,
        done:   make(chan bool),
    }
    go es.serve()
    return es
}

// main
func main() {
    log.println("starting echo server")
    es := newechoserver("127.0.0.1:18081")
    // run the server for 1 second
    time.sleep(1 * time.second)
    // close the server
    log.println("stopping echo server")
    es.stop()
}

這會印出

2012/11/16 12:53:35 Starting echo server
2012/11/16 12:53:36 Stopping echo server
2012/11/16 12:53:36 Accept failed: accept tcp 127.0.0.1:18081: use of closed network connection

我想隱藏 accept failed 訊息，但顯然我不想掩蓋 accept 可以報告的其他錯誤。我當然可以查看 use of closed network connection 的錯誤測試，但這真的很難看。我可以設置一個標誌，表示我即將關閉並忽略錯誤（如果已設定），我想 - 有更好的方法嗎？

解決方法

在accept() 呼叫之後檢查循環中的一些「是時候停止」標誌，然後從main 翻轉它，然後連接到您的偵聽連接埠以取得伺服器套接字「不被卡住」。這與舊的「自管道技巧」非常相似。

以上是如何在 Go 中停止監聽伺服器的詳細內容。更多資訊請關注PHP中文網其他相關文章！