unistack-org/micro-broker-kgo
4
0
Fork 2
Code Issues 3 Pull Requests 1 Actions 8 Packages Projects Releases 45 Wiki Activity

allow rebalance on unsubscribe #125

Merged
vtolstov merged 1 commits from unsubfix into v3 2023-05-25 15:12:55 +03:00
Conversation 0 Commits 1 Files Changed 1 +3 -3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
subscriber.go
View File

@ -52,10 +52,11 @@ func (s *subscriber) Unsubscribe(ctx context.Context) error {
return nil return nil
} }
select { select {
case <-ctx.Done(): // case <-ctx.Done():
return ctx.Err() // return ctx.Err()
default: default:
s.c.PauseFetchTopics(s.topic) s.c.PauseFetchTopics(s.topic)
s.c.CloseAllowingRebalance()
kc := make(map[string][]int32) kc := make(map[string][]int32)
for ctp := range s.consumers { for ctp := range s.consumers {
kc[ctp.t] = append(kc[ctp.t], ctp.p) kc[ctp.t] = append(kc[ctp.t], ctp.p)
@ -80,7 +81,6 @@ func (s *subscriber) poll(ctx context.Context) {
s.c.CloseAllowingRebalance() s.c.CloseAllowingRebalance()
return return
case <-s.done: case <-s.done:
s.c.CloseAllowingRebalance()
return return
default: default:
fetches := s.c.PollRecords(ctx, maxInflight) fetches := s.c.PollRecords(ctx, maxInflight)