AWS Lambda集成Scala Cats Effect IO异常问题排查

在AWS Lambda上运行基于Scala语言并使用Cats Effect IO编写的函数时遇到异常：此前普通Scala函数能正常运行，但使用如下IO编写的函数仅第一个任务执行完成后就结束了。

问题代码 #

def execute: IO[Unit] = {
  for {
    task1 <- IO.pure(println("task 1"))
    task2 <- IO.pure(println("task 2"))
    task3 <- IO.pure(println("task 3"))
  } yield ()
}

Lambda日志 #

INIT_START Runtime Version: java:17.v42 Runtime Version ARN:arn:aws:lambda:eu-central-1::runtime:07e1a019fe488fb718021df9129caf01fd13d22d401104a151e929a91a82af7b
START RequestId:b032972f-4c47-4c9d-be22-c83141360bd7 Version: $LATEST
task 1 
END RequestId: b032972f-4c47-4c9d-be22-c83141360bd7 
REPORT RequestId: b032972f-4c47-4c9d-be22-c83141360bd7 Duration: 2211.47 ms Billed Duration: 2212 ms Memory Size: 512 MB Max Memory Used: 119 MB Init Duration: 577.36 ms

起初怀疑是多线程问题，但for推导式的顺序执行逻辑和IO的特性排除了这个可能，需要明确问题根源。

1. IO构造方式错误 #

IO.pure(println("task 1"))的写法存在本质问题：IO.pure用于包裹纯值，它的参数会被立即求值，也就是说println会在IO对象创建时就执行，而非IO被运行时。如果Lambda运行时在IO构造过程中提前终止进程，就会导致后续的println无法执行。

正确的写法是用IO(...)包裹副作用操作，它会把逻辑延迟到IO被主动运行时才执行：

def execute: IO[Unit] = {
  for {
    _ <- IO(println("task 1"))
    _ <- IO(println("task 2"))
    _ <- IO(println("task 3"))
  } yield ()
}

2. 未主动触发IO执行 #

Cats Effect的IO是惰性的，它只是一段执行逻辑的描述，不会自动运行。AWS Lambda运行时无法识别IO类型的返回值，也不会自动帮你运行IO逻辑。必须在handler方法中显式调用unsafeRunSync()（或其他IO运行方法），确保所有IO逻辑执行完毕后再结束函数。

示例handler代码：

import cats.effect.IO
import com.amazonaws.services.lambda.runtime.{Context, RequestHandler}

class LambdaHandler extends RequestHandler[String, Unit] {
  override def handleRequest(input: String, context: Context): Unit = {
    // 主动运行IO，确保所有任务执行完成
    execute.unsafeRunSync()
  }

  def execute: IO[Unit] = {
    for {
      _ <- IO(println("task 1"))
      _ <- IO(println("task 2"))
      _ <- IO(println("task 3"))
    } yield ()
  }
}

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