| /* |
| * Copyright (c) 1996, 2005, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package sun.rmi.transport; |
| |
| import java.io.IOException; |
| import java.rmi.server.UID; |
| import sun.rmi.server.MarshalOutputStream; |
| |
| /** |
| * Special stream to keep track of refs being marshaled as return |
| * results to determine whether a special ack needs to be sent |
| * to the distributed collector. |
| * |
| * @author Ann Wollrath |
| */ |
| class ConnectionOutputStream extends MarshalOutputStream { |
| |
| /** connection associated with ConnectionOutputStream */ |
| private final Connection conn; |
| /** indicates whether output stream is used to marshal results */ |
| private final boolean resultStream; |
| /** identifier for gc ack*/ |
| private final UID ackID; |
| |
| /** to store refs to returned remote object until DGC ack is received */ |
| private DGCAckHandler dgcAckHandler = null; |
| |
| /** |
| * Constructs an marshal output stream using the underlying |
| * stream associated with the connection, the parameter c. |
| * @param c is the Connection object associated with the |
| * ConnectionOutputStream object being constructed |
| * @param resultStream indicates whether this stream is used |
| * to marshal return results |
| */ |
| ConnectionOutputStream(Connection conn, boolean resultStream) |
| throws IOException |
| { |
| super(conn.getOutputStream()); |
| this.conn = conn; |
| this.resultStream = resultStream; |
| ackID = resultStream ? new UID() : null; |
| } |
| |
| void writeID() throws IOException { |
| assert resultStream; |
| ackID.write(this); |
| } |
| |
| /** |
| * Returns true if this output stream is used to marshal return |
| * results; otherwise returns false. |
| */ |
| boolean isResultStream() { |
| return resultStream; |
| } |
| |
| /** |
| * Saves a reference to the specified object in this stream's |
| * DGCAckHandler. |
| **/ |
| void saveObject(Object obj) { |
| // should always be accessed from same thread |
| if (dgcAckHandler == null) { |
| dgcAckHandler = new DGCAckHandler(ackID); |
| } |
| dgcAckHandler.add(obj); |
| } |
| |
| /** |
| * Returns this stream's DGCAckHandler, or null if it doesn't have |
| * one (saveObject was not invoked). This method should only be |
| * invoked after all objects have been written to the stream, |
| * because future objects written may yet cause a DGCAckHandler to |
| * be created (by invoking saveObject). |
| **/ |
| DGCAckHandler getDGCAckHandler() { |
| return dgcAckHandler; |
| } |
| |
| void done() { |
| if (dgcAckHandler != null) { |
| dgcAckHandler.startTimer(); |
| } |
| } |
| } |