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 java.io.IOException;
27:
28: import javax.servlet.ServletInputStream;
29: import javax.servlet.http.HttpServletRequest;
30: import javax.servlet.http.HttpServletRequestWrapper;
31:
32: /**
33: * HttpServletRequestWrapper that returns a
34: * {@link org.riotfamily.forms.fileupload.CountingServletInputStream CountingInputStream}.
35: */
36: public class HttpUploadRequest extends HttpServletRequestWrapper {
37:
38: private ServletInputStream wrappedStream;
39:
40: private CountingServletInputStream countingInputStream;
41:
42: public HttpUploadRequest(HttpServletRequest request) {
43: super (request);
44: this .countingInputStream = new CountingServletInputStream();
45: }
46:
47: public CountingServletInputStream getCountingInputStream() {
48: return countingInputStream;
49: }
50:
51: public ServletInputStream getInputStream() throws IOException {
52: if (wrappedStream == null) {
53: wrappedStream = super.getInputStream();
54: countingInputStream.setSourceStream(wrappedStream);
55: }
56: return countingInputStream;
57: }
58:
59: }
|