unistack-org/micro
3
0
Fork 3
Code Issues 14 Pull Requests 2 Actions Packages Projects Releases 242 Wiki Activity

unsubscribe can async (#1596)

Browse Source 
Co-authored-by: huangshaojie <huangshaojie@corp.netease.com>
Co-authored-by: Asim Aslam <asim@aslam.me>
This commit is contained in:
Socket 2020-04-30 17:42:13 +08:00 committed by GitHub
parent 7792dbc34d
commit 46d09ec2bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server/grpc/grpc.go
View File

@ -785,15 +785,21 @@ func (g *grpcServer) Deregister() error {
g.registered = false g.registered = false
wg := sync.WaitGroup{}
for sb, subs := range g.subscribers { for sb, subs := range g.subscribers {
for _, sub := range subs { for _, sub := range subs {
if logger.V(logger.InfoLevel, logger.DefaultLogger) { wg.Add(1)
logger.Infof("Unsubscribing from topic: %s", sub.Topic()) go func(s broker.Subscriber) {
} defer wg.Done()
sub.Unsubscribe() if logger.V(logger.InfoLevel, logger.DefaultLogger) {
logger.Infof("Unsubscribing from topic: %s", s.Topic())
}
s.Unsubscribe()
}(sub)
} }
g.subscribers[sb] = nil g.subscribers[sb] = nil
} }
wg.Wait()
g.Unlock() g.Unlock()
return nil return nil