| 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 _CloseToken { | |
| 6 /// This token is sent from [IsolateSink]s to [IsolateStream]s to ask them to | |
| 7 /// close themselves. | |
| 8 const _CloseToken(); | |
| 9 } | |
| 10 | |
| 11 patch bool _isCloseToken(var object) { | |
| 12 // TODO(floitsch): can we compare against const _CloseToken()? | |
| 13 return object is _CloseToken; | |
| 14 } | |
| 15 | |
| 16 patch class MessageBox { | |
| 17 /* patch */ MessageBox.oneShot() : this._oneShot(new ReceivePort()); | |
| 18 MessageBox._oneShot(ReceivePort receivePort) | |
| 19 : stream = new IsolateStream._fromOriginalReceivePortOneShot(receivePort), | |
| 20 sink = new _IsolateSink._fromPort(receivePort.toSendPort()); | |
| 21 | |
| 22 /* patch */ MessageBox() : this._(new ReceivePort()); | |
| 23 MessageBox._(ReceivePort receivePort) | |
| 24 : stream = new IsolateStream._fromOriginalReceivePort(receivePort), | |
| 25 sink = new _IsolateSink._fromPort(receivePort.toSendPort()); | |
| 26 } | |
| 27 | |
| 28 class _IsolateSink implements IsolateSink { | |
| 29 bool _isClosed = false; | |
| 30 final SendPort _port; | |
| 31 _IsolateSink._fromPort(this._port); | |
| 32 | |
| 33 void add(dynamic message) { | |
| 34 _port.send(message); | |
| 35 } | |
| 36 | |
| 37 void addError(Object errorEvent) { | |
| 38 throw new UnimplementedError("addError on isolate streams"); | |
| 39 } | |
| 40 | |
| 41 void close() { | |
| 42 if (_isClosed) return; | |
| 43 add(const _CloseToken()); | |
| 44 _isClosed = true; | |
| 45 } | |
| 46 | |
| 47 bool operator==(var other) { | |
| 48 return other is IsolateSink && _port == other._port; | |
| 49 } | |
| 50 | |
| 51 int get hashCode => _port.hashCode + 499; | |
| 52 } | |
| 53 | |
| 54 patch IsolateSink streamSpawnFunction( | |
| 55 void topLevelFunction(), | |
| 56 [bool unhandledExceptionCallback(IsolateUnhandledException e)]) { | |
| 57 SendPort sendPort = spawnFunction(topLevelFunction, | |
| 58 unhandledExceptionCallback); | |
| 59 return new _IsolateSink._fromPort(sendPort); | |
| 60 } | |
| 61 | |
| 62 patch class ReceivePort { | 5 patch class ReceivePort { |
| 63 /* patch */ factory ReceivePort() { | 6 /* patch */ factory ReceivePort() { |
| 64 return new _ReceivePortImpl(); | 7 _ReceivePortImpl result = new _ReceivePortImpl(); |
| 8 // TODO(floitsch): remove the hack to close receive-ports on cancel. |
| 9 result._controller = new StreamController(onCancel: result._close); |
| 10 return result; |
| 65 } | 11 } |
| 66 } | 12 } |
| 67 | 13 |
| 68 class _ReceivePortImpl implements ReceivePort { | 14 class _ReceivePortImpl extends Stream implements ReceivePort { |
| 69 factory _ReceivePortImpl() native "ReceivePortImpl_factory"; | 15 factory _ReceivePortImpl() native "ReceivePortImpl_factory"; |
| 70 | 16 |
| 17 // Deprecated. |
| 71 receive(void onMessage(var message, SendPort replyTo)) { | 18 receive(void onMessage(var message, SendPort replyTo)) { |
| 72 _onMessage = onMessage; | 19 _onMessage = onMessage; |
| 73 } | 20 } |
| 74 | 21 |
| 22 // Deprecated. |
| 75 close() { | 23 close() { |
| 24 _close(); |
| 25 } |
| 26 |
| 27 _close() { |
| 76 _portMap.remove(_id); | 28 _portMap.remove(_id); |
| 77 _closeInternal(_id); | 29 _closeInternal(_id); |
| 78 } | 30 } |
| 79 | 31 |
| 80 SendPort toSendPort() { | 32 SendPort get sendPort { |
| 81 return new _SendPortImpl(_id); | 33 return new _SendPortImpl(_id); |
| 82 } | 34 } |
| 83 | 35 |
| 36 StreamSubscription listen(void onData(var message), |
| 37 { Function onError, |
| 38 void onDone(), |
| 39 bool cancelOnError }) { |
| 40 return _controller.stream.listen(onData); |
| 41 } |
| 42 |
| 84 /**** Internal implementation details ****/ | 43 /**** Internal implementation details ****/ |
| 85 // Called from the VM to create a new ReceivePort instance. | 44 // Called from the VM to create a new ReceivePort instance. |
| 86 static _ReceivePortImpl _get_or_create(int id) { | 45 static _ReceivePortImpl _get_or_create(int id) { |
| 87 if (_portMap != null) { | 46 if (_portMap != null) { |
| 88 _ReceivePortImpl port = _portMap[id]; | 47 _ReceivePortImpl port = _portMap[id]; |
| 89 if (port != null) { | 48 if (port != null) { |
| 90 return port; | 49 return port; |
| 91 } | 50 } |
| 92 } | 51 } |
| 93 return new _ReceivePortImpl._internal(id); | 52 return new _ReceivePortImpl._internal(id); |
| 94 } | 53 } |
| 95 | 54 |
| 96 _ReceivePortImpl._internal(int id) : _id = id { | 55 _ReceivePortImpl._internal(int id) : _id = id { |
| 97 if (_portMap == null) { | 56 if (_portMap == null) { |
| 98 _portMap = new Map(); | 57 _portMap = new Map(); |
| 99 } | 58 } |
| 100 _portMap[id] = this; | 59 _portMap[id] = this; |
| 101 } | 60 } |
| 102 | 61 |
| 103 // Called from the VM to retrieve the ReceivePort for a message. | 62 // Called from the VM to retrieve the ReceivePort for a message. |
| 104 static _ReceivePortImpl _lookupReceivePort(int id) { | 63 static _ReceivePortImpl _lookupReceivePort(int id) { |
| 105 assert(_portMap != null); | 64 assert(_portMap != null); |
| 106 return _portMap[id]; | 65 return _portMap[id]; |
| 107 } | 66 } |
| 108 | 67 |
| 109 // Called from the VM to dispatch to the handler. | 68 // Called from the VM to dispatch to the handler. |
| 110 static void _handleMessage(_ReceivePortImpl port, int replyId, var message) { | 69 static void _handleMessage(_ReceivePortImpl port, int replyId, var message) { |
| 111 assert(port != null); | 70 assert(port != null); |
| 112 SendPort replyTo = (replyId == 0) ? null : new _SendPortImpl(replyId); | 71 SendPort replyTo = (replyId == 0) ? null : new _SendPortImpl(replyId); |
| 113 (port._onMessage)(message, replyTo); | 72 if (port._onMessage != null) { |
| 73 (port._onMessage)(message, replyTo); |
| 74 } else { |
| 75 port._controller.add(message); |
| 76 } |
| 114 } | 77 } |
| 115 | 78 |
| 116 // Call into the VM to close the VM maintained mappings. | 79 // Call into the VM to close the VM maintained mappings. |
| 117 static _closeInternal(int id) native "ReceivePortImpl_closeInternal"; | 80 static _closeInternal(int id) native "ReceivePortImpl_closeInternal"; |
| 118 | 81 |
| 119 final int _id; | 82 final int _id; |
| 83 StreamController _controller; |
| 120 var _onMessage; | 84 var _onMessage; |
| 121 | 85 |
| 122 // id to ReceivePort mapping. | 86 // id to ReceivePort mapping. |
| 123 static Map _portMap; | 87 static Map _portMap; |
| 124 } | 88 } |
| 125 | 89 |
| 126 | 90 |
| 127 class _SendPortImpl implements SendPort { | 91 class _SendPortImpl implements SendPort { |
| 128 /*--- public interface ---*/ | 92 /*--- public interface ---*/ |
| 129 void send(var message, [SendPort replyTo = null]) { | 93 void send(var message, [SendPort replyTo = null]) { |
| 130 this._sendNow(message, replyTo); | 94 this._sendNow(message, replyTo); |
| 131 } | 95 } |
| 132 | 96 |
| 133 void _sendNow(var message, SendPort replyTo) { | 97 void _sendNow(var message, SendPort replyTo) { |
| 134 int replyId = (replyTo == null) ? 0 : replyTo._id; | 98 int replyId = (replyTo == null) ? 0 : replyTo._id; |
| 135 _sendInternal(_id, replyId, message); | 99 _sendInternal(_id, replyId, message); |
| 136 } | 100 } |
| 137 | 101 |
| 102 /// Deprecated. |
| 138 Future call(var message) { | 103 Future call(var message) { |
| 139 final completer = new Completer.sync(); | 104 final completer = new Completer.sync(); |
| 140 final port = new _ReceivePortImpl(); | 105 final port = new _ReceivePortImpl(); |
| 141 send(message, port.toSendPort()); | 106 send(message, port.toSendPort()); |
| 142 port.receive((value, ignoreReplyTo) { | 107 port.receive((value, ignoreReplyTo) { |
| 143 port.close(); | 108 port.close(); |
| 144 if (value is Exception) { | 109 if (value is Exception) { |
| 145 completer.completeError(value); | 110 completer.completeError(value); |
| 146 } else { | 111 } else { |
| 147 completer.complete(value); | 112 completer.complete(value); |
| (...skipping 24 matching lines...) Expand all Loading... |
| 172 static _sendInternal(int sendId, int replyId, var message) | 137 static _sendInternal(int sendId, int replyId, var message) |
| 173 native "SendPortImpl_sendInternal_"; | 138 native "SendPortImpl_sendInternal_"; |
| 174 | 139 |
| 175 final int _id; | 140 final int _id; |
| 176 } | 141 } |
| 177 | 142 |
| 178 _getPortInternal() native "isolate_getPortInternal"; | 143 _getPortInternal() native "isolate_getPortInternal"; |
| 179 | 144 |
| 180 ReceivePort _portInternal; | 145 ReceivePort _portInternal; |
| 181 | 146 |
| 147 typedef _ZeroArgFunction(); |
| 148 |
| 149 /** |
| 150 * Takes the real entry point as argument and invokes it with the initial |
| 151 * message. |
| 152 * |
| 153 * The initial message is (currently) received through the global port variable. |
| 154 */ |
| 155 void _startIsolate(Function entryPoint) { |
| 156 bool first = true; |
| 157 _Isolate.port.receive((message, replyTo) { |
| 158 assert(first); |
| 159 first = false; |
| 160 var initialMessage = message[0]; |
| 161 var reply = message[1]; |
| 162 reply.send("started"); |
| 163 if (entryPoint is _ZeroArgFunction) { |
| 164 entryPoint(); |
| 165 } else { |
| 166 entryPoint(initialMessage); |
| 167 } |
| 168 }); |
| 169 } |
| 170 |
| 171 patch class Isolate { |
| 172 /* patch */ static Future<Isolate> spawn( |
| 173 void entryPoint(message), var message, { bool startPaused: false }) { |
| 174 if (startPaused) throw new UnimplementedError("spawn paused isolate"); |
| 175 return new Future<Isolate>.sync(() { |
| 176 // The VM will invoke [_startIsolate] with entryPoint as argument. |
| 177 SendPort controlPort = _Isolate._spawnFunction(entryPoint); |
| 178 ReceivePort readyPort = new ReceivePort(); |
| 179 controlPort.send([message, readyPort.sendPort]); |
| 180 Completer completer = new Completer<Isolate>(); |
| 181 readyPort.receive((ignored1, ignored2) { |
| 182 readyPort.close(); |
| 183 completer.complete(new Isolate._fromControlPort(controlPort)); |
| 184 }); |
| 185 return completer.future; |
| 186 }); |
| 187 } |
| 188 |
| 189 /* patch */ static Future<Isolate> spawnUri(Uri uri, var message) { |
| 190 throw new UnimplementedError("Isolate.spawnUri"); |
| 191 } |
| 192 } |
| 193 |
| 182 patch class _Isolate { | 194 patch class _Isolate { |
| 183 /* patch */ static ReceivePort get port { | 195 /* patch */ static ReceivePort get port { |
| 184 if (_portInternal == null) { | 196 if (_portInternal == null) { |
| 185 _portInternal = _getPortInternal(); | 197 _portInternal = _getPortInternal(); |
| 186 } | 198 } |
| 187 return _portInternal; | 199 return _portInternal; |
| 188 } | 200 } |
| 189 | 201 |
| 190 /* patch */ static SendPort spawnFunction(void topLevelFunction(), | 202 static SendPort _spawnFunction(Function topLevelFunction) |
| 191 [bool unhandledExceptionCallback(IsolateUnhandledException e)]) | |
| 192 native "isolate_spawnFunction"; | 203 native "isolate_spawnFunction"; |
| 193 | 204 |
| 194 /* patch */ static SendPort spawnUri(String uri) native "isolate_spawnUri"; | 205 /* patch */ static SendPort spawnUri(String uri) native "isolate_spawnUri"; |
| 195 } | 206 } |
| OLD | NEW |