如何让Go的http.Server闲置一段时间后退出？

大家好，今天本人给大家带来文章《如何让Go的http.Server闲置一段时间后退出？》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我正在使用标准库 net/http 包在 Go 中编写一个 Web 服务器，该包利用了 systemd 套接字激活。

我已经掌握了基础知识，以便在第一次与侦听套接字建立连接时启动服务器，并且可以在收到信号时执行正常关闭（即 systemctl stop 将在不切断活动请求的情况下工作）。

我希望服务器在闲置一段时间后自动退出。类似于以下内容：

1. 当最后一个活动请求完成时，启动一个计时器（例如 30 秒）。
2. 如果在此期间有任何新请求到达，请停止计时器。
3. 如果计时器到期，请执行正常关闭。

这个想法是释放服务器正在使用的资源，因为我们知道当新客户端出现时，systemd 会再次启动我们。

我不确定第 (1) 和 (2) 部分。理想情况下，我想要一个不涉及修改所有注册处理程序的解决方案。

解决方案

使用 @ceriselimón 的建议，以下辅助类型似乎可以解决问题：

type idletracker struct {
    mu     sync.mutex
    active map[net.conn]bool
    idle   time.duration
    timer  *time.timer
}

func newidletracker(idle time.duration) *idletracker {
    return &idletracker{
        active: make(map[net.conn]bool),
        idle:   idle,
        timer:  time.newtimer(idle),
    }
}

func (t *idletracker) connstate(conn net.conn, state http.connstate) {
    t.mu.lock()
    defer t.mu.unlock()

    oldactive := len(t.active)
    switch state {
    case http.statenew, http.stateactive, http.statehijacked:
        t.active[conn] = true
        // stop the timer if we transitioned to idle
        if oldactive == 0 {
            t.timer.stop()
        }
    case http.stateidle, http.stateclosed:
        delete(t.active, conn)
        // restart the timer if we've become idle
        if oldactive > 0 && len(t.active) == 0 {
            t.timer.reset(t.idle)
        }
    }
}

func (t *idletracker) done() <-chan time.time {
    return t.timer.c
}

将其 connstate 方法分配给服务器的 connstate 成员将跟踪服务器是否繁忙，并在我们空闲了所请求的时间量时向我们发出信号：

idle := NewIdleTracker(5 * time.Second)
server.ConnState = idle.ConnState

go func() {
    <-idle.Done()
    if err := server.Shutdown(context.Background()); err != nil {
        log.Fatalf("error shutting down: %v\n", err)
    }
}()

理论要掌握，实操不能落！以上关于《如何让Go的http.Server闲置一段时间后退出？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！