能否用CloudFront Functions替代Lambda@Edge改写WebSocket API请求路径？

完全可以用CloudFront Functions替代Lambda@Edge完成这个URL改写操作，而且更适合WebSocket这种对延迟敏感的场景——CloudFront Functions运行在CloudFront全球边缘节点，延迟远低于Lambda@Edge（尤其是需要调用区域Lambda的源请求场景），同时支持WebSocket初始HTTP升级请求的处理，能轻松完成这类轻量的URI改写任务。

具体实现步骤 #

1. 创建CloudFront Functions函数 #

编写一个轻量函数，根据请求URI中的标识片段，将其改写为API网关要求的阶段路径/dev，示例代码如下：

function handler(event) {
    const request = event.request;
    const uri = request.uri;
    
    // 匹配foo-socket路径，替换为API网关阶段路径
    if (uri.startsWith('/dev/foo-socket')) {
        request.uri = '/dev';
    }
    // 匹配bar-socket路径，替换为API网关阶段路径
    else if (uri.startsWith('/dev/bar-socket')) {
        request.uri = '/dev';
    }
    
    return request;
}

如果需要处理带子路径的请求（如/dev/foo-socket/xxx），可根据实际需求调整替换逻辑，WebSocket API网关通常仅需阶段路径/dev，所以直接替换即可。

2. 发布函数到生产环境 #

在AWS控制台完成函数创建后，先通过模拟WebSocket升级请求测试逻辑正确性，再将函数发布到LIVE阶段。

3. 关联函数到CloudFront行为 #

进入CloudFront分发的行为设置，分别为/dev/foo-socket/*和/dev/bar-socket/*两个行为，在Viewer Request阶段关联刚才创建的CloudFront Functions。

关键注意事项 #

• CloudFront Functions仅支持Viewer Request和Viewer Response阶段，这里利用「先匹配行为选择对应源，再通过Viewer Request函数修改转发URI」的逻辑，完全满足需求
• 可调整正则匹配逻辑覆盖更多路径场景（如带/不带尾部斜杠的情况）
• CloudFront Functions执行时间限制为1ms、内存限制2MB，这类轻量字符串替换操作完全符合限制要求

内容的提问来源于stack exchange，提问作者cmgchess