diff --git a/pkg/server/server.go b/pkg/server/server.go index 69857b5..11f3e4d 100644 --- a/pkg/server/server.go +++ b/pkg/server/server.go @@ -35,7 +35,7 @@ type CacheServer struct { } type CacheData struct { - Body []byte + Body []byte Headers map[string]string } @@ -77,7 +77,7 @@ func (server CacheServer) Start(stopChan chan int) { if cacheHeadersEnabled == "true" { headers := make(map[string]string) - for h,v := range r.Header{ + for h, v := range r.Header { headers[h] = strings.Join(v, ";") } cacheData.Headers = headers @@ -136,14 +136,14 @@ func determinatePort() string { } func (server CacheServer) gzipWriter(b []byte) *bytes.Buffer { - buf := bytes.NewBuffer([]byte{}) - gzipWriter := gzip.NewWriter(buf) + var buf bytes.Buffer + gzipWriter := gzip.NewWriter(&buf) _, err := gzipWriter.Write(b) if err != nil { server.Logger.Error("Gzip Writer Encountered With an Error", zap.Error(err)) } gzipWriter.Close() - return buf + return &buf } func (server CacheServer) CacheHandler(w http.ResponseWriter, r *http.Request) { diff --git a/pkg/tests/server_bench_test.go b/pkg/tests/server_bench_test.go index f37f577..6386a0a 100644 --- a/pkg/tests/server_bench_test.go +++ b/pkg/tests/server_bench_test.go @@ -8,8 +8,11 @@ import ( func BenchmarkServerHash(b *testing.B) { os.Setenv("CACHE_KEY_PREFIX", "test") - var cacheServer *server.CacheServer = new(server.CacheServer) + var cacheServer = new(server.CacheServer) + + b.ResetTimer() + for n := 0; n < b.N; n++ { cacheServer.HashURL("adsfadsdfasdfas") } -} \ No newline at end of file +} diff --git a/server.test b/server.test new file mode 100755 index 0000000..a3a91b7 Binary files /dev/null and b/server.test differ