1 files changed, 8 insertions, 0 deletions
diff --git a/internal/net/wsconn.go b/internal/net/wsconn.go
index 4990c0b..34663ba 100644
--- a/internal/net/wsconn.go
+++ b/internal/net/wsconn.go
@@ -1,6 +1,8 @@
 package net
 
 import (
+	"time"
+
 	"github.com/gorilla/websocket"
 )
 
@@ -25,6 +27,12 @@ func (w *wsConn) Write(b []byte) (int, error) {
 	return len(b), err
 }
 
+// SetWriteDeadline delegates to the underlying websocket connection, which
+// supports deadlines natively.
+func (w *wsConn) SetWriteDeadline(ts time.Time) error {
+	return w.conn.SetWriteDeadline(ts)
+}
+
 func (w *wsConn) Close() error {
 	return w.conn.Close()
 }