mirror of
https://git.asonix.dog/asonix/background-jobs.git
synced 2024-11-21 19:40:59 +00:00
Add ActixJob to handle spawinging ?Send futures
This commit is contained in:
parent
799391811c
commit
ca1c073666
6 changed files with 116 additions and 2 deletions
|
@ -14,7 +14,7 @@ actix = "0.10.0-alpha.2"
|
|||
actix-rt = "1.0.0"
|
||||
anyhow = "1.0"
|
||||
async-trait = "0.1.24"
|
||||
background-jobs-core = { version = "0.7", path = "../jobs-core" }
|
||||
background-jobs-core = { version = "0.7", path = "../jobs-core", features = ["with-actix"] }
|
||||
chrono = "0.4"
|
||||
log = "0.4"
|
||||
num_cpus = "1.10.0"
|
||||
|
|
|
@ -139,6 +139,8 @@ mod worker;
|
|||
|
||||
use self::{every::every, server::Server, worker::local_worker};
|
||||
|
||||
pub use background_jobs_core::ActixJob;
|
||||
|
||||
/// Create a new Server
|
||||
///
|
||||
/// In previous versions of this library, the server itself was run on it's own dedicated threads
|
||||
|
|
|
@ -9,7 +9,12 @@ keywords = ["jobs", "processor"]
|
|||
readme = "../README.md"
|
||||
edition = "2018"
|
||||
|
||||
[features]
|
||||
default = []
|
||||
with-actix = ["actix", "tokio"]
|
||||
|
||||
[dependencies]
|
||||
actix = { version = "0.10.0-alpha.1", optional = true }
|
||||
anyhow = "1.0"
|
||||
async-trait = "0.1.24"
|
||||
chrono = { version = "0.4", features = ["serde"] }
|
||||
|
@ -18,4 +23,5 @@ log = "0.4"
|
|||
serde = { version = "1.0", features = ["derive"] }
|
||||
serde_json = "1.0"
|
||||
thiserror = "1.0"
|
||||
tokio = { version = "0.2.13", optional = true }
|
||||
uuid = { version = "0.8.1", features = ["serde", "v4"] }
|
||||
|
|
101
jobs-core/src/actix_job.rs
Normal file
101
jobs-core/src/actix_job.rs
Normal file
|
@ -0,0 +1,101 @@
|
|||
use crate::{Backoff, Job, MaxRetries, Processor};
|
||||
use anyhow::Error;
|
||||
use log::error;
|
||||
use serde::{de::DeserializeOwned, ser::Serialize};
|
||||
use std::{future::Future, pin::Pin};
|
||||
use tokio::sync::oneshot;
|
||||
|
||||
/// The ActixJob trait defines parameters pertaining to an instance of background job
|
||||
///
|
||||
/// This trait is specific to Actix, and will automatically implement the Job trait with the
|
||||
/// proper translation from ?Send futures to Send futures
|
||||
pub trait ActixJob: Serialize + DeserializeOwned + 'static {
|
||||
/// The processor this job is associated with. The job's processor can be used to create a
|
||||
/// JobInfo from a job, which is used to serialize the job into a storage mechanism.
|
||||
type Processor: Processor<Job = Self>;
|
||||
|
||||
/// The application state provided to this job at runtime.
|
||||
type State: Clone + 'static;
|
||||
|
||||
/// The future returned by this job
|
||||
type Future: Future<Output = Result<(), Error>>;
|
||||
|
||||
/// Users of this library must define what it means to run a job.
|
||||
///
|
||||
/// This should contain all the logic needed to complete a job. If that means queuing more
|
||||
/// jobs, sending an email, shelling out (don't shell out), or doing otherwise lengthy
|
||||
/// processes, that logic should all be called from inside this method.
|
||||
///
|
||||
/// The state passed into this job is initialized at the start of the application. The state
|
||||
/// argument could be useful for containing a hook into something like r2d2, or the address of
|
||||
/// an actor in an actix-based system.
|
||||
fn run(self, state: Self::State) -> Self::Future;
|
||||
|
||||
/// If this job should not use the default queue for its processor, this can be overridden in
|
||||
/// user-code.
|
||||
///
|
||||
/// Jobs will only be processed by processors that are registered, and if a queue is supplied
|
||||
/// here that is not associated with a valid processor for this job, it will never be
|
||||
/// processed.
|
||||
fn queue(&self) -> Option<&str> {
|
||||
None
|
||||
}
|
||||
|
||||
/// If this job should not use the default maximum retry count for its processor, this can be
|
||||
/// overridden in user-code.
|
||||
fn max_retries(&self) -> Option<MaxRetries> {
|
||||
None
|
||||
}
|
||||
|
||||
/// If this job should not use the default backoff strategy for its processor, this can be
|
||||
/// overridden in user-code.
|
||||
fn backoff_strategy(&self) -> Option<Backoff> {
|
||||
None
|
||||
}
|
||||
|
||||
/// Define the maximum number of milliseconds this job should be allowed to run before being
|
||||
/// considered dead.
|
||||
///
|
||||
/// This is important for allowing the job server to reap processes that were started but never
|
||||
/// completed.
|
||||
fn timeout(&self) -> Option<i64> {
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> Job for T
|
||||
where
|
||||
T: ActixJob,
|
||||
{
|
||||
type Processor = T::Processor;
|
||||
type State = T::State;
|
||||
type Future = Pin<Box<dyn Future<Output = Result<(), Error>> + Send>>;
|
||||
|
||||
fn run(self, state: Self::State) -> Self::Future {
|
||||
let (tx, rx) = oneshot::channel();
|
||||
|
||||
actix::spawn(async move {
|
||||
if let Err(_) = tx.send(ActixJob::run(self, state).await) {
|
||||
error!("Job dropped");
|
||||
}
|
||||
});
|
||||
|
||||
Box::pin(async move { rx.await? })
|
||||
}
|
||||
|
||||
fn queue(&self) -> Option<&str> {
|
||||
ActixJob::queue(self)
|
||||
}
|
||||
|
||||
fn max_retries(&self) -> Option<MaxRetries> {
|
||||
ActixJob::max_retries(self)
|
||||
}
|
||||
|
||||
fn backoff_strategy(&self) -> Option<Backoff> {
|
||||
ActixJob::backoff_strategy(self)
|
||||
}
|
||||
|
||||
fn timeout(&self) -> Option<i64> {
|
||||
ActixJob::timeout(self)
|
||||
}
|
||||
}
|
|
@ -8,6 +8,8 @@
|
|||
|
||||
use anyhow::Error;
|
||||
|
||||
#[cfg(feature = "with-actix")]
|
||||
mod actix_job;
|
||||
mod job;
|
||||
mod job_info;
|
||||
mod processor;
|
||||
|
@ -24,6 +26,9 @@ pub use crate::{
|
|||
storage::{memory_storage, Storage},
|
||||
};
|
||||
|
||||
#[cfg(feature = "with-actix")]
|
||||
pub use actix_job::ActixJob;
|
||||
|
||||
#[derive(Debug, thiserror::Error)]
|
||||
/// The error type returned by a `Processor`'s `process` method
|
||||
pub enum JobError {
|
||||
|
|
|
@ -208,7 +208,7 @@ pub use background_jobs_core::{
|
|||
};
|
||||
|
||||
#[cfg(feature = "background-jobs-actix")]
|
||||
pub use background_jobs_actix::{create_server, QueueHandle, WorkerConfig};
|
||||
pub use background_jobs_actix::{create_server, ActixJob, QueueHandle, WorkerConfig};
|
||||
|
||||
#[cfg(feature = "background-jobs-sled-storage")]
|
||||
pub mod sled_storage {
|
||||
|
|
Loading…
Reference in a new issue