From 249adccd5a56bdb96efbf7c5d92241539d6b0cfb Mon Sep 17 00:00:00 2001
From: "daniel.naude" <danieln@bob.co.za>
Date: Tue, 2 Jan 2024 08:54:30 +0200
Subject: [PATCH] Set Headers

---
 handler_utils/debug.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/handler_utils/debug.go b/handler_utils/debug.go
index 06817f6..ba930cb 100644
--- a/handler_utils/debug.go
+++ b/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))
 }
-- 
GitLab