卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34758本站已运行393

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

如何在 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 翻转它,然后连接到您的侦听端口以获取服务器套接字“不被卡住”。这与旧的“自管道技巧”非常相似。

卓越飞翔博客
上一篇: 如何使用 Python 检查 sum.golang.org 中的 go.mod 哈希值?
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