当前位置：首页 > news >正文

大安移动网站建设大学建网站

news 2025/11/2 5:51:57

大安移动网站建设,大学建网站,做淘宝客网站要多少钱,长沙网站制作首页使用Go开发web服务时很多情况下都会使用号称比标准库快10x的FastHttp, 但fasthttp(版本: 20180529.0.0)至今也没有提供优雅关闭的方法，默认情况下退出服务只能kill。下面谈几个实现方案。首先，要明确所谓的优雅关闭是要求我们在调用close()时要做到以下…

使用Go开发web服务时很多情况下都会使用号称比标准库快10x的FastHttp, 但fasthttp(版本: 20180529.0.0)至今也没有提供优雅关闭的方法，默认情况下退出服务只能kill。下面谈几个实现方案。

首先，要明确所谓的优雅关闭是要求我们在调用close()时要做到以下几点：

拒绝接受新连接
等待正在处理的请求完成，然后关闭连接
关闭剩余空闲的连接

要做到第一点，我们需要重写一下net.Listener实现，例如叫GraceListener, 在此结构体中组合一个真正干活的Listener，覆盖Close()方法，在此方法中先将干活的Listener关闭，此时就不会再监听新请求了，然后再block当前routine直到所有连接全部关闭为止。代码如下：

type graceListener struct {net.Listener
}func (gl *graceListener) Close() error {err := gl.Close()if nil != err {return err}// block, 直到所有连接关闭
}

fastServ := &fasthttp.Server{Concurrency:  100,Handler:      xxxFunc,LogAllErrors: true,}ln, err := net.Listen("tcp4", ":8080")if nil != err {// err}graceLn := &graceListener{Listener: ln,}fastServ.Serve(graceLn)

接下来看看如何满足后面两条要求。这里有两种方案，第一种最简单的方案是，保存一个全局的sync.WaitGroup指针，在你的请求处理函数中，先调用wg.Add(1), 然后defer wg.Done(), 最后在上面的Close()方法中使用wg.Wait()即可。这里建议最好使用select给等待加个超时功能，即如果超过指定时间还没有退出则强制退出：

// 此方法一直block到所有请求退出或超时
func WaitForGracefullyClose() error {select {case <-waitAllRoutineDone():return nilcase <-time.After(maxWait):return fmt.Errorf("force shutdown after %v", maxWait)}}// 等待所有请求处理routine完成;
// 此方法返回只有1个缓冲的channel, 只有当所有routine结束时channel才会有元素
func waitAllRoutineDone() chan struct{} {flagChan := make(chan struct{}, 1)go func() {wg.Wait()flagChan <- struct{}{}}()return flagChan
}

此外还有第二种方案，那就是在自己的graceListener中添加一个计数器用于统计当前的连接数，重写Accept()方法，将计数器+1，再定义一个自己的套壳net.Conn实现，重写Close()方法，在里面将计数器-1。这里要注意线程安全问题，最好使用atomic包进行操作。最后在graceListener#Close()中关闭Listener后等待计数器归零。这种方案有以下几个缺点：