Java 类org.springframework.web.servlet.FlashMap 实例源码

项目：spring4-understanding 文件：RedirectView.java

/**
 * Convert model to request parameters and redirect to the given URL.
 * @see #appendQueryProperties
 * @see #sendRedirect
 */
@Override
protected void renderMergedOutputModel(Map<String, Object> model, HttpServletRequest request,
        HttpServletResponse response) throws IOException {

    String targetUrl = createTargetUrl(model, request);
    targetUrl = updateTargetUrl(targetUrl, model, request, response);

    FlashMap flashMap = RequestContextUtils.getOutputFlashMap(request);
    if (!CollectionUtils.isEmpty(flashMap)) {
        UriComponents uriComponents = UriComponentsBuilder.fromUriString(targetUrl).build();
        flashMap.setTargetRequestPath(uriComponents.getPath());
        flashMap.addTargetRequestParams(uriComponents.getQueryParams());
        FlashMapManager flashMapManager = RequestContextUtils.getFlashMapManager(request);
        if (flashMapManager == null) {
            throw new IllegalStateException("FlashMapManager not found despite output FlashMap having been set");
        }
        flashMapManager.saveOutputFlashMap(flashMap, request, response);
    }

    sendRedirect(request, response, targetUrl, this.http10Compatible);
}

项目：spring4-understanding 文件：AbstractFlashMapManager.java

/**
 * Return a FlashMap contained in the given list that matches the request.
 * @return a matching FlashMap or {@code null}
 */
private FlashMap getMatchingFlashMap(List<FlashMap> allMaps, HttpServletRequest request) {
    List<FlashMap> result = new LinkedList<FlashMap>();
    for (FlashMap flashMap : allMaps) {
        if (isFlashMapForRequest(flashMap, request)) {
            result.add(flashMap);
        }
    }
    if (!result.isEmpty()) {
        Collections.sort(result);
        if (logger.isDebugEnabled()) {
            logger.debug("Found matching FlashMap(s): " + result);
        }
        return result.get(0);
    }
    return null;
}

项目：spring4-understanding 文件：AbstractFlashMapManager.java

/**
 * Whether the given FlashMap matches the current request.
 * Uses the expected request path and query parameters saved in the FlashMap.
 */
protected boolean isFlashMapForRequest(FlashMap flashMap, HttpServletRequest request) {
    String expectedPath = flashMap.getTargetRequestPath();
    if (expectedPath != null) {
        String requestUri = getUrlPathHelper().getOriginatingRequestUri(request);
        if (!requestUri.equals(expectedPath) && !requestUri.equals(expectedPath + "/")) {
            return false;
        }
    }
    UriComponents uriComponents = ServletUriComponentsBuilder.fromRequest(request).build();
    MultiValueMap<String, String> actualParams = uriComponents.getQueryParams();
    MultiValueMap<String, String> expectedParams = flashMap.getTargetRequestParams();
    for (String expectedName : expectedParams.keySet()) {
        List<String> actualValues = actualParams.get(expectedName);
        if (actualValues == null) {
            return false;
        }
        for (String expectedValue : expectedParams.get(expectedName)) {
            if (!actualValues.contains(expectedValue)) {
                return false;
            }
        }
    }
    return true;
}

项目：spring4-understanding 文件：RequestMappingHandlerAdapterTests.java

@Test
public void setAlwaysUseRedirectAttributes() throws Exception {
    HandlerMethodArgumentResolver redirectAttributesResolver = new RedirectAttributesMethodArgumentResolver();
    HandlerMethodArgumentResolver modelResolver = new ModelMethodProcessor();
    HandlerMethodReturnValueHandler viewHandler = new ViewNameMethodReturnValueHandler();

    this.handlerAdapter.setArgumentResolvers(Arrays.asList(redirectAttributesResolver, modelResolver));
    this.handlerAdapter.setReturnValueHandlers(Collections.singletonList(viewHandler));
    this.handlerAdapter.setIgnoreDefaultModelOnRedirect(true);
    this.handlerAdapter.afterPropertiesSet();

    this.request.setAttribute(DispatcherServlet.OUTPUT_FLASH_MAP_ATTRIBUTE, new FlashMap());

    HandlerMethod handlerMethod = handlerMethod(new RedirectAttributeController(), "handle", Model.class);
    ModelAndView mav = this.handlerAdapter.handle(request, response, handlerMethod);

    assertTrue("Without RedirectAttributes arg, model should be empty", mav.getModel().isEmpty());
}

项目：spring4-understanding 文件：FlashMapManagerTests.java

@Test
public void retrieveAndUpdateMatchByParams() {
    FlashMap flashMap = new FlashMap();
    flashMap.put("key", "value");
    flashMap.addTargetRequestParam("number", "one");

    this.flashMapManager.setFlashMaps(Arrays.asList(flashMap));

    this.request.setQueryString("number=");
    FlashMap inputFlashMap = this.flashMapManager.retrieveAndUpdate(this.request, this.response);

    assertNull(inputFlashMap);
    assertEquals("FlashMap should not have been removed", 1, this.flashMapManager.getFlashMaps().size());

    this.request.setQueryString("number=two");
    inputFlashMap = this.flashMapManager.retrieveAndUpdate(this.request, this.response);

    assertNull(inputFlashMap);
    assertEquals("FlashMap should not have been removed", 1, this.flashMapManager.getFlashMaps().size());

    this.request.setQueryString("number=one");
    inputFlashMap = this.flashMapManager.retrieveAndUpdate(this.request, this.response);

    assertEquals(flashMap, inputFlashMap);
    assertEquals("Input FlashMap should have been removed", 0, this.flashMapManager.getFlashMaps().size());
}

项目：spring4-understanding 文件：FlashMapManagerTests.java

@Test
public void retrieveAndUpdateMatchWithMultiValueParam() {
    FlashMap flashMap = new FlashMap();
    flashMap.put("name", "value");
    flashMap.addTargetRequestParam("id", "1");
    flashMap.addTargetRequestParam("id", "2");

    this.flashMapManager.setFlashMaps(Arrays.asList(flashMap));

    this.request.setQueryString("id=1");
    FlashMap inputFlashMap = this.flashMapManager.retrieveAndUpdate(this.request, this.response);

    assertNull(inputFlashMap);
    assertEquals("FlashMap should not have been removed", 1, this.flashMapManager.getFlashMaps().size());

    this.request.setQueryString("id=1&id=2");
    inputFlashMap = this.flashMapManager.retrieveAndUpdate(this.request, this.response);

    assertEquals(flashMap, inputFlashMap);
    assertEquals("Input FlashMap should have been removed", 0, this.flashMapManager.getFlashMaps().size());
}

项目：spring4-understanding 文件：FlashMapManagerTests.java

@Test
public void retrieveAndUpdateSortMultipleMatches() {
    FlashMap emptyFlashMap = new FlashMap();

    FlashMap flashMapOne = new FlashMap();
    flashMapOne.put("key1", "value1");
    flashMapOne.setTargetRequestPath("/one");

    FlashMap flashMapTwo = new FlashMap();
    flashMapTwo.put("key1", "value1");
    flashMapTwo.put("key2", "value2");
    flashMapTwo.setTargetRequestPath("/one/two");

    this.flashMapManager.setFlashMaps(Arrays.asList(emptyFlashMap, flashMapOne, flashMapTwo));

    this.request.setRequestURI("/one/two");
    FlashMap inputFlashMap = this.flashMapManager.retrieveAndUpdate(this.request, this.response);

    assertEquals(flashMapTwo, inputFlashMap);
    assertEquals("Input FlashMap should have been removed", 2, this.flashMapManager.getFlashMaps().size());
}

项目：spring4-understanding 文件：FlashMapManagerTests.java

@Test
public void saveOutputFlashMapDecodeParameters() throws Exception {

    FlashMap flashMap = new FlashMap();
    flashMap.put("key", "value");
    flashMap.setTargetRequestPath("/path");
    flashMap.addTargetRequestParam("param", "%D0%90%D0%90");
    flashMap.addTargetRequestParam("param", "%D0%91%D0%91");
    flashMap.addTargetRequestParam("param", "%D0%92%D0%92");
    flashMap.addTargetRequestParam("%3A%2F%3F%23%5B%5D%40", "value");

    this.request.setCharacterEncoding("UTF-8");
    this.flashMapManager.saveOutputFlashMap(flashMap, this.request, this.response);

    MockHttpServletRequest requestAfterRedirect = new MockHttpServletRequest("GET", "/path");
    requestAfterRedirect.setQueryString("param=%D0%90%D0%90&param=%D0%91%D0%91&param=%D0%92%D0%92&%3A%2F%3F%23%5B%5D%40=value");
    requestAfterRedirect.addParameter("param", "\u0410\u0410");
    requestAfterRedirect.addParameter("param", "\u0411\u0411");
    requestAfterRedirect.addParameter("param", "\u0412\u0412");
    requestAfterRedirect.addParameter(":/?#[]@", "value");

    flashMap = this.flashMapManager.retrieveAndUpdate(requestAfterRedirect, new MockHttpServletResponse());
    assertNotNull(flashMap);
    assertEquals(1, flashMap.size());
    assertEquals("value", flashMap.get("key"));
}

项目：spring4-understanding 文件：FlashMapManagerTests.java

@Test
public void flashAttributesWithQueryParamsWithSpace() throws Exception {

    String encodedValue = URLEncoder.encode("1 2", "UTF-8");

    FlashMap flashMap = new FlashMap();
    flashMap.put("key", "value");
    flashMap.setTargetRequestPath("/path");
    flashMap.addTargetRequestParam("param", encodedValue);

    this.request.setCharacterEncoding("UTF-8");
    this.flashMapManager.saveOutputFlashMap(flashMap, this.request, this.response);

    MockHttpServletRequest requestAfterRedirect = new MockHttpServletRequest("GET", "/path");
    requestAfterRedirect.setQueryString("param=" + encodedValue);
    requestAfterRedirect.addParameter("param", "1 2");

    flashMap = this.flashMapManager.retrieveAndUpdate(requestAfterRedirect, new MockHttpServletResponse());
    assertNotNull(flashMap);
    assertEquals(1, flashMap.size());
    assertEquals("value", flashMap.get("key"));
}

项目：entelect-spring-webapp-template 文件：GlobalExceptionControllerAdvice.java

@ExceptionHandler({Exception.class, RuntimeException.class})
public ModelAndView handleAllExceptions(Exception ex,
                                        HttpServletRequest request) {
    log.error(String.format("An unhandled error occurred on page %s.", request.getRequestURL()) ,ex);

    if (config.isDevOn()) {
        FlashMap flashMap = RequestContextUtils.getOutputFlashMap(request);
        flashMap.put("exceptionName", ex.getClass().getSimpleName());
        flashMap.put("exception", ex.getMessage());

        StringWriter errors = new StringWriter();
        ex.printStackTrace(new PrintWriter(errors));
        flashMap.put("stackTrace", errors.toString());
    }
    return new ModelAndView("redirect:/error");
}