-
Notifications
You must be signed in to change notification settings - Fork 43
/
CustomerVerticle.java
116 lines (104 loc) · 4.15 KB
/
CustomerVerticle.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
110
111
112
113
114
115
116
package com.zanclus.api;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.zanclus.data.access.CustomerDAO;
import com.zanclus.data.entities.Customer;
import io.vertx.core.AbstractVerticle;
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.ext.web.handler.BodyHandler;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;
/**
* A {@link io.vertx.core.Verticle} which handles requests for the REST endpoints
*/
@Component
@Slf4j
public class CustomerVerticle {
@Autowired
private CustomerDAO dao;
@Autowired
private ObjectMapper mapper;
@Autowired
private Vertx vertx;
/**
* Set up the Vert.x Web routes and start the HTTP server
* @throws Exception
*/
@PostConstruct
public void start() throws Exception {
Router router = Router.router(vertx);
router.route().handler(BodyHandler.create());
router.get("/v1/customer/:id")
.produces("application/json")
.blockingHandler(this::getCustomerById);
router.put("/v1/customer")
.consumes("application/json")
.produces("application/json")
.blockingHandler(this::addCustomer);
router.get("/v1/customer")
.produces("application/json")
.blockingHandler(this::getAllCustomers);
vertx.createHttpServer().requestHandler(router::accept).listen(8080);
}
/**
* Add a new customer to the database
* @param rc The {@link RoutingContext} for the request
*/
private void addCustomer(RoutingContext rc) {
try {
String body = rc.getBodyAsString();
Customer customer = mapper.readValue(body, Customer.class);
Customer saved = dao.save(customer);
if (saved!=null) {
rc.response().setStatusMessage("Accepted").setStatusCode(202).end(mapper.writeValueAsString(saved));
} else {
rc.response().setStatusMessage("Bad Request").setStatusCode(400).end("Bad Request");
}
} catch (IOException e) {
rc.response().setStatusMessage("Server Error").setStatusCode(500).end("Server Error");
log.error("Server error", e);
}
}
/**
* Get a {@link Customer} from the database as indicated by the {@code id}
* @param rc The {@link RoutingContext} for the request
*/
private void getCustomerById(RoutingContext rc) {
log.info("Request for single customer");
Long id = Long.parseLong(rc.request().getParam("id"));
try {
Customer customer = dao.findOne(id);
if (customer==null) {
rc.response().setStatusMessage("Not Found").setStatusCode(404).end("Not Found");
} else {
rc.response().setStatusMessage("OK").setStatusCode(200).end(mapper.writeValueAsString(customer));
}
} catch (JsonProcessingException jpe) {
rc.response().setStatusMessage("Server Error").setStatusCode(500).end("Server Error");
log.error("Server error", jpe);
}
}
/**
* Get a {@link List} of all {@link Customer} records in the database
* @param rc The {@link RoutingContext} for the request
*/
private void getAllCustomers(RoutingContext rc) {
log.info("Request for all customers");
List<Customer> customers = StreamSupport.stream(dao.findAll().spliterator(), false).collect(Collectors.toList());
try {
rc.response().setStatusMessage("OK").setStatusCode(200).end(mapper.writeValueAsString(customers));
} catch (JsonProcessingException jpe) {
rc.response().setStatusMessage("Server Error").setStatusCode(500).end("Server Error");
log.error("Server error", jpe);
}
}
}