add waitGroups for waiting finish all connects #131
25
tcp.go
25
tcp.go
@ -2,7 +2,6 @@
|
||||
package tcp // import "go.unistack.org/micro-server-tcp/v3"
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
"net"
|
||||
@ -16,7 +15,6 @@ import (
|
||||
"go.unistack.org/micro/v3/register"
|
||||
"go.unistack.org/micro/v3/server"
|
||||
"golang.org/x/net/netutil"
|
||||
"golang.org/x/sync/errgroup"
|
||||
)
|
||||
|
||||
type tcpServer struct {
|
||||
@ -28,8 +26,6 @@ type tcpServer struct {
|
||||
sync.RWMutex
|
||||
registered bool
|
||||
init bool
|
||||
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
func (h *tcpServer) newCodec(ct string) (codec.Codec, error) {
|
||||
@ -431,24 +427,17 @@ func (h *tcpServer) gracefulStop() {
|
||||
tm := time.NewTimer(h.opts.GracefulTimeout)
|
||||
defer tm.Stop()
|
||||
|
||||
g, gctx := errgroup.WithContext(context.Background())
|
||||
|
||||
g.Go(func() error {
|
||||
done := make(chan struct{})
|
||||
go func() {
|
||||
h.opts.Wait.Wait()
|
||||
vtolstov marked this conversation as resolved
Outdated
|
||||
return nil
|
||||
})
|
||||
done <- struct{}{}
|
||||
}()
|
||||
|
||||
g.Go(func() error {
|
||||
select {
|
||||
case <-tm.C:
|
||||
return nil
|
||||
case <-gctx.Done():
|
||||
return gctx.Err()
|
||||
case <-done:
|
||||
devstigneev
commented
Тут мы ждем либо у нас концертны все закрылись и вызвался cancel() Тут мы ждем либо у нас концертны все закрылись и вызвался cancel()
либо ctx.Done выполнился по GracefulTimeout
devstigneev
commented
если я использую time.Timer, не совсем смог прийти к решению, что я его в ручную остановил и мог без препятствий завершить метод если я использую time.Timer, не совсем смог прийти к решению, что я его в ручную остановил и мог без препятствий завершить метод
devstigneev
commented
но возникает другая проблема, если Wait - залочился, то горутина с ней утекла но возникает другая проблема, если Wait - залочился, то горутина с ней утекла
|
||||
close(done)
|
||||
}
|
||||
})
|
||||
|
||||
g.Wait()
|
||||
return
|
||||
}
|
||||
|
||||
func (h *tcpServer) String() string {
|
||||
@ -501,8 +490,8 @@ func (h *tcpServer) serve(ln net.Listener, hd Handler) {
|
||||
|
||||
h.opts.Wait.Add(1)
|
||||
go func() {
|
||||
defer h.opts.Wait.Done()
|
||||
hd.Serve(c)
|
||||
h.opts.Wait.Done()
|
||||
}()
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user
смотри, лучше сделать канал и если вейтгруппа кончилась- прямо там закрывать канал, тогда в селекте мы сразу попадем на нужный кейс
да, я прикинул потом, что закрытие нужно сместить