Add support for batch request (#3)

Assumes each line in request body
should be logged as a separate log line.

text\ntext\ntext
{json}\n{json}\n{json}

main.go

@@ -1,9 +1,9 @@
 package main
 
 import (
+	"bufio"
 	"encoding/json"
 	"fmt"
-	"io/ioutil"
 	"net/http"
 	"os"
 
@@ -55,25 +55,31 @@ func Logger(logger *log.Logger) func(http.ResponseWriter, *http.Request) {
 			ObjectType: vars[ObjectType],
 			ObjectName: vars[ObjectName],
 		}
-		body, err := ioutil.ReadAll(request.Body)
-		if err != nil {
-			response.WriteHeader(http.StatusBadRequest)
-			response.Write([]byte(err.Error()))
-			return
-		}
-
-		if request.Header.Get("Content-Type") == "application/json" {
-			var jsonBody map[string]interface{}
-			err := json.Unmarshal(body, &jsonBody)
+		bodies := bufio.NewScanner(request.Body)
+		bodies.Split(bufio.ScanLines)
+		/*
+			body, err := ioutil.ReadAll(request.Body)
 			if err != nil {
-				LogRaw(logger, metadata, string(body))
 				response.WriteHeader(http.StatusBadRequest)
 				response.Write([]byte(err.Error()))
 				return
 			}
-			LogJSON(logger, metadata, jsonBody)
-		} else {
-			LogRaw(logger, metadata, string(body))
+		*/
+		for bodies.Scan() {
+			body := bodies.Bytes()
+			if request.Header.Get("Content-Type") == "application/json" {
+				var jsonBody map[string]interface{}
+				err := json.Unmarshal(body, &jsonBody)
+				if err != nil {
+					LogRaw(logger, metadata, string(body))
+					response.WriteHeader(http.StatusBadRequest)
+					response.Write([]byte(err.Error()))
+					return
+				}
+				LogJSON(logger, metadata, jsonBody)
+			} else {
+				LogRaw(logger, metadata, string(body))
+			}
 		}
 		response.WriteHeader(http.StatusOK)
 	}

main_test.go

@@ -20,6 +20,14 @@ func TestRawOutput(t *testing.T) {
 	assert.Contains(t, log, "test-on")
 }
 
+func TestRawMultiLineOutput(t *testing.T) {
+	log, status := run(t, "text/plain", "a-b-c\nd-e-f")
+	assert.Equal(t, http.StatusOK, status)
+
+	assert.Contains(t, log, "\"msg\":\"a-b-c\"")
+	assert.Contains(t, log, "\"msg\":\"d-e-f\"")
+}
+
 func TestJSONOutput(t *testing.T) {
 	log, status := run(t, "application/json", "{\"a\":1}")
 	assert.Equal(t, http.StatusOK, status)
@@ -30,6 +38,14 @@ func TestJSONOutput(t *testing.T) {
 	assert.Contains(t, log, "test-on")
 }
 
+func TestJSONMultiOutput(t *testing.T) {
+	log, status := run(t, "application/json", "{\"a\":1}\n{\"b\":1}\n")
+	assert.Equal(t, http.StatusOK, status)
+
+	assert.Contains(t, log, "{\"a\":1,")
+	assert.Contains(t, log, "{\"b\":1,")
+}
+
 func TestJSONOutputMsgFromOriginal(t *testing.T) {
 	log, status := run(t, "application/json", "{\"msg\":\"a\"}")
 	assert.Equal(t, http.StatusOK, status)