postgres-example: copy from basic-example

This commit is contained in:
asonix 2024-01-10 15:08:15 -06:00
parent ae90774bb1
commit e2e784f012
2 changed files with 114 additions and 4 deletions

View file

@ -6,5 +6,10 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies] [dependencies]
actix-rt = "2.9.0"
anyhow = "1.0.79"
background-jobs = { version = "0.17.0", features = ["background-jobs-postgres"], path = "../.." } background-jobs = { version = "0.17.0", features = ["background-jobs-postgres"], path = "../.." }
serde = { version = "1.0.195", features = ["derive"] }
tokio = { version = "1.35.1", features = ["full"] } tokio = { version = "1.35.1", features = ["full"] }
tracing = "0.1.40"
tracing-subscriber = { version = "0.3.18", features = ["env-filter"] }

View file

@ -1,9 +1,114 @@
use background_jobs::postgres::Storage; use actix_rt::Arbiter;
use background_jobs::{
postgres::Storage, ActixSpawner, MaxRetries, UnsendJob as Job, WorkerConfig,
};
// use background_jobs_sled_storage::Storage;
use std::{
future::{ready, Ready},
time::{Duration, SystemTime},
};
use tracing::info;
use tracing_subscriber::EnvFilter;
#[tokio::main] const DEFAULT_QUEUE: &str = "default";
async fn main() -> Result<(), Box<dyn std::error::Error>> {
#[derive(Clone, Debug)]
pub struct MyState {
pub app_name: String,
}
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
pub struct MyJob {
some_u64: u64,
other_u64: u64,
}
#[actix_rt::main]
async fn main() -> anyhow::Result<()> {
let env_filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
tracing_subscriber::fmt::fmt()
.with_env_filter(env_filter)
.init();
// Set up our Storage
let storage = let storage =
Storage::connect("postgres://postgres:postgres@localhost:5432/db".parse()?).await?; Storage::connect("postgres://postgres:postgres@localhost:5432/db".parse()?).await?;
println!("Hello, world!");
let arbiter = Arbiter::new();
// Configure and start our workers
let queue_handle =
WorkerConfig::new_in_arbiter(arbiter.handle(), storage, |_| MyState::new("My App"))
.register::<MyJob>()
.set_worker_count(DEFAULT_QUEUE, 16)
.start();
// Queue our jobs
queue_handle.queue(MyJob::new(1, 2)).await?;
queue_handle.queue(MyJob::new(3, 4)).await?;
queue_handle.queue(MyJob::new(5, 6)).await?;
for i in 0..20 {
queue_handle
.schedule(
MyJob::new(7 + i, 8 + i),
SystemTime::now() + Duration::from_secs(i),
)
.await?;
}
// Block on Actix
actix_rt::signal::ctrl_c().await?;
arbiter.stop();
let _ = arbiter.join();
Ok(()) Ok(())
} }
impl MyState {
pub fn new(app_name: &str) -> Self {
MyState {
app_name: app_name.to_owned(),
}
}
}
impl MyJob {
pub fn new(some_u64: u64, other_u64: u64) -> Self {
MyJob {
some_u64,
other_u64,
}
}
}
impl Job for MyJob {
type State = MyState;
type Future = Ready<anyhow::Result<()>>;
type Spawner = ActixSpawner;
// The name of the job. It is super important that each job has a unique name,
// because otherwise one job will overwrite another job when they're being
// registered.
const NAME: &'static str = "MyJob";
// The queue that this processor belongs to
//
// Workers have the option to subscribe to specific queues, so this is important to
// determine which worker will call the processor
//
// Jobs can optionally override the queue they're spawned on
const QUEUE: &'static str = DEFAULT_QUEUE;
// The number of times background-jobs should try to retry a job before giving up
//
// Jobs can optionally override this value
const MAX_RETRIES: MaxRetries = MaxRetries::Count(1);
fn run(self, state: MyState) -> Self::Future {
info!("{}: args, {:?}", state.app_name, self);
ready(Ok(()))
}
}