From 1ae8d7af7f15da6a0b496359d10624ab9598c843 Mon Sep 17 00:00:00 2001
From: Anmol Sethi <hi@nhooyr.io>
Date: Fri, 25 Oct 2019 12:34:23 -0400
Subject: [PATCH] Fix for gin

Closes #166
---
 handshake.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/handshake.go b/handshake.go
index d1a9fba4..d6006b8d 100644
--- a/handshake.go
+++ b/handshake.go
@@ -141,6 +141,13 @@ func accept(w http.ResponseWriter, r *http.Request, opts *AcceptOptions) (*Conn,
 
 	w.WriteHeader(http.StatusSwitchingProtocols)
 
+	// Fix for gin, see #166
+	if w, ok := w.(interface {
+		WriteHeaderNow()
+	}); ok {
+		w.WriteHeaderNow()
+	}
+
 	netConn, brw, err := hj.Hijack()
 	if err != nil {
 		err = fmt.Errorf("failed to hijack connection: %w", err)