123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165 |
- import org.vertx.java.core.Handler;
- import org.vertx.java.core.buffer.Buffer;
- import org.vertx.java.core.eventbus.Message;
- import org.vertx.java.core.http.HttpServerRequest;
- import org.vertx.java.core.http.HttpServerResponse;
- import org.vertx.java.core.json.JsonArray;
- import org.vertx.java.core.json.JsonObject;
- import org.vertx.java.core.json.impl.Json;
- import org.vertx.java.platform.Verticle;
- import java.text.DateFormat;
- import java.text.SimpleDateFormat;
- import java.util.Collections;
- import java.util.Date;
- import java.util.Random;
- import java.util.concurrent.ThreadLocalRandom;
- public class WebServer extends Verticle implements Handler<HttpServerRequest> {
- private Buffer helloWorldBuffer = new Buffer("Hello, World!");
- private String helloWorldContentLength = String.valueOf(helloWorldBuffer.length());
- private DateFormat DATE_FORMAT = new SimpleDateFormat("EEE, dd MMM yyyyy HH:mm:ss z");
- private String dateString;
- @Override
- public void start() {
- vertx.createHttpServer().requestHandler(WebServer.this).listen(8080);
- vertx.setPeriodic(1000, new Handler<Long>() {
- @Override
- public void handle(Long timerID) {
- formatDate();
- }
- });
- formatDate();
- }
- @Override
- public void handle(HttpServerRequest req) {
- String path = req.path();
- switch (path) {
- case "/plaintext":
- handlePlainText(req);
- break;
- case "/json":
- handleJson(req);
- break;
- case "/db":
- handleDbMongo(req);
- break;
- case "/queries":
- handleQueriesMongo(req);
- break;
- default:
- req.response().setStatusCode(404);
- req.response().end();
- }
- }
- private void formatDate() {
- dateString = DATE_FORMAT.format(new Date());
- }
- private void handlePlainText(HttpServerRequest req) {
- HttpServerResponse resp = req.response();
- setHeaders(resp, "text/plain", helloWorldContentLength);
- resp.end(helloWorldBuffer);
- }
- private void handleJson(HttpServerRequest req) {
- Buffer buff = new Buffer(Json.encode(Collections.singletonMap("message", "Hello, world!")));
- HttpServerResponse resp = req.response();
- setHeaders(resp, "application/json", String.valueOf(buff.length()));
- resp.end(buff);
- }
- private void handleDbMongo(final HttpServerRequest req) {
- findRandom(ThreadLocalRandom.current(), new Handler<Message<JsonObject>>() {
- @Override
- public void handle(Message<JsonObject> reply) {
- JsonObject world = getResultFromReply(reply);
- String result = world.encode();
- sendResponse(req, result);
- }
- });
- }
- private JsonObject getResultFromReply(Message<JsonObject> reply) {
- JsonObject body = reply.body();
- JsonObject world = body.getObject("result");
- Object id = world.removeField("_id");
- world.putValue("id", id);
- return world;
- }
- private void handleQueriesMongo(final HttpServerRequest req) {
- int queriesParam = 1;
- try {
- queriesParam = Integer.parseInt(req.params().get("queries"));
- } catch (NumberFormatException e) {
- e.printStackTrace();
- }
- if (queriesParam < 1)
- {
- queriesParam = 1;
- }
- if (queriesParam > 500)
- {
- queriesParam = 500;
- }
- final MongoHandler dbh = new MongoHandler(req, queriesParam);
- final Random random = ThreadLocalRandom.current();
- for (int i = 0; i < queriesParam; i++) {
- findRandom(random, dbh);
- }
- }
- private void findRandom(Random random, Handler<Message<JsonObject>> handler) {
- vertx.eventBus().send(
- "hello.persistor",
- new JsonObject()
- .putString("action", "findone")
- .putString("collection", "World")
- .putObject("matcher", new JsonObject().putNumber("_id", (random.nextInt(10000) + 1))),
- handler);
- }
- private void sendResponse(HttpServerRequest req, String result) {
- Buffer buff = new Buffer(result);
- HttpServerResponse resp = req.response();
- setHeaders(resp, "application/json", String.valueOf(buff.length()));
- resp.end(buff);
- }
- private void setHeaders(HttpServerResponse resp, String contentType, String contentLength) {
- resp.putHeader("Content-Type", contentType);
- resp.putHeader("Content-Length", contentLength);
- resp.putHeader("Server", "vert.x");
- resp.putHeader("Date", dateString);
- }
- private class MongoHandler implements Handler<Message<JsonObject>> {
- private final HttpServerRequest req;
- private final int queries;
- private final JsonArray worlds;
- public MongoHandler(HttpServerRequest request, int queriesParam) {
- req = request;
- queries = queriesParam;
- worlds = new JsonArray();
- }
- @Override
- public void handle(Message<JsonObject> reply) {
- JsonObject world = getResultFromReply(reply);
- worlds.add(world);
- if (worlds.size() == this.queries) {
- // All queries have completed; send the response.
- String result = worlds.encode();
- sendResponse(req, result);
- }
- }
- }
- }
|