如何捕获Java编写的AWS Lambda函数中主线程抛出的OutOfMemoryError？

这个问题得分两种场景来说，我给你掰扯清楚：

1. 你自己代码里的main线程抛出OOM #

如果是你在Lambda代码里自己写的main方法（比如有些特殊场景下你手动调用了main逻辑），那完全可以捕获。毕竟这部分线程是你自己代码能掌控的：

• 最简单的方式就是在main方法里用try-catch包裹可能爆内存的代码，注意要直接抓OutOfMemoryError或者更上层的Throwable——因为OOM属于Error，不属于Exception体系，抓Exception是抓不到的：

public class MyLambdaHandler implements RequestHandler<Object, Object> {
    public static void main(String[] args) {
        try {
            // 举个例子：创建远超Lambda内存配额的大数组
            byte[] hugeArray = new byte[1024 * 1024 * 1024 * 5]; // 5GB，一般Lambda内存没这么大
        } catch (OutOfMemoryError oom) {
            // 这里可以打日志、做一些极简的资源清理
            System.err.println("逮到OOM了：" + oom.getMessage());
            // 注意：这时候别再尝试分配内存了，大概率会直接再爆一次
        }
    }

    @Override
    public Object handleRequest(Object input, Context context) {
        // Lambda handler里调用自己的main方法
        main(new String[]{});
        return "处理完成";
    }
}

• 也可以给这个main线程设置全局的未捕获异常处理器，哪怕代码里没写try-catch，也能接住OOM：

public static void main(String[] args) {
    Thread.currentThread().setUncaughtExceptionHandler((thread, throwable) -> {
        if (throwable instanceof OutOfMemoryError) {
            System.err.println("main线程里出现未捕获的OOM：" + throwable.getMessage());
            // 这里只能做些最基础的日志，别搞复杂操作，内存已经不够用了
        }
    });
    // 触发OOM的代码逻辑
}

2. Lambda运行时的主线程抛出OOM #

如果是AWS Lambda运行环境自己的主线程（就是负责加载你的handler、调度执行任务的那个系统线程）爆了OOM，那你完全没办法直接捕获。这个线程是AWS管的，你的代码碰不到它的异常处理逻辑，也没法包裹它的执行流程。

这种情况下，OOM一发生，Lambda会直接终止你的函数实例，然后把错误信息写到CloudWatch日志里（比如java.lang.OutOfMemoryError: Java heap space）。你能做的就是去CloudWatch监控这类错误，要么调高Lambda的内存配额（Lambda的内存和CPU是绑定的，内存越大堆空间也越大），要么优化你的代码减少内存占用——比如别一次性加载大量数据到内存，及时释放无用对象之类的。

最后提个醒 #

• 捕获到OOM之后，千万别再尝试分配新内存了，JVM堆已经炸了，再分配只会再次触发OOM。
• Lambda的内存配额默认是128MB，最高能调到10GB，根据你的业务需求合理调整，别浪费也别抠门。
• 最好从根源上避免OOM：用流式处理代替批量加载、复用对象池、清理无用引用，比事后捕获靠谱多了。

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