Set Headers

handler_utils/debug.go

@@ -5,13 +5,14 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"github.com/aws/aws-lambda-go/events"
-	"github.com/aws/aws-lambda-go/lambda"
-	"gitlab.bob.co.za/bob-public-utils/bobgroup-go-utils/utils"
 	"io"
 	"net/http"
 	"os"
 	"strings"
+
+	"github.com/aws/aws-lambda-go/events"
+	"github.com/aws/aws-lambda-go/lambda"
+	"gitlab.bob.co.za/bob-public-utils/bobgroup-go-utils/utils"
 )
 
 // ========================================================================
@@ -67,7 +68,10 @@ func ServeHTTPFunctions(ctx context.Context, lambdaHandler lambda.Handler, w htt
 		return
 	}
 
-	w.Header().Set("Content-Type", "application/json")
+	for key, value := range response["headers"].(map[string]interface{}) {
+		w.Header().Set(key, value.(string))
+	}
+
 	w.WriteHeader(int(response["statusCode"].(float64)))
 	_, _ = io.WriteString(w, response["body"].(string))
 }