forked from Elara6331/itd
Properly close multiplexed streams
This commit is contained in:
parent
94ec82c4a6
commit
86a77f6f1f
20
api/drpc.go
20
api/drpc.go
@ -51,6 +51,7 @@ func (m *muxConn) Invoke(ctx context.Context, rpc string, enc drpc.Encoding, in,
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
dconn := drpcconn.New(conn)
|
dconn := drpcconn.New(conn)
|
||||||
|
defer dconn.Close()
|
||||||
return dconn.Invoke(ctx, rpc, enc, in, out)
|
return dconn.Invoke(ctx, rpc, enc, in, out)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -59,7 +60,22 @@ func (m *muxConn) NewStream(ctx context.Context, rpc string, enc drpc.Encoding)
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
dconn := drpcconn.New(conn)
|
dconn := drpcconn.New(conn)
|
||||||
return dconn.NewStream(ctx, rpc, enc)
|
|
||||||
|
go func() {
|
||||||
|
<-dconn.Closed()
|
||||||
|
conn.Close()
|
||||||
|
}()
|
||||||
|
|
||||||
|
s, err := dconn.NewStream(ctx, rpc, enc)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
<-s.Context().Done()
|
||||||
|
dconn.Close()
|
||||||
|
}()
|
||||||
|
|
||||||
|
return s, nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user