KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > riotfamily > forms > fileupload > CountingServletInputStream


1 /* ***** BEGIN LICENSE BLOCK *****
2  * Version: MPL 1.1
3  * The contents of this file are subject to the Mozilla Public License Version
4  * 1.1 (the "License"); you may not use this file except in compliance with
5  * the License. You may obtain a copy of the License at
6  * http://www.mozilla.org/MPL/
7  *
8  * Software distributed under the License is distributed on an "AS IS" basis,
9  * 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 JavaDoc;
27
28 import javax.servlet.ServletInputStream JavaDoc;
29
30 /**
31  * ServletInputStream that counts the number of bytes read. This class is used
32  * to provide progress information for file uploads.
33  */

34 public class CountingServletInputStream extends ServletInputStream JavaDoc {
35
36     private ServletInputStream JavaDoc sourceStream;
37
38     private long bytesRead;
39     
40     public CountingServletInputStream() {
41     }
42
43     public void setSourceStream(ServletInputStream JavaDoc sourceStream) {
44         this.sourceStream = sourceStream;
45     }
46     
47     public int read() throws IOException JavaDoc {
48         if (sourceStream == null) {
49             throw new IllegalMonitorStateException JavaDoc("No sourceStream set");
50         }
51         int i = sourceStream.read();
52         if (i >= 0) {
53             bytesRead ++;
54         }
55         return i;
56     }
57
58     public long getBytesRead() {
59         return bytesRead;
60     }
61 }
62
Popular Tags