如何在 Go 中停止监听服务器-Golang-PHP中文网

如何在 Go 中停止监听服务器

王林

发布： 2024-02-09 12:09:22

转载

757人浏览过

如何在 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 的错误测试，但这真的很难看。我可以设置一个标志，表示我即将关闭并忽略错误（如果已设置），我想 - 有更好的方法吗？