[java-support] 02/02: IDP-1793 Use Suppliers for HttpRequest/Response
Rod Widdowson
rdw at steadingsoftware.com
Sat Aug 6 15:24:30 UTC 2022
This is an automated email from the git hooks/post-receive script.
rdw pushed a commit to branch main
in repository java-support.
View the commit online:
http://git.shibboleth.net/view/?p=java-support.git;a=commit;h=78cdc039574026d157477aaf4c7ab46b1c9bfa2b
commit 78cdc039574026d157477aaf4c7ab46b1c9bfa2b
Author: Rod Widdowson <rdw at steadingsoftware.com>
AuthorDate: Sat Aug 6 16:06:09 2022 +0100
IDP-1793 Use Suppliers for HttpRequest/Response
https://shibboleth.atlassian.net/browse/IDP-1793
The CookieManager gains setters for a Supplier<HttpServletRequest>
and Response. There are used throughout the implementation.
---
.../utilities/java/support/net/CookieManager.java | 74 ++++++++++++++++------
.../java/support/net/CookieManagerTest.java | 24 ++++---
2 files changed, 68 insertions(+), 30 deletions(-)
diff --git a/src/main/java/net/shibboleth/utilities/java/support/net/CookieManager.java b/src/main/java/net/shibboleth/utilities/java/support/net/CookieManager.java
index 882ffac..52d6c8f 100644
--- a/src/main/java/net/shibboleth/utilities/java/support/net/CookieManager.java
+++ b/src/main/java/net/shibboleth/utilities/java/support/net/CookieManager.java
@@ -17,18 +17,25 @@
package net.shibboleth.utilities.java.support.net;
+import java.util.function.Supplier;
+
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import jakarta.servlet.http.Cookie;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
import net.shibboleth.utilities.java.support.annotation.constraint.NonnullAfterInit;
import net.shibboleth.utilities.java.support.annotation.constraint.NotEmpty;
import net.shibboleth.utilities.java.support.component.AbstractInitializableComponent;
import net.shibboleth.utilities.java.support.component.ComponentInitializationException;
import net.shibboleth.utilities.java.support.logic.Constraint;
+import net.shibboleth.utilities.java.support.primitive.DeprecationSupport;
import net.shibboleth.utilities.java.support.primitive.StringSupport;
+import net.shibboleth.utilities.java.support.primitive.DeprecationSupport.ObjectType;
/**
* A helper class for managing one or more cookies on behalf of a component.
@@ -39,17 +46,20 @@ import net.shibboleth.utilities.java.support.primitive.StringSupport;
*/
public final class CookieManager extends AbstractInitializableComponent {
+ /** Log. */
+ private final Logger log = LoggerFactory.getLogger(CookieManager.class);
+
/** Path of cookie. */
@Nullable private String cookiePath;
/** Domain of cookie. */
@Nullable private String cookieDomain;
- /** Servlet request to read from. */
- @NonnullAfterInit private HttpServletRequest httpRequest;
+ /** Supplier for the servlet request to read from. */
+ @NonnullAfterInit private Supplier<HttpServletRequest> httpRequestSupplier;
- /** Servlet response to write to. */
- @NonnullAfterInit private HttpServletResponse httpResponse;
+ /** Supplier for the servlet response to write to. */
+ @NonnullAfterInit private Supplier<HttpServletResponse> httpResponseSupplier;
/** Is cookie secure? */
private boolean secure;
@@ -92,27 +102,50 @@ public final class CookieManager extends AbstractInitializableComponent {
}
/**
- * Set the servlet request to read from.
- *
- * @param request servlet request
+ * Set the Supplier for the servlet request to read from.
+ *
+ * @param requestSupplier servlet request supplier
*/
- public void setHttpServletRequest(@Nonnull final HttpServletRequest request) {
+ public void setHttpServletRequestSupplier(@Nonnull final Supplier<HttpServletRequest> requestSupplier) {
checkSetterPreconditions();
-
- httpRequest = Constraint.isNotNull(request, "HttpServletRequest cannot be null");
+ httpRequestSupplier = Constraint.isNotNull(requestSupplier, "HttpServletRequest cannot be null");
}
/**
- * Set the servlet response to write to.
- *
- * @param response servlet response
+ * Get the current HTTP request if available.
+ *
+ * @return current HTTP request
+ */
+ @NonnullAfterInit private HttpServletRequest getHttpServletRequest() {
+ if (httpRequestSupplier == null) {
+ return null;
+ }
+ return httpRequestSupplier.get();
+ }
+
+ /**
+ * Set the supplier for the servlet response to write to.
+ *
+ * @param responseSupplier servlet response
*/
- public void setHttpServletResponse(@Nonnull final HttpServletResponse response) {
+ public void setHttpServletResponseSupplier(@Nonnull final Supplier<HttpServletResponse> responseSupplier) {
checkSetterPreconditions();
-
- httpResponse = Constraint.isNotNull(response, "HttpServletResponse cannot be null");
+ httpResponseSupplier = Constraint.isNotNull(responseSupplier, "HttpServletResponse cannot be null");
}
+ /**
+ * Get the current HTTP response if available.
+ *
+ * @return current HTTP response or null
+ */
+ @NonnullAfterInit private HttpServletResponse getHttpServletResponse() {
+ if (httpResponseSupplier == null) {
+ return null;
+ }
+ return httpResponseSupplier.get();
+ }
+
+
/**
* Set the SSL-only flag.
*
@@ -151,7 +184,7 @@ public final class CookieManager extends AbstractInitializableComponent {
protected void doInitialize() throws ComponentInitializationException {
super.doInitialize();
- if (httpRequest == null || httpResponse == null) {
+ if (getHttpServletRequest() == null || getHttpServletResponse() == null) {
throw new ComponentInitializationException("Servlet request and response must be set");
}
}
@@ -174,7 +207,7 @@ public final class CookieManager extends AbstractInitializableComponent {
cookie.setHttpOnly(httpOnly);
cookie.setMaxAge(maxAge);
- httpResponse.addCookie(cookie);
+ getHttpServletResponse().addCookie(cookie);
}
/**
@@ -194,7 +227,7 @@ public final class CookieManager extends AbstractInitializableComponent {
cookie.setHttpOnly(httpOnly);
cookie.setMaxAge(0);
- httpResponse.addCookie(cookie);
+ getHttpServletResponse().addCookie(cookie);
}
/**
@@ -226,7 +259,7 @@ public final class CookieManager extends AbstractInitializableComponent {
@Nullable public String getCookieValue(@Nonnull @NotEmpty final String name, @Nullable final String defValue) {
checkComponentActive();
- final Cookie[] cookies = httpRequest.getCookies();
+ final Cookie[] cookies = getHttpServletRequest().getCookies();
if (cookies != null) {
for (final Cookie cookie : cookies) {
if (cookie.getName().equals(name)) {
@@ -244,6 +277,7 @@ public final class CookieManager extends AbstractInitializableComponent {
* @return the cookie path
*/
@Nonnull @NotEmpty private String contextPathToCookiePath() {
+ final HttpServletRequest httpRequest = getHttpServletRequest();
return "".equals(httpRequest.getContextPath()) ? "/" : httpRequest.getContextPath();
}
diff --git a/src/test/java/net/shibboleth/utilities/java/support/net/CookieManagerTest.java b/src/test/java/net/shibboleth/utilities/java/support/net/CookieManagerTest.java
index 7c8256d..4a69397 100644
--- a/src/test/java/net/shibboleth/utilities/java/support/net/CookieManagerTest.java
+++ b/src/test/java/net/shibboleth/utilities/java/support/net/CookieManagerTest.java
@@ -17,6 +17,8 @@
package net.shibboleth.utilities.java.support.net;
+import java.util.function.Supplier;
+
import net.shibboleth.utilities.java.support.component.ComponentInitializationException;
import org.springframework.mock.web.MockHttpServletRequest;
@@ -25,6 +27,8 @@ import org.testng.Assert;
import org.testng.annotations.Test;
import jakarta.servlet.http.Cookie;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
/** {@link CookieManager} unit test. */
public class CookieManagerTest {
@@ -44,8 +48,8 @@ public class CookieManagerTest {
MockHttpServletResponse response = new MockHttpServletResponse();
CookieManager cm = new CookieManager();
- cm.setHttpServletRequest(request);
- cm.setHttpServletResponse(response);
+ cm.setHttpServletRequestSupplier(new Supplier<>() { public HttpServletRequest get() {return request;}});
+ cm.setHttpServletResponseSupplier(new Supplier<>() { public HttpServletResponse get() {return response;}});
cm.initialize();
}
@@ -54,13 +58,13 @@ public class CookieManagerTest {
MockHttpServletResponse response = new MockHttpServletResponse();
CookieManager cm = new CookieManager();
- cm.setHttpServletRequest(request);
- cm.setHttpServletResponse(response);
+ cm.setHttpServletRequestSupplier(new Supplier<>() { public HttpServletRequest get() {return request;}});
+ cm.setHttpServletResponseSupplier(new Supplier<>() { public HttpServletResponse get() {return response;}});
cm.setCookiePath("/idp");
cm.initialize();
-
+
cm.addCookie("foo", "bar");
-
+
Cookie cookie = response.getCookie("foo");
Assert.assertNotNull(cookie);
Assert.assertEquals(cookie.getValue(), "bar");
@@ -76,8 +80,8 @@ public class CookieManagerTest {
MockHttpServletResponse response = new MockHttpServletResponse();
CookieManager cm = new CookieManager();
- cm.setHttpServletRequest(request);
- cm.setHttpServletResponse(response);
+ cm.setHttpServletRequestSupplier(new Supplier<>() { public HttpServletRequest get() {return request;}});
+ cm.setHttpServletResponseSupplier(new Supplier<>() { public HttpServletResponse get() {return response;}});
cm.initialize();
cm.addCookie("foo", "bar");
@@ -98,8 +102,8 @@ public class CookieManagerTest {
MockHttpServletResponse response = new MockHttpServletResponse();
CookieManager cm = new CookieManager();
- cm.setHttpServletRequest(request);
- cm.setHttpServletResponse(response);
+ cm.setHttpServletRequestSupplier(new Supplier<>() { public HttpServletRequest get() {return request;}});
+ cm.setHttpServletResponseSupplier(new Supplier<>() { public HttpServletResponse get() {return response;}});
cm.initialize();
cm.unsetCookie("foo");
--
To stop receiving notification emails like this one, please contact
the administrator of this repository.
More information about the commits
mailing list