OLD | NEW |
---|---|
1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file |
2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
4 | 4 |
5 class _SocketInputStream implements SocketInputStream { | 5 class _SocketInputStream implements SocketInputStream { |
6 _SocketInputStream(Socket socket) : _socket = socket { | 6 _SocketInputStream(Socket socket) : _socket = socket { |
7 if (_socket._id == -1) _closed = true; | 7 if (_socket._id == -1) _closed = true; |
8 _socket.onClosed = _onClosed; | 8 _socket.onClosed = _onClosed; |
9 } | 9 } |
10 | 10 |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
105 | 105 |
106 bool writeFrom(List<int> buffer, [int offset = 0, int len]) { | 106 bool writeFrom(List<int> buffer, [int offset = 0, int len]) { |
107 return _write( | 107 return _write( |
108 buffer, offset, (len == null) ? buffer.length - offset : len, true); | 108 buffer, offset, (len == null) ? buffer.length - offset : len, true); |
109 } | 109 } |
110 | 110 |
111 void close() { | 111 void close() { |
112 if (!_pendingWrites.isEmpty()) { | 112 if (!_pendingWrites.isEmpty()) { |
113 // Mark the socket for close when all data is written. | 113 // Mark the socket for close when all data is written. |
114 _closing = true; | 114 _closing = true; |
115 _socket._onWrite = _onWrite; | 115 _setupWaitOnWrite(); |
Mads Ager (google)
2012/05/02 08:18:17
I found the name a bit confusing. How about callin
Søren Gjesse
2012/05/02 10:22:45
Done.
| |
116 } else { | 116 } else { |
117 // Close the socket for writing. | 117 // Close the socket for writing. |
118 _socket._closeWrite(); | 118 _socket._closeWrite(); |
119 _closed = true; | 119 _closed = true; |
120 } | 120 } |
121 } | 121 } |
122 | 122 |
123 void destroy() { | 123 void destroy() { |
124 _socket.onWrite = null; | 124 _socket.onWrite = null; |
125 _pendingWrites.clear(); | 125 _pendingWrites.clear(); |
126 _socket.close(); | 126 _socket.close(); |
127 _closed = true; | 127 _closed = true; |
128 } | 128 } |
129 | 129 |
130 void set onNoPendingWrites(void callback()) { | 130 void set onNoPendingWrites(void callback()) { |
131 if (_noPendingWritesTimer != null) { | |
132 _noPendingWritesTimer.cancel(); | |
133 _noPendingWritesTimer = null; | |
134 } | |
131 _onNoPendingWrites = callback; | 135 _onNoPendingWrites = callback; |
132 if (_onNoPendingWrites != null) { | 136 if (_onNoPendingWrites != null) { |
133 _socket._onWrite = _onWrite; | 137 if (_pendingWrites.isEmpty()) { |
138 _noPendingWritesTimer = new Timer(0, (t) { | |
139 if (_onNoPendingWrites != null) _onNoPendingWrites(); | |
140 }); | |
141 } else { | |
142 _setupWaitOnWrite(); | |
143 } | |
134 } | 144 } |
135 } | 145 } |
136 | 146 |
137 bool _write(List<int> buffer, int offset, int len, bool copyBuffer) { | 147 bool _write(List<int> buffer, int offset, int len, bool copyBuffer) { |
138 if (_closing || _closed) throw new StreamException("Stream closed"); | 148 if (_closing || _closed) throw new StreamException("Stream closed"); |
139 int bytesWritten = 0; | 149 int bytesWritten = 0; |
140 if (_pendingWrites.isEmpty()) { | 150 if (_pendingWrites.isEmpty()) { |
141 // If nothing is buffered write as much as possible and buffer | 151 // If nothing is buffered write as much as possible and buffer |
142 // the rest. | 152 // the rest. |
143 bytesWritten = _socket.writeList(buffer, offset, len); | 153 bytesWritten = _socket.writeList(buffer, offset, len); |
144 if (bytesWritten == len) return true; | 154 if (bytesWritten == len) return true; |
145 } | 155 } |
146 | 156 |
147 // Place remaining data on the pending writes queue. | 157 // Place remaining data on the pending writes queue. |
148 int notWrittenOffset = offset + bytesWritten; | 158 int notWrittenOffset = offset + bytesWritten; |
149 if (copyBuffer) { | 159 if (copyBuffer) { |
150 List<int> newBuffer = | 160 List<int> newBuffer = |
151 buffer.getRange(notWrittenOffset, len - bytesWritten); | 161 buffer.getRange(notWrittenOffset, len - bytesWritten); |
152 _pendingWrites.add(newBuffer); | 162 _pendingWrites.add(newBuffer); |
153 } else { | 163 } else { |
154 assert(offset + len == buffer.length); | 164 assert(offset + len == buffer.length); |
155 _pendingWrites.add(buffer, notWrittenOffset); | 165 _pendingWrites.add(buffer, notWrittenOffset); |
156 } | 166 } |
157 _socket._onWrite = _onWrite; | 167 _setupWaitOnWrite(); |
158 return false; | 168 return false; |
159 } | 169 } |
160 | 170 |
161 void _onWrite() { | 171 void _onWrite() { |
162 // Write as much buffered data to the socket as possible. | 172 // Write as much buffered data to the socket as possible. |
163 while (!_pendingWrites.isEmpty()) { | 173 while (!_pendingWrites.isEmpty()) { |
164 List<int> buffer = _pendingWrites.first; | 174 List<int> buffer = _pendingWrites.first; |
165 int offset = _pendingWrites.index; | 175 int offset = _pendingWrites.index; |
166 int bytesToWrite = buffer.length - offset; | 176 int bytesToWrite = buffer.length - offset; |
167 int bytesWritten = _socket.writeList(buffer, offset, bytesToWrite); | 177 int bytesWritten = _socket.writeList(buffer, offset, bytesToWrite); |
(...skipping 11 matching lines...) Expand all Loading... | |
179 } else { | 189 } else { |
180 if (_onNoPendingWrites != null) _onNoPendingWrites(); | 190 if (_onNoPendingWrites != null) _onNoPendingWrites(); |
181 } | 191 } |
182 if (_onNoPendingWrites == null) { | 192 if (_onNoPendingWrites == null) { |
183 _socket._onWrite = null; | 193 _socket._onWrite = null; |
184 } else { | 194 } else { |
185 _socket._onWrite = _onWrite; | 195 _socket._onWrite = _onWrite; |
186 } | 196 } |
187 } | 197 } |
188 | 198 |
199 void _setupWaitOnWrite() { | |
Mads Ager (google)
2012/05/02 08:18:17
Can we add a few comments? This sets up a write ha
Søren Gjesse
2012/05/02 10:22:45
Done.
| |
200 if (_noPendingWritesTimer != null) { | |
201 _noPendingWritesTimer.cancel(); | |
202 _noPendingWritesTimer = null; | |
203 } | |
204 _socket._onWrite = _onWrite; | |
205 } | |
206 | |
189 bool _onSocketError(e) { | 207 bool _onSocketError(e) { |
190 close(); | 208 close(); |
191 if (_onError != null) { | 209 if (_onError != null) { |
192 _onError(e); | 210 _onError(e); |
193 return true; | 211 return true; |
194 } else { | 212 } else { |
195 return false; | 213 return false; |
196 } | 214 } |
197 } | 215 } |
198 | 216 |
199 Socket _socket; | 217 Socket _socket; |
200 _BufferList _pendingWrites; | 218 _BufferList _pendingWrites; |
201 var _onNoPendingWrites; | 219 Function _onNoPendingWrites; |
220 Timer _noPendingWritesTimer; | |
202 bool _closing = false; | 221 bool _closing = false; |
203 bool _closed = false; | 222 bool _closed = false; |
204 } | 223 } |
OLD | NEW |