rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-29 19:10:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes to protocol based on https://github.com/neondatabase/neon/pull/8039

Browse Source
This commit is contained in:
Andrey Rudenko
2024-06-13 10:52:41 +02:00
parent ff55d6f4df
commit 3b2159310f
2 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
compute_tools/src/compute.rs
View File

@@ -14,6 +14,7 @@ use std::time;
use std::time::Instant;
use std::time::SystemTime;
use anyhow::bail;
use anyhow::{Context, Result};
use chrono::{DateTime, Utc};
use futures::future::join_all;
@@ -21,6 +22,7 @@ use futures::stream::FuturesUnordered;
use futures::StreamExt;
use nix::unistd::Pid;
use postgres::error::SqlState;
use postgres::SimpleQueryMessage;
use postgres::{Client, NoTls};
use tracing::{debug, error, info, instrument, warn};
use utils::id::{TenantId, TimelineId};
@@ -1452,9 +1454,20 @@ fn lsn_lease_request(configs: &[postgres::Config], cmd: &str) -> Result<u128> {
.iter()
.map(|config| {
let mut client = config.connect(NoTls)?;
let msg = client.query_one(cmd, &[])?;
let bytes: &[u8] = msg.try_get("valid_until")?;
Ok(u128::from_be_bytes(bytes.try_into()?))
let res = client.simple_query(cmd)?;
let msg = match res.get(0) {
Some(msg) => msg,
None => bail!("empty response"),
};
let row = match msg {
SimpleQueryMessage::Row(row) => row,
_ => bail!("error parsing lsn lease response"),
};
let valid_until_str = match row.get("valid_until") {
Some(valid_until) => valid_until,
None => bail!("valid_until not found"),
};
Ok(u128::from_str(valid_until_str)?)
})
.collect::<Result<Vec<u128>>>()?
.into_iter()