Что ожидает wg.Wait() ?

@Timurs · Регистрация: 23.02.2016

Студворк — интернет-сервис помощи студентам

Есть такая библиотека. В структуре сервера есть поле wg sync.WaitGroup, но я никак не могу понять как они с ним работают.
https://github.com/fvbock/endl... endless.go

Например, у меня есть некоторая функция, которую я вызываю 3 раза передавая адреса с разными портами.

Go
1
2
3
4
5
func startServer(addr string) error {
    router := gin.Default()
    server := endless.NewServer(addr, router)
    return server.ListenAndServe()
}

Мне интересно каким образом закроется вызов этой функции. Метод server.ListenAndServe() запустит горутину srv.handleSignals() и вернет метод srv.Serve()

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
func (srv *endlessServer) ListenAndServe() (err error) {
    addr := srv.Addr
    if addr == "" {
        addr = ":http"
    }
 
    go srv.handleSignals()
 
    l, err := srv.getListener(addr)
    if err != nil {
        log.Println(err)
        return
    }
 
    srv.EndlessListener = newEndlessListener(l, srv)
 
    if srv.isChild {
        syscall.Kill(syscall.Getppid(), syscall.SIGTERM)
    }
 
    srv.BeforeBegin(srv.Addr)
 
    return srv.Serve()
}

Горутина srv.handleSignals() насколько я понимаю открывает бесконечный цикл

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
func (srv *endlessServer) handleSignals() {
    var sig os.Signal
 
    signal.Notify(
        srv.sigChan,
        hookableSignals...,
    )
 
    pid := syscall.Getpid()
    for {
        sig = <-srv.sigChan
        srv.signalHooks(PRE_SIGNAL, sig)
        switch sig {
        case syscall.SIGHUP:
            log.Println(pid, "Received SIGHUP. forking.")
            err := srv.fork()
            if err != nil {
                log.Println("Fork err:", err)
            }
        case syscall.SIGUSR1:
            log.Println(pid, "Received SIGUSR1.")
        case syscall.SIGUSR2:
            log.Println(pid, "Received SIGUSR2.")
            srv.hammerTime(0 * time.Second)
        case syscall.SIGINT:
            log.Println(pid, "Received SIGINT.")
            srv.shutdown()
        case syscall.SIGTERM:
            log.Println(pid, "Received SIGTERM.")
            srv.shutdown()
        case syscall.SIGTSTP:
            log.Println(pid, "Received SIGTSTP.")
        default:
            log.Printf("Received %v: nothing i care about...\n", sig)
        }
        srv.signalHooks(POST_SIGNAL, sig)
    }
}

Я нажимаю ctrl+c, поэтому иду в srv.shutdown(), который устанавливает стейт серверу в STATE_SHUTTING_DOWN и запускает еще одну горутину

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
func (srv *endlessServer) shutdown() {
    if srv.getState() != STATE_RUNNING {
        return
    }
 
    srv.setState(STATE_SHUTTING_DOWN)
    if DefaultHammerTime >= 0 {
        go srv.hammerTime(DefaultHammerTime)
    }
    // disable keep-alives on existing connections
    srv.SetKeepAlivesEnabled(false)
    err := srv.EndlessListener.Close()
    if err != nil {
        log.Println(syscall.Getpid(), "Listener.Close() error:", err)
    } else {
        log.Println(syscall.Getpid(), srv.EndlessListener.Addr(), "Listener closed.")
    }
}

Которая делает что-то совсем непонятное

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
func (srv *endlessServer) hammerTime(d time.Duration) {
    defer func() {
        // we are calling srv.wg.Done() until it panics which means we called
        // Done() when the counter was already at 0 and we're done.
        // (and thus Serve() will return and the parent will exit)
        if r := recover(); r != nil {
            log.Println("WaitGroup at 0", r)
        }
    }()
    if srv.getState() != STATE_SHUTTING_DOWN {
        return
    }
    time.Sleep(d)
    log.Println("[STOP - Hammer Time] Forcefully shutting down parent")
    for {
        if srv.getState() == STATE_TERMINATE {
            break
        }
        srv.wg.Done()
        runtime.Gosched()
    }
}

Ну и srv.Serve() запускает wg.Wait()

Go
1
2
3
4
5
6
7
8
9
func (srv *endlessServer) Serve() (err error) {
    defer log.Println(syscall.Getpid(), "Serve() returning...")
    srv.setState(STATE_RUNNING)
    err = srv.Server.Serve(srv.EndlessListener)
    log.Println(syscall.Getpid(), "Waiting for connections to finish...")
    srv.wg.Wait()
    srv.setState(STATE_TERMINATE)
    return
}

Вопрос собственно такой, где wg.Add(), завершения каких горутин srv.wg.Wait() ждет?

И еще такой, если в hammerTime() прописать такое

Go
1
2
3
4
5
        // ...
    println("State server before hammer time: ", srv.getState())
    time.Sleep(d)
    println("State server after hammer time: ", srv.getState())
        // ...

