mirrors/PMTiles
1
0
Fork 0
mirror of https://github.com/protomaps/PMTiles.git synced 2026-02-04 02:41:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor AWS handler with modular handlerRaw

Browse Source
This commit is contained in:
Brandon Liu
2022-12-21 16:30:56 +08:00
parent 80c02e1833
commit 283b9f9d24
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
serverless/aws/src/index.ts
View File

@@ -136,9 +136,10 @@ const apiResp = (
// Assumes event is a API Gateway V2 or Lambda Function URL formatted dict // Assumes event is a API Gateway V2 or Lambda Function URL formatted dict
// and returns API Gateway V2 / Lambda Function dict responses // and returns API Gateway V2 / Lambda Function dict responses
// Does not work with CloudFront events/Lambda@Edge; see README // Does not work with CloudFront events/Lambda@Edge; see README
export const handler = async ( export const handlerRaw = async (
event: APIGatewayProxyEventV2, event: APIGatewayProxyEventV2,
context: Context context: Context,
tilePostprocess?: (a: ArrayBuffer, t: TileType) => ArrayBuffer
): Promise<APIGatewayProxyResult> => { ): Promise<APIGatewayProxyResult> => {
var path; var path;
var is_api_gateway; var is_api_gateway;
@@ -216,10 +217,16 @@ export const handler = async (
break; break;
} }
let data = tile_result.data;
if (tilePostprocess) {
data = tilePostprocess(data, header.tileType);
}
if (is_api_gateway) { if (is_api_gateway) {
// this is wasted work, but we need to force API Gateway to interpret the Lambda response as binary // this is wasted work, but we need to force API Gateway to interpret the Lambda response as binary
// without depending on clients sending matching Accept: headers in the request. // without depending on clients sending matching Accept: headers in the request.
const recompressed_data = zlib.gzipSync(tile_result.data); const recompressed_data = zlib.gzipSync(data);
headers["Content-Encoding"] = "gzip"; headers["Content-Encoding"] = "gzip";
return apiResp( return apiResp(
200, 200,
@@ -231,7 +238,7 @@ export const handler = async (
// returns uncompressed response // returns uncompressed response
return apiResp( return apiResp(
200, 200,
Buffer.from(tile_result.data).toString("base64"), Buffer.from(data).toString("base64"),
true, true,
headers headers
); );
@@ -247,3 +254,10 @@ export const handler = async (
} }
return apiResp(404, "Invalid URL", false, headers); return apiResp(404, "Invalid URL", false, headers);
}; };
export const handler = async (
event: APIGatewayProxyEventV2,
context: Context
): Promise<APIGatewayProxyResult> => {
return handlerRaw(event, context);
};