修复map加锁无效

ouqiang · Jun 26, 2017 · 3d61c76 · 3d61c76
1 parent 3e8c3a2
commit 3d61c76
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/rpc/grpcpool/grpc_pool.go b/modules/rpc/grpcpool/grpc_pool.go
@@ -32,16 +32,18 @@ type GRPCPool struct {
 
 func (p *GRPCPool) Get(addr string) (*grpc.ClientConn, error)  {
     p.RLock()
-    defer p.RUnlock()
     pool, ok := p.conns[addr]
+    p.RUnlock()
     if !ok {
         err := p.newCommonPool(addr)
         if err != nil {
             return nil, err
         }
     }
 
+    p.RLock()
     pool = p.conns[addr]
+    p.RUnlock()
     conn, err := pool.Get()
     if err != nil {
         return nil, err