rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-16 09:52:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

use chrono::DateTime for Poisoned errors

Browse Source
This commit is contained in:
Christian Schwarz
2024-03-21 15:31:24 +00:00
parent 3a5994bb43
commit 3dfc7de99b
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
libs/utils/src/poison.rs
View File

@@ -24,8 +24,6 @@
//! # });
//! ```
use std::time::Instant;
use tracing::warn;
pub struct Poison<T> {
@@ -38,7 +36,7 @@ pub struct Poison<T> {
enum State {
Clean,
Armed,
Poisoned { at: Instant },
Poisoned { at: chrono::DateTime<chrono::Utc> },
}
impl<T> Poison<T> {
@@ -88,7 +86,7 @@ impl<'a, T> Guard<'a, T> {
self.0.state = State::Clean;
}
State::Poisoned { at } => {
unreachable!("we fail check_and_arm() if it's in that state: {at:?}")
unreachable!("we fail check_and_arm() if it's in that state: {at}")
}
}
}
@@ -102,12 +100,12 @@ impl<'a, T> Drop for Guard<'a, T> {
}
State::Armed => {
// still armed => poison it
let at = Instant::now();
let at = chrono::Utc::now();
self.0.state = State::Poisoned { at };
warn!(at=?at, "poisoning {}", self.0.what);
}
State::Poisoned { at } => {
unreachable!("we fail check_and_arm() if it's in that state: {at:?}")
unreachable!("we fail check_and_arm() if it's in that state: {at}")
}
}
}
@@ -115,6 +113,9 @@ impl<'a, T> Drop for Guard<'a, T> {
#[derive(thiserror::Error, Debug)]
pub enum Error {
#[error("poisoned at {at:?}: {what}")]
Poisoned { what: &'static str, at: Instant },
#[error("poisoned at {at}: {what}")]
Poisoned {
what: &'static str,
at: chrono::DateTime<chrono::Utc>,
},
}