blob: c818a59115c5324852f50d961225f5ad2babed64 [file] [log] [blame]
// Copyright (c) 2016, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
import "dart:async";
import "dart:isolate";
import "package:async/async.dart";
import "package:stack_trace/stack_trace.dart";
import "package:stream_channel/isolate_channel.dart";
import "package:stream_channel/stream_channel.dart";
import "package:test_api/src/util/remote_exception.dart"; // ignore: implementation_imports
import 'package:test_api/src/utils.dart'; // ignore: implementation_imports
/// A sink transformer that wraps data and error events so that errors can be
/// decoded after being JSON-serialized.
final _transformer = StreamSinkTransformer<dynamic, dynamic>.fromHandlers(
handleData: (data, sink) {
sink.add({"type": "data", "data": data});
}, handleError: (error, stackTrace, sink) {
{"type": "error", "error": RemoteException.serialize(error, stackTrace)});
/// Runs the body of a hybrid isolate and communicates its messages, errors, and
/// prints to the main isolate.
/// The [getMain] function returns the `hybridMain()` method. It's wrapped in a
/// closure so that, if the method undefined, we can catch the error and notify
/// the caller of it.
/// The [data] argument contains two values: a [SendPort] that communicates with
/// the main isolate, and a message to pass to `hybridMain()`.
void listen(Function getMain(), List data) {
var channel = IsolateChannel.connectSend(data.first as SendPort);
var message = data.last;
Chain.capture(() {
runZoned(() {
dynamic /*Function*/ main;
try {
main = getMain();
} on NoSuchMethodError catch (_) {
_sendError(channel, "No top-level hybridMain() function defined.");
} catch (error, stackTrace) {
_sendError(channel, error, stackTrace);
if (main is! Function) {
_sendError(channel, "Top-level hybridMain is not a function.");
} else if (main is! Function(StreamChannel) &&
main is! Function(StreamChannel, Null)) {
"Top-level hybridMain() function must take one or two arguments.");
// Wrap [channel] before passing it to user code so that we can wrap
// errors and distinguish user data events from control events sent by the
// listener.
var transformedChannel = channel.transformSink(_transformer);
if (main is Function(StreamChannel)) {
} else {
main(transformedChannel, message);
}, zoneSpecification: ZoneSpecification(print: (_, __, ___, line) {
channel.sink.add({"type": "print", "line": line});
}, onError: (error, stackTrace) async {
_sendError(channel, error, stackTrace);
await channel.sink.close();
Zone.current.handleUncaughtError(error, stackTrace);
/// Sends a message over [channel] indicating an error from user code.
void _sendError(StreamChannel channel, error, [StackTrace stackTrace]) {
"type": "error",
"error": RemoteException.serialize(error, stackTrace ?? Chain.current())