KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > apache > mina > handler > support > IoSessionOutputStream


1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements. See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership. The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License. You may obtain a copy of the License at
9  *
10  * http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied. See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  *
19  */

20 package org.apache.mina.handler.support;
21
22 import java.io.IOException JavaDoc;
23 import java.io.OutputStream JavaDoc;
24
25 import org.apache.mina.common.ByteBuffer;
26 import org.apache.mina.common.IoSession;
27 import org.apache.mina.common.WriteFuture;
28
29 /**
30  * An {@link OutputStream} that forwards all write operations to
31  * the associated {@link IoSession}.
32  *
33  * @author The Apache Directory Project (mina-dev@directory.apache.org)
34  * @version $Rev: 555855 $, $Date: 2007-07-13 12:19:00 +0900 (금, 13 7월 2007) $
35  */

36 public class IoSessionOutputStream extends OutputStream JavaDoc {
37     private final IoSession session;
38
39     private WriteFuture lastWriteFuture;
40
41     public IoSessionOutputStream(IoSession session) {
42         this.session = session;
43     }
44
45     @Override JavaDoc
46     public void close() throws IOException JavaDoc {
47         try {
48             flush();
49         } finally {
50             session.close().join();
51         }
52     }
53
54     private void checkClosed() throws IOException JavaDoc {
55         if (!session.isConnected()) {
56             throw new IOException JavaDoc("The session has been closed.");
57         }
58     }
59
60     private synchronized void write(ByteBuffer buf) throws IOException JavaDoc {
61         checkClosed();
62         lastWriteFuture = session.write(buf);
63     }
64
65     @Override JavaDoc
66     public void write(byte[] b, int off, int len) throws IOException JavaDoc {
67         write(ByteBuffer.wrap(b.clone(), off, len));
68     }
69
70     @Override JavaDoc
71     public void write(int b) throws IOException JavaDoc {
72         ByteBuffer buf = ByteBuffer.allocate(1);
73         buf.put((byte) b);
74         buf.flip();
75         write(buf);
76     }
77
78     @Override JavaDoc
79     public synchronized void flush() throws IOException JavaDoc {
80         if (lastWriteFuture == null) {
81             return;
82         }
83
84         lastWriteFuture.join();
85         if (!lastWriteFuture.isWritten()) {
86             throw new IOException JavaDoc(
87                     "The bytes could not be written to the session");
88         }
89     }
90 }
91
Popular Tags