得益于 Spray marshalling 系统的巧妙设计,有一种非常优雅的方式可以实现这一目标。代码 ( gist ) 说明了这一点:
case class User(name: String, no: Int)
// Json marshaller
object UnMarshalling extends DefaultJsonProtocol {
val jsonUser = jsonFormat2(User)
val textUser = Unmarshaller[User](`text/plain`) {
case HttpEntity.NonEmpty(contentType, data) =>
val res = data.asString.drop(5).dropRight(1).split(',')
User(res(0),res(1).toInt)
}
implicit val userMarshal = Unmarshaller.oneOf(jsonUser, textUser)
}
class UnMarshalTest extends FunSpec with ScalatestRouteTest with Matchers {
import UnMarshalling._
// Marshals response according to the Accept header media type
val putOrder = path("user") {
put {
// Namespace clash with ScalaTestRoutes.entity
MarshallingDirectives.entity(as[User]) {
user =>
complete(s"no=${user.no}")
}
}
}
describe("Our route should") {
val json = """ {"name" : "bender", "no" : 1234} """
it("submit a json") {
Put("/user", HttpEntity(`application/json`,json)) ~> putOrder ~> check {
responseAs[String] should equal("no=1234")
}
}
it("Submit text") {
Put("/user", HttpEntity(`text/plain`,"""User(Zoidberg,322)""")) ~> putOrder ~> check {
responseAs[String] should equal("no=322")
}
}
}
}