LNVPS/api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: delete unpaid vms

Browse Source
This commit is contained in:
kieran 2025-01-06 15:35:59 +00:00
parent 265d91dd83
commit de55d34086
No known key found for this signature in database
GPG Key ID: DE71CEB3925BE941
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/worker.rs
View File

@ -13,7 +13,7 @@ use lnvps_db::LNVpsDb;
use log::{debug, error, info}; use log::{debug, error, info};
use nostr::{EventBuilder, PublicKey}; use nostr::{EventBuilder, PublicKey};
use nostr_sdk::Client; use nostr_sdk::Client;
use std::ops::Add; use std::ops::{Add, Sub};
use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender}; use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender};
#[derive(Debug)] #[derive(Debug)]
@ -209,6 +209,7 @@ impl Worker {
} }
} }
// check VM status from db vm list
let db_vms = self.db.list_vms().await?; let db_vms = self.db.list_vms().await?;
for vm in db_vms { for vm in db_vms {
let state = if let Some(s) = self.vm_state_cache.get_state(vm.id).await { let state = if let Some(s) = self.vm_state_cache.get_state(vm.id).await {
@ -221,9 +222,16 @@ impl Worker {
None None
}; };
if state.is_none() && vm.expires > Utc::now() { // create VM if not spawned yet
if vm.expires > Utc::now() && state.is_none() {
self.check_vm(vm.id).await?; self.check_vm(vm.id).await?;
} }
// delete vm if not paid (in new state)
if vm.expires < Utc::now().sub(Days::new(1)) && state.is_none() {
info!("Deleting unpaid VM {}", vm.id);
self.provisioner.delete_vm(vm.id).await?;
}
} }
self.last_check_vms = Utc::now(); self.last_check_vms = Utc::now();