github
/
v2ray-core
mirror of https://github.com/v2ray/v2ray-core
1
0
Fork 0
Code Issues Releases Wiki Activity

wait for done

pull/432/head
Darien Raymond 2017-04-05 13:46:11 +02:00
parent 32842bc14b
commit 55328921b5
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/proxyman/mux/mux.go
View File

@ -182,6 +182,13 @@ func fetchInput(ctx context.Context, s *session, output buf.Writer) {
} }
} }
func waitForDone(ctx context.Context, s *session) {
<-ctx.Done()
s.closeUplink()
s.closeDownlink()
s.output.Close()
}
func (m *Client) Dispatch(ctx context.Context, outboundRay ray.OutboundRay) bool { func (m *Client) Dispatch(ctx context.Context, outboundRay ray.OutboundRay) bool {
m.access.Lock() m.access.Lock()
defer m.access.Unlock() defer m.access.Unlock()
@ -210,6 +217,7 @@ func (m *Client) Dispatch(ctx context.Context, outboundRay ray.OutboundRay) bool
} }
m.sessions[id] = s m.sessions[id] = s
go fetchInput(ctx, s, m.inboundRay.InboundInput()) go fetchInput(ctx, s, m.inboundRay.InboundInput())
go waitForDone(ctx, s)
return true return true
} }