То можно заметить, что после time.Sleep(d) состояние сервера меняется на STATE_TERMINATE, за счет чего так получается?

@Timurs · 11.11.2022, 16:20 **[ТС]**

Так, кажется, что разобрался.

1.
У сервера есть поле EndlessListener типа net.Listener

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
// A Listener is a generic network listener for stream-oriented protocols.
//
// Multiple goroutines may invoke methods on a Listener simultaneously.
type Listener interface {
    // Accept waits for and returns the next connection to the listener.
    Accept() (Conn, error)
 
    // Close closes the listener.
    // Any blocked Accept operations will be unblocked and return errors.
    Close() error
 
    // Addr returns the listener's network address.
    Addr() Addr
}

В коде пакета endless создается структура

Go
1
2
3
4
5
type endlessListener struct {
    net.Listener
    stopped bool
    server  *endlessServer
}

При помощи которой реализуется интерфейс net.Listener
Для примера Accept()

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
func (el *endlessListener) Accept() (c net.Conn, err error) {
    println("Ожидаю ваших соединений...")
    tc, err := el.Listener.(*net.TCPListener).AcceptTCP()
    if err != nil {
        println("Пришло время прощаться, листенер закрывается.")
        println("До новых встреч.")
        return
    }
    tc.SetKeepAlive(true)                  // see http.tcpKeepAliveListener
    tc.SetKeepAlivePeriod(3 * time.Minute) // see http.tcpKeepAliveListener
 
    c = endlessConn{
        Conn:   tc,
        server: el.server,
    }
 
    el.server.wg.Add(1)
    return
}

2.
В методе server.ListenAndServe() экземпляр структуры присваивается полю сервера EndlessListener, то есть создаем свой листенер.

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
func (srv *endlessServer) ListenAndServe() (err error) {
    addr := srv.Addr
    if addr == "" {
        addr = ":http"
    }
 
    go srv.handleSignals()
 
    l, err := srv.getListener(addr)
    if err != nil {
        log.Println(err)
        return
    }
        
        println("Как понимаю утиная типизация?")
    srv.EndlessListener = newEndlessListener(l, srv)
 
    if srv.isChild {
        syscall.Kill(syscall.Getppid(), syscall.SIGTERM)
    }
 
    srv.BeforeBegin(srv.Addr)
 
    return srv.Serve()
}

3.
В методе server.Serve() вызывается метод Serve(net.Listener) из пакета http, в который прокидывается реализация своего листенера.

Go
1
2
3
4
5
6
7
8
9
10
11
12
func (srv *endlessServer) Serve() (err error) {
    defer log.Println(syscall.Getpid(), "Serve() returning...")
    srv.setState(STATE_RUNNING)
    fmt.Println("Прокидывание листенера")
    err = srv.Server.Serve(srv.EndlessListener)
    println("Я закрыл соединение: ", err.Error())
    log.Println(syscall.Getpid(), "Waiting for connections to finish...")
    srv.wg.Wait()
    srv.setState(STATE_TERMINATE)
    println("Всё, я выключаю сервер!")
    return
}

Который внутри себя вызывает метод Accept()

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
func (srv *Server) Serve(l net.Listener) error {
    if fn := testHookServerServe; fn != nil {
        fn(srv, l) // call hook with unwrapped listener
    }
 
    origListener := l
    l = &onceCloseListener{Listener: l}
    defer l.Close()
 
    if err := srv.setupHTTP2_Serve(); err != nil {
        return err
    }
 
    if !srv.trackListener(&l, true) {
        return ErrServerClosed
    }
    defer srv.trackListener(&l, false)
 
    baseCtx := context.Background()
    if srv.BaseContext != nil {
        baseCtx = srv.BaseContext(origListener)
        if baseCtx == nil {
            panic("BaseContext returned a nil context")
        }
    }
 
    var tempDelay time.Duration // how long to sleep on accept failure
 
    ctx := context.WithValue(baseCtx, ServerContextKey, srv)
    for {
        rw, err := l.Accept()
        if err != nil {
            select {
            case <-srv.getDoneChan():
                return ErrServerClosed
            default:
            }
            if ne, ok := err.(net.Error); ok && ne.Temporary() {
                if tempDelay == 0 {
                    tempDelay = 5 * time.Millisecond
                } else {
                    tempDelay *= 2
                }
                if max := 1 * time.Second; tempDelay > max {
                    tempDelay = max
                }
                srv.logf("http: Accept error: %v; retrying in %v", err, tempDelay)
                time.Sleep(tempDelay)
                continue
            }
            return err
        }
        connCtx := ctx
        if cc := srv.ConnContext; cc != nil {
            connCtx = cc(connCtx, rw)
            if connCtx == nil {
                panic("ConnContext returned nil")
            }
        }
        tempDelay = 0
        c := srv.newConn(rw)
        c.setState(c.rwc, StateNew, runHooks) // before Serve can return
        go c.serve(connCtx)
    }
}

Вызвав его программа ожидает запросов

