Make NewResponseWriter take an io.Writer

1 files changed, 4 insertions, 3 deletions

diff --git a/server.go b/server.go
index 3e069ae..20d57bf 100644
--- a/server.go
+++ b/server.go
@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"crypto/tls"
 	"errors"
+	"io"
 	"log"
 	"net"
 	"strconv"
@@ -239,10 +240,10 @@ type ResponseWriter struct {
 	mediatype   string
 }
 
-// NewResponseWriter returns a ResponseWriter that will write to conn.
-func NewResponseWriter(conn net.Conn) *ResponseWriter {
+// NewResponseWriter returns a ResponseWriter that uses the provided io.Writer.
+func NewResponseWriter(w io.Writer) *ResponseWriter {
 	return &ResponseWriter{
-		b: bufio.NewWriter(conn),
+		b: bufio.NewWriter(w),
 	}
 }