initial commit

gzipresponsewriter.go

@@ -0,0 +1,36 @@
+package gziphttp
+
+import (
+	"compress/gzip"
+	"net/http"
+)
+
+type GZipResponseWriter struct {
+	w  http.ResponseWriter
+	gz *gzip.Writer
+}
+
+func New(w http.ResponseWriter) *GZipResponseWriter {
+	w.Header().Add("Content-Type", "text/html")
+	w.Header().Add("Content-Encoding", "gzip")
+	return &GZipResponseWriter{
+		w:  w,
+		gz: gzip.NewWriter(w),
+	}
+}
+
+func (gz *GZipResponseWriter) Header() http.Header {
+	return gz.w.Header()
+}
+
+func (gz *GZipResponseWriter) WriteHeader(statusCode int) {
+	gz.w.WriteHeader(statusCode)
+}
+
+func (gz *GZipResponseWriter) Write(b []byte) (int, error) {
+	return gz.gz.Write(b)
+}
+
+func (gz *GZipResponseWriter) Close() error {
+	return gz.gz.Close()
+}

gzipresponsewriter_test.go

@@ -0,0 +1,34 @@
+package gziphttp
+
+import (
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestGZipResponseWriter(t *testing.T) {
+	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		gz := New(w)
+		defer gz.Close()
+		fmt.Fprintln(gz, "Hello world")
+	}))
+	defer s.Close()
+
+	req := httptest.NewRequest("GET", s.URL, nil)
+	req.RequestURI = ""
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		t.Fatal(resp.StatusCode)
+	}
+	if b, err := ioutil.ReadAll(resp.Body); err != nil {
+		t.Fatal(err)
+	} else if s := string(b); s != "Hello world\n" {
+		t.Fatalf("%q", s)
+	}
+}