KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > com > go > trove > net > BufferedSocket


1 /* ====================================================================
2  * Trove - Copyright (c) 1997-2000 Walt Disney Internet Group
3  * ====================================================================
4  * The Tea Software License, Version 1.1
5  *
6  * Copyright (c) 2000 Walt Disney Internet Group. All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  * notice, this list of conditions and the following disclaimer.
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  * notice, this list of conditions and the following disclaimer in
17  * the documentation and/or other materials provided with the
18  * distribution.
19  *
20  * 3. The end-user documentation included with the redistribution,
21  * if any, must include the following acknowledgment:
22  * "This product includes software developed by the
23  * Walt Disney Internet Group (http://opensource.go.com/)."
24  * Alternately, this acknowledgment may appear in the software itself,
25  * if and wherever such third-party acknowledgments normally appear.
26  *
27  * 4. The names "Tea", "TeaServlet", "Kettle", "Trove" and "BeanDoc" must
28  * not be used to endorse or promote products derived from this
29  * software without prior written permission. For written
30  * permission, please contact opensource@dig.com.
31  *
32  * 5. Products derived from this software may not be called "Tea",
33  * "TeaServlet", "Kettle" or "Trove", nor may "Tea", "TeaServlet",
34  * "Kettle", "Trove" or "BeanDoc" appear in their name, without prior
35  * written permission of the Walt Disney Internet Group.
36  *
37  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
38  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
39  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
40  * DISCLAIMED. IN NO EVENT SHALL THE WALT DISNEY INTERNET GROUP OR ITS
41  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
42  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
43  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
44  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
45  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
46  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
47  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
48  * ====================================================================
49  *
50  * For more information about Tea, please see http://opensource.go.com/.
51  */

52
53 package com.go.trove.net;
54
55 import java.io.*;
56 import java.net.Socket JavaDoc;
57 import com.go.trove.io.*;
58
59 /******************************************************************************
60  * A convenient class for supporting buffering on socket I/O streams.
61  *
62  * @author Brian S O'Neill
63  * @version
64  * <!--$$Revision:--> 1 <!-- $-->, <!--$$JustDate:--> 00/12/06 <!-- $-->
65  */

66 public class BufferedSocket extends FilteredSocket {
67     private final int mInBufSize;
68     private final int mOutBufSize;
69
70     private InputStream mIn;
71     private OutputStream mOut;
72
73     public BufferedSocket(Socket JavaDoc s) throws IOException {
74         this(s, -1, -1);
75     }
76
77     /**
78      * @param inputBufferSize specify 0 for no buffering, -1 for default
79      * @param outputBufferSize specify 0 for no buffering, -1 for default
80      */

81     public BufferedSocket(Socket JavaDoc s,
82                           int inputBufferSize,
83                           int outputBufferSize) throws IOException {
84         super(s);
85         mInBufSize = inputBufferSize;
86         mOutBufSize = outputBufferSize;
87     }
88     
89     public synchronized InputStream getInputStream() throws IOException {
90         if (mIn == null) {
91             InputStream in = super.getInputStream();
92
93             if (mInBufSize < 0) {
94                 mIn = new FastBufferedInputStream(in);
95             }
96             else if (mInBufSize > 0) {
97                 mIn = new FastBufferedInputStream(in, mInBufSize);
98             }
99             else {
100                 mIn = in;
101             }
102         }
103         return mIn;
104     }
105     
106     public synchronized OutputStream getOutputStream() throws IOException {
107         if (mOut == null) {
108             OutputStream out = super.getOutputStream();
109
110             if (mOutBufSize < 0) {
111                 mOut = new FastBufferedOutputStream(out);
112             }
113             else if (mOutBufSize > 0) {
114                 mOut = new FastBufferedOutputStream(out, mOutBufSize);
115             }
116             else {
117                 mOut = out;
118             }
119         }
120         return mOut;
121     }
122     
123     public synchronized void close() throws IOException {
124         // Ensure buffered output is flushed.
125
if (mOutBufSize != 0 && mOut != null) {
126             mOut.close();
127         }
128         super.close();
129     }
130 }
131
Popular Tags