-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
MovieRatingService.java
109 lines (95 loc) · 3.95 KB
/
MovieRatingService.java
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
package io.vertx.example.virtualthreads;
import io.vertx.core.AbstractVerticle;
import io.vertx.core.DeploymentOptions;
import io.vertx.core.ThreadingModel;
import io.vertx.core.Vertx;
import io.vertx.core.json.JsonObject;
import io.vertx.ext.web.Router;
import io.vertx.ext.web.RoutingContext;
import io.vertx.jdbcclient.JDBCPool;
import io.vertx.sqlclient.*;
import java.util.Arrays;
import java.util.List;
import static io.vertx.core.Future.await;
public class MovieRatingService extends AbstractVerticle {
public static void main(String[] args) throws Exception {
var vertx = Vertx.vertx();
vertx.deployVerticle(MovieRatingService.class, new DeploymentOptions().setThreadingModel(ThreadingModel.VIRTUAL_THREAD))
.toCompletionStage()
.toCompletableFuture()
.get();
}
private static final List<String> DB_STATEMENTS = Arrays.asList(
"CREATE TABLE MOVIE (ID VARCHAR(16) PRIMARY KEY, TITLE VARCHAR(256) NOT NULL)",
"CREATE TABLE RATING (ID INTEGER IDENTITY PRIMARY KEY, value INTEGER, MOVIE_ID VARCHAR(16))",
"INSERT INTO MOVIE (ID, TITLE) VALUES 'starwars', 'Star Wars'",
"INSERT INTO MOVIE (ID, TITLE) VALUES 'indianajones', 'Indiana Jones'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 1, 'starwars'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 5, 'starwars'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 9, 'starwars'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 10, 'starwars'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 4, 'indianajones'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 7, 'indianajones'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 3, 'indianajones'",
"INSERT INTO RATING (VALUE, MOVIE_ID) VALUES 9, 'indianajones'"
);
private SqlClient client;
@Override
public void start() {
client = JDBCPool.pool(vertx, new JsonObject()
.put("url", "jdbc:hsqldb:mem:test?shutdown=true")
.put("driver_class", "org.hsqldb.jdbcDriver")
.put("max_pool_size-loop", 30)
);
DB_STATEMENTS.forEach(statement -> {
await(client.query(statement).execute());
});
// Build Vert.x Web router
var router = Router.router(vertx);
router.get("/movie/:id").handler(this::getMovie);
router.get("/rateMovie/:id").handler(this::rateMovie);
router.get("/getRating/:id").handler(this::getRating);
// Start the server
await(vertx.createHttpServer()
.requestHandler(router)
.listen(config().getInteger("http.port", 8080)));
}
// Send info about a movie
private void getMovie(RoutingContext ctx) {
var id = ctx.pathParam("id");
var rows = await(client.preparedQuery("SELECT TITLE FROM MOVIE WHERE ID=?").execute(Tuple.of(id)));
if (rows.size() == 1) {
ctx.response()
.putHeader("Content-Type", "application/json")
.end(new JsonObject()
.put("id", id)
.put("title", rows.iterator().next().getString("TITLE"))
.encode());
} else {
ctx.response().setStatusCode(404).end();
}
}
// Rate a movie
private void rateMovie(RoutingContext ctx) {
var movie = ctx.pathParam("id");
var rating = Integer.parseInt(ctx.queryParam("rating").get(0));
RowSet<Row> rows = await(client.preparedQuery("SELECT TITLE FROM MOVIE WHERE ID=?").execute(Tuple.of(movie)));
if (rows.size() == 1) {
await(client.preparedQuery("INSERT INTO RATING (VALUE, MOVIE_ID) VALUES ?, ?").execute(Tuple.of(rating, movie)));
ctx.response().setStatusCode(200).end();
} else {
ctx.response().setStatusCode(404).end();
}
}
// Get the current rating of a movie
private void getRating(RoutingContext ctx) {
var id = ctx.pathParam("id");
var rows = await(client.preparedQuery("SELECT AVG(VALUE) AS VALUE FROM RATING WHERE MOVIE_ID=?").execute(Tuple.of(id)));
ctx.response()
.putHeader("Content-Type", "application/json")
.end(new JsonObject()
.put("id", id)
.put("title", rows.iterator().next().getDouble("VALUE"))
.encode());
}
}