Update server and handler to work with streamed responses

httpd/handler.go

@@ -30,6 +30,10 @@ const (
 	DefaultChunkSize = 10000
 )
 
+const (
+	DefaultChunkSize = 10000
+)
+
 // TODO: Standard response headers (see: HeaderHandler)
 // TODO: Compression (see: CompressionHeaderHandler)
 
@@ -201,6 +205,8 @@ func (h *Handler) serveQuery(w http.ResponseWriter, r *http.Request, user *influ
 	chunkSize := DefaultChunkSize
 	if chunked {
 		if cs, err := strconv.ParseInt(q.Get("chunk_size"), 10, 64); err == nil {
+			chunkSize = DefaultChunkSize
+		} else {
 			chunkSize = int(cs)
 		}
 	}

httpd/handler_test.go

@@ -11,6 +11,7 @@ import (
 	"net/http/httptest"
 	"net/url"
 	"os"
+	"reflect"
 	"regexp"
 	"strings"
 	"testing"