| 1 | | | import 'dart:async'; |
| 2 | | | import 'dart:convert'; |
| 3 | | | import 'dart:isolate' as vm; |
| 4 | | | |
| 5 | | | import 'package:logger/web.dart'; |
| 6 | | | import 'package:meta/meta.dart'; |
| 7 | | | |
| 8 | | | import '../../channel.dart'; |
| 9 | | | import '../../exceptions/exception_manager.dart'; |
| 10 | | | import '../../exceptions/squadron_error.dart'; |
| 11 | | | import '../../exceptions/squadron_exception.dart'; |
| 12 | | | import '../../exceptions/task_terminated_exception.dart'; |
| 13 | | | import '../../exceptions/worker_exception.dart'; |
| 14 | | | import '../../tokens/_squadron_cancelation_token.dart'; |
| 15 | | | import '../../utils.dart'; |
| 16 | | | import '../../worker/worker_message.dart'; |
| 17 | | | import '../../worker/worker_request.dart'; |
| 18 | | | import '../../worker/worker_response.dart'; |
| 19 | | | import '../xplat/_disconnected_channel.dart'; |
| 20 | | | import '../xplat/_forward_stream_controller.dart'; |
| 21 | | | import '../xplat/_result_stream.dart'; |
| 22 | | | import '_typedefs.dart'; |
| 23 | | | |
| 24 | | | part '_channel_forward.dart'; |
| 25 | | | part '_channel_impl.dart'; |
| 26 | | | |
| 27 | | | // Stub implementations. |
| 28 | | | |
| 29 | | | /// Starts an [Isolate] using the [entryPoint] and sends a start |
| 30 | | | /// [WorkerRequest] with [startArguments]. The future completes after the |
| 31 | | | /// worker [Isolate]'s main program has provided the [SendPort] via |
| 32 | | | /// [WorkerChannel.connect]. |
| 33 | | 11 | Future<Channel> openChannel( |
| 34 | | | EntryPoint entryPoint, |
| 35 | | | ExceptionManager exceptionManager, |
| 36 | | | Logger? logger, |
| 37 | | | List startArguments, |
| 38 | | | PlatformThreadHook? hook, |
| 39 | | | ) async { |
| 40 | | 11 | final completer = Completer<_VmChannel>(); |
| 41 | | | Channel? channel; |
| 42 | | | |
| 43 | | 11 | void $failure(Object error, [StackTrace? stackTrace]) { |
| 44 | | 11 | if (!completer.isCompleted) { |
| 45 | | 6 | completer.completeError(SquadronException.from(error, stackTrace)); |
| 46 | | | } |
| 47 | | | } |
| 48 | | | |
| 49 | | 10 | void $success(_VmChannel channel) { |
| 50 | | 10 | if (!completer.isCompleted) { |
| 51 | | 10 | completer.complete(channel); |
| 52 | | | } |
| 53 | | | } |
| 54 | | | |
| 55 | | 11 | final receiver = vm.ReceivePort(); |
| 56 | | 11 | final exitPort = vm.ReceivePort(); |
| 57 | | 11 | final errorPort = vm.ReceivePort(); |
| 58 | | | |
| 59 | | 22 | exitPort.listen((message) { |
| 60 | | 22 | $failure(SquadronErrorImpl.create('Connection to worker failed')); |
| 61 | | 6 | logger?.t('Isolate terminated with message $message.'); |
| 62 | | 10 | channel?.close(); |
| 63 | | 11 | receiver.close(); |
| 64 | | 11 | errorPort.close(); |
| 65 | | 11 | exitPort.close(); |
| 66 | | | }); |
| 67 | | | |
| 68 | | 11 | errorPort.listen((message) { |
| 69 | | | SquadronException? error; |
| 70 | | | try { |
| 71 | | 0 | final data = jsonDecode(message[0]); |
| 72 | | 0 | if (data is List) { |
| 73 | | 0 | error = exceptionManager.deserialize(data.cast<String>()); |
| 74 | | | } |
| 75 | | | } catch (_) { |
| 76 | | | // not a String representing a SquadronException |
| 77 | | | } |
| 78 | | | |
| 79 | | 0 | error ??= WorkerException( |
| 80 | | 0 | message[0], |
| 81 | | 0 | SquadronException.loadStackTrace(message[1]), |
| 82 | | | ); |
| 83 | | | |
| 84 | | 0 | logger?.d(() => 'Unhandled error from Isolate: ${error?.message}.'); |
| 85 | | 0 | $failure(error); |
| 86 | | | }); |
| 87 | | | |
| 88 | | 11 | final disconnected = DisconnectedChannel(exceptionManager, logger); |
| 89 | | | |
| 90 | | 21 | receiver.listen((message) { |
| 91 | | 10 | final response = WorkerResponse.from(message); |
| 92 | | 10 | if (!response.unwrapInPlace(disconnected)) { |
| 93 | | | return; |
| 94 | | | } |
| 95 | | | |
| 96 | | 10 | final error = response.error; |
| 97 | | | if (error != null) { |
| 98 | | 3 | logger?.e(() => 'Connection to Isolate failed: $error'); |
| 99 | | 2 | $failure(error); |
| 100 | | 10 | } else if (response.endOfStream) { |
| 101 | | 1 | logger?.w('Disconnecting from Isolate'); |
| 102 | | 1 | channel?.close(); |
| 103 | | 10 | } else if (!completer.isCompleted) { |
| 104 | | 3 | logger?.t('Connected to Isolate'); |
| 105 | | | final platformChannel = |
| 106 | | 20 | _VmChannel._(response.result, logger, exceptionManager); |
| 107 | | | channel = platformChannel; |
| 108 | | 10 | $success(platformChannel); |
| 109 | | | } else { |
| 110 | | 0 | logger?.e(() => 'Unexpected response: $response'); |
| 111 | | | } |
| 112 | | | }); |
| 113 | | | |
| 114 | | 22 | final startRequest = WorkerRequest.start(receiver.sendPort, startArguments); |
| 115 | | 11 | startRequest.wrapInPlace(); |
| 116 | | 11 | final isolate = await vm.Isolate.spawn( |
| 117 | | | entryPoint, |
| 118 | | | startRequest, |
| 119 | | | errorsAreFatal: false, |
| 120 | | 11 | onExit: exitPort.sendPort, |
| 121 | | 11 | onError: errorPort.sendPort, |
| 122 | | | ); |
| 123 | | | |
| 124 | | | try { |
| 125 | | 11 | final channel = await completer.future; |
| 126 | | 10 | channel._thread = isolate; |
| 127 | | | if (hook != null) { |
| 128 | | 2 | await hook.call(isolate); |
| 129 | | | } |
| 130 | | 3 | logger?.t('Created Isolate'); |
| 131 | | | return channel; |
| 132 | | | } catch (ex) { |
| 133 | | 0 | logger?.e(() => 'Connection to Isolate failed: $ex'); |
| 134 | | 3 | isolate.kill(priority: vm.Isolate.beforeNextEvent); |
| 135 | | | rethrow; |
| 136 | | | } |
| 137 | | | } |
| 138 | | | |
| 139 | | 6 | @internal |
| 140 | | | int getActiveConnections(Channel channel) => |
| 141 | | 18 | (channel is _VmChannel) ? channel._activeConnections.length : 0; |
| 142 | | | |
| 143 | | 2 | @internal |
| 144 | | | void terminateChannel(Channel channel, TaskTerminatedException ex) { |
| 145 | | 2 | if (channel is _VmChannel) { |
| 146 | | 4 | channel._thread?.kill(priority: vm.Isolate.immediate); |
| 147 | | 2 | final pendingConnections = channel._activeConnections; |
| 148 | | 4 | for (var c in pendingConnections) { |
| 149 | | 4 | c.subscription?.cancel(); |
| 150 | | 2 | c.safeAddError(ex); |
| 151 | | 2 | c.close(); |
| 152 | | | } |
| 153 | | | } |
| 154 | | | } |
| 155 | | | |
| 156 | | | /// Creates a [_VmChannel] from a [SendPort]. |
| 157 | | 3 | Channel? deserialize( |
| 158 | | | PlatformChannel? channelInfo, [ |
| 159 | | | Logger? logger, |
| 160 | | | ExceptionManager? exceptionManager, |
| 161 | | | ]) => |
| 162 | | | (channelInfo == null) |
| 163 | | | ? null |
| 164 | | 3 | : _VmChannel._( |
| 165 | | | channelInfo, |
| 166 | | | logger, |
| 167 | | 3 | exceptionManager ?? ExceptionManager(), |
| 168 | | | ); |