mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-17 02:12:56 +00:00
e8d2843df6
- Automatically set a node's availability to Active if it is responsive in startup_reconcile - Impose a 5s timeout of HTTP request to list location conf, so that an unresponsive node can't hang it for minutes - Do several retries if the request fails with a retryable error, to be tolerant of concurrent pageserver & storage controller restarts - Add a readiness hook for use with k8s so that we can tell when the startup reconciliaton is done and the service is fully ready to do work. - Add /metrics to the list of un-authenticated endpoints (this is unrelated but we're touching the line in this PR already, and it fixes auth error spam in deployed container.) - A test for the above. Closes: #6670
54 lines
1.3 KiB
Rust
54 lines
1.3 KiB
Rust
use tokio_util::task::{task_tracker::TaskTrackerToken, TaskTracker};
|
|
|
|
/// While a reference is kept around, the associated [`Barrier::wait`] will wait.
|
|
///
|
|
/// Can be cloned, moved and kept around in futures as "guard objects".
|
|
#[derive(Clone)]
|
|
pub struct Completion(TaskTrackerToken);
|
|
|
|
/// Barrier will wait until all clones of [`Completion`] have been dropped.
|
|
#[derive(Clone)]
|
|
pub struct Barrier(TaskTracker);
|
|
|
|
impl Default for Barrier {
|
|
fn default() -> Self {
|
|
let (_, rx) = channel();
|
|
rx
|
|
}
|
|
}
|
|
|
|
impl Barrier {
|
|
pub async fn wait(self) {
|
|
self.0.wait().await;
|
|
}
|
|
|
|
pub async fn maybe_wait(barrier: Option<Barrier>) {
|
|
if let Some(b) = barrier {
|
|
b.wait().await
|
|
}
|
|
}
|
|
|
|
/// Return true if a call to wait() would complete immediately
|
|
pub fn is_ready(&self) -> bool {
|
|
futures::future::FutureExt::now_or_never(self.0.wait()).is_some()
|
|
}
|
|
}
|
|
|
|
impl PartialEq for Barrier {
|
|
fn eq(&self, other: &Self) -> bool {
|
|
TaskTracker::ptr_eq(&self.0, &other.0)
|
|
}
|
|
}
|
|
|
|
impl Eq for Barrier {}
|
|
|
|
/// Create new Guard and Barrier pair.
|
|
pub fn channel() -> (Completion, Barrier) {
|
|
let tracker = TaskTracker::new();
|
|
// otherwise wait never exits
|
|
tracker.close();
|
|
|
|
let token = tracker.token();
|
|
(Completion(token), Barrier(tracker))
|
|
}
|