Skip to content

fix wasm + no-std cargo interactions #864

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 11 commits into from
55 changes: 29 additions & 26 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,24 +29,6 @@ jobs:
toolchain: ${{ matrix.rust }}
override: true

- name: Cache cargo registry
uses: actions/cache@v2
with:
path: ~/.cargo/registry
key: ${{ matrix.os }}-${{ matrix.rust }}-cargo-registry-${{ hashFiles('**/Cargo.toml') }}

- name: Cache cargo index
uses: actions/cache@v2
with:
path: ~/.cargo/git
key: ${{ matrix.os }}-${{ matrix.rust }}-cargo-index-${{ hashFiles('**/Cargo.toml') }}

- name: Cache cargo build
uses: actions/cache@v2
with:
path: target
key: ${{ matrix.os }}-${{ matrix.rust }}-cargo-build-target-${{ hashFiles('**/Cargo.toml') }}

- name: check
uses: actions-rs/cargo@v1
with:
Expand All @@ -59,14 +41,6 @@ jobs:
command: check
args: --features unstable --all --bins --examples --tests

- name: check wasm
uses: actions-rs/cargo@v1
with:
command: check
target: wasm32-unknown-unknown
override: true
args: --features unstable --all --bins --tests

- name: check bench
uses: actions-rs/cargo@v1
if: matrix.rust == 'nightly'
Expand Down Expand Up @@ -160,6 +134,35 @@ jobs:
- name: test
run: cross test --all --features unstable --target ${{ matrix.target }}

check_wasm:
name: Check wasm targets
runs-on: ubuntu-latest
strategy:
matrix:
rust: [nightly, beta, stable]

steps:
- uses: actions/checkout@master

- name: Install rust with wasm32-unknown-unknown
uses: actions-rs/toolchain@v1
with:
toolchain: ${{ matrix.rust }}
target: wasm32-unknown-unknown
override: true

- name: check
uses: actions-rs/cargo@v1
with:
command: check
args: --target wasm32-unknown-unknown --features wasm-bindgen

- name: check unstable
uses: actions-rs/cargo@v1
with:
command: check
args: --target wasm32-unknown-unknown --tests --all --features unstable,wasm-bindgen

check_fmt_and_docs:
name: Checking fmt and docs
runs-on: ubuntu-latest
Expand Down
14 changes: 7 additions & 7 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,12 @@ std = [
"once_cell",
"pin-utils",
"slab",
"async-mutex",
]
wasm-bindgen = [
"futures-timer/wasm-bindgen",
"wasm-bindgen-futures",
"futures-channel",
"async-mutex",
]
alloc = [
"futures-core/alloc",
Expand All @@ -75,6 +78,8 @@ pin-project-lite = { version = "0.1.4", optional = true }
pin-utils = { version = "0.1.0-alpha.4", optional = true }
slab = { version = "0.4.2", optional = true }
futures-timer = { version = "3.0.2", optional = true }
wasm-bindgen-futures = { version = "0.4.10", optional = true }
futures-channel = { version = "0.3.4", optional = true }

# Devdepencency, but they are not allowed to be optional :/
surf = { version = "1.0.3", optional = true }
Expand All @@ -85,13 +90,7 @@ async-io = { version = "0.1.8", optional = true }
blocking = { version = "0.5.2", optional = true }
futures-lite = { version = "0.1.8", optional = true }

[target.'cfg(target_arch = "wasm32")'.dependencies]
futures-timer = { version = "3.0.2", optional = true, features = ["wasm-bindgen"] }
wasm-bindgen-futures = { version = "0.4.10", optional = true }
futures-channel = { version = "0.3.4", optional = true }

[target.'cfg(target_arch = "wasm32")'.dev-dependencies]
wasm-bindgen-test = "0.3.10"

[dependencies.tokio]
version = "0.2"
Expand All @@ -105,6 +104,7 @@ rand = "0.7.3"
tempfile = "3.1.0"
futures = "0.3.4"
rand_xorshift = "0.2.0"
wasm-bindgen-test = "0.3.10"

[[test]]
name = "stream"
Expand Down
11 changes: 6 additions & 5 deletions src/task/builder.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
use std::cell::Cell;
use std::future::Future;
use std::pin::Pin;
use std::sync::Arc;
Expand All @@ -7,7 +6,7 @@ use std::task::{Context, Poll};
use pin_project_lite::pin_project;

use crate::io;
use crate::task::{self, JoinHandle, Task, TaskLocalsWrapper};
use crate::task::{JoinHandle, Task, TaskLocalsWrapper};

/// Task builder that configures the settings of a new task.
#[derive(Debug, Default)]
Expand Down Expand Up @@ -61,7 +60,7 @@ impl Builder {
});

let task = wrapped.tag.task().clone();
let handle = task::executor::spawn(wrapped);
let handle = crate::task::executor::spawn(wrapped);

Ok(JoinHandle::new(handle, task))
}
Expand All @@ -81,7 +80,7 @@ impl Builder {
});

let task = wrapped.tag.task().clone();
let handle = task::executor::local(wrapped);
let handle = crate::task::executor::local(wrapped);

Ok(JoinHandle::new(handle, task))
}
Expand Down Expand Up @@ -143,6 +142,8 @@ impl Builder {
where
F: Future<Output = T>,
{
use std::cell::Cell;

let wrapped = self.build(future);

// Log this `block_on` operation.
Expand All @@ -167,7 +168,7 @@ impl Builder {
TaskLocalsWrapper::set_current(&wrapped.tag, || {
let res = if should_run {
// The first call should run the executor
task::executor::run(wrapped)
crate::task::executor::run(wrapped)
} else {
futures_lite::future::block_on(wrapped)
};
Expand Down
10 changes: 10 additions & 0 deletions src/task/join_handle.rs
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,17 @@ impl<T> Drop for JoinHandle<T> {
impl<T> Future for JoinHandle<T> {
type Output = T;

#[cfg(not(target_os = "unknown"))]
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
Pin::new(&mut self.handle.as_mut().unwrap()).poll(cx)
}

#[cfg(target_arch = "wasm32")]
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
match Pin::new(&mut self.handle.as_mut().unwrap()).poll(cx) {
Poll::Ready(Ok(t)) => Poll::Ready(t),
Poll::Ready(Err(_)) => unreachable!("channel must not be canceled"),
Poll::Pending => Poll::Pending,
}
}
}
6 changes: 2 additions & 4 deletions tests/collect.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#[cfg(feature = "unstable")]
#[test]
fn test_send() -> async_std::io::Result<()> {
fn test_send() {
use async_std::prelude::*;
use async_std::{stream, task};

Expand All @@ -14,7 +14,5 @@ fn test_send() -> async_std::io::Result<()> {

// This line triggers a compilation error
test_send_trait(&fut);

Ok(())
})
});
}