fix(dcpool): invalid takeout caused by double lock

iyear · Oct 3, 2024 · ec66ab1 · ec66ab1
1 parent c874919
commit ec66ab1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/dcpool/dcpool.go b/core/dcpool/dcpool.go
@@ -48,13 +48,13 @@ func (p *pool) current() int {
 }
 
 func (p *pool) Client(ctx context.Context, dc int) *tg.Client {
+	p.mu.Lock()
+	defer p.mu.Unlock()
+
 	return tg.NewClient(p.invoker(ctx, dc))
 }
 
 func (p *pool) invoker(ctx context.Context, dc int) tg.Invoker {
-	p.mu.Lock()
-	defer p.mu.Unlock()
-
 	if i, ok := p.invokers[dc]; ok {
 		return i
 	}