mirror
/
websocket
mirror of https://github.com/gorilla/websocket.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

use done chan for waiting pipe

This commit is contained in:
Cooper Oh 2024-07-19 00:32:35 +09:00
parent 5e557d257e
commit d249ef1a3b
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
client_server_test.go
View File

@ -109,17 +109,16 @@ func (s *cstProxyServer) ServeHTTP(w http.ResponseWriter, req *http.Request) {
_, _ = fmt.Fprintf(conn, "HTTP/1.1 200 Connection established\r\n\r\n") _, _ = fmt.Fprintf(conn, "HTTP/1.1 200 Connection established\r\n\r\n")
wg := sync.WaitGroup{} done := make(chan struct{}, 2)
wg.Add(2)
go func() { go func() {
defer wg.Done()
_, _ = io.Copy(upstream, conn) _, _ = io.Copy(upstream, conn)
done <- struct{}{}
}() }()
go func() { go func() {
defer wg.Done()
_, _ = io.Copy(conn, upstream) _, _ = io.Copy(conn, upstream)
done <- struct{}{}
}() }()
wg.Wait() <-done
} }
func newProxyServer() *httptest.Server { func newProxyServer() *httptest.Server {