1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/*
* Copyright 1996-2005 Sun Microsystems, Inc. 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. Sun designates this
* particular file as subject to the "Classpath" exception as provided
* by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
* CA 95054 USA or visit www.sun.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();
}
}
}
|