Merge pull request #58 from shogo82148/support-response-streaming

support AWS Lambda response streaming

examples/function-urls/template.yaml

@@ -10,4 +10,8 @@ Resources:
       Runtime: provided.al2
       Timeout: 3
       FunctionUrlConfig:
-         AuthType: NONE
+        AuthType: NONE
+        InvokeMode: RESPONSE_STREAM
+      Environment:
+        Variables:
+           RIDGENATIVE_INVOKE_MODE: RESPONSE_STREAM

invoke.go

@@ -3,6 +3,7 @@ package ridgenative
 import (
 	"context"
 	"encoding/json"
+	"io"
 	"net/http"
 )
 
@@ -29,3 +30,23 @@ func callBytesHandlerFunc(ctx context.Context, payload []byte, h handlerFunc) (r
 	}
 	return json.Marshal(resp)
 }
+
+func callHandlerFuncSteaming(ctx context.Context, payload []byte, h handlerFuncSteaming) (response io.ReadCloser, contentType string, err error) {
+	defer func() {
+		if v := recover(); v != nil {
+			err = lambdaPanicResponse(v)
+		}
+	}()
+
+	var req *request
+	if err := json.Unmarshal(payload, &req); err != nil {
+		return nil, "", err
+	}
+
+	r, w := io.Pipe()
+	contentType, err = h(ctx, req, w)
+	if err != nil {
+		return nil, "", err
+	}
+	return r, contentType, nil
+}