unistack-org/micro-server-tcp
3
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases 12 Wiki Activity

add waitGroups for waiting finish all connects #131

Merged
vtolstov merged 7 commits from devstigneev/micro-server-tcp:issue_130 into v3 2024-03-13 15:36:35 +03:00
Conversation 7 Commits 7 Files Changed 5 +10 -21
1 changed files with 10 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 8661d62ce8 - Show all commits

25
tcp.go
View File

@ -2,7 +2,6 @@
package tcp // import "go.unistack.org/micro-server-tcp/v3" package tcp // import "go.unistack.org/micro-server-tcp/v3"
import ( import (
"context"
"crypto/tls" "crypto/tls"
"fmt" "fmt"
"net" "net"
@ -16,7 +15,6 @@ import (
"go.unistack.org/micro/v3/register" "go.unistack.org/micro/v3/register"
"go.unistack.org/micro/v3/server" "go.unistack.org/micro/v3/server"
"golang.org/x/net/netutil" "golang.org/x/net/netutil"
"golang.org/x/sync/errgroup"
) )
type tcpServer struct { type tcpServer struct {
@ -28,8 +26,6 @@ type tcpServer struct {
sync.RWMutex sync.RWMutex
registered bool registered bool
init bool init bool
wg sync.WaitGroup
} }
func (h *tcpServer) newCodec(ct string) (codec.Codec, error) { func (h *tcpServer) newCodec(ct string) (codec.Codec, error) {
@ -431,24 +427,17 @@ func (h *tcpServer) gracefulStop() {
tm := time.NewTimer(h.opts.GracefulTimeout) tm := time.NewTimer(h.opts.GracefulTimeout)
defer tm.Stop() defer tm.Stop()
g, gctx := errgroup.WithContext(context.Background()) done := make(chan struct{})
go func() {
g.Go(func() error {
h.opts.Wait.Wait() h.opts.Wait.Wait()
vtolstov marked this conversation as resolved Outdated
vtolstov commented 2024-03-08 15:15:46 +03:00
Outdated
Review
Copy Link

смотри, лучше сделать канал и если вейтгруппа кончилась- прямо там закрывать канал, тогда в селекте мы сразу попадем на нужный кейс

смотри, лучше сделать канал и если вейтгруппа кончилась- прямо там закрывать канал, тогда в селекте мы сразу попадем на нужный кейс
devstigneev commented 2024-03-08 15:36:45 +03:00
Outdated
Review
Copy Link

да, я прикинул потом, что закрытие нужно сместить

да, я прикинул потом, что закрытие нужно сместить
return nil done <- struct{}{}
}) }()
g.Go(func() error {
select { select {
case <-tm.C: case <-tm.C:
return nil case <-done:
devstigneev commented 2024-03-06 13:02:16 +03:00
Outdated
Review
Copy Link

Тут мы ждем либо у нас концертны все закрылись и вызвался cancel()
либо ctx.Done выполнился по GracefulTimeout

Тут мы ждем либо у нас концертны все закрылись и вызвался cancel() либо ctx.Done выполнился по GracefulTimeout
devstigneev commented 2024-03-06 13:04:05 +03:00
Outdated
Review
Copy Link

если я использую time.Timer, не совсем смог прийти к решению, что я его в ручную остановил и мог без препятствий завершить метод

если я использую time.Timer, не совсем смог прийти к решению, что я его в ручную остановил и мог без препятствий завершить метод
devstigneev commented 2024-03-06 13:05:05 +03:00
Outdated
Review
Copy Link

но возникает другая проблема, если Wait - залочился, то горутина с ней утекла

но возникает другая проблема, если Wait - залочился, то горутина с ней утекла
case <-gctx.Done(): close(done)
return gctx.Err()
} }
})
g.Wait()
return
} }
func (h *tcpServer) String() string { func (h *tcpServer) String() string {
@ -501,8 +490,8 @@ func (h *tcpServer) serve(ln net.Listener, hd Handler) {
h.opts.Wait.Add(1) h.opts.Wait.Add(1)
go func() { go func() {
defer h.opts.Wait.Done()
hd.Serve(c) hd.Serve(c)
h.opts.Wait.Done()
}() }()
} }
} }