Akka Http如何实现UUID校验与Future处理的自定义指令

你原有代码的编译错误来自path指令的使用规则：path(Segment) { ... }结构的大括号内要求返回Route类型，但你返回的provide(result)属于Directive1[UUID]类型，二者类型不匹配。通过flatMap做指令转换即可解决该问题。

该指令自动提取路径段、校验UUID合法性，非法直接返回400 Bad Request：

import akka.http.scaladsl.server.Directive1
import akka.http.scaladsl.server.Directives._
import akka.http.scaladsl.model.StatusCodes
import java.util.UUID
import scala.util.Try

def extractUUID: Directive1[UUID] = path(Segment).flatMap { maybeUuid =>
  Try(UUID.fromString(maybeUuid)) match {
    case Success(validUuid) => provide(validUuid)
    case Failure(_) => complete(StatusCodes.BadRequest, "UUID格式非法")
  }
}

注意你参考的示例中使用Await.result是错误写法，会阻塞Akka HTTP调度线程、严重降低服务吞吐量，以下是基于原生非阻塞API的实现，异常自动返回500 Internal Server Error：

import scala.concurrent.Future
import scala.util.{Success, Failure}

def futureSuccess[T](future: => Future[T]): Directive1[T] = {
  onComplete(future).flatMap {
    case Success(value) => provide(value)
    case Failure(_) => complete(StatusCodes.InternalServerError, "服务内部错误")
  }
}

这里使用传名参数=> Future[T]保证Future的惰性执行，符合Akka HTTP指令的执行特性。

路由写法可完全符合预期，且不同CRUD接口可完全复用上述两个指令：

// 创建商品接口
private val createProduct = post {
  path("product" / extractUUID) { validUUID =>
    entity(as[ProductData]) { productData =>
      futureSuccess(addProduct(validUUID, productData)) { _ =>
        complete(StatusCodes.OK)
      }
    }
  }
}

// 查询商品接口示例（直接复用指令）
private val getProduct = get {
  path("product" / extractUUID) { validUUID =>
    futureSuccess(getProduct(validUUID)) { product =>
      complete(StatusCodes.OK, product)
    }
  }
}

内容的提问来源于stack exchange，提问作者M.G.