01: /* ***** BEGIN LICENSE BLOCK *****
02: * Version: MPL 1.1
03: * The contents of this file are subject to the Mozilla Public License Version
04: * 1.1 (the "License"); you may not use this file except in compliance with
05: * the License. You may obtain a copy of the License at
06: * http://www.mozilla.org/MPL/
07: *
08: * Software distributed under the License is distributed on an "AS IS" basis,
09: * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
10: * for the specific language governing rights and limitations under the
11: * License.
12: *
13: * The Original Code is Riot.
14: *
15: * The Initial Developer of the Original Code is
16: * Neteye GmbH.
17: * Portions created by the Initial Developer are Copyright (C) 2006
18: * the Initial Developer. All Rights Reserved.
19: *
20: * Contributor(s):
21: * Felix Gnass [fgnass at neteye dot de]
22: *
23: * ***** END LICENSE BLOCK ***** */
24: package org.riotfamily.forms.fileupload;
25:
26: import javax.servlet.http.HttpServletRequest;
27:
28: import org.apache.commons.logging.Log;
29: import org.apache.commons.logging.LogFactory;
30: import org.springframework.web.multipart.MultipartException;
31: import org.springframework.web.multipart.MultipartHttpServletRequest;
32: import org.springframework.web.multipart.MultipartResolver;
33:
34: /**
35: * MultipartResolver implementation to track the progress of file uploads.
36: * Acts as wrapper so that the actual work is performed by the implementation
37: * passed as constructor argument. Instead of the original request a
38: * {@link org.riotfamily.forms.fileupload.HttpUploadRequest HttpUploadRequest} is
39: * passed to the underlying resolver which provides the progress information.
40: */
41: public class FormsMultipartResolver implements MultipartResolver {
42:
43: private static final String UPLOAD_ID_ATTR = FormsMultipartResolver.class
44: .getName()
45: + ".uploadId";
46:
47: private Log log = LogFactory.getLog(FormsMultipartResolver.class);
48:
49: private MultipartResolver resolver;
50:
51: private String uploadIdParam = "uploadId";
52:
53: public FormsMultipartResolver(MultipartResolver resolver) {
54: this .resolver = resolver;
55: }
56:
57: public boolean isMultipart(HttpServletRequest request) {
58: return resolver.isMultipart(request);
59: }
60:
61: public MultipartHttpServletRequest resolveMultipart(
62: HttpServletRequest request) throws MultipartException {
63:
64: String uploadId = request.getParameter(uploadIdParam);
65: if (uploadId == null) {
66: log.debug("Parameter '" + uploadIdParam + "' not set");
67: return resolver.resolveMultipart(request);
68: }
69: log.debug("Upload: " + uploadId);
70: HttpUploadRequest uploadRequest = new HttpUploadRequest(request);
71: request.setAttribute(UPLOAD_ID_ATTR, uploadId);
72: UploadStatus.add(uploadId, uploadRequest);
73: return resolver.resolveMultipart(uploadRequest);
74: }
75:
76: public void cleanupMultipart(MultipartHttpServletRequest request) {
77: UploadStatus.clearStatus(request.getAttribute(UPLOAD_ID_ATTR));
78: resolver.cleanupMultipart(request);
79: }
80:
81: }
|