From d9a690a98f252dff4c66d20184cd61e4280759a8 Mon Sep 17 00:00:00 2001 From: Adnan Maolood Date: Fri, 18 Dec 2020 01:47:29 -0500 Subject: [PATCH] Make NewResponseWriter take an io.Writer --- server.go | 7 ++++--- 1 file 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), } }