-
Notifications
You must be signed in to change notification settings - Fork 3
/
server.scala
110 lines (103 loc) · 3.64 KB
/
server.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package com.robert42.ft
import Requests._
import java.net.InetSocketAddress
import java.util.{NoSuchElementException => NoSuchElement}
import org.jboss.netty.handler.codec.http.HttpMethod._
import com.twitter.util.Future
import com.twitter.finagle.http.{Http, RichHttp, Request, Response}
import com.twitter.finagle.http.Status._
import com.twitter.finagle.http.Version.Http11
import com.twitter.finagle.http.path._
import com.twitter.finagle.{Service, SimpleFilter}
import com.twitter.finagle.builder.{Server, ServerBuilder}
import net.liftweb.common.Logger
/**
* A rather sophisticated REST server that handles exceptions.
*/
object RestServer extends Logger {
/**
* A simple Filter that catches exceptions and converts them to appropriate
* HTTP responses.
*/
class HandleExceptions extends SimpleFilter[Request, Response] {
def apply(request: Request, service: Service[Request, Response]) = {
// `handle` asynchronously handles exceptions.
service(request) handle { case error =>
val statusCode = error match {
case _: IllegalArgumentException =>
Forbidden
case _ =>
InternalServerError
}
Responses error error.getMessage
}
}
}
/**
* The service itself.
*/
class Respond extends Service[Request, Response] with Logger {
def apply(request: Request) = {
try {
request.method -> Path(request.path) match {
case GET -> Root / "todos" => Future.value {
val data = Todos.allAsJson
debug("data: %s" format data)
Responses.json(data, acceptsGzip(request))
}
case GET -> Root / "todos" / id => Future.value {
val todo = Todos get id
val data = todo.toJson
debug("data: %s" format data)
Responses.json(data, acceptsGzip(request))
}
case POST -> Root / "todos" => Future.value {
val content = request.contentString
val todo = Todos.fromJson(content, create = true)
val data = todo.toJson
Responses.json(data, acceptsGzip(request))
}
case PUT -> Root / "todos" / id => Future.value {
val content = request.contentString
val todo = Todos.fromJson(content, update = true)
val data = todo.toJson
debug("data: %s" format data)
Responses.json(data, acceptsGzip(request))
}
case DELETE -> Root / "todos" / id => Future.value {
Todos remove id
debug("data: %s" format id)
Response()
}
case _ =>
Future value Response(Http11, NotFound)
}
} catch {
case e: NoSuchElement => Future value Response(Http11, NotFound)
case e: Exception => Future.value {
val message = Option(e.getMessage) getOrElse "Something went wrong."
error("\nMessage: %s\nStack trace:\n%s"
.format(message, e.getStackTraceString))
Responses.error(message, acceptsGzip(request))
}
}
}
}
def main(args: Array[String]) {
MongoConfig.init
val handleExceptions = new HandleExceptions
val respond = new Respond
// Compose the Filters and Service together:
val service = handleExceptions andThen respond
val port = Option(System getenv "PORT") match {
case Some(port) => port.toInt
case None => 8888
}
val server = ServerBuilder()
.codec(RichHttp[Request](Http()))
.bindTo(new InetSocketAddress(port))
.name("restserver")
.build(service)
info("Server started on port: %s" format port)
}
}