diff --git a/server.go b/server.go index 785228a..041e888 100644 --- a/server.go +++ b/server.go @@ -69,7 +69,14 @@ func (s *Server) Connect(w http.ResponseWriter, r *http.Request) { xfer := func(dst io.WriteCloser, src io.ReadCloser) { defer dst.Close() defer src.Close() - io.Copy(dst, src) + io.Copy( + throttledWriter{ + ctx: r.Context(), + w: dst, + limiter: s.limiter, + }, + src, + ) } go xfer(dest, client)