| 104 | 104 | | } |
| 105 | 105 | | } |
| 106 | 106 | | |
| 107 | + | /// What a free workspace is told when an upload would take its artifacts |
| 108 | + | /// and cache past `cap`, holding `held` and adding `adding`. None while it |
| 109 | + | /// fits, or with no cap. |
| 110 | + | pub fn storage_refusal(held: u64, adding: u64, cap: u64) -> Option<String> { |
| 111 | + | (cap > 0 && held.saturating_add(adding) > cap).then(|| { |
| 112 | + | format!( |
| 113 | + | "This workspace's artifacts and cache hold {} of the {} a workspace on the free plan keeps, so this upload is not kept. Delete artifacts you no longer need, let old cache entries expire, or start the g1t plan, which has no such cap.", |
| 114 | + | shown_bytes(held), |
| 115 | + | shown_bytes(cap) |
| 116 | + | ) |
| 117 | + | }) |
| 118 | + | } |
| 119 | + | |
| 120 | + | impl crate::Actions { |
| 121 | + | /// Why an upload of `adding` bytes to the workspace's artifacts or |
| 122 | + | /// cache is refused: only past FREE_ACTIONS_STORAGE_BYTES, and only |
| 123 | + | /// without the plan. Billing is asked only once the workspace is past |
| 124 | + | /// the cap, and when it does not answer the upload goes ahead. |
| 125 | + | pub(crate) async fn storage_refused(&self, namespace: &str, adding: u64) -> worker::Result<Option<String>> { |
| 126 | + | let cap = self.limits.free_actions_storage_bytes; |
| 127 | + | if cap == 0 { |
| 128 | + | return Ok(None); |
| 129 | + | } |
| 130 | + | #[derive(serde::Deserialize)] |
| 131 | + | struct Held { |
| 132 | + | bytes: Option<f64>, |
| 133 | + | } |
| 134 | + | let held = self |
| 135 | + | .db |
| 136 | + | .prepare( |
| 137 | + | "SELECT SUM(size) AS bytes FROM ( |
| 138 | + | SELECT size FROM cache_entries WHERE namespace = ?1 AND status IN ('ready', 'pending') |
| 139 | + | UNION ALL SELECT size FROM artifacts WHERE namespace = ?1 AND status IN ('ready', 'pending') |
| 140 | + | )", |
| 141 | + | ) |
| 142 | + | .bind(&[namespace.into()])? |
| 143 | + | .first::<Held>(None) |
| 144 | + | .await? |
| 145 | + | .and_then(|held| held.bytes) |
| 146 | + | .unwrap_or(0.0) as u64; |
| 147 | + | let Some(why) = storage_refusal(held, adding, cap) else { |
| 148 | + | return Ok(None); |
| 149 | + | }; |
| 150 | + | Ok(match self.standing(namespace).await { |
| 151 | + | Some(standing) if !standing.has_plan => Some(why), |
| 152 | + | _ => None, |
| 153 | + | }) |
| 154 | + | } |
| 155 | + | } |
| 156 | + | |
| 157 | + | #[cfg(test)] |
| 158 | + | mod storage_tests { |
| 159 | + | use super::storage_refusal; |
| 160 | + | |
| 161 | + | const GB: u64 = 1024 * 1024 * 1024; |
| 162 | + | |
| 163 | + | #[test] |
| 164 | + | fn a_free_workspace_keeps_two_gigabytes_of_artifacts_and_cache() { |
| 165 | + | assert_eq!(storage_refusal(GB, GB, 2 * GB), None); |
| 166 | + | let refused = storage_refusal(2 * GB - 10, 100, 2 * GB).unwrap(); |
| 167 | + | assert!(refused.contains("of the 2 GB"), "{refused}"); |
| 168 | + | assert!(refused.contains("start the g1t plan"), "{refused}"); |
| 169 | + | assert_eq!(storage_refusal(100 * GB, GB, 0), None); |
| 170 | + | } |
| 171 | + | } |
| 172 | + | |
| 107 | 173 | | #[cfg(test)] |
| 108 | 174 | | mod room_tests { |
| 109 | 175 | | use super::self_hosted_room; |