Go
1
2
3
4
5
6
7
8
    println("Ожидаю ваших соединений...")
    tc, err := el.Listener.(*net.TCPListener).AcceptTCP()
    tc, err := el.Listener.(*net.TCPListener).AcceptTCP()
    if err != nil {
        println("Пришло время прощаться, листенер закрывается.")
        println("До новых встреч.")
        return
    }

4.
Если нажать ctrl+c не совершая никаких запросов, то вызывается shutdown(), закрывая листенер

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
func (srv *endlessServer) shutdown(num int) {
    if srv.getState() != STATE_RUNNING {
        println("Not shutdown: ", srv.getState())
        return
    }
 
    srv.setState(STATE_SHUTTING_DOWN)
    if DefaultHammerTime >= 0 {
        go srv.hammerTime(DefaultHammerTime)
    }
    // disable keep-alives on existing connections
        println("Отключение постоянного http-соединения")
    srv.SetKeepAlivesEnabled(false)
        println("и закрытие листенера")
    err := srv.EndlessListener.Close()
    if err != nil {
        log.Println(syscall.Getpid(), "Listener.Close() error:", err)
    } else {
        log.Println(syscall.Getpid(), srv.EndlessListener.Addr(), "Listener closed.")
    }
}

5.
Тот в свою очередь возвращает ошибку в Accept() и el.server.wg.Add(1) не выполняется.

Go
1
2
3
4
5
    println("Ожидаю ваших соединений...")
    tc, err := el.Listener.(*net.TCPListener).AcceptTCP()
    if err != nil {
        return
    }

6.
А значит в Serve() вейтгруппа никого не ждет srv.wg.Wait(). Далее устанавливается серверу состояние терминейт и происходит выход из функций Serve(), ListenAndServe() и startServer(...)

Go
1
2
3
4
5
6
7
8
9
10
11
12
func (srv *endlessServer) Serve() (err error) {
    defer log.Println(syscall.Getpid(), "Serve() returning...")
    srv.setState(STATE_RUNNING)
    fmt.Println("EL: ", srv.EndlessListener)
    err = srv.Server.Serve(srv.EndlessListener)
    println("Я закрыл соединение: ", err.Error())
    log.Println(syscall.Getpid(), "Waiting for connections to finish...")
    srv.wg.Wait()
    srv.setState(STATE_TERMINATE)
    println("Всё, я выключаю сервер!")
    return
}

Новые блоги и статьи Все статьи Все блоги /
SDL3 для Web (WebAssembly): Основы отладки веб-приложений на SDL3 по USB и Wi-Fi, запущенных в браузере мобильных устройств 8Observer8 07.02.2026 Содержание блога Браузер Chrome имеет средства для отладки мобильных веб-приложений по USB. В этой пошаговой инструкции ограничимся работой с консолью. Вывод в консоль - это часть процесса. . .	SDL3 для Web (WebAssembly): Обработчик клика мыши в браузере ПК и касания экрана в браузере на мобильном устройстве 8Observer8 02.02.2026 Содержание блога Для начала пошагово создадим рабочий пример для подготовки к экспериментам в браузере ПК и в браузере мобильного устройства. Потом напишем обработчик клика мыши и обработчик. . .	Философия технологии iceja 01.02.2026 На мой взгляд у человека в технических проектах остается роль генерального директора. Все остальное нейронки делают уже лучше человека. Они не могут нести предпринимательские риски, не могут. . .	SDL3 для Web (WebAssembly): Вывод текста со шрифтом TTF с помощью SDL3_ttf 8Observer8 01.02.2026 Содержание блога В этой пошаговой инструкции создадим с нуля веб-приложение, которое выводит текст в окне браузера. Запустим на Android на локальном сервере. Загрузим Release на бесплатный. . .
SDL3 для Web (WebAssembly): Сборка C/C++ проекта из консоли 8Observer8 30.01.2026 Содержание блога Если вы откроете примеры для начинающих на официальном репозитории SDL3 в папке: examples, то вы увидите, что все примеры используют следующие четыре обязательные функции, а. . .	SDL3 для Web (WebAssembly): Установка Emscripten SDK (emsdk) и CMake для сборки C и C++ приложений в Wasm 8Observer8 30.01.2026 Содержание блога Для того чтобы скачать Emscripten SDK (emsdk) необходимо сначало скачать и уставить Git: Install for Windows. Следуйте стандартной процедуре установки Git через установщик. . . .	SDL3 для Android: Подключение Box2D v3, физика и отрисовка коллайдеров 8Observer8 29.01.2026 Содержание блога Box2D - это библиотека для 2D физики для анимаций и игр. С её помощью можно определять были ли коллизии между конкретными объектами. Версия v3 была полностью переписана на Си, в. . .	Инструменты COM: Сохранение данный из VARIANT в файл и загрузка из файла в VARIANT bedvit 28.01.2026 Сохранение базовых типов COM и массивов (одномерных или двухмерных) любой вложенности (деревья) в файл, с возможностью выбора алгоритмов сжатия и шифрования. Часть библиотеки BedvitCOM Использованы. . .