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