fix: delete_after_days
This commit is contained in:
parent
91cf173058
commit
3a5093d3c3
@ -52,7 +52,12 @@ async fn main() -> Result<(), Error> {
|
|||||||
.get_provisioner(db.clone(), lnd.clone(), exchange.clone());
|
.get_provisioner(db.clone(), lnd.clone(), exchange.clone());
|
||||||
worker_provisioner.init().await?;
|
worker_provisioner.init().await?;
|
||||||
|
|
||||||
let mut worker = Worker::new(db.clone(), worker_provisioner, status.clone());
|
let mut worker = Worker::new(
|
||||||
|
db.clone(),
|
||||||
|
worker_provisioner,
|
||||||
|
settings.delete_after,
|
||||||
|
status.clone(),
|
||||||
|
);
|
||||||
let sender = worker.sender();
|
let sender = worker.sender();
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
loop {
|
loop {
|
||||||
|
@ -21,6 +21,7 @@ pub enum WorkJob {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub struct Worker {
|
pub struct Worker {
|
||||||
|
delete_after: u16,
|
||||||
db: Box<dyn LNVpsDb>,
|
db: Box<dyn LNVpsDb>,
|
||||||
provisioner: Box<dyn Provisioner>,
|
provisioner: Box<dyn Provisioner>,
|
||||||
vm_state_cache: VmStateCache,
|
vm_state_cache: VmStateCache,
|
||||||
@ -32,6 +33,7 @@ impl Worker {
|
|||||||
pub fn new<D: LNVpsDb + Clone + 'static, P: Provisioner + 'static>(
|
pub fn new<D: LNVpsDb + Clone + 'static, P: Provisioner + 'static>(
|
||||||
db: D,
|
db: D,
|
||||||
provisioner: P,
|
provisioner: P,
|
||||||
|
delete_after: u16,
|
||||||
vm_state_cache: VmStateCache,
|
vm_state_cache: VmStateCache,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
let (tx, rx) = unbounded_channel();
|
let (tx, rx) = unbounded_channel();
|
||||||
@ -39,6 +41,7 @@ impl Worker {
|
|||||||
db: Box::new(db),
|
db: Box::new(db),
|
||||||
provisioner: Box::new(provisioner),
|
provisioner: Box::new(provisioner),
|
||||||
vm_state_cache,
|
vm_state_cache,
|
||||||
|
delete_after,
|
||||||
tx,
|
tx,
|
||||||
rx,
|
rx,
|
||||||
}
|
}
|
||||||
@ -73,7 +76,7 @@ impl Worker {
|
|||||||
self.provisioner.stop_vm(db_vm.id).await?;
|
self.provisioner.stop_vm(db_vm.id).await?;
|
||||||
}
|
}
|
||||||
// Delete VM if expired > 3 days
|
// Delete VM if expired > 3 days
|
||||||
if db_vm.expires.add(Days::new(3)) < Utc::now() {
|
if db_vm.expires.add(Days::new(self.delete_after as u64)) < Utc::now() {
|
||||||
info!("Deleting expired VM {}", db_vm.id);
|
info!("Deleting expired VM {}", db_vm.id);
|
||||||
self.provisioner.delete_vm(db_vm.id).await?;
|
self.provisioner.delete_vm(db_vm.id).await?;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user