Как я могу заставить http выйти из Go сервера после простоя в течение определенного периода времени? - PullRequest
Новая
       21

Как я могу заставить http выйти из Go сервера после простоя в течение определенного периода времени?

0 голосов
/ 07 июня 2019

Я пишу веб-сервер на Go, используя стандартную библиотеку net/http, в которой используется активация сокета systemd.

У меня есть базовые принципы работы: сервер запускается при первом подключенииподключен к прослушивающему сокету, и я могу выполнить плавное отключение при получении сигнала (т. е. systemctl stop будет работать без отключения активных запросов).

Я хотел бы, чтобы сервер автоматически завершал работу, когда он имеетпростоял в течение некоторого периода времени.Что-то вроде следующего:

  1. , когда последний активный запрос завершится, запустите таймер, скажем, на 30 секунд.
  2. если в течение этого периода поступит какой-либо новый запрос, остановите таймер.
  3. если таймер истекает, выполните постепенное отключение.

Идея состоит в том, чтобы освободить ресурсы, которые использовал сервер, зная, что systemd снова запустит нас, когда появится новый клиент.

Это части (1) и (2), в которых я не уверен.В идеале мне бы хотелось, чтобы решение не включало также изменение всех зарегистрированных обработчиков.

1 Ответ

0 голосов
/ 08 июня 2019

Используя предложение @ 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)
    }
}()
...