Examples with ServerResponse io.helidon.webserver.ServerResponse used on opensource projects

Example 21 with ServerResponse

use of io.helidon.webserver.ServerResponse in project helidon by oracle.

the class AccessLogSupportTest method testCustomFormat.

@Test
void testCustomFormat() {
    AccessLogSupport accessLog = AccessLogSupport.builder().add(TimestampLogEntry.builder().formatter(DateTimeFormatter.ofPattern("yyyyMMddHHmmss")).build()).add(HeaderLogEntry.create("Referer")).build();
    ServerRequest request = mock(ServerRequest.class);
    Context context = Context.create();
    when(request.remoteAddress()).thenReturn(REMOTE_IP);
    when(request.context()).thenReturn(context);
    when(request.method()).thenReturn(Http.Method.PUT);
    HttpRequest.Path path = mock(HttpRequest.Path.class);
    when(path.toRawString()).thenReturn(PATH);
    when(request.path()).thenReturn(path);
    when(request.version()).thenReturn(Http.Version.V1_1);
    RequestHeaders headers = mock(RequestHeaders.class);
    when(headers.all("Referer")).thenReturn(Arrays.asList("first", "second"));
    when(request.headers()).thenReturn(headers);
    ServerResponse response = mock(ServerResponse.class);
    when(response.status()).thenReturn(Http.Status.I_AM_A_TEAPOT);
    String logRecord = accessLog.createLogRecord(request, response, BEGIN_TIME, 0L, END_TIME, TIME_TAKEN_MICROS * 1000);
    String expected = "20071203101530 \"first,second\"";
    assertThat(logRecord, is(expected));
}

Example 22 with ServerResponse

use of io.helidon.webserver.ServerResponse in project helidon by oracle.

the class AccessLogSupportTest method testCommonFormat.

@Test
void testCommonFormat() {
    AccessLogSupport accessLog = AccessLogSupport.builder().commonLogFormat().build();
    ServerRequest request = mock(ServerRequest.class);
    Context context = Context.create();
    when(request.remoteAddress()).thenReturn(REMOTE_IP);
    when(request.context()).thenReturn(context);
    when(request.method()).thenReturn(Http.Method.PUT);
    HttpRequest.Path path = mock(HttpRequest.Path.class);
    when(path.toRawString()).thenReturn(PATH);
    when(request.path()).thenReturn(path);
    when(request.version()).thenReturn(Http.Version.V1_1);
    ServerResponse response = mock(ServerResponse.class);
    when(response.status()).thenReturn(Http.Status.I_AM_A_TEAPOT);
    AccessLogContext accessLogContext = mock(AccessLogContext.class);
    when(accessLogContext.requestDateTime()).thenReturn(BEGIN_TIME);
    String expectedTimestamp = TimestampLogEntry.create().doApply(accessLogContext);
    String logRecord = accessLog.createLogRecord(request, response, BEGIN_TIME, 0L, END_TIME, TIME_TAKEN_MICROS * 1000);
    // 192.168.0.104 - [18/Jun/2019:23:10:44 +0200] "GET /greet/test HTTP/1.1" 200 55 2248
    String expected = REMOTE_IP + " - - " + expectedTimestamp + " \"" + METHOD + " " + PATH + " " + HTTP_VERSION + "\" " + STATUS_CODE + " " + CONTENT_LENGTH;
    assertThat(logRecord, is(expected));
}

Example 23 with ServerResponse

use of io.helidon.webserver.ServerResponse in project helidon by oracle.

the class AccessLogSupport method handle.

private void handle(ServerRequest req, ServerResponse res) {
    // Check if this path should be excluded from access log
    if (excludePaths.size() > 0) {
        for (PathMatcher matcher : excludePaths) {
            PathMatcher.Result r = matcher.match(req.path().toString());
            if (r.matches()) {
                req.next();
                return;
            }
        }
    }
    ZonedDateTime now = ZonedDateTime.now(clock);
    long nanoNow = System.nanoTime();
    logFormat.forEach(entry -> entry.accept(req, res));
    res.whenSent().thenAccept(aResponse -> log(req, aResponse, now, nanoNow)).exceptionally(throwable -> {
        log(req, res, now, nanoNow);
        return null;
    });
    req.next();
}

Example 24 with ServerResponse

use of io.helidon.webserver.ServerResponse in project helidon by oracle.

the class VerifyService method getPokemonById.

// Get Pokemon by ID and return its data.
private void getPokemonById(final ServerRequest request, final ServerResponse response) {
    try {
        final String idStr = AbstractService.param(request, QUERY_ID_PARAM);
        final int id = Integer.parseInt(idStr);
        final JsonObjectBuilder pokemonBuilder = Json.createObjectBuilder();
        dbClient.execute(exec -> exec.namedGet("get-pokemon-by-id", id)).thenAccept(data -> data.ifPresentOrElse(row -> {
            final JsonArrayBuilder typesBuilder = Json.createArrayBuilder();
            pokemonBuilder.add("name", row.column("name").as(String.class));
            pokemonBuilder.add("id", row.column("id").as(Integer.class));
            dbClient.execute(exec -> exec.namedQuery("get-pokemon-types", id)).forEach(typeRow -> typesBuilder.add(typeRow.as(JsonObject.class))).onComplete(() -> {
                pokemonBuilder.add("types", typesBuilder.build());
                response.send(AppResponse.okStatus(pokemonBuilder.build()));
            }).exceptionally(t -> {
                response.send(exceptionStatus(t));
                return null;
            });
        }, () -> response.send(AppResponse.okStatus(JsonObject.EMPTY_JSON_OBJECT)))).exceptionally(t -> {
            response.send(exceptionStatus(t));
            return null;
        });
    } catch (RemoteTestException ex) {
        response.send(exceptionStatus(ex));
    }
}

Example 25 with ServerResponse

use of io.helidon.webserver.ServerResponse in project helidon by oracle.

the class MapperService method executeUpdateTest.

private void executeUpdateTest(final ServerRequest request, final ServerResponse response, final String testName, final TestDMLFunction test) {
    LOGGER.fine(() -> String.format("Running Mapper.%s on server", testName));
    try {
        String name = param(request, QUERY_NAME_PARAM);
        String idStr = param(request, QUERY_ID_PARAM);
        int id = Integer.parseInt(idStr);
        Pokemon srcPokemon = Pokemon.POKEMONS.get(id);
        Pokemon updatedPokemon = new Pokemon(id, name, srcPokemon.getTypesArray());
        test.apply(updatedPokemon).thenAccept(result -> response.send(AppResponse.okStatus(Json.createValue(result)))).exceptionally(t -> {
            response.send(exceptionStatus(t));
            return null;
        });
    } catch (RemoteTestException | NumberFormatException ex) {
        LOGGER.fine(() -> String.format("Error in Mapper.%s on server", testName));
        response.send(exceptionStatus(ex));
    }
}