lune-org/lune
1
1
Fork 0
mirror of https://github.com/lune-org/lune.git synced 2025-04-04 10:30:54 +01:00
Code Issues Projects Releases Packages Wiki Activity

Figure out how to solve background tasks in new scheduler

Browse source
This commit is contained in:
Filip Tibell 2023-08-20 21:53:21 -05:00
parent 70caf89295
commit 3f5371a7c1
4 changed files with 72 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/lune/builtins/net/mod.rs
View file

@ -168,7 +168,9 @@ where
Ok(bound) => bound,
};
// Start up our web server
sched.schedule_future_background(async move {
// TODO: Spawn a scheduler background task here,
// and communicate using an mpsc channel instead
sched.schedule_future_background_local(async move {
bound
.http1_only(true) // Web sockets can only use http1
.http1_keepalive(true) // Web sockets must be kept alive

2
src/lune/builtins/net/server.rs
View file

@ -53,7 +53,7 @@ impl Service<Request<Body>> for NetServiceInner {
let sched = lua
.app_data_ref::<&Scheduler>()
.expect("Lua struct is missing scheduler");
sched.schedule_future_background(async move {
sched.schedule_future_background_local(async move {
// Create our new full websocket object, then
// schedule our handler to get called asap
let res = async move {

23
src/lune/builtins/process/mod.rs
View file

@ -7,7 +7,6 @@ use std::{
use dunce::canonicalize;
use mlua::prelude::*;
use os_str_bytes::RawOsString;
use tokio::task;
use crate::lune::{scheduler::Scheduler, util::TableBuilder};
@ -164,12 +163,24 @@ async fn process_spawn(
lua: &Lua,
(program, args, options): (String, Option<Vec<String>>, ProcessSpawnOptions),
) -> LuaResult<LuaTable> {
// Spawn the new process in the background,
// letting the tokio runtime place it on a
// different thread if necessary, and wait
let (status, stdout, stderr) = task::spawn(spawn_command(program, args, options))
/*
Spawn the new process in the background, letting the tokio
runtime place it on a different thread if possible / necessary
Note that we have to use our scheduler here, we can't
use anything like tokio::task::spawn because our lua
scheduler will not drive those futures to completion
*/
let sched = lua
.app_data_ref::<&Scheduler>()
.expect("Lua struct is missing scheduler");
let fut = spawn_command(program, args, options);
let recv = sched.schedule_future_background(fut);
let (status, stdout, stderr) = recv
.await
.expect("Spawned process should not be cancellable")?;
.expect("Failed to receive result of spawned process")?;
// NOTE: If an exit code was not given by the child process,
// we default to 1 if it yielded any error output, otherwise 0

56
src/lune/scheduler/impl_async.rs
View file

@ -1,5 +1,9 @@
use futures_util::Future;
use mlua::prelude::*;
use tokio::{
sync::oneshot::{self, Receiver},
task,
};
use super::{IntoLuaThread, Scheduler};
@ -29,18 +33,60 @@ where
/**
Schedules a plain future to run in the background.
Note that this will keep the scheduler alive even
if the future does not spawn any new lua threads.
This will spawn the future both on the scheduler and
potentially on a different thread using [`task::spawn`],
meaning the provided future must implement [`Send`].
Returns a [`Receiver`] which may be `await`-ed
to retrieve the result of the spawned future.
This [`Receiver`] may be safely ignored if the result of the
spawned future is not needed, the future will run either way.
*/
pub fn schedule_future_background<F>(&self, fut: F)
pub fn schedule_future_background<F, FR>(&self, fut: F) -> Receiver<FR>
where
F: Future<Output = ()> + 'static,
F: Future<Output = FR> + Send + 'static,
FR: Send + 'static,
{
let (tx, rx) = oneshot::channel();
let handle = task::spawn(async move {
let res = fut.await;
tx.send(res).ok();
});
let futs = self
.futures_background
.try_lock()
.expect("Failed to lock futures queue for background tasks");
futs.push(Box::pin(fut))
futs.push(Box::pin(async move {
handle.await.ok();
}));
rx
}
/**
Equivalent to [`schedule_future_background`], except the
future is only spawned on the scheduler, on the main thread.
*/
pub fn schedule_future_background_local<F, FR>(&self, fut: F) -> Receiver<FR>
where
F: Future<Output = FR> + 'static,
FR: 'static,
{
let (tx, rx) = oneshot::channel();
let futs = self
.futures_background
.try_lock()
.expect("Failed to lock futures queue for background tasks");
futs.push(Box::pin(async move {
let res = fut.await;
tx.send(res).ok();
}));
rx
}
/**