SPR-8448 Remove URI template vars from the model before appending to query string in RedirectView
This commit is contained in:
parent
f288060ad8
commit
23f31d453f
|
|
@ -22,6 +22,7 @@ import java.lang.reflect.Array;
|
||||||
import java.net.URI;
|
import java.net.URI;
|
||||||
import java.net.URISyntaxException;
|
import java.net.URISyntaxException;
|
||||||
import java.net.URLEncoder;
|
import java.net.URLEncoder;
|
||||||
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
|
@ -30,15 +31,14 @@ import java.util.Iterator;
|
||||||
import java.util.LinkedHashMap;
|
import java.util.LinkedHashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Set;
|
|
||||||
|
|
||||||
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletRequest;
|
||||||
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
|
||||||
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.BeanUtils;
|
||||||
import org.springframework.http.HttpStatus;
|
import org.springframework.http.HttpStatus;
|
||||||
import org.springframework.ui.Model;
|
|
||||||
import org.springframework.util.ObjectUtils;
|
import org.springframework.util.ObjectUtils;
|
||||||
|
import org.springframework.web.servlet.HandlerMapping;
|
||||||
import org.springframework.web.servlet.View;
|
import org.springframework.web.servlet.View;
|
||||||
import org.springframework.web.util.UriTemplate;
|
import org.springframework.web.util.UriTemplate;
|
||||||
import org.springframework.web.util.UriUtils;
|
import org.springframework.web.util.UriUtils;
|
||||||
|
|
@ -232,18 +232,23 @@ public class RedirectView extends AbstractUrlBasedView {
|
||||||
enc = WebUtils.DEFAULT_CHARACTER_ENCODING;
|
enc = WebUtils.DEFAULT_CHARACTER_ENCODING;
|
||||||
}
|
}
|
||||||
|
|
||||||
UriTemplate uriTemplate = createUriTemplate(targetUrl, enc);
|
UriTemplate redirectUri = createUriTemplate(targetUrl, enc);
|
||||||
if (uriTemplate.getVariableNames().size() > 0) {
|
if (redirectUri.getVariableNames().size() > 0) {
|
||||||
targetUrl = new StringBuilder(uriTemplate.expand(model).toString());
|
targetUrl = new StringBuilder(redirectUri.expand(model).toString());
|
||||||
model = removeKeys(model, uriTemplate.getVariableNames());
|
model = removeKeys(model, redirectUri.getVariableNames());
|
||||||
}
|
}
|
||||||
if (this.exposeModelAttributes) {
|
if (this.exposeModelAttributes) {
|
||||||
|
List<String> uriTemplateVarNames = getUriTemplateVarNames(request);
|
||||||
|
if (!uriTemplateVarNames.isEmpty()) {
|
||||||
|
model = removeKeys(model, uriTemplateVarNames);
|
||||||
|
}
|
||||||
appendQueryProperties(targetUrl, model, enc);
|
appendQueryProperties(targetUrl, model, enc);
|
||||||
}
|
}
|
||||||
|
|
||||||
sendRedirect(request, response, targetUrl.toString(), this.http10Compatible);
|
sendRedirect(request, response, targetUrl.toString(), this.http10Compatible);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("serial")
|
||||||
private UriTemplate createUriTemplate(StringBuilder targetUrl, final String encoding) {
|
private UriTemplate createUriTemplate(StringBuilder targetUrl, final String encoding) {
|
||||||
return new UriTemplate(targetUrl.toString()) {
|
return new UriTemplate(targetUrl.toString()) {
|
||||||
@Override
|
@Override
|
||||||
|
|
@ -268,6 +273,16 @@ public class RedirectView extends AbstractUrlBasedView {
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns URI template variable names for the current request; or an empty list.
|
||||||
|
*/
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
private List<String> getUriTemplateVarNames(HttpServletRequest request) {
|
||||||
|
String key = HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE;
|
||||||
|
Map<String, String> map = (Map<String, String>) request.getAttribute(key);
|
||||||
|
return (map != null) ? new ArrayList<String>(map.keySet()) : Collections.<String>emptyList();
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Append query properties to the redirect URL.
|
* Append query properties to the redirect URL.
|
||||||
* Stringifies, URL-encodes and formats model attributes as query properties.
|
* Stringifies, URL-encodes and formats model attributes as query properties.
|
||||||
|
|
|
||||||
|
|
@ -25,6 +25,7 @@ import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.springframework.mock.web.MockHttpServletRequest;
|
import org.springframework.mock.web.MockHttpServletRequest;
|
||||||
import org.springframework.mock.web.MockHttpServletResponse;
|
import org.springframework.mock.web.MockHttpServletResponse;
|
||||||
|
import org.springframework.web.servlet.HandlerMapping;
|
||||||
|
|
||||||
public class RedirectViewUriTemplateTests {
|
public class RedirectViewUriTemplateTests {
|
||||||
|
|
||||||
|
|
@ -39,7 +40,7 @@ public class RedirectViewUriTemplateTests {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void pathVar() throws Exception {
|
public void uriTemplateVar() throws Exception {
|
||||||
Map<String, Object> model = new HashMap<String, Object>();
|
Map<String, Object> model = new HashMap<String, Object>();
|
||||||
model.put("foo", "bar");
|
model.put("foo", "bar");
|
||||||
|
|
||||||
|
|
@ -51,7 +52,7 @@ public class RedirectViewUriTemplateTests {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void pathVarAndArrayParam() throws Exception {
|
public void uriTemplateVarAndArrayParam() throws Exception {
|
||||||
Map<String, Object> model = new HashMap<String, Object>();
|
Map<String, Object> model = new HashMap<String, Object>();
|
||||||
model.put("foo", "bar");
|
model.put("foo", "bar");
|
||||||
model.put("fooArr", new String[] { "baz", "bazz" });
|
model.put("fooArr", new String[] { "baz", "bazz" });
|
||||||
|
|
@ -63,7 +64,7 @@ public class RedirectViewUriTemplateTests {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void pathVarWithObjectConversion() throws Exception {
|
public void uriTemplateVarWithObjectConversion() throws Exception {
|
||||||
Map<String, Object> model = new HashMap<String, Object>();
|
Map<String, Object> model = new HashMap<String, Object>();
|
||||||
model.put("foo", new Long(611));
|
model.put("foo", new Long(611));
|
||||||
|
|
||||||
|
|
@ -73,4 +74,22 @@ public class RedirectViewUriTemplateTests {
|
||||||
assertEquals("/foo/611", response.getRedirectedUrl());
|
assertEquals("/foo/611", response.getRedirectedUrl());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void doNotAppendUriTemplateVarFromCurrentRequest() throws Exception {
|
||||||
|
Map<String, Object> model = new HashMap<String, Object>();
|
||||||
|
model.put("name1", "value1");
|
||||||
|
model.put("name2", "value2");
|
||||||
|
model.put("name3", "value3");
|
||||||
|
|
||||||
|
Map<String, String> uriTemplatVars = new HashMap<String, String>();
|
||||||
|
uriTemplatVars.put("name1", "value1");
|
||||||
|
uriTemplatVars.put("name2", "value2");
|
||||||
|
request.setAttribute(HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE, uriTemplatVars);
|
||||||
|
|
||||||
|
String url = "http://url.somewhere.com";
|
||||||
|
RedirectView redirectView = new RedirectView(url + "/{name2}");
|
||||||
|
redirectView.renderMergedOutputModel(model, request, response);
|
||||||
|
|
||||||
|
assertEquals(url + "/value2?name3=value3", response.getRedirectedUrl());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue