Merge pull request #37 from FireTail-io/buffer-channels

Buffer channels

Author: TheTeaCat

main.go

@@ -25,6 +25,10 @@ func main() {
 		log.Println("Firetail extension starting in debug mode.")
 	}
 
+	// Log the value of AWS_LAMBDA_EXEC_WRAPPER and AWS_LAMBDA_RUNTIME_API for debugging
+	log.Println("AWS_LAMBDA_EXEC_WRAPPER:", os.Getenv("AWS_LAMBDA_EXEC_WRAPPER"))
+	log.Println("AWS_LAMBDA_RUNTIME_API:", os.Getenv("AWS_LAMBDA_RUNTIME_API"))
+
 	// This context will be cancelled whenever a SIGTERM or SIGINT signal is received
 	// We'll use it for our requests to the extensions API & to shutdown the log server
 	ctx := getContext()

proxy/proxy.go

@@ -36,9 +36,9 @@ func NewProxyServer() (*ProxyServer, error) {
 	ps := &ProxyServer{
 		runtimeEndpoint:       os.Getenv("AWS_LAMBDA_RUNTIME_API"),
 		port:                  port,
-		eventsChannel:         make(chan *http.Response),
-		lambdaResponseChannel: make(chan *http.Request),
-		RecordsChannel:        make(chan firetail.Record),
+		eventsChannel:         make(chan *http.Response, 1),
+		lambdaResponseChannel: make(chan *http.Request, 1),
+		RecordsChannel:        make(chan firetail.Record, 100),
 	}
 
 	r := chi.NewRouter()

proxy/proxy_handler.go

@@ -2,6 +2,7 @@ package proxy
 
 import (
 	"io"
+	"log"
 	"net/http"
 	"net/url"
 	"strings"
@@ -34,6 +35,7 @@ func getProxyHandler(urlMappingFunc func(r *http.Request) (*url.URL, error), req
 
 		// Send the request to the requestChannel with the copied body if the channel was provided
 		if requestChannel != nil {
+			log.Println("Captured lambda response", requestBodyCopy.String())
 			r.Body = io.NopCloser(strings.NewReader(requestBodyCopy.String()))
 			*requestChannel <- r
 		}
@@ -57,6 +59,7 @@ func getProxyHandler(urlMappingFunc func(r *http.Request) (*url.URL, error), req
 
 		// Send the response to the responseChannel with the copied body if the channel was provided
 		if responseChannel != nil {
+			log.Println("Captured event", responseBodyCopy.String())
 			resp.Body = io.NopCloser(strings.NewReader(responseBodyCopy.String()))
 			*responseChannel <- resp
 		}