Java Code Examples for org.springframework.web.context.request.NativeWebRequest#setAttribute()

@SuppressWarnings("unchecked")
@Override
public Object resolveArgument(MethodParameter parameter,
		ModelAndViewContainer mavContainer, NativeWebRequest webRequest,
		WebDataBinderFactory binderFactory) throws Exception {
	Assert.notNull(parameter, "Parameter must not be null");
	if (webRequest.getAttribute(NOTIFICATION_REQUEST_ATTRIBUTE_NAME,
			RequestAttributes.SCOPE_REQUEST) == null) {
		webRequest.setAttribute(NOTIFICATION_REQUEST_ATTRIBUTE_NAME,
				this.messageConverter.read(JsonNode.class,
						createInputMessage(webRequest)),
				RequestAttributes.SCOPE_REQUEST);
	}

	JsonNode content = (JsonNode) webRequest.getAttribute(
			NOTIFICATION_REQUEST_ATTRIBUTE_NAME, RequestAttributes.SCOPE_REQUEST);
	return doResolveArgumentFromNotificationMessage(content,
			createInputMessage(webRequest), parameter.getParameterType());
}
 

@Override
@SuppressWarnings("unchecked")
protected void handleResolvedValue(@Nullable Object arg, String name, MethodParameter parameter,
		@Nullable ModelAndViewContainer mavContainer, NativeWebRequest request) {

	String key = View.PATH_VARIABLES;
	int scope = RequestAttributes.SCOPE_REQUEST;
	Map<String, Object> pathVars = (Map<String, Object>) request.getAttribute(key, scope);
	if (pathVars == null) {
		pathVars = new HashMap<>();
		request.setAttribute(key, pathVars, scope);
	}
	pathVars.put(name, arg);
}
 

@Override
@SuppressWarnings("unchecked")
protected void handleResolvedValue(@Nullable Object arg, String name, MethodParameter parameter,
		@Nullable ModelAndViewContainer mavContainer, NativeWebRequest request) {

	String key = View.PATH_VARIABLES;
	int scope = RequestAttributes.SCOPE_REQUEST;
	Map<String, Object> pathVars = (Map<String, Object>) request.getAttribute(key, scope);
	if (pathVars == null) {
		pathVars = new HashMap<>();
		request.setAttribute(key, pathVars, scope);
	}
	pathVars.put(name, arg);
}
 

private String getRequestBody(NativeWebRequest webRequest) {
    HttpServletRequest servletRequest = webRequest.getNativeRequest(HttpServletRequest.class);

    String jsonBody = (String) webRequest.getAttribute(JSON_BODY_ATTRIBUTE, NativeWebRequest.SCOPE_REQUEST);
    if (jsonBody == null) {
        try {
            jsonBody = IOUtils.toString(servletRequest.getInputStream(), "UTF-8");
            webRequest.setAttribute(JSON_BODY_ATTRIBUTE, jsonBody, NativeWebRequest.SCOPE_REQUEST);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
    return jsonBody;
}
 

@Override
@SuppressWarnings("unchecked")
protected void handleResolvedValue(Object arg, String name, MethodParameter parameter,
		ModelAndViewContainer mavContainer, NativeWebRequest request) {

	String key = View.PATH_VARIABLES;
	int scope = RequestAttributes.SCOPE_REQUEST;
	Map<String, Object> pathVars = (Map<String, Object>) request.getAttribute(key, scope);
	if (pathVars == null) {
		pathVars = new HashMap<String, Object>();
		request.setAttribute(key, pathVars, scope);
	}
	pathVars.put(name, arg);
}
 

@Override
@SuppressWarnings("unchecked")
protected void handleResolvedValue(Object arg, String name, MethodParameter parameter,
		ModelAndViewContainer mavContainer, NativeWebRequest request) {

	String key = View.PATH_VARIABLES;
	int scope = RequestAttributes.SCOPE_REQUEST;
	Map<String, Object> pathVars = (Map<String, Object>) request.getAttribute(key, scope);
	if (pathVars == null) {
		pathVars = new HashMap<String, Object>();
		request.setAttribute(key, pathVars, scope);
	}
	pathVars.put(name, arg);
}
 

public WebRequestQueryContext(NativeWebRequest request) {
	this.contextMap = (HashMap<String, Object>) request.getAttribute(ATTRIBUTE_KEY, NativeWebRequest.SCOPE_REQUEST);
	if (this.contextMap == null) {
		this.contextMap = new HashMap<>();
		request.setAttribute(ATTRIBUTE_KEY, contextMap, NativeWebRequest.SCOPE_REQUEST);
	}
}
 

default ResponseEntity<Problem> create(final Throwable throwable, final Problem problem,
        final NativeWebRequest request, final HttpHeaders headers) {

    final HttpStatus status = HttpStatus.valueOf(Optional.ofNullable(problem.getStatus())
            .orElse(Status.INTERNAL_SERVER_ERROR)
            .getStatusCode());

    log(throwable, problem, request, status);

    if (status == HttpStatus.INTERNAL_SERVER_ERROR) {
        request.setAttribute(ERROR_EXCEPTION, throwable, SCOPE_REQUEST);
    }

    return process(negotiate(request).map(contentType ->
            ResponseEntity
                    .status(status)
                    .headers(headers)
                    .contentType(contentType)
                    .body(problem))
            .orElseGet(throwingSupplier(() -> {
                final ResponseEntity<Problem> fallback = fallback(throwable, problem, request, headers);

                if (fallback.getBody() == null) {
                    /*
                     * Ugly hack to workaround an issue with Tomcat and Spring as described in
                     * https://github.com/zalando/problem-spring-web/issues/84.
                     *
                     * The default fallback in case content negotiation failed is a 406 Not Acceptable without
                     * a body. Tomcat will then display its error page since no body was written and the response
                     * was not committed. In order to force Spring to flush/commit one would need to provide a
                     * body but that in turn would fail because Spring would then fail to negotiate the correct
                     * content type.
                     *
                     * Writing the status code, headers and flushing the body manually is a dirty way to bypass
                     * both parties, Tomcat and Spring, at the same time.
                     */
                    final ServerHttpResponse response = new ServletServerHttpResponse(
                            request.getNativeResponse(HttpServletResponse.class));

                    response.setStatusCode(fallback.getStatusCode());
                    response.getHeaders().putAll(fallback.getHeaders());
                    response.getBody(); // just so we're actually flushing the body...
                    response.flush();
                }

                return fallback;
            })), request);
}