|
| 1 | +package org.vosao.servlet; |
| 2 | + |
| 3 | +import java.io.IOException; |
| 4 | + |
| 5 | +import javax.servlet.ServletConfig; |
| 6 | +import javax.servlet.ServletException; |
| 7 | +import javax.servlet.http.HttpServletRequest; |
| 8 | +import javax.servlet.http.HttpServletResponse; |
| 9 | + |
| 10 | +import com.bradmcevoy.http.HttpManager; |
| 11 | +import com.bradmcevoy.http.Request; |
| 12 | +import com.bradmcevoy.http.Response; |
| 13 | +import com.bradmcevoy.http.ServletResponse; |
| 14 | + |
| 15 | +public class WebdavServlet extends BaseSpringServlet { |
| 16 | + |
| 17 | + ServletConfig config; |
| 18 | + HttpManager httpManager; |
| 19 | + |
| 20 | + private static final ThreadLocal<HttpServletRequest> originalRequest = |
| 21 | + new ThreadLocal<HttpServletRequest>(); |
| 22 | + private static final ThreadLocal<HttpServletResponse> originalResponse = |
| 23 | + new ThreadLocal<HttpServletResponse>(); |
| 24 | + |
| 25 | + public static HttpServletRequest request() { |
| 26 | + return originalRequest.get(); |
| 27 | + } |
| 28 | + |
| 29 | + public static HttpServletResponse response() { |
| 30 | + return originalResponse.get(); |
| 31 | + } |
| 32 | + |
| 33 | + public static void forward(String url) { |
| 34 | + try { |
| 35 | + request().getRequestDispatcher(url).forward(originalRequest.get(), |
| 36 | + originalResponse.get()); |
| 37 | + } catch (IOException ex) { |
| 38 | + throw new RuntimeException(ex); |
| 39 | + } catch (ServletException ex) { |
| 40 | + throw new RuntimeException(ex); |
| 41 | + } |
| 42 | + } |
| 43 | + |
| 44 | + public void init(ServletConfig config) throws ServletException { |
| 45 | + try { |
| 46 | + this.config = config; |
| 47 | + httpManager = (HttpManager) getSpringBean("milton.http.manager"); |
| 48 | + } catch (Throwable ex) { |
| 49 | + logger.error("Exception starting milton servlet " + ex.getMessage()); |
| 50 | + throw new RuntimeException(ex); |
| 51 | + } |
| 52 | + } |
| 53 | + |
| 54 | + public void service(javax.servlet.ServletRequest servletRequest, javax.servlet.ServletResponse servletResponse) throws ServletException, IOException { |
| 55 | + HttpServletRequest req = (HttpServletRequest) servletRequest; |
| 56 | + HttpServletResponse resp = (HttpServletResponse) servletResponse; |
| 57 | + try { |
| 58 | + originalRequest.set(req); |
| 59 | + originalResponse.set(resp); |
| 60 | + Request request = new WebdavServletRequest(req); |
| 61 | + Response response = new ServletResponse(resp); |
| 62 | + httpManager.process(request, response); |
| 63 | + } finally { |
| 64 | + originalRequest.remove(); |
| 65 | + originalResponse.remove(); |
| 66 | + servletResponse.getOutputStream().flush(); |
| 67 | + servletResponse.flushBuffer(); |
| 68 | + } |
| 69 | + } |
| 70 | + |
| 71 | + public String getServletInfo() { |
| 72 | + return "WebdavServlet"; |
| 73 | + } |
| 74 | + |
| 75 | + public ServletConfig getServletConfig() { |
| 76 | + return config; |
| 77 | + } |
| 78 | + |
| 79 | + public void destroy() { |
| 80 | + |
| 81 | + } |
| 82 | + } |
| 83 | + |
| 84 | + |
0 commit comments