io.finch.benchmarks.service.FinchUserService.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finch-benchmarks_2.10 Show documentation

benchmarks

The newest version!

package io.finch.benchmarks.service

import com.twitter.finagle.http.{Request, Response}
import com.twitter.finagle.Service
import io.finch._
import io.finch.request._
import io.finch.response.EncodeResponse

class FinchUserService(implicit
  userDecoder: DecodeRequest[User],
  newUserInfoDecoder: DecodeRequest[NewUserInfo],
  userEncoder: EncodeResponse[User],
  usersEncoder: EncodeResponse[List[User]],
  encodeErr: EncodeResponse[Map[String, String]]
) extends UserService {

  val getUser: Endpoint[User] = get("users" / long) { id: Long =>
    db.get(id).map {
      case Some(user) => Ok(user)
      case None => BadRequest("error" -> s"No user $id")
    }
  }

  val getUsers: Endpoint[List[User]] = get("users") {
    Ok(db.all)
  }

  val postUser: Endpoint[Unit] = post("users" ? body.as[NewUserInfo]) { u: NewUserInfo =>
    db.add(u.name, u.age).map(id => Created.withHeader("Location" -> s"/users/$id"))
  }

  val deleteUser: Endpoint[String] = delete("users") {
    db.delete().map(count => Ok(s"$count users deleted"))
  }

  val putUser: Endpoint[Unit] = put("users" ? body.as[User]) { u: User =>
    db.update(u).map(_ => NoContent())
  } handle {
    case UserNotFound(id) => BadRequest("error" -> s"No user $id")
  }

  val backend: Service[Request, Response] = (
    getUsers :+: getUser :+: postUser :+: deleteUser :+: putUser
  ).handle({
    case _ => InternalServerError()
  }).toService
}