Merge Worker and Isolate types (#2078)

Reduces generics.
This commit is contained in:
Ryan Dahl 2019-04-08 17:10:00 -04:00 committed by GitHub
parent 734cf781c6
commit 2debbdacb9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 359 additions and 631 deletions

View file

@ -17,24 +17,12 @@ impl CliBehavior {
} }
} }
impl IsolateStateContainer for &CliBehavior {
fn state(&self) -> Arc<IsolateState> {
self.state.clone()
}
}
impl IsolateStateContainer for CliBehavior {
fn state(&self) -> Arc<IsolateState> {
self.state.clone()
}
}
impl Behavior for CliBehavior { impl Behavior for CliBehavior {
fn dispatch( fn dispatch(
&mut self, &mut self,
control: &[u8], control: &[u8],
zero_copy: deno_buf, zero_copy: deno_buf,
) -> (bool, Box<Op>) { ) -> (bool, Box<Op>) {
ops::dispatch_all(self, control, zero_copy, ops::op_selector_std) ops::dispatch_all(&self.state, control, zero_copy, ops::op_selector_std)
} }
} }

View file

@ -1,22 +1,17 @@
// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license. // Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
use crate::flags::DenoFlags; use crate::cli_behavior::CliBehavior;
use crate::isolate_state::*; use crate::isolate_state::*;
use crate::js_errors; use crate::js_errors;
use crate::js_errors::JSErrorColor; use crate::js_errors::JSErrorColor;
use crate::msg; use crate::msg;
use crate::ops;
use crate::resources; use crate::resources;
use crate::resources::ResourceId; use crate::resources::ResourceId;
use crate::startup_data; use crate::startup_data;
use crate::tokio_util; use crate::tokio_util;
use crate::workers; use crate::worker::Worker;
use crate::workers::WorkerBehavior; use deno::js_check;
use crate::workers::WorkerInit;
use deno::deno_buf;
use deno::Behavior;
use deno::Buf; use deno::Buf;
use deno::JSError; use deno::JSError;
use deno::Op;
use futures::future::*; use futures::future::*;
use futures::sync::oneshot; use futures::sync::oneshot;
use futures::Future; use futures::Future;
@ -44,51 +39,6 @@ lazy_static! {
static ref C_RUNTIME: Mutex<Runtime> = Mutex::new(Runtime::new().unwrap()); static ref C_RUNTIME: Mutex<Runtime> = Mutex::new(Runtime::new().unwrap());
} }
pub struct CompilerBehavior {
pub state: Arc<IsolateState>,
}
impl CompilerBehavior {
pub fn new(flags: DenoFlags, argv_rest: Vec<String>) -> Self {
Self {
state: Arc::new(IsolateState::new(flags, argv_rest, None, true)),
}
}
}
impl IsolateStateContainer for CompilerBehavior {
fn state(&self) -> Arc<IsolateState> {
self.state.clone()
}
}
impl IsolateStateContainer for &CompilerBehavior {
fn state(&self) -> Arc<IsolateState> {
self.state.clone()
}
}
impl Behavior for CompilerBehavior {
fn dispatch(
&mut self,
control: &[u8],
zero_copy: deno_buf,
) -> (bool, Box<Op>) {
ops::dispatch_all(self, control, zero_copy, ops::op_selector_compiler)
}
}
impl WorkerBehavior for CompilerBehavior {
fn set_internal_channels(&mut self, worker_channels: WorkerChannels) {
self.state = Arc::new(IsolateState::new(
self.state.flags.clone(),
self.state.argv.clone(),
Some(worker_channels),
true,
));
}
}
// This corresponds to JS ModuleMetaData. // This corresponds to JS ModuleMetaData.
// TODO Rename one or the other so they correspond. // TODO Rename one or the other so they correspond.
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
@ -142,21 +92,26 @@ fn lazy_start(parent_state: Arc<IsolateState>) -> ResourceId {
let mut cell = C_RID.lock().unwrap(); let mut cell = C_RID.lock().unwrap();
cell cell
.get_or_insert_with(|| { .get_or_insert_with(|| {
let worker_result = workers::spawn( let child_state = Arc::new(IsolateState::new(
startup_data::compiler_isolate_init(),
CompilerBehavior::new(
parent_state.flags.clone(), parent_state.flags.clone(),
parent_state.argv.clone(), parent_state.argv.clone(),
), ));
"TS", let rid = child_state.resource.rid;
WorkerInit::Script("compilerMain()".to_string()), let resource = child_state.resource.clone();
let behavior = CliBehavior::new(child_state);
let mut worker = Worker::new(
"TS".to_string(),
startup_data::compiler_isolate_init(),
behavior,
); );
match worker_result {
Ok(worker) => { js_check(worker.execute("denoMain()"));
let rid = worker.resource.rid; js_check(worker.execute("workerMain()"));
js_check(worker.execute("compilerMain()"));
let mut runtime = C_RUNTIME.lock().unwrap(); let mut runtime = C_RUNTIME.lock().unwrap();
runtime.spawn(lazy(move || { runtime.spawn(lazy(move || {
let resource = worker.resource.clone();
worker.then(move |result| -> Result<(), ()> { worker.then(move |result| -> Result<(), ()> {
// Close resource so the future created by // Close resource so the future created by
// handle_worker_message_stream exits // handle_worker_message_stream exits
@ -191,25 +146,13 @@ fn lazy_start(parent_state: Arc<IsolateState>) -> ResourceId {
}).map_err(|_| ()) }).map_err(|_| ())
})); }));
rid rid
}
Err(err) => {
println!("{}", err.to_string());
std::process::exit(1);
}
}
}).clone() }).clone()
} }
fn req( fn req(specifier: &str, referrer: &str, cmd_id: u32) -> Buf {
specifier: &str,
referrer: &str,
is_worker_main: bool,
cmd_id: u32,
) -> Buf {
json!({ json!({
"specifier": specifier, "specifier": specifier,
"referrer": referrer, "referrer": referrer,
"isWorker": is_worker_main,
"cmdId": cmd_id, "cmdId": cmd_id,
}).to_string() }).to_string()
.into_boxed_str() .into_boxed_str()
@ -228,7 +171,7 @@ pub fn compile_async(
); );
let cmd_id = new_cmd_id(); let cmd_id = new_cmd_id();
let req_msg = req(&specifier, &referrer, parent_state.is_worker, cmd_id); let req_msg = req(&specifier, &referrer, cmd_id);
let module_meta_data_ = module_meta_data.clone(); let module_meta_data_ = module_meta_data.clone();
let compiler_rid = lazy_start(parent_state.clone()); let compiler_rid = lazy_start(parent_state.clone());
@ -362,7 +305,7 @@ mod tests {
fn test_parse_cmd_id() { fn test_parse_cmd_id() {
let cmd_id = new_cmd_id(); let cmd_id = new_cmd_id();
let msg = req("Hello", "World", false, cmd_id); let msg = req("Hello", "World", cmd_id);
let res_json = std::str::from_utf8(&msg).unwrap(); let res_json = std::str::from_utf8(&msg).unwrap();

View file

@ -5,25 +5,23 @@ use crate::flags;
use crate::global_timer::GlobalTimer; use crate::global_timer::GlobalTimer;
use crate::modules::Modules; use crate::modules::Modules;
use crate::permissions::DenoPermissions; use crate::permissions::DenoPermissions;
use crate::resources;
use crate::resources::ResourceId; use crate::resources::ResourceId;
use crate::workers::UserWorkerBehavior; use crate::worker::Worker;
use crate::workers::Worker;
use deno::Buf; use deno::Buf;
use futures::future::Shared; use futures::future::Shared;
use futures::sync::mpsc as async_mpsc;
use std; use std;
use std::collections::HashMap; use std::collections::HashMap;
use std::env; use std::env;
use std::sync::atomic::{AtomicUsize, Ordering}; use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::Arc;
use std::sync::Mutex; use std::sync::Mutex;
use std::time::Instant; use std::time::Instant;
use tokio::sync::mpsc as async_mpsc;
pub type WorkerSender = async_mpsc::Sender<Buf>; pub type WorkerSender = async_mpsc::Sender<Buf>;
pub type WorkerReceiver = async_mpsc::Receiver<Buf>; pub type WorkerReceiver = async_mpsc::Receiver<Buf>;
pub type WorkerChannels = (WorkerSender, WorkerReceiver); pub type WorkerChannels = (WorkerSender, WorkerReceiver);
pub type UserWorkerTable = pub type UserWorkerTable = HashMap<ResourceId, Shared<Worker>>;
HashMap<ResourceId, Shared<Worker<UserWorkerBehavior>>>;
// AtomicU64 is currently unstable // AtomicU64 is currently unstable
#[derive(Default)] #[derive(Default)]
@ -48,22 +46,23 @@ pub struct IsolateState {
pub flags: flags::DenoFlags, pub flags: flags::DenoFlags,
pub metrics: Metrics, pub metrics: Metrics,
pub modules: Mutex<Modules>, pub modules: Mutex<Modules>,
pub worker_channels: Option<Mutex<WorkerChannels>>, pub worker_channels: Mutex<WorkerChannels>,
pub global_timer: Mutex<GlobalTimer>, pub global_timer: Mutex<GlobalTimer>,
pub workers: Mutex<UserWorkerTable>, pub workers: Mutex<UserWorkerTable>,
pub is_worker: bool,
pub start_time: Instant, pub start_time: Instant,
pub resource: resources::Resource,
} }
impl IsolateState { impl IsolateState {
pub fn new( pub fn new(flags: flags::DenoFlags, argv_rest: Vec<String>) -> Self {
flags: flags::DenoFlags,
argv_rest: Vec<String>,
worker_channels: Option<WorkerChannels>,
is_worker: bool,
) -> Self {
let custom_root = env::var("DENO_DIR").map(|s| s.into()).ok(); let custom_root = env::var("DENO_DIR").map(|s| s.into()).ok();
let (worker_in_tx, worker_in_rx) = async_mpsc::channel::<Buf>(1);
let (worker_out_tx, worker_out_rx) = async_mpsc::channel::<Buf>(1);
let internal_channels = (worker_out_tx, worker_in_rx);
let external_channels = (worker_in_tx, worker_out_rx);
let resource = resources::add_worker(external_channels);
Self { Self {
dir: deno_dir::DenoDir::new(custom_root).unwrap(), dir: deno_dir::DenoDir::new(custom_root).unwrap(),
argv: argv_rest, argv: argv_rest,
@ -71,11 +70,11 @@ impl IsolateState {
flags, flags,
metrics: Metrics::default(), metrics: Metrics::default(),
modules: Mutex::new(Modules::new()), modules: Mutex::new(Modules::new()),
worker_channels: worker_channels.map(Mutex::new), worker_channels: Mutex::new(internal_channels),
global_timer: Mutex::new(GlobalTimer::new()), global_timer: Mutex::new(GlobalTimer::new()),
workers: Mutex::new(UserWorkerTable::new()), workers: Mutex::new(UserWorkerTable::new()),
is_worker,
start_time: Instant::now(), start_time: Instant::now(),
resource,
} }
} }
@ -126,7 +125,7 @@ impl IsolateState {
let argv = vec![String::from("./deno"), String::from("hello.js")]; let argv = vec![String::from("./deno"), String::from("hello.js")];
// For debugging: argv.push_back(String::from("-D")); // For debugging: argv.push_back(String::from("-D"));
let (flags, rest_argv) = flags::set_flags(argv).unwrap(); let (flags, rest_argv) = flags::set_flags(argv).unwrap();
IsolateState::new(flags, rest_argv, None, false) IsolateState::new(flags, rest_argv)
} }
pub fn metrics_op_dispatched( pub fn metrics_op_dispatched(
@ -153,8 +152,3 @@ impl IsolateState {
.fetch_add(bytes_received, Ordering::SeqCst); .fetch_add(bytes_received, Ordering::SeqCst);
} }
} }
/// Provides state getter function
pub trait IsolateStateContainer {
fn state(&self) -> Arc<IsolateState>;
}

View file

@ -20,7 +20,6 @@ mod fs;
mod global_timer; mod global_timer;
mod http_body; mod http_body;
mod http_util; mod http_util;
pub mod isolate;
pub mod isolate_state; pub mod isolate_state;
pub mod js_errors; pub mod js_errors;
pub mod modules; pub mod modules;
@ -35,12 +34,12 @@ mod startup_data;
mod tokio_util; mod tokio_util;
mod tokio_write; mod tokio_write;
pub mod version; pub mod version;
pub mod workers; pub mod worker;
use crate::cli_behavior::CliBehavior; use crate::cli_behavior::CliBehavior;
use crate::errors::RustOrJsError; use crate::errors::RustOrJsError;
use crate::isolate::Isolate;
use crate::isolate_state::IsolateState; use crate::isolate_state::IsolateState;
use crate::worker::Worker;
use futures::lazy; use futures::lazy;
use futures::Future; use futures::Future;
use log::{LevelFilter, Metadata, Record}; use log::{LevelFilter, Metadata, Record};
@ -105,27 +104,28 @@ fn main() {
let should_prefetch = flags.prefetch || flags.info; let should_prefetch = flags.prefetch || flags.info;
let should_display_info = flags.info; let should_display_info = flags.info;
let state = Arc::new(IsolateState::new(flags, rest_argv, None, false)); let state = Arc::new(IsolateState::new(flags, rest_argv));
let state_ = state.clone(); let state_ = state.clone();
let cli = CliBehavior::new(state_); let cli = CliBehavior::new(state_);
let mut isolate = Isolate::new(startup_data::deno_isolate_init(), cli); let mut main_worker =
Worker::new("main".to_string(), startup_data::deno_isolate_init(), cli);
let main_future = lazy(move || { let main_future = lazy(move || {
// Setup runtime. // Setup runtime.
js_check(isolate.execute("denoMain()")); js_check(main_worker.execute("denoMain()"));
// Execute main module. // Execute main module.
if let Some(main_module) = state.main_module() { if let Some(main_module) = state.main_module() {
debug!("main_module {}", main_module); debug!("main_module {}", main_module);
js_check(isolate.execute_mod(&main_module, should_prefetch)); js_check(main_worker.execute_mod(&main_module, should_prefetch));
if should_display_info { if should_display_info {
// Display file info and exit. Do not run file // Display file info and exit. Do not run file
isolate.print_file_info(&main_module); main_worker.print_file_info(&main_module);
std::process::exit(0); std::process::exit(0);
} }
} }
isolate.then(|result| { main_worker.then(|result| {
js_check(result); js_check(result);
Ok(()) Ok(())
}) })

View file

@ -1,11 +1,12 @@
// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license. // Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
use atty; use atty;
use crate::ansi; use crate::ansi;
use crate::cli_behavior::CliBehavior;
use crate::errors; use crate::errors;
use crate::errors::{DenoError, DenoResult, ErrorKind}; use crate::errors::{DenoError, DenoResult, ErrorKind};
use crate::fs as deno_fs; use crate::fs as deno_fs;
use crate::http_util; use crate::http_util;
use crate::isolate_state::{IsolateState, IsolateStateContainer}; use crate::isolate_state::IsolateState;
use crate::js_errors::apply_source_map; use crate::js_errors::apply_source_map;
use crate::js_errors::JSErrorColor; use crate::js_errors::JSErrorColor;
use crate::msg; use crate::msg;
@ -19,8 +20,9 @@ use crate::startup_data;
use crate::tokio_util; use crate::tokio_util;
use crate::tokio_write; use crate::tokio_write;
use crate::version; use crate::version;
use crate::workers; use crate::worker::Worker;
use deno::deno_buf; use deno::deno_buf;
use deno::js_check;
use deno::Buf; use deno::Buf;
use deno::JSError; use deno::JSError;
use deno::Op; use deno::Op;
@ -60,7 +62,7 @@ pub type OpWithError = dyn Future<Item = Buf, Error = DenoError> + Send;
// TODO Ideally we wouldn't have to box the OpWithError being returned. // TODO Ideally we wouldn't have to box the OpWithError being returned.
// The box is just to make it easier to get a prototype refactor working. // The box is just to make it easier to get a prototype refactor working.
type OpCreator = type OpCreator =
fn(sc: &IsolateStateContainer, base: &msg::Base<'_>, data: deno_buf) fn(state: &Arc<IsolateState>, base: &msg::Base<'_>, data: deno_buf)
-> Box<OpWithError>; -> Box<OpWithError>;
type OpSelector = fn(inner_type: msg::Any) -> Option<OpCreator>; type OpSelector = fn(inner_type: msg::Any) -> Option<OpCreator>;
@ -75,7 +77,7 @@ fn empty_buf() -> Buf {
/// control corresponds to the first argument of Deno.core.dispatch(). /// control corresponds to the first argument of Deno.core.dispatch().
/// data corresponds to the second argument of Deno.core.dispatch(). /// data corresponds to the second argument of Deno.core.dispatch().
pub fn dispatch_all( pub fn dispatch_all(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
control: &[u8], control: &[u8],
zero_copy: deno_buf, zero_copy: deno_buf,
op_selector: OpSelector, op_selector: OpSelector,
@ -92,10 +94,9 @@ pub fn dispatch_all(
None => panic!("Unhandled message {}", msg::enum_name_any(inner_type)), None => panic!("Unhandled message {}", msg::enum_name_any(inner_type)),
}; };
let state = sc.state().clone(); let op: Box<OpWithError> = op_func(state, &base, zero_copy);
let op: Box<OpWithError> = op_func(sc, &base, zero_copy);
let state = state.clone();
state.metrics_op_dispatched(bytes_sent_control, bytes_sent_zero_copy); state.metrics_op_dispatched(bytes_sent_control, bytes_sent_zero_copy);
let boxed_op = Box::new( let boxed_op = Box::new(
@ -143,23 +144,6 @@ pub fn dispatch_all(
(base.sync(), boxed_op) (base.sync(), boxed_op)
} }
/// Superset of op_selector_worker for compiler isolates
pub fn op_selector_compiler(inner_type: msg::Any) -> Option<OpCreator> {
match inner_type {
msg::Any::FetchModuleMetaData => Some(op_fetch_module_meta_data),
_ => op_selector_worker(inner_type),
}
}
/// Superset of op_selector_std for worker isolates
pub fn op_selector_worker(inner_type: msg::Any) -> Option<OpCreator> {
match inner_type {
msg::Any::WorkerGetMessage => Some(op_worker_get_message),
msg::Any::WorkerPostMessage => Some(op_worker_post_message),
_ => op_selector_std(inner_type),
}
}
/// Standard ops set for most isolates /// Standard ops set for most isolates
pub fn op_selector_std(inner_type: msg::Any) -> Option<OpCreator> { pub fn op_selector_std(inner_type: msg::Any) -> Option<OpCreator> {
match inner_type { match inner_type {
@ -208,6 +192,14 @@ pub fn op_selector_std(inner_type: msg::Any) -> Option<OpCreator> {
msg::Any::HostGetMessage => Some(op_host_get_message), msg::Any::HostGetMessage => Some(op_host_get_message),
msg::Any::HostPostMessage => Some(op_host_post_message), msg::Any::HostPostMessage => Some(op_host_post_message),
msg::Any::Write => Some(op_write), msg::Any::Write => Some(op_write),
// TODO(ry) split these out so that only the appropriate Workers can access
// them. Only the compiler worker should be able to access
// FetchModuleMetaData.
msg::Any::FetchModuleMetaData => Some(op_fetch_module_meta_data),
msg::Any::WorkerGetMessage => Some(op_worker_get_message),
msg::Any::WorkerPostMessage => Some(op_worker_post_message),
_ => None, _ => None,
} }
} }
@ -217,19 +209,19 @@ pub fn op_selector_std(inner_type: msg::Any) -> Option<OpCreator> {
// If the High precision flag is not set, the // If the High precision flag is not set, the
// nanoseconds are rounded on 2ms. // nanoseconds are rounded on 2ms.
fn op_now( fn op_now(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
let seconds = sc.state().start_time.elapsed().as_secs(); let seconds = state.start_time.elapsed().as_secs();
let mut subsec_nanos = sc.state().start_time.elapsed().subsec_nanos(); let mut subsec_nanos = state.start_time.elapsed().subsec_nanos();
let reduced_time_precision = 2000000; // 2ms in nanoseconds let reduced_time_precision = 2000000; // 2ms in nanoseconds
// If the permission is not enabled // If the permission is not enabled
// Round the nano result on 2 milliseconds // Round the nano result on 2 milliseconds
// see: https://developer.mozilla.org/en-US/docs/Web/API/DOMHighResTimeStamp#Reduced_time_precision // see: https://developer.mozilla.org/en-US/docs/Web/API/DOMHighResTimeStamp#Reduced_time_precision
if !sc.state().permissions.allows_high_precision() { if !state.permissions.allows_high_precision() {
subsec_nanos -= subsec_nanos % reduced_time_precision subsec_nanos -= subsec_nanos % reduced_time_precision
} }
@ -253,7 +245,7 @@ fn op_now(
} }
fn op_is_tty( fn op_is_tty(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
_data: deno_buf, _data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -278,7 +270,7 @@ fn op_is_tty(
} }
fn op_exit( fn op_exit(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
_data: deno_buf, _data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -287,14 +279,14 @@ fn op_exit(
} }
fn op_start( fn op_start(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
let mut builder = FlatBufferBuilder::new(); let mut builder = FlatBufferBuilder::new();
let state = sc.state(); let state = state;
let argv = state.argv.iter().map(|s| s.as_str()).collect::<Vec<_>>(); let argv = state.argv.iter().map(|s| s.as_str()).collect::<Vec<_>>();
let argv_off = builder.create_vector_of_strings(argv.as_slice()); let argv_off = builder.create_vector_of_strings(argv.as_slice());
@ -311,7 +303,7 @@ fn op_start(
let deno_version = version::DENO; let deno_version = version::DENO;
let deno_version_off = builder.create_string(deno_version); let deno_version_off = builder.create_string(deno_version);
let main_module = sc.state().main_module().map(|m| builder.create_string(&m)); let main_module = state.main_module().map(|m| builder.create_string(&m));
let inner = msg::StartRes::create( let inner = msg::StartRes::create(
&mut builder, &mut builder,
@ -320,9 +312,9 @@ fn op_start(
pid: std::process::id(), pid: std::process::id(),
argv: Some(argv_off), argv: Some(argv_off),
main_module, main_module,
debug_flag: sc.state().flags.log_debug, debug_flag: state.flags.log_debug,
types_flag: sc.state().flags.types, types_flag: state.flags.types,
version_flag: sc.state().flags.version, version_flag: state.flags.version,
v8_version: Some(v8_version_off), v8_version: Some(v8_version_off),
deno_version: Some(deno_version_off), deno_version: Some(deno_version_off),
no_color: !ansi::use_color(), no_color: !ansi::use_color(),
@ -343,7 +335,7 @@ fn op_start(
} }
fn op_format_error( fn op_format_error(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -352,7 +344,7 @@ fn op_format_error(
let orig_error = String::from(inner.error().unwrap()); let orig_error = String::from(inner.error().unwrap());
let js_error = JSError::from_v8_exception(&orig_error).unwrap(); let js_error = JSError::from_v8_exception(&orig_error).unwrap();
let js_error_mapped = apply_source_map(&js_error, &sc.state().dir); let js_error_mapped = apply_source_map(&js_error, &state.dir);
let js_error_string = JSErrorColor(&js_error_mapped).to_string(); let js_error_string = JSErrorColor(&js_error_mapped).to_string();
let mut builder = FlatBufferBuilder::new(); let mut builder = FlatBufferBuilder::new();
@ -402,7 +394,7 @@ pub fn odd_future(err: DenoError) -> Box<OpWithError> {
// https://github.com/denoland/deno/blob/golang/os.go#L100-L154 // https://github.com/denoland/deno/blob/golang/os.go#L100-L154
fn op_fetch_module_meta_data( fn op_fetch_module_meta_data(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -412,19 +404,14 @@ fn op_fetch_module_meta_data(
let specifier = inner.specifier().unwrap(); let specifier = inner.specifier().unwrap();
let referrer = inner.referrer().unwrap(); let referrer = inner.referrer().unwrap();
assert_eq!( assert_eq!(state.dir.root.join("gen"), state.dir.gen, "Sanity check");
sc.state().dir.root.join("gen"),
sc.state().dir.gen,
"Sanity check"
);
let use_cache = !sc.state().flags.reload; let use_cache = !state.flags.reload;
Box::new(futures::future::result(|| -> OpResult { Box::new(futures::future::result(|| -> OpResult {
let builder = &mut FlatBufferBuilder::new(); let builder = &mut FlatBufferBuilder::new();
// TODO(ry) Use fetch_module_meta_data_async. // TODO(ry) Use fetch_module_meta_data_async.
let out = sc let out = state
.state()
.dir .dir
.fetch_module_meta_data(specifier, referrer, use_cache)?; .fetch_module_meta_data(specifier, referrer, use_cache)?;
let data_off = builder.create_vector(out.source_code.as_slice()); let data_off = builder.create_vector(out.source_code.as_slice());
@ -448,7 +435,7 @@ fn op_fetch_module_meta_data(
} }
fn op_chdir( fn op_chdir(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -462,20 +449,20 @@ fn op_chdir(
} }
fn op_global_timer_stop( fn op_global_timer_stop(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert!(base.sync()); assert!(base.sync());
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
let state = sc.state(); let state = state;
let mut t = state.global_timer.lock().unwrap(); let mut t = state.global_timer.lock().unwrap();
t.cancel(); t.cancel();
ok_future(empty_buf()) ok_future(empty_buf())
} }
fn op_global_timer( fn op_global_timer(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -486,7 +473,7 @@ fn op_global_timer(
let val = inner.timeout(); let val = inner.timeout();
assert!(val >= 0); assert!(val >= 0);
let state = sc.state(); let state = state;
let mut t = state.global_timer.lock().unwrap(); let mut t = state.global_timer.lock().unwrap();
let deadline = Instant::now() + Duration::from_millis(val as u64); let deadline = Instant::now() + Duration::from_millis(val as u64);
let f = t.new_timeout(deadline); let f = t.new_timeout(deadline);
@ -508,7 +495,7 @@ fn op_global_timer(
} }
fn op_set_env( fn op_set_env(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -516,7 +503,7 @@ fn op_set_env(
let inner = base.inner_as_set_env().unwrap(); let inner = base.inner_as_set_env().unwrap();
let key = inner.key().unwrap(); let key = inner.key().unwrap();
let value = inner.value().unwrap(); let value = inner.value().unwrap();
if let Err(e) = sc.state().check_env() { if let Err(e) = state.check_env() {
return odd_future(e); return odd_future(e);
} }
std::env::set_var(key, value); std::env::set_var(key, value);
@ -524,14 +511,14 @@ fn op_set_env(
} }
fn op_env( fn op_env(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
if let Err(e) = sc.state().check_env() { if let Err(e) = state.check_env() {
return odd_future(e); return odd_future(e);
} }
@ -556,7 +543,7 @@ fn op_env(
} }
fn op_permissions( fn op_permissions(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -566,12 +553,12 @@ fn op_permissions(
let inner = msg::PermissionsRes::create( let inner = msg::PermissionsRes::create(
builder, builder,
&msg::PermissionsResArgs { &msg::PermissionsResArgs {
run: sc.state().permissions.allows_run(), run: state.permissions.allows_run(),
read: sc.state().permissions.allows_read(), read: state.permissions.allows_read(),
write: sc.state().permissions.allows_write(), write: state.permissions.allows_write(),
net: sc.state().permissions.allows_net(), net: state.permissions.allows_net(),
env: sc.state().permissions.allows_env(), env: state.permissions.allows_env(),
high_precision: sc.state().permissions.allows_high_precision(), high_precision: state.permissions.allows_high_precision(),
}, },
); );
ok_future(serialize_response( ok_future(serialize_response(
@ -586,7 +573,7 @@ fn op_permissions(
} }
fn op_revoke_permission( fn op_revoke_permission(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -594,12 +581,12 @@ fn op_revoke_permission(
let inner = base.inner_as_permission_revoke().unwrap(); let inner = base.inner_as_permission_revoke().unwrap();
let permission = inner.permission().unwrap(); let permission = inner.permission().unwrap();
let result = match permission { let result = match permission {
"run" => sc.state().permissions.revoke_run(), "run" => state.permissions.revoke_run(),
"read" => sc.state().permissions.revoke_read(), "read" => state.permissions.revoke_read(),
"write" => sc.state().permissions.revoke_write(), "write" => state.permissions.revoke_write(),
"net" => sc.state().permissions.revoke_net(), "net" => state.permissions.revoke_net(),
"env" => sc.state().permissions.revoke_env(), "env" => state.permissions.revoke_env(),
"highPrecision" => sc.state().permissions.revoke_high_precision(), "highPrecision" => state.permissions.revoke_high_precision(),
_ => Ok(()), _ => Ok(()),
}; };
if let Err(e) = result { if let Err(e) = result {
@ -609,7 +596,7 @@ fn op_revoke_permission(
} }
fn op_fetch( fn op_fetch(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -632,7 +619,7 @@ fn op_fetch(
} }
let req = maybe_req.unwrap(); let req = maybe_req.unwrap();
if let Err(e) = sc.state().check_net(url) { if let Err(e) = state.check_net(url) {
return odd_future(e); return odd_future(e);
} }
@ -696,7 +683,7 @@ where
} }
fn op_make_temp_dir( fn op_make_temp_dir(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -706,7 +693,7 @@ fn op_make_temp_dir(
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
// FIXME // FIXME
if let Err(e) = sc.state().check_write("make_temp") { if let Err(e) = state.check_write("make_temp") {
return odd_future(e); return odd_future(e);
} }
@ -745,7 +732,7 @@ fn op_make_temp_dir(
} }
fn op_mkdir( fn op_mkdir(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -755,7 +742,7 @@ fn op_mkdir(
let recursive = inner.recursive(); let recursive = inner.recursive();
let mode = inner.mode(); let mode = inner.mode();
if let Err(e) = sc.state().check_write(&path) { if let Err(e) = state.check_write(&path) {
return odd_future(e); return odd_future(e);
} }
@ -767,7 +754,7 @@ fn op_mkdir(
} }
fn op_chmod( fn op_chmod(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -776,7 +763,7 @@ fn op_chmod(
let _mode = inner.mode(); let _mode = inner.mode();
let path = String::from(inner.path().unwrap()); let path = String::from(inner.path().unwrap());
if let Err(e) = sc.state().check_write(&path) { if let Err(e) = state.check_write(&path) {
return odd_future(e); return odd_future(e);
} }
@ -806,7 +793,7 @@ fn op_chmod(
} }
fn op_open( fn op_open(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -855,20 +842,20 @@ fn op_open(
match mode { match mode {
"r" => { "r" => {
if let Err(e) = sc.state().check_read(&filename_str) { if let Err(e) = state.check_read(&filename_str) {
return odd_future(e); return odd_future(e);
} }
} }
"w" | "a" | "x" => { "w" | "a" | "x" => {
if let Err(e) = sc.state().check_write(&filename_str) { if let Err(e) = state.check_write(&filename_str) {
return odd_future(e); return odd_future(e);
} }
} }
&_ => { &_ => {
if let Err(e) = sc.state().check_read(&filename_str) { if let Err(e) = state.check_read(&filename_str) {
return odd_future(e); return odd_future(e);
} }
if let Err(e) = sc.state().check_write(&filename_str) { if let Err(e) = state.check_write(&filename_str) {
return odd_future(e); return odd_future(e);
} }
} }
@ -896,7 +883,7 @@ fn op_open(
} }
fn op_close( fn op_close(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -913,7 +900,7 @@ fn op_close(
} }
fn op_shutdown( fn op_shutdown(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -939,7 +926,7 @@ fn op_shutdown(
} }
fn op_read( fn op_read(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -977,7 +964,7 @@ fn op_read(
} }
fn op_write( fn op_write(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1014,7 +1001,7 @@ fn op_write(
} }
fn op_seek( fn op_seek(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1036,7 +1023,7 @@ fn op_seek(
} }
fn op_remove( fn op_remove(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1046,7 +1033,7 @@ fn op_remove(
let path = PathBuf::from(path_); let path = PathBuf::from(path_);
let recursive = inner.recursive(); let recursive = inner.recursive();
if let Err(e) = sc.state().check_write(path.to_str().unwrap()) { if let Err(e) = state.check_write(path.to_str().unwrap()) {
return odd_future(e); return odd_future(e);
} }
@ -1065,7 +1052,7 @@ fn op_remove(
} }
fn op_copy_file( fn op_copy_file(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1076,10 +1063,10 @@ fn op_copy_file(
let to_ = inner.to().unwrap(); let to_ = inner.to().unwrap();
let to = PathBuf::from(to_); let to = PathBuf::from(to_);
if let Err(e) = sc.state().check_read(&from_) { if let Err(e) = state.check_read(&from_) {
return odd_future(e); return odd_future(e);
} }
if let Err(e) = sc.state().check_write(&to_) { if let Err(e) = state.check_write(&to_) {
return odd_future(e); return odd_future(e);
} }
@ -1121,7 +1108,7 @@ fn get_mode(_perm: &fs::Permissions) -> u32 {
} }
fn op_cwd( fn op_cwd(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1147,7 +1134,7 @@ fn op_cwd(
} }
fn op_stat( fn op_stat(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1158,7 +1145,7 @@ fn op_stat(
let filename = PathBuf::from(filename_); let filename = PathBuf::from(filename_);
let lstat = inner.lstat(); let lstat = inner.lstat();
if let Err(e) = sc.state().check_read(&filename_) { if let Err(e) = state.check_read(&filename_) {
return odd_future(e); return odd_future(e);
} }
@ -1199,7 +1186,7 @@ fn op_stat(
} }
fn op_read_dir( fn op_read_dir(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1208,7 +1195,7 @@ fn op_read_dir(
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
let path = String::from(inner.path().unwrap()); let path = String::from(inner.path().unwrap());
if let Err(e) = sc.state().check_read(&path) { if let Err(e) = state.check_read(&path) {
return odd_future(e); return odd_future(e);
} }
@ -1260,7 +1247,7 @@ fn op_read_dir(
} }
fn op_rename( fn op_rename(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1269,7 +1256,7 @@ fn op_rename(
let oldpath = PathBuf::from(inner.oldpath().unwrap()); let oldpath = PathBuf::from(inner.oldpath().unwrap());
let newpath_ = inner.newpath().unwrap(); let newpath_ = inner.newpath().unwrap();
let newpath = PathBuf::from(newpath_); let newpath = PathBuf::from(newpath_);
if let Err(e) = sc.state().check_write(&newpath_) { if let Err(e) = state.check_write(&newpath_) {
return odd_future(e); return odd_future(e);
} }
blocking(base.sync(), move || -> OpResult { blocking(base.sync(), move || -> OpResult {
@ -1280,7 +1267,7 @@ fn op_rename(
} }
fn op_link( fn op_link(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1290,7 +1277,7 @@ fn op_link(
let newname_ = inner.newname().unwrap(); let newname_ = inner.newname().unwrap();
let newname = PathBuf::from(newname_); let newname = PathBuf::from(newname_);
if let Err(e) = sc.state().check_write(&newname_) { if let Err(e) = state.check_write(&newname_) {
return odd_future(e); return odd_future(e);
} }
@ -1302,7 +1289,7 @@ fn op_link(
} }
fn op_symlink( fn op_symlink(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1312,7 +1299,7 @@ fn op_symlink(
let newname_ = inner.newname().unwrap(); let newname_ = inner.newname().unwrap();
let newname = PathBuf::from(newname_); let newname = PathBuf::from(newname_);
if let Err(e) = sc.state().check_write(&newname_) { if let Err(e) = state.check_write(&newname_) {
return odd_future(e); return odd_future(e);
} }
// TODO Use type for Windows. // TODO Use type for Windows.
@ -1331,7 +1318,7 @@ fn op_symlink(
} }
fn op_read_link( fn op_read_link(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1341,7 +1328,7 @@ fn op_read_link(
let name_ = inner.name().unwrap(); let name_ = inner.name().unwrap();
let name = PathBuf::from(name_); let name = PathBuf::from(name_);
if let Err(e) = sc.state().check_read(&name_) { if let Err(e) = state.check_read(&name_) {
return odd_future(e); return odd_future(e);
} }
@ -1369,7 +1356,7 @@ fn op_read_link(
} }
fn op_repl_start( fn op_repl_start(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1379,7 +1366,7 @@ fn op_repl_start(
let history_file = String::from(inner.history_file().unwrap()); let history_file = String::from(inner.history_file().unwrap());
debug!("op_repl_start {}", history_file); debug!("op_repl_start {}", history_file);
let history_path = repl::history_path(&sc.state().dir, &history_file); let history_path = repl::history_path(&state.dir, &history_file);
let repl = repl::Repl::new(history_path); let repl = repl::Repl::new(history_path);
let resource = resources::add_repl(repl); let resource = resources::add_repl(repl);
@ -1400,7 +1387,7 @@ fn op_repl_start(
} }
fn op_repl_readline( fn op_repl_readline(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1436,7 +1423,7 @@ fn op_repl_readline(
} }
fn op_truncate( fn op_truncate(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1446,7 +1433,7 @@ fn op_truncate(
let filename = String::from(inner.name().unwrap()); let filename = String::from(inner.name().unwrap());
let len = inner.len(); let len = inner.len();
if let Err(e) = sc.state().check_write(&filename) { if let Err(e) = state.check_write(&filename) {
return odd_future(e); return odd_future(e);
} }
@ -1459,12 +1446,12 @@ fn op_truncate(
} }
fn op_listen( fn op_listen(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
if let Err(e) = sc.state().check_net("listen") { if let Err(e) = state.check_net("listen") {
return odd_future(e); return odd_future(e);
} }
@ -1521,12 +1508,12 @@ fn new_conn(cmd_id: u32, tcp_stream: TcpStream) -> OpResult {
} }
fn op_accept( fn op_accept(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
if let Err(e) = sc.state().check_net("accept") { if let Err(e) = state.check_net("accept") {
return odd_future(e); return odd_future(e);
} }
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
@ -1547,12 +1534,12 @@ fn op_accept(
} }
fn op_dial( fn op_dial(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert_eq!(data.len(), 0); assert_eq!(data.len(), 0);
if let Err(e) = sc.state().check_net("dial") { if let Err(e) = state.check_net("dial") {
return odd_future(e); return odd_future(e);
} }
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
@ -1573,7 +1560,7 @@ fn op_dial(
} }
fn op_metrics( fn op_metrics(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1583,7 +1570,7 @@ fn op_metrics(
let builder = &mut FlatBufferBuilder::new(); let builder = &mut FlatBufferBuilder::new();
let inner = msg::MetricsRes::create( let inner = msg::MetricsRes::create(
builder, builder,
&msg::MetricsResArgs::from(&sc.state().metrics), &msg::MetricsResArgs::from(&state.metrics),
); );
ok_future(serialize_response( ok_future(serialize_response(
cmd_id, cmd_id,
@ -1597,7 +1584,7 @@ fn op_metrics(
} }
fn op_resources( fn op_resources(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1649,14 +1636,14 @@ fn subprocess_stdio_map(v: msg::ProcessStdio) -> std::process::Stdio {
} }
fn op_run( fn op_run(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
assert!(base.sync()); assert!(base.sync());
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
if let Err(e) = sc.state().check_run() { if let Err(e) = state.check_run() {
return odd_future(e); return odd_future(e);
} }
@ -1722,7 +1709,7 @@ fn op_run(
} }
fn op_run_status( fn op_run_status(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1731,7 +1718,7 @@ fn op_run_status(
let inner = base.inner_as_run_status().unwrap(); let inner = base.inner_as_run_status().unwrap();
let rid = inner.rid(); let rid = inner.rid();
if let Err(e) = sc.state().check_run() { if let Err(e) = state.check_run() {
return odd_future(e); return odd_future(e);
} }
@ -1786,20 +1773,16 @@ impl Future for GetMessageFuture {
type Error = (); type Error = ();
fn poll(&mut self) -> Result<Async<Self::Item>, Self::Error> { fn poll(&mut self) -> Result<Async<Self::Item>, Self::Error> {
assert!(self.state.worker_channels.is_some()); let mut wc = self.state.worker_channels.lock().unwrap();
match self.state.worker_channels { wc.1
None => panic!("expected worker_channels"), .poll()
Some(ref wc) => { .map_err(|err| panic!("worker_channel recv err {:?}", err))
let mut wc = wc.lock().unwrap();
wc.1.poll()
}
}
} }
} }
/// Get message from host as guest worker /// Get message from host as guest worker
fn op_worker_get_message( fn op_worker_get_message(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1807,7 +1790,7 @@ fn op_worker_get_message(
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
let op = GetMessageFuture { let op = GetMessageFuture {
state: sc.state().clone(), state: state.clone(),
}; };
let op = op.map_err(move |_| -> DenoError { unimplemented!() }); let op = op.map_err(move |_| -> DenoError { unimplemented!() });
let op = op.and_then(move |maybe_buf| -> DenoResult<Buf> { let op = op.and_then(move |maybe_buf| -> DenoResult<Buf> {
@ -1834,7 +1817,7 @@ fn op_worker_get_message(
/// Post message to host as guest worker /// Post message to host as guest worker
fn op_worker_post_message( fn op_worker_post_message(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1842,13 +1825,9 @@ fn op_worker_post_message(
let d = Vec::from(data.as_ref()).into_boxed_slice(); let d = Vec::from(data.as_ref()).into_boxed_slice();
assert!(sc.state().worker_channels.is_some()); let tx = {
let tx = match sc.state().worker_channels { let wc = state.worker_channels.lock().unwrap();
None => panic!("expected worker_channels"),
Some(ref wc) => {
let wc = wc.lock().unwrap();
wc.0.clone() wc.0.clone()
}
}; };
let op = tx.send(d); let op = tx.send(d);
let op = op.map_err(|e| errors::new(ErrorKind::Other, e.to_string())); let op = op.map_err(|e| errors::new(ErrorKind::Other, e.to_string()));
@ -1868,7 +1847,7 @@ fn op_worker_post_message(
/// Create worker as the host /// Create worker as the host
fn op_create_worker( fn op_create_worker(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1878,20 +1857,24 @@ fn op_create_worker(
let specifier = inner.specifier().unwrap(); let specifier = inner.specifier().unwrap();
Box::new(futures::future::result(move || -> OpResult { Box::new(futures::future::result(move || -> OpResult {
let parent_state = sc.state().clone(); let parent_state = state.clone();
let behavior = workers::UserWorkerBehavior::new(
let child_state = Arc::new(IsolateState::new(
parent_state.flags.clone(), parent_state.flags.clone(),
parent_state.argv.clone(), parent_state.argv.clone(),
); ));
match workers::spawn( let rid = child_state.resource.rid;
startup_data::deno_isolate_init(), let behavior = CliBehavior::new(child_state);
behavior, let name = format!("USER-WORKER-{}", specifier);
&format!("USER-WORKER-{}", specifier),
workers::WorkerInit::Module(specifier.to_string()), let mut worker =
) { Worker::new(name, startup_data::deno_isolate_init(), behavior);
Ok(worker) => { js_check(worker.execute("denoMain()"));
js_check(worker.execute("workerMain()"));
let result = worker.execute_mod(specifier, false);
match result {
Ok(_) => {
let mut workers_tl = parent_state.workers.lock().unwrap(); let mut workers_tl = parent_state.workers.lock().unwrap();
let rid = worker.resource.rid;
workers_tl.insert(rid, worker.shared()); workers_tl.insert(rid, worker.shared());
let builder = &mut FlatBufferBuilder::new(); let builder = &mut FlatBufferBuilder::new();
let msg_inner = msg::CreateWorkerRes::create( let msg_inner = msg::CreateWorkerRes::create(
@ -1916,7 +1899,7 @@ fn op_create_worker(
/// Return when the worker closes /// Return when the worker closes
fn op_host_get_worker_closed( fn op_host_get_worker_closed(
sc: &IsolateStateContainer, state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1924,7 +1907,7 @@ fn op_host_get_worker_closed(
let cmd_id = base.cmd_id(); let cmd_id = base.cmd_id();
let inner = base.inner_as_host_get_worker_closed().unwrap(); let inner = base.inner_as_host_get_worker_closed().unwrap();
let rid = inner.rid(); let rid = inner.rid();
let state = sc.state().clone(); let state = state.clone();
let shared_worker_future = { let shared_worker_future = {
let workers_tl = state.workers.lock().unwrap(); let workers_tl = state.workers.lock().unwrap();
@ -1947,7 +1930,7 @@ fn op_host_get_worker_closed(
/// Get message from guest worker as host /// Get message from guest worker as host
fn op_host_get_message( fn op_host_get_message(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {
@ -1981,7 +1964,7 @@ fn op_host_get_message(
/// Post message to guest worker as host /// Post message to guest worker as host
fn op_host_post_message( fn op_host_post_message(
_sc: &IsolateStateContainer, _state: &Arc<IsolateState>,
base: &msg::Base<'_>, base: &msg::Base<'_>,
data: deno_buf, data: deno_buf,
) -> Box<OpWithError> { ) -> Box<OpWithError> {

View file

@ -35,6 +35,7 @@ use std::sync::{Arc, Mutex};
use tokio; use tokio;
use tokio::io::{AsyncRead, AsyncWrite}; use tokio::io::{AsyncRead, AsyncWrite};
use tokio::net::TcpStream; use tokio::net::TcpStream;
use tokio::sync::mpsc;
use tokio_process; use tokio_process;
pub type ResourceId = u32; // Sometimes referred to RID. pub type ResourceId = u32; // Sometimes referred to RID.
@ -309,7 +310,7 @@ pub fn add_worker(wc: WorkerChannels) -> Resource {
pub fn post_message_to_worker( pub fn post_message_to_worker(
rid: ResourceId, rid: ResourceId,
buf: Buf, buf: Buf,
) -> futures::sink::Send<futures::sync::mpsc::Sender<Buf>> { ) -> futures::sink::Send<mpsc::Sender<Buf>> {
let mut table = RESOURCE_TABLE.lock().unwrap(); let mut table = RESOURCE_TABLE.lock().unwrap();
let maybe_repr = table.get_mut(&rid); let maybe_repr = table.get_mut(&rid);
match maybe_repr { match maybe_repr {
@ -334,9 +335,10 @@ impl Future for WorkerReceiver {
let mut table = RESOURCE_TABLE.lock().unwrap(); let mut table = RESOURCE_TABLE.lock().unwrap();
let maybe_repr = table.get_mut(&self.rid); let maybe_repr = table.get_mut(&self.rid);
match maybe_repr { match maybe_repr {
Some(Repr::Worker(ref mut wc)) => wc.1.poll().map_err(|()| { Some(Repr::Worker(ref mut wc)) => wc
errors::new(errors::ErrorKind::Other, "recv msg error".to_string()) .1
}), .poll()
.map_err(|err| errors::new(errors::ErrorKind::Other, err.to_string())),
_ => Err(bad_resource()), _ => Err(bad_resource()),
} }
} }
@ -359,9 +361,10 @@ impl Stream for WorkerReceiverStream {
let mut table = RESOURCE_TABLE.lock().unwrap(); let mut table = RESOURCE_TABLE.lock().unwrap();
let maybe_repr = table.get_mut(&self.rid); let maybe_repr = table.get_mut(&self.rid);
match maybe_repr { match maybe_repr {
Some(Repr::Worker(ref mut wc)) => wc.1.poll().map_err(|()| { Some(Repr::Worker(ref mut wc)) => wc
errors::new(errors::ErrorKind::Other, "recv msg error".to_string()) .1
}), .poll()
.map_err(|err| errors::new(errors::ErrorKind::Other, err.to_string())),
_ => Err(bad_resource()), _ => Err(bad_resource()),
} }
} }

View file

@ -1,17 +1,16 @@
// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license. // Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
use crate::cli_behavior::CliBehavior;
use crate::compiler::compile_async; use crate::compiler::compile_async;
use crate::compiler::ModuleMetaData; use crate::compiler::ModuleMetaData;
use crate::errors::DenoError; use crate::errors::DenoError;
use crate::errors::RustOrJsError; use crate::errors::RustOrJsError;
use crate::isolate_state::IsolateState; use crate::isolate_state::IsolateState;
use crate::isolate_state::IsolateStateContainer;
use crate::js_errors; use crate::js_errors;
use crate::js_errors::JSErrorColor; use crate::js_errors::JSErrorColor;
use crate::msg; use crate::msg;
use crate::tokio_util; use crate::tokio_util;
use deno; use deno;
use deno::deno_mod; use deno::deno_mod;
use deno::Behavior;
use deno::JSError; use deno::JSError;
use deno::StartupData; use deno::StartupData;
use futures::future::Either; use futures::future::Either;
@ -20,23 +19,22 @@ use futures::Future;
use std::sync::atomic::Ordering; use std::sync::atomic::Ordering;
use std::sync::Arc; use std::sync::Arc;
pub trait DenoBehavior: Behavior + IsolateStateContainer + Send {}
impl<T> DenoBehavior for T where T: Behavior + IsolateStateContainer + Send {}
type CoreIsolate<B> = deno::Isolate<B>;
/// Wraps deno::Isolate to provide source maps, ops for the CLI, and /// Wraps deno::Isolate to provide source maps, ops for the CLI, and
/// high-level module loading /// high-level module loading
pub struct Isolate<B: Behavior> { pub struct Worker {
inner: CoreIsolate<B>, inner: deno::Isolate<CliBehavior>,
state: Arc<IsolateState>, state: Arc<IsolateState>,
} }
impl<B: DenoBehavior> Isolate<B> { impl Worker {
pub fn new(startup_data: StartupData, behavior: B) -> Isolate<B> { pub fn new(
let state = behavior.state().clone(); _name: String,
startup_data: StartupData,
behavior: CliBehavior,
) -> Worker {
let state = behavior.state.clone();
Self { Self {
inner: CoreIsolate::new(startup_data, behavior), inner: deno::Isolate::new(startup_data, behavior),
state, state,
} }
} }
@ -196,7 +194,7 @@ impl<B: DenoBehavior> Isolate<B> {
} }
} }
impl<B: DenoBehavior> Future for Isolate<B> { impl Future for Worker {
type Item = (); type Item = ();
type Error = JSError; type Error = JSError;
@ -255,8 +253,14 @@ mod tests {
use super::*; use super::*;
use crate::cli_behavior::CliBehavior; use crate::cli_behavior::CliBehavior;
use crate::flags; use crate::flags;
use crate::isolate_state::IsolateState;
use crate::resources;
use crate::startup_data;
use crate::tokio_util;
use deno::js_check;
use futures::future::lazy; use futures::future::lazy;
use std::sync::atomic::Ordering; use std::sync::atomic::Ordering;
use std::thread;
#[test] #[test]
fn execute_mod() { fn execute_mod() {
@ -268,15 +272,15 @@ mod tests {
let argv = vec![String::from("./deno"), filename.clone()]; let argv = vec![String::from("./deno"), filename.clone()];
let (flags, rest_argv) = flags::set_flags(argv).unwrap(); let (flags, rest_argv) = flags::set_flags(argv).unwrap();
let state = Arc::new(IsolateState::new(flags, rest_argv, None, false)); let state = Arc::new(IsolateState::new(flags, rest_argv));
let state_ = state.clone(); let state_ = state.clone();
tokio_util::run(lazy(move || { tokio_util::run(lazy(move || {
let cli = CliBehavior::new(state.clone()); let cli = CliBehavior::new(state.clone());
let mut isolate = Isolate::new(StartupData::None, cli); let mut worker = Worker::new("TEST".to_string(), StartupData::None, cli);
if let Err(err) = isolate.execute_mod(&filename, false) { if let Err(err) = worker.execute_mod(&filename, false) {
eprintln!("execute_mod err {:?}", err); eprintln!("execute_mod err {:?}", err);
} }
tokio_util::panic_on_error(isolate) tokio_util::panic_on_error(worker)
})); }));
let metrics = &state_.metrics; let metrics = &state_.metrics;
@ -291,18 +295,113 @@ mod tests {
let argv = vec![String::from("./deno"), filename.clone()]; let argv = vec![String::from("./deno"), filename.clone()];
let (flags, rest_argv) = flags::set_flags(argv).unwrap(); let (flags, rest_argv) = flags::set_flags(argv).unwrap();
let state = Arc::new(IsolateState::new(flags, rest_argv, None, false)); let state = Arc::new(IsolateState::new(flags, rest_argv));
let state_ = state.clone(); let state_ = state.clone();
tokio_util::run(lazy(move || { tokio_util::run(lazy(move || {
let cli = CliBehavior::new(state.clone()); let cli = CliBehavior::new(state.clone());
let mut isolate = Isolate::new(StartupData::None, cli); let mut worker = Worker::new("TEST".to_string(), StartupData::None, cli);
if let Err(err) = isolate.execute_mod(&filename, false) { if let Err(err) = worker.execute_mod(&filename, false) {
eprintln!("execute_mod err {:?}", err); eprintln!("execute_mod err {:?}", err);
} }
tokio_util::panic_on_error(isolate) tokio_util::panic_on_error(worker)
})); }));
let metrics = &state_.metrics; let metrics = &state_.metrics;
assert_eq!(metrics.resolve_count.load(Ordering::SeqCst), 2); assert_eq!(metrics.resolve_count.load(Ordering::SeqCst), 2);
} }
fn create_test_worker() -> Worker {
let state = Arc::new(IsolateState::mock());
let cli = CliBehavior::new(state.clone());
let mut worker =
Worker::new("TEST".to_string(), startup_data::deno_isolate_init(), cli);
js_check(worker.execute("denoMain()"));
js_check(worker.execute("workerMain()"));
worker
}
#[test]
fn test_worker_messages() {
tokio_util::init(|| {
let mut worker = create_test_worker();
let source = r#"
onmessage = function(e) {
console.log("msg from main script", e.data);
if (e.data == "exit") {
close();
return;
} else {
console.assert(e.data === "hi");
}
postMessage([1, 2, 3]);
console.log("after postMessage");
}
"#;
js_check(worker.execute(source));
let resource = worker.state.resource.clone();
let resource_ = resource.clone();
tokio::spawn(lazy(move || {
worker.then(move |r| -> Result<(), ()> {
resource_.close();
js_check(r);
Ok(())
})
}));
let msg = json!("hi").to_string().into_boxed_str().into_boxed_bytes();
let r = resources::post_message_to_worker(resource.rid, msg).wait();
assert!(r.is_ok());
let maybe_msg = resources::get_message_from_worker(resource.rid)
.wait()
.unwrap();
assert!(maybe_msg.is_some());
// Check if message received is [1, 2, 3] in json
assert_eq!(*maybe_msg.unwrap(), *b"[1,2,3]");
let msg = json!("exit")
.to_string()
.into_boxed_str()
.into_boxed_bytes();
let r = resources::post_message_to_worker(resource.rid, msg).wait();
assert!(r.is_ok());
})
}
#[test]
fn removed_from_resource_table_on_close() {
tokio_util::init(|| {
let mut worker = create_test_worker();
js_check(
worker.execute("onmessage = () => { delete window['onmessage']; }"),
);
let resource = worker.state.resource.clone();
let rid = resource.rid;
tokio::spawn(lazy(move || {
worker.then(move |r| -> Result<(), ()> {
resource.close();
println!("workers.rs after resource close");
js_check(r);
Ok(())
})
}));
assert_eq!(resources::get_type(rid), Some("worker".to_string()));
let msg = json!("hi").to_string().into_boxed_str().into_boxed_bytes();
let r = resources::post_message_to_worker(rid, msg).wait();
assert!(r.is_ok());
debug!("rid {:?}", rid);
// TODO Need a way to get a future for when a resource closes.
// For now, just sleep for a bit.
thread::sleep(std::time::Duration::from_millis(1000));
assert_eq!(resources::get_type(rid), None);
})
}
} }

View file

@ -1,284 +0,0 @@
// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
use crate::errors::*;
use crate::flags::DenoFlags;
use crate::isolate::{DenoBehavior, Isolate};
use crate::isolate_state::IsolateState;
use crate::isolate_state::IsolateStateContainer;
use crate::isolate_state::WorkerChannels;
use crate::ops;
use crate::resources;
use deno::deno_buf;
use deno::Behavior;
use deno::Buf;
use deno::JSError;
use deno::Op;
use deno::StartupData;
use futures::sync::mpsc;
use futures::Future;
use futures::Poll;
use std::sync::Arc;
pub struct UserWorkerBehavior {
pub state: Arc<IsolateState>,
}
impl UserWorkerBehavior {
pub fn new(flags: DenoFlags, argv_rest: Vec<String>) -> Self {
Self {
state: Arc::new(IsolateState::new(flags, argv_rest, None, true)),
}
}
}
impl IsolateStateContainer for UserWorkerBehavior {
fn state(&self) -> Arc<IsolateState> {
self.state.clone()
}
}
impl IsolateStateContainer for &UserWorkerBehavior {
fn state(&self) -> Arc<IsolateState> {
self.state.clone()
}
}
impl Behavior for UserWorkerBehavior {
fn dispatch(
&mut self,
control: &[u8],
zero_copy: deno_buf,
) -> (bool, Box<Op>) {
ops::dispatch_all(self, control, zero_copy, ops::op_selector_worker)
}
}
impl WorkerBehavior for UserWorkerBehavior {
fn set_internal_channels(&mut self, worker_channels: WorkerChannels) {
self.state = Arc::new(IsolateState::new(
self.state.flags.clone(),
self.state.argv.clone(),
Some(worker_channels),
true,
));
}
}
/// Behavior trait specific to workers
pub trait WorkerBehavior: DenoBehavior {
/// Used to setup internal channels at worker creation.
/// This is intended to be temporary fix.
/// TODO(afinch7) come up with a better solution to set worker channels
fn set_internal_channels(&mut self, worker_channels: WorkerChannels);
}
/// Rust interface for WebWorkers.
pub struct Worker<B: WorkerBehavior> {
isolate: Isolate<B>,
pub resource: resources::Resource,
}
impl<B: WorkerBehavior> Worker<B> {
pub fn new(startup_data: StartupData, mut behavior: B) -> Self {
let (worker_in_tx, worker_in_rx) = mpsc::channel::<Buf>(1);
let (worker_out_tx, worker_out_rx) = mpsc::channel::<Buf>(1);
let internal_channels = (worker_out_tx, worker_in_rx);
let external_channels = (worker_in_tx, worker_out_rx);
behavior.set_internal_channels(internal_channels);
let isolate = Isolate::new(startup_data, behavior);
Worker {
isolate,
resource: resources::add_worker(external_channels),
}
}
pub fn execute(&mut self, js_source: &str) -> Result<(), JSError> {
self.isolate.execute(js_source)
}
pub fn execute_mod(
&mut self,
js_filename: &str,
is_prefetch: bool,
) -> Result<(), RustOrJsError> {
self.isolate.execute_mod(js_filename, is_prefetch)
}
}
impl<B: WorkerBehavior> Future for Worker<B> {
type Item = ();
type Error = JSError;
fn poll(&mut self) -> Poll<(), JSError> {
self.isolate.poll()
}
}
/// Method and data used to initalize a worker
pub enum WorkerInit {
Script(String),
Module(String),
}
pub fn spawn<B: WorkerBehavior + 'static>(
startup_data: StartupData,
behavior: B,
worker_debug_name: &str,
init: WorkerInit,
) -> Result<Worker<B>, RustOrJsError> {
let state = behavior.state().clone();
let mut worker = Worker::new(startup_data, behavior);
worker
.execute(&format!("denoMain('{}')", worker_debug_name))
.expect("worker workerInit failed");
worker
.execute("workerMain()")
.expect("worker workerMain failed");
let init_result = match init {
WorkerInit::Script(script) => match worker.execute(&script) {
Ok(_) => Ok(()),
Err(e) => Err(RustOrJsError::Js(e)),
},
WorkerInit::Module(specifier) => {
let should_prefetch = state.flags.prefetch || state.flags.info;
match state.dir.resolve_module_url(&specifier, ".") {
Err(err) => Err(RustOrJsError::Rust(DenoError::from(err))),
Ok(module_url) => {
worker.execute_mod(&module_url.to_string(), should_prefetch)
}
}
}
};
match init_result {
Ok(_) => Ok(worker),
Err(err) => Err(err),
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::compiler::CompilerBehavior;
use crate::isolate_state::IsolateState;
use crate::js_errors::JSErrorColor;
use crate::startup_data;
use crate::tokio_util;
use futures::future::lazy;
use std::thread;
#[test]
fn test_spawn() {
tokio_util::init(|| {
let worker_result = spawn(
startup_data::compiler_isolate_init(),
CompilerBehavior::new(
IsolateState::mock().flags.clone(),
IsolateState::mock().argv.clone(),
),
"TEST",
WorkerInit::Script(
r#"
onmessage = function(e) {
console.log("msg from main script", e.data);
if (e.data == "exit") {
close();
return;
} else {
console.assert(e.data === "hi");
}
postMessage([1, 2, 3]);
console.log("after postMessage");
}
"#.into(),
),
);
assert!(worker_result.is_ok());
let worker = worker_result.unwrap();
let resource = worker.resource.clone();
let resource_ = resource.clone();
tokio::spawn(lazy(move || {
worker.then(move |r| -> Result<(), ()> {
resource_.close();
debug!("workers.rs after resource close");
if let Err(err) = r {
panic!("{}", JSErrorColor(&err).to_string());
}
Ok(())
})
}));
let msg = json!("hi").to_string().into_boxed_str().into_boxed_bytes();
let r = resources::post_message_to_worker(resource.rid, msg).wait();
assert!(r.is_ok());
let maybe_msg = resources::get_message_from_worker(resource.rid)
.wait()
.unwrap();
assert!(maybe_msg.is_some());
// Check if message received is [1, 2, 3] in json
assert_eq!(*maybe_msg.unwrap(), *b"[1,2,3]");
let msg = json!("exit")
.to_string()
.into_boxed_str()
.into_boxed_bytes();
let r = resources::post_message_to_worker(resource.rid, msg).wait();
assert!(r.is_ok());
})
}
#[test]
fn removed_from_resource_table_on_close() {
tokio_util::init(|| {
let worker_result = spawn(
startup_data::compiler_isolate_init(),
CompilerBehavior::new(
IsolateState::mock().flags.clone(),
IsolateState::mock().argv.clone(),
),
"TEST",
WorkerInit::Script("onmessage = () => close();".into()),
);
assert!(worker_result.is_ok());
let worker = worker_result.unwrap();
let resource = worker.resource.clone();
let resource_ = resource.clone();
tokio::spawn(lazy(move || {
worker.then(move |r| -> Result<(), ()> {
resource_.close();
debug!("workers.rs after resource close");
if let Err(err) = r {
panic!("{}", JSErrorColor(&err).to_string());
}
Ok(())
})
}));
assert_eq!(
resources::get_type(resource.rid),
Some("worker".to_string())
);
let msg = json!("hi").to_string().into_boxed_str().into_boxed_bytes();
let r = resources::post_message_to_worker(resource.rid, msg).wait();
assert!(r.is_ok());
println!("rid {:?}", resource.rid);
// TODO Need a way to get a future for when a resource closes.
// For now, just sleep for a bit.
// resource.close();
thread::sleep(std::time::Duration::from_millis(1000));
assert_eq!(resources::get_type(resource.rid), None);
})
}
}

View file

@ -46,7 +46,6 @@ type SourceMap = string;
interface CompilerLookup { interface CompilerLookup {
specifier: ModuleSpecifier; specifier: ModuleSpecifier;
referrer: ContainingFile; referrer: ContainingFile;
isWorker: boolean;
cmdId: number; cmdId: number;
} }

View file

@ -129,10 +129,13 @@ export async function workerMain(): Promise<void> {
log("workerMain got null message. quitting."); log("workerMain got null message. quitting.");
break; break;
} }
if (window["onmessage"]) { if (window["onmessage"]) {
const event = { data }; const event = { data };
window.onmessage(event); window.onmessage(event);
} else { }
if (!window["onmessage"]) {
break; break;
} }
} }