Store servlet in servlet request, too.

[demoscenemusic.git] / src / main / java / net / pterodactylus / demoscenemusic / page / ServletRequest.java

diff --git a/src/main/java/net/pterodactylus/demoscenemusic/page/ServletRequest.java b/src/main/java/net/pterodactylus/demoscenemusic/page/ServletRequest.java
index 64e66c1..74d0450 100644 (file)
--- a/src/main/java/net/pterodactylus/demoscenemusic/page/ServletRequest.java
+++ b/src/main/java/net/pterodactylus/demoscenemusic/page/ServletRequest.java
@@ -20,6 +20,7 @@ package net.pterodactylus.demoscenemusic.page;
 import java.net.URI;
 import java.net.URISyntaxException;
 
+import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 
 import net.pterodactylus.util.web.Method;
@@ -33,15 +34,17 @@ import net.pterodactylus.util.web.Request;
 public class ServletRequest extends Request {
 
        private final HttpServletRequest servletRequest;
+       private final HttpServlet servlet;
 
        /**
         * @param uri
         * @param method
         * @throws URISyntaxException
         */
-       public ServletRequest(javax.servlet.http.HttpServletRequest servletRequest) throws URISyntaxException {
+       public ServletRequest(javax.servlet.http.HttpServletRequest servletRequest, HttpServlet servlet) throws URISyntaxException {
                super(new URI(servletRequest.getRequestURI()), Method.valueOf(servletRequest.getMethod().toUpperCase()));
                this.servletRequest = servletRequest;
+               this.servlet = servlet;
        }
 
        /**
@@ -51,4 +54,11 @@ public class ServletRequest extends Request {
                return servletRequest;
        }
 
+       /**
+        * @return the servlet
+        */
+       public HttpServlet getServlet() {
+               return servlet;
+       }
+
 }