unistack-org/micro-server-tcp
4
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases 10 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 2 Commits 7 Files Changed 5 +13 -10
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit bbe2425174 - Show all commits

4
subscriber.go
View File

@ -184,11 +184,11 @@ func validateSubscriber(sub server.Subscriber) error {
return nil
}
func (s *tcpServer) createSubHandler(sb *tcpSubscriber, opts server.Options) broker.Handler {
func (h *tcpServer) createSubHandler(sb *tcpSubscriber, opts server.Options) broker.Handler {
return func(p broker.Event) error {
msg := p.Message()
ct := msg.Header["Content-Type"]
cf, err := s.newCodec(ct)
cf, err := h.newCodec(ct)
if err != nil {
return err
}

19
tcp.go
View File

@ -404,6 +404,7 @@ func (h *tcpServer) Start() error {
}
}
h.gracefulStop()
ch <- ts.Close()
// deregister
@ -420,19 +421,21 @@ func (h *tcpServer) Start() error {
}
func (h *tcpServer) Stop() error {
ch := make(chan error)
h.exit <- ch
return <-ch
}
func (h *tcpServer) gracefulStop() {
ctx, cancel := context.WithTimeout(context.Background(), h.opts.GracefulTimeout)
go func() {
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

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

да, я прикинул потом, что закрытие нужно сместить
defer cancel()
h.wg.Wait()
h.opts.Wait.Wait()
cancel()
}()
// wait timeout or finish all connects
<-ctx.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 - залочился, то горутина с ней утекла
ch := make(chan error)
h.exit <- ch
return <-ch
}
func (h *tcpServer) String() string {
@ -483,9 +486,9 @@ func (h *tcpServer) serve(ln net.Listener, hd Handler) {
return
}
h.wg.Add(1)
h.opts.Wait.Add(1)
go func() {
defer h.wg.Done()
defer h.opts.Wait.Done()
hd.Serve(c)
}()
}