Merge pull request #40 from FireTail-io/ipv6

Ipv6

Author: rileyfiretail

logging/default_batch_callback.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"log"
 	"net/http"
 )
 
@@ -53,5 +54,8 @@ func getDefaultBatchCallback(options BatchLoggerOptions) func([][]byte) {
 				break
 			}
 		}
+		if err != nil {
+			log.Println("Error sending logs to Firetail API: ", err)
+		}
 	}
 }

middlewares/http/middleware.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 	"io"
 	"io/ioutil"
+	"net"
 	"net/http"
 	"net/http/httptest"
 	"strings"
@@ -53,14 +54,18 @@ func GetMiddleware(options *Options) (func(next http.Handler) http.Handler, erro
 	middleware := func(next http.Handler) http.Handler {
 		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			// Create a LogEntry populated with everything we know right now
+			ip, _, err := net.SplitHostPort(r.RemoteAddr)
+			if err != nil {
+				ip = strings.TrimSuffix(strings.TrimPrefix(r.RemoteAddr[:strings.LastIndex(r.RemoteAddr, ":")], "["), "]")
+			}
 			logEntry := logging.LogEntry{
 				Version:     logging.The100Alpha,
 				DateCreated: time.Now().UnixMilli(),
 				Request: logging.Request{
 					HTTPProtocol: logging.HTTPProtocol(r.Proto),
 					Headers:      r.Header,
 					Method:       logging.Method(r.Method),
-					IP:           strings.Split(r.RemoteAddr, ":")[0],
+					IP:           ip,
 					Resource:     r.URL.Path, // We'll fill this in later if we have a router
 				},
 			}