@@ -9,20 +9,17 @@ import (
99 "net/http"
1010 "net/url"
1111 "strings"
12+ "github.com/aws/aws-lambda-go/events"
1213)
1314
1415func NewALBPayloadBuilder (enableMultiValue bool ) PayloadBuilder {
15- if enableMultiValue {
16- return ALBMultiValuePayloadBuilder {}
17- } else {
18- return ALBPayloadBuilder {}
16+ return ALBPayloadBuilder {
17+ EnableMultiValue : enableMultiValue ,
1918 }
2019}
2120
2221type ALBPayloadBuilder struct {
23- }
24-
25- type ALBMultiValuePayloadBuilder struct {
22+ EnableMultiValue bool
2623}
2724
2825func (pb ALBPayloadBuilder ) BuildRequest (r * http.Request ) ([]byte , error ) {
@@ -36,98 +33,51 @@ func (pb ALBPayloadBuilder) BuildRequest(r *http.Request) ([]byte, error) {
3633 return nil , err
3734 }
3835
39- event := struct {
40- HttpMethod string `json:"httpMethod"`
41- Path string `json:"path"`
42- QueryStringParameters map [string ]string `json:"queryStringParameters"`
43- Headers map [string ]string `json:"headers"`
44- Body string `json:"body"`
45- IsBase64Encoded bool `json:"isBase64Encoded"`
46- }{
47- HttpMethod : r .Method ,
36+ event := events.ALBTargetGroupRequest {
37+ HTTPMethod : r .Method ,
4838 Path : r .URL .Path ,
49- QueryStringParameters : ArrayMapToFirstElementMap (queryParams ),
50- Headers : ArrayMapToFirstElementMap (r .Header ),
5139 Body : body ,
5240 IsBase64Encoded : isBase64Encoded ,
41+ RequestContext : events.ALBTargetGroupRequestContext {
42+ ELB : events.ELBContext {
43+ TargetGroupArn : "arn:aws:elasticloadbalancing:us-east-2:123456789012:targetgroup/lambda-local-proxy-dummy/1234567890123456" ,
44+ },
45+ },
5346 }
5447
55- return json .Marshal (event )
56- }
57-
58- func (pb ALBMultiValuePayloadBuilder ) BuildRequest (r * http.Request ) ([]byte , error ) {
59- queryParams , err := url .ParseQuery (r .URL .RawQuery )
60- if err != nil {
61- return nil , err
62- }
63-
64- body , isBase64Encoded , err := ReadBodyAsString (r )
65- if err != nil {
66- return nil , err
67- }
68-
69- event := struct {
70- HttpMethod string `json:"httpMethod"`
71- Path string `json:"path"`
72- MultiValueQueryStringParameters map [string ][]string `json:"multiValueQueryStringParameters"`
73- MultiValueHeaders map [string ][]string `json:"multiValueHeaders"`
74- Body string `json:"body"`
75- IsBase64Encoded bool `json:"isBase64Encoded"`
76- }{
77- HttpMethod : r .Method ,
78- Path : r .URL .Path ,
79- MultiValueQueryStringParameters : queryParams ,
80- MultiValueHeaders : r .Header ,
81- Body : body ,
82- IsBase64Encoded : isBase64Encoded ,
48+ if pb .EnableMultiValue {
49+ event .MultiValueQueryStringParameters = queryParams
50+ event .MultiValueHeaders = r .Header
51+ } else {
52+ event .QueryStringParameters = ArrayMapToFirstElementMap (queryParams )
53+ event .Headers = ArrayMapToFirstElementMap (r .Header )
8354 }
8455
8556 return json .Marshal (event )
8657}
8758
8859func (pb ALBPayloadBuilder ) BuildResponse (blob []byte ) (int , []byte , map [string ][]string , error ) {
89- resp := struct {
90- StatusCode int `json:"statusCode"`
91- Headers map [string ]string `json:"headers"`
92- Body string `json:"body"`
93- IsBase64Encoded bool `json:"isBase64Encoded"`
94- }{}
60+ resp := events.ALBTargetGroupResponse {}
9561
9662 err := json .Unmarshal (blob , & resp )
9763
64+ var body []byte
9865 if err != nil {
9966 return BuildErrorResponse (err , "Invalid JSON response" )
10067 } else if resp .IsBase64Encoded {
10168 binary , err := base64 .StdEncoding .DecodeString (resp .Body )
10269 if err != nil {
10370 return BuildErrorResponse (err , "Invalid body in JSON response" )
10471 }
105- return resp . StatusCode , binary , MapToArrayMap ( resp . Headers ), nil
72+ body = binary
10673 } else {
107- return resp . StatusCode , []byte (resp .Body ), MapToArrayMap ( resp . Headers ), nil
74+ body = []byte (resp .Body )
10875 }
109- }
110-
111- func (pb ALBMultiValuePayloadBuilder ) BuildResponse (blob []byte ) (int , []byte , map [string ][]string , error ) {
112- resp := struct {
113- StatusCode int `json:"statusCode"`
114- MultiValueHeaders map [string ][]string `json:"multiValueHeaders"`
115- Body string `json:"body"`
116- IsBase64Encoded bool `json:"isBase64Encoded"`
117- }{}
118-
119- err := json .Unmarshal (blob , & resp )
12076
121- if err != nil {
122- return BuildErrorResponse (err , "Invalid JSON response" )
123- } else if resp .IsBase64Encoded {
124- binary , err := base64 .StdEncoding .DecodeString (resp .Body )
125- if err != nil {
126- return BuildErrorResponse (err , "Invalid body in JSON response" )
127- }
128- return resp .StatusCode , binary , resp .MultiValueHeaders , nil
77+ if pb .EnableMultiValue {
78+ return resp .StatusCode , body , resp .MultiValueHeaders , nil
12979 } else {
130- return resp .StatusCode , [] byte ( resp . Body ), resp .MultiValueHeaders , nil
80+ return resp .StatusCode , body , MapToArrayMap ( resp .Headers ) , nil
13181 }
13282}
13383
0 commit comments