From 3092e9f62f6988391184f52057787f63d1088839 Mon Sep 17 00:00:00 2001
From: bel <bel@bel.bel>
Date: Fri, 12 Aug 2022 23:35:00 -0600
Subject: [PATCH] throttle https

---
 server.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

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)