时间戳

首页 > 解决方案 > 闲置一段时间后如何让Go的http.Server退出？

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

问题描述

我正在net/http使用使用 systemd 套接字激活的标准库包在 Go 中编写 Web 服务器。

我有基本的工作，以便服务器在第一次连接到侦听套接字时启动，并且我可以在收到信号时执行正常关闭（即这样systemctl stop可以在不切断活动请求的情况下工作）。

我想要的是服务器在空闲一段时间后自动退出。类似于以下内容：

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

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

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

标签： httpgosystemd

解决方案

使用@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)
    }
}()

推荐阅读