mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-06 21:12:55 +00:00
920040e402
Updates storage components to edition 2024. We like to stay on the latest edition if possible. There is no functional changes, however some code changes had to be done to accommodate the edition's breaking changes. The PR has two commits: * the first commit updates storage crates to edition 2024 and appeases `cargo clippy` by changing code. i have accidentially ran the formatter on some files that had other edits. * the second commit performs a `cargo fmt` I would recommend a closer review of the first commit and a less close review of the second one (as it just runs `cargo fmt`). part of https://github.com/neondatabase/neon/issues/10918
670 lines
22 KiB
Rust
670 lines
22 KiB
Rust
use std::cell::{RefCell, RefMut, UnsafeCell};
|
|
use std::ffi::CStr;
|
|
use std::sync::Arc;
|
|
|
|
use bytes::Bytes;
|
|
use desim::executor::{self, PollSome};
|
|
use desim::network::TCP;
|
|
use desim::node_os::NodeOs;
|
|
use desim::proto::{AnyMessage, NetEvent, NodeEvent};
|
|
use desim::world::NodeId;
|
|
use tracing::debug;
|
|
use utils::lsn::Lsn;
|
|
use walproposer::api_bindings::Level;
|
|
use walproposer::bindings::{
|
|
NeonWALReadResult, SafekeeperStateDesiredEvents, WL_SOCKET_READABLE, WL_SOCKET_WRITEABLE,
|
|
};
|
|
use walproposer::walproposer::{ApiImpl, Config};
|
|
|
|
use super::walproposer_disk::DiskWalProposer;
|
|
|
|
/// Special state for each wp->sk connection.
|
|
struct SafekeeperConn {
|
|
host: String,
|
|
port: String,
|
|
node_id: NodeId,
|
|
// socket is Some(..) equals to connection is established
|
|
socket: Option<TCP>,
|
|
// connection is in progress
|
|
is_connecting: bool,
|
|
// START_WAL_PUSH is in progress
|
|
is_start_wal_push: bool,
|
|
// pointer to Safekeeper in walproposer for callbacks
|
|
raw_ptr: *mut walproposer::bindings::Safekeeper,
|
|
}
|
|
|
|
impl SafekeeperConn {
|
|
pub fn new(host: String, port: String) -> Self {
|
|
// port number is the same as NodeId
|
|
let port_num = port.parse::<u32>().unwrap();
|
|
Self {
|
|
host,
|
|
port,
|
|
node_id: port_num,
|
|
socket: None,
|
|
is_connecting: false,
|
|
is_start_wal_push: false,
|
|
raw_ptr: std::ptr::null_mut(),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Simulation version of a postgres WaitEventSet. At pos 0 there is always
|
|
/// a special NodeEvents channel, which is used as a latch.
|
|
struct EventSet {
|
|
os: NodeOs,
|
|
// all pollable channels, 0 is always NodeEvent channel
|
|
chans: Vec<Box<dyn PollSome>>,
|
|
// 0 is always nullptr
|
|
sk_ptrs: Vec<*mut walproposer::bindings::Safekeeper>,
|
|
// event mask for each channel
|
|
masks: Vec<u32>,
|
|
}
|
|
|
|
impl EventSet {
|
|
pub fn new(os: NodeOs) -> Self {
|
|
let node_events = os.node_events();
|
|
Self {
|
|
os,
|
|
chans: vec![Box::new(node_events)],
|
|
sk_ptrs: vec![std::ptr::null_mut()],
|
|
masks: vec![WL_SOCKET_READABLE],
|
|
}
|
|
}
|
|
|
|
/// Leaves all readable channels at the beginning of the array.
|
|
fn sort_readable(&mut self) -> usize {
|
|
let mut cnt = 1;
|
|
for i in 1..self.chans.len() {
|
|
if self.masks[i] & WL_SOCKET_READABLE != 0 {
|
|
self.chans.swap(i, cnt);
|
|
self.sk_ptrs.swap(i, cnt);
|
|
self.masks.swap(i, cnt);
|
|
cnt += 1;
|
|
}
|
|
}
|
|
cnt
|
|
}
|
|
|
|
fn update_event_set(&mut self, conn: &SafekeeperConn, event_mask: u32) {
|
|
let index = self
|
|
.sk_ptrs
|
|
.iter()
|
|
.position(|&ptr| ptr == conn.raw_ptr)
|
|
.expect("safekeeper should exist in event set");
|
|
self.masks[index] = event_mask;
|
|
}
|
|
|
|
fn add_safekeeper(&mut self, sk: &SafekeeperConn, event_mask: u32) {
|
|
for ptr in self.sk_ptrs.iter() {
|
|
assert!(*ptr != sk.raw_ptr);
|
|
}
|
|
|
|
self.chans.push(Box::new(
|
|
sk.socket
|
|
.as_ref()
|
|
.expect("socket should not be closed")
|
|
.recv_chan(),
|
|
));
|
|
self.sk_ptrs.push(sk.raw_ptr);
|
|
self.masks.push(event_mask);
|
|
}
|
|
|
|
fn remove_safekeeper(&mut self, sk: &SafekeeperConn) {
|
|
let index = self.sk_ptrs.iter().position(|&ptr| ptr == sk.raw_ptr);
|
|
if index.is_none() {
|
|
debug!("remove_safekeeper: sk={:?} not found", sk.raw_ptr);
|
|
return;
|
|
}
|
|
let index = index.unwrap();
|
|
|
|
self.chans.remove(index);
|
|
self.sk_ptrs.remove(index);
|
|
self.masks.remove(index);
|
|
|
|
// to simulate the actual behaviour
|
|
self.refresh_event_set();
|
|
}
|
|
|
|
/// Updates all masks to match the result of a SafekeeperStateDesiredEvents.
|
|
fn refresh_event_set(&mut self) {
|
|
for (i, mask) in self.masks.iter_mut().enumerate() {
|
|
if i == 0 {
|
|
continue;
|
|
}
|
|
|
|
let mut mask_sk: u32 = 0;
|
|
let mut mask_nwr: u32 = 0;
|
|
unsafe { SafekeeperStateDesiredEvents(self.sk_ptrs[i], &mut mask_sk, &mut mask_nwr) };
|
|
|
|
if mask_sk != *mask {
|
|
debug!(
|
|
"refresh_event_set: sk={:?}, old_mask={:#b}, new_mask={:#b}",
|
|
self.sk_ptrs[i], *mask, mask_sk
|
|
);
|
|
*mask = mask_sk;
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Wait for events on all channels.
|
|
fn wait(&mut self, timeout_millis: i64) -> walproposer::walproposer::WaitResult {
|
|
// all channels are always writeable
|
|
for (i, mask) in self.masks.iter().enumerate() {
|
|
if *mask & WL_SOCKET_WRITEABLE != 0 {
|
|
return walproposer::walproposer::WaitResult::Network(
|
|
self.sk_ptrs[i],
|
|
WL_SOCKET_WRITEABLE,
|
|
);
|
|
}
|
|
}
|
|
|
|
let cnt = self.sort_readable();
|
|
|
|
let slice = &self.chans[0..cnt];
|
|
match executor::epoll_chans(slice, timeout_millis) {
|
|
None => walproposer::walproposer::WaitResult::Timeout,
|
|
Some(0) => {
|
|
let msg = self.os.node_events().must_recv();
|
|
match msg {
|
|
NodeEvent::Internal(AnyMessage::Just32(0)) => {
|
|
// got a notification about new WAL available
|
|
}
|
|
NodeEvent::Internal(_) => unreachable!(),
|
|
NodeEvent::Accept(_) => unreachable!(),
|
|
}
|
|
walproposer::walproposer::WaitResult::Latch
|
|
}
|
|
Some(index) => walproposer::walproposer::WaitResult::Network(
|
|
self.sk_ptrs[index],
|
|
WL_SOCKET_READABLE,
|
|
),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// This struct handles all calls from walproposer into walproposer_api.
|
|
pub struct SimulationApi {
|
|
os: NodeOs,
|
|
safekeepers: RefCell<Vec<SafekeeperConn>>,
|
|
disk: Arc<DiskWalProposer>,
|
|
redo_start_lsn: Option<Lsn>,
|
|
last_logged_commit_lsn: u64,
|
|
shmem: UnsafeCell<walproposer::bindings::WalproposerShmemState>,
|
|
config: Config,
|
|
event_set: RefCell<Option<EventSet>>,
|
|
}
|
|
|
|
pub struct Args {
|
|
pub os: NodeOs,
|
|
pub config: Config,
|
|
pub disk: Arc<DiskWalProposer>,
|
|
pub redo_start_lsn: Option<Lsn>,
|
|
}
|
|
|
|
impl SimulationApi {
|
|
pub fn new(args: Args) -> Self {
|
|
// initialize connection state for each safekeeper
|
|
let sk_conns = args
|
|
.config
|
|
.safekeepers_list
|
|
.iter()
|
|
.map(|s| {
|
|
SafekeeperConn::new(
|
|
s.split(':').next().unwrap().to_string(),
|
|
s.split(':').nth(1).unwrap().to_string(),
|
|
)
|
|
})
|
|
.collect::<Vec<_>>();
|
|
|
|
Self {
|
|
os: args.os,
|
|
safekeepers: RefCell::new(sk_conns),
|
|
disk: args.disk,
|
|
redo_start_lsn: args.redo_start_lsn,
|
|
last_logged_commit_lsn: 0,
|
|
shmem: UnsafeCell::new(walproposer::api_bindings::empty_shmem()),
|
|
config: args.config,
|
|
event_set: RefCell::new(None),
|
|
}
|
|
}
|
|
|
|
/// Get SafekeeperConn for the given Safekeeper.
|
|
fn get_conn(&self, sk: &mut walproposer::bindings::Safekeeper) -> RefMut<'_, SafekeeperConn> {
|
|
let sk_port = unsafe { CStr::from_ptr(sk.port).to_str().unwrap() };
|
|
let state = self.safekeepers.borrow_mut();
|
|
RefMut::map(state, |v| {
|
|
v.iter_mut()
|
|
.find(|conn| conn.port == sk_port)
|
|
.expect("safekeeper conn not found by port")
|
|
})
|
|
}
|
|
}
|
|
|
|
impl ApiImpl for SimulationApi {
|
|
fn get_current_timestamp(&self) -> i64 {
|
|
debug!("get_current_timestamp");
|
|
// PG TimestampTZ is microseconds, but simulation unit is assumed to be
|
|
// milliseconds, so add 10^3
|
|
self.os.now() as i64 * 1000
|
|
}
|
|
|
|
fn update_donor(&self, donor: &mut walproposer::bindings::Safekeeper, donor_lsn: u64) {
|
|
let mut shmem = unsafe { *self.get_shmem_state() };
|
|
shmem.propEpochStartLsn.value = donor_lsn;
|
|
shmem.donor_conninfo = donor.conninfo;
|
|
}
|
|
|
|
fn conn_status(
|
|
&self,
|
|
_: &mut walproposer::bindings::Safekeeper,
|
|
) -> walproposer::bindings::WalProposerConnStatusType {
|
|
debug!("conn_status");
|
|
// break the connection with a 10% chance
|
|
if self.os.random(100) < 10 {
|
|
walproposer::bindings::WalProposerConnStatusType_WP_CONNECTION_BAD
|
|
} else {
|
|
walproposer::bindings::WalProposerConnStatusType_WP_CONNECTION_OK
|
|
}
|
|
}
|
|
|
|
fn conn_connect_start(&self, sk: &mut walproposer::bindings::Safekeeper) {
|
|
debug!("conn_connect_start");
|
|
let mut conn = self.get_conn(sk);
|
|
|
|
assert!(conn.socket.is_none());
|
|
let socket = self.os.open_tcp(conn.node_id);
|
|
conn.socket = Some(socket);
|
|
conn.raw_ptr = sk;
|
|
conn.is_connecting = true;
|
|
}
|
|
|
|
fn conn_connect_poll(
|
|
&self,
|
|
_: &mut walproposer::bindings::Safekeeper,
|
|
) -> walproposer::bindings::WalProposerConnectPollStatusType {
|
|
debug!("conn_connect_poll");
|
|
// TODO: break the connection here
|
|
walproposer::bindings::WalProposerConnectPollStatusType_WP_CONN_POLLING_OK
|
|
}
|
|
|
|
fn conn_send_query(&self, sk: &mut walproposer::bindings::Safekeeper, query: &str) -> bool {
|
|
debug!("conn_send_query: {}", query);
|
|
self.get_conn(sk).is_start_wal_push = true;
|
|
true
|
|
}
|
|
|
|
fn conn_get_query_result(
|
|
&self,
|
|
_: &mut walproposer::bindings::Safekeeper,
|
|
) -> walproposer::bindings::WalProposerExecStatusType {
|
|
debug!("conn_get_query_result");
|
|
// TODO: break the connection here
|
|
walproposer::bindings::WalProposerExecStatusType_WP_EXEC_SUCCESS_COPYBOTH
|
|
}
|
|
|
|
fn conn_async_read(
|
|
&self,
|
|
sk: &mut walproposer::bindings::Safekeeper,
|
|
vec: &mut Vec<u8>,
|
|
) -> walproposer::bindings::PGAsyncReadResult {
|
|
debug!("conn_async_read");
|
|
let mut conn = self.get_conn(sk);
|
|
|
|
let socket = if let Some(socket) = conn.socket.as_mut() {
|
|
socket
|
|
} else {
|
|
// socket is already closed
|
|
return walproposer::bindings::PGAsyncReadResult_PG_ASYNC_READ_FAIL;
|
|
};
|
|
|
|
let msg = socket.recv_chan().try_recv();
|
|
|
|
match msg {
|
|
None => {
|
|
// no message is ready
|
|
walproposer::bindings::PGAsyncReadResult_PG_ASYNC_READ_TRY_AGAIN
|
|
}
|
|
Some(NetEvent::Closed) => {
|
|
// connection is closed
|
|
debug!("conn_async_read: connection is closed");
|
|
conn.socket = None;
|
|
walproposer::bindings::PGAsyncReadResult_PG_ASYNC_READ_FAIL
|
|
}
|
|
Some(NetEvent::Message(msg)) => {
|
|
// got a message
|
|
let b = match msg {
|
|
desim::proto::AnyMessage::Bytes(b) => b,
|
|
_ => unreachable!(),
|
|
};
|
|
vec.extend_from_slice(&b);
|
|
walproposer::bindings::PGAsyncReadResult_PG_ASYNC_READ_SUCCESS
|
|
}
|
|
}
|
|
}
|
|
|
|
fn conn_blocking_write(&self, sk: &mut walproposer::bindings::Safekeeper, buf: &[u8]) -> bool {
|
|
let mut conn = self.get_conn(sk);
|
|
debug!("conn_blocking_write to {}: {:?}", conn.node_id, buf);
|
|
let socket = conn.socket.as_mut().unwrap();
|
|
socket.send(desim::proto::AnyMessage::Bytes(Bytes::copy_from_slice(buf)));
|
|
true
|
|
}
|
|
|
|
fn conn_async_write(
|
|
&self,
|
|
sk: &mut walproposer::bindings::Safekeeper,
|
|
buf: &[u8],
|
|
) -> walproposer::bindings::PGAsyncWriteResult {
|
|
let mut conn = self.get_conn(sk);
|
|
debug!("conn_async_write to {}: {:?}", conn.node_id, buf);
|
|
if let Some(socket) = conn.socket.as_mut() {
|
|
socket.send(desim::proto::AnyMessage::Bytes(Bytes::copy_from_slice(buf)));
|
|
} else {
|
|
// connection is already closed
|
|
debug!("conn_async_write: writing to a closed socket!");
|
|
// TODO: maybe we should return error here?
|
|
}
|
|
walproposer::bindings::PGAsyncWriteResult_PG_ASYNC_WRITE_SUCCESS
|
|
}
|
|
|
|
fn wal_reader_allocate(&self, _: &mut walproposer::bindings::Safekeeper) -> NeonWALReadResult {
|
|
debug!("wal_reader_allocate");
|
|
walproposer::bindings::NeonWALReadResult_NEON_WALREAD_SUCCESS
|
|
}
|
|
|
|
fn wal_read(
|
|
&self,
|
|
_sk: &mut walproposer::bindings::Safekeeper,
|
|
buf: &mut [u8],
|
|
startpos: u64,
|
|
) -> NeonWALReadResult {
|
|
self.disk.lock().read(startpos, buf);
|
|
walproposer::bindings::NeonWALReadResult_NEON_WALREAD_SUCCESS
|
|
}
|
|
|
|
fn init_event_set(&self, _: &mut walproposer::bindings::WalProposer) {
|
|
debug!("init_event_set");
|
|
let new_event_set = EventSet::new(self.os.clone());
|
|
let old_event_set = self.event_set.replace(Some(new_event_set));
|
|
assert!(old_event_set.is_none());
|
|
}
|
|
|
|
fn update_event_set(&self, sk: &mut walproposer::bindings::Safekeeper, event_mask: u32) {
|
|
debug!(
|
|
"update_event_set, sk={:?}, events_mask={:#b}",
|
|
sk as *mut walproposer::bindings::Safekeeper, event_mask
|
|
);
|
|
let conn = self.get_conn(sk);
|
|
|
|
self.event_set
|
|
.borrow_mut()
|
|
.as_mut()
|
|
.unwrap()
|
|
.update_event_set(&conn, event_mask);
|
|
}
|
|
|
|
fn add_safekeeper_event_set(
|
|
&self,
|
|
sk: &mut walproposer::bindings::Safekeeper,
|
|
event_mask: u32,
|
|
) {
|
|
debug!(
|
|
"add_safekeeper_event_set, sk={:?}, events_mask={:#b}",
|
|
sk as *mut walproposer::bindings::Safekeeper, event_mask
|
|
);
|
|
|
|
self.event_set
|
|
.borrow_mut()
|
|
.as_mut()
|
|
.unwrap()
|
|
.add_safekeeper(&self.get_conn(sk), event_mask);
|
|
}
|
|
|
|
fn rm_safekeeper_event_set(&self, sk: &mut walproposer::bindings::Safekeeper) {
|
|
debug!(
|
|
"rm_safekeeper_event_set, sk={:?}",
|
|
sk as *mut walproposer::bindings::Safekeeper,
|
|
);
|
|
|
|
self.event_set
|
|
.borrow_mut()
|
|
.as_mut()
|
|
.unwrap()
|
|
.remove_safekeeper(&self.get_conn(sk));
|
|
}
|
|
|
|
fn active_state_update_event_set(&self, sk: &mut walproposer::bindings::Safekeeper) {
|
|
debug!("active_state_update_event_set");
|
|
|
|
assert!(sk.state == walproposer::bindings::SafekeeperState_SS_ACTIVE);
|
|
self.event_set
|
|
.borrow_mut()
|
|
.as_mut()
|
|
.unwrap()
|
|
.refresh_event_set();
|
|
}
|
|
|
|
fn wal_reader_events(&self, _sk: &mut walproposer::bindings::Safekeeper) -> u32 {
|
|
0
|
|
}
|
|
|
|
fn wait_event_set(
|
|
&self,
|
|
_: &mut walproposer::bindings::WalProposer,
|
|
timeout_millis: i64,
|
|
) -> walproposer::walproposer::WaitResult {
|
|
// TODO: handle multiple stages as part of the simulation (e.g. connect, start_wal_push, etc)
|
|
let mut conns = self.safekeepers.borrow_mut();
|
|
for conn in conns.iter_mut() {
|
|
if conn.socket.is_some() && conn.is_connecting {
|
|
conn.is_connecting = false;
|
|
debug!("wait_event_set, connecting to {}:{}", conn.host, conn.port);
|
|
return walproposer::walproposer::WaitResult::Network(
|
|
conn.raw_ptr,
|
|
WL_SOCKET_READABLE | WL_SOCKET_WRITEABLE,
|
|
);
|
|
}
|
|
if conn.socket.is_some() && conn.is_start_wal_push {
|
|
conn.is_start_wal_push = false;
|
|
debug!(
|
|
"wait_event_set, start wal push to {}:{}",
|
|
conn.host, conn.port
|
|
);
|
|
return walproposer::walproposer::WaitResult::Network(
|
|
conn.raw_ptr,
|
|
WL_SOCKET_READABLE,
|
|
);
|
|
}
|
|
}
|
|
drop(conns);
|
|
|
|
let res = self
|
|
.event_set
|
|
.borrow_mut()
|
|
.as_mut()
|
|
.unwrap()
|
|
.wait(timeout_millis);
|
|
|
|
debug!(
|
|
"wait_event_set, timeout_millis={}, res={:?}",
|
|
timeout_millis, res,
|
|
);
|
|
res
|
|
}
|
|
|
|
fn strong_random(&self, buf: &mut [u8]) -> bool {
|
|
debug!("strong_random");
|
|
buf.fill(0);
|
|
true
|
|
}
|
|
|
|
fn finish_sync_safekeepers(&self, lsn: u64) {
|
|
debug!("finish_sync_safekeepers, lsn={}", lsn);
|
|
executor::exit(0, Lsn(lsn).to_string());
|
|
}
|
|
|
|
fn log_internal(&self, _wp: &mut walproposer::bindings::WalProposer, level: Level, msg: &str) {
|
|
debug!("wp_log[{}] {}", level, msg);
|
|
if level == Level::Fatal || level == Level::Panic {
|
|
if msg.contains("rejects our connection request with term") {
|
|
// collected quorum with lower term, then got rejected by next connected safekeeper
|
|
executor::exit(1, msg.to_owned());
|
|
}
|
|
if msg.contains("collected propEpochStartLsn") && msg.contains(", but basebackup LSN ")
|
|
{
|
|
// sync-safekeepers collected wrong quorum, walproposer collected another quorum
|
|
executor::exit(1, msg.to_owned());
|
|
}
|
|
if msg.contains("failed to download WAL for logical replicaiton") {
|
|
// Recovery connection broken and recovery was failed
|
|
executor::exit(1, msg.to_owned());
|
|
}
|
|
if msg.contains("missing majority of votes, collected") {
|
|
// Voting bug when safekeeper disconnects after voting
|
|
executor::exit(1, msg.to_owned());
|
|
}
|
|
panic!("unknown FATAL error from walproposer: {}", msg);
|
|
}
|
|
}
|
|
|
|
fn after_election(&self, wp: &mut walproposer::bindings::WalProposer) {
|
|
let prop_lsn = wp.propEpochStartLsn;
|
|
let prop_term = wp.propTerm;
|
|
|
|
let mut prev_lsn: u64 = 0;
|
|
let mut prev_term: u64 = 0;
|
|
|
|
unsafe {
|
|
let history = wp.propTermHistory.entries;
|
|
let len = wp.propTermHistory.n_entries as usize;
|
|
if len > 1 {
|
|
let entry = *history.wrapping_add(len - 2);
|
|
prev_lsn = entry.lsn;
|
|
prev_term = entry.term;
|
|
}
|
|
}
|
|
|
|
let msg = format!(
|
|
"prop_elected;{};{};{};{}",
|
|
prop_lsn, prop_term, prev_lsn, prev_term
|
|
);
|
|
|
|
debug!(msg);
|
|
self.os.log_event(msg);
|
|
}
|
|
|
|
fn get_redo_start_lsn(&self) -> u64 {
|
|
debug!("get_redo_start_lsn -> {:?}", self.redo_start_lsn);
|
|
self.redo_start_lsn.expect("redo_start_lsn is not set").0
|
|
}
|
|
|
|
fn get_shmem_state(&self) -> *mut walproposer::bindings::WalproposerShmemState {
|
|
self.shmem.get()
|
|
}
|
|
|
|
fn start_streaming(
|
|
&self,
|
|
startpos: u64,
|
|
callback: &walproposer::walproposer::StreamingCallback,
|
|
) {
|
|
let disk = &self.disk;
|
|
let disk_lsn = disk.lock().flush_rec_ptr().0;
|
|
debug!("start_streaming at {} (disk_lsn={})", startpos, disk_lsn);
|
|
if startpos < disk_lsn {
|
|
debug!(
|
|
"startpos < disk_lsn, it means we wrote some transaction even before streaming started"
|
|
);
|
|
}
|
|
assert!(startpos <= disk_lsn);
|
|
let mut broadcasted = Lsn(startpos);
|
|
|
|
loop {
|
|
let available = disk.lock().flush_rec_ptr();
|
|
assert!(available >= broadcasted);
|
|
callback.broadcast(broadcasted, available);
|
|
broadcasted = available;
|
|
callback.poll();
|
|
}
|
|
}
|
|
|
|
fn process_safekeeper_feedback(
|
|
&mut self,
|
|
wp: &mut walproposer::bindings::WalProposer,
|
|
_sk: &mut walproposer::bindings::Safekeeper,
|
|
) {
|
|
debug!("process_safekeeper_feedback, commit_lsn={}", wp.commitLsn);
|
|
if wp.commitLsn > self.last_logged_commit_lsn {
|
|
self.os.log_event(format!("commit_lsn;{}", wp.commitLsn));
|
|
self.last_logged_commit_lsn = wp.commitLsn;
|
|
}
|
|
}
|
|
|
|
fn get_flush_rec_ptr(&self) -> u64 {
|
|
let lsn = self.disk.lock().flush_rec_ptr();
|
|
debug!("get_flush_rec_ptr: {}", lsn);
|
|
lsn.0
|
|
}
|
|
|
|
fn recovery_download(
|
|
&self,
|
|
wp: &mut walproposer::bindings::WalProposer,
|
|
sk: &mut walproposer::bindings::Safekeeper,
|
|
) -> bool {
|
|
let mut startpos = wp.truncateLsn;
|
|
let endpos = wp.propEpochStartLsn;
|
|
|
|
if startpos == endpos {
|
|
debug!("recovery_download: nothing to download");
|
|
return true;
|
|
}
|
|
|
|
debug!("recovery_download from {} to {}", startpos, endpos,);
|
|
|
|
let replication_prompt = format!(
|
|
"START_REPLICATION {} {} {} {}",
|
|
self.config.ttid.tenant_id, self.config.ttid.timeline_id, startpos, endpos,
|
|
);
|
|
let async_conn = self.get_conn(sk);
|
|
|
|
let conn = self.os.open_tcp(async_conn.node_id);
|
|
conn.send(desim::proto::AnyMessage::Bytes(replication_prompt.into()));
|
|
|
|
let chan = conn.recv_chan();
|
|
while startpos < endpos {
|
|
let event = chan.recv();
|
|
match event {
|
|
NetEvent::Closed => {
|
|
debug!("connection closed in recovery");
|
|
break;
|
|
}
|
|
NetEvent::Message(AnyMessage::Bytes(b)) => {
|
|
debug!("got recovery bytes from safekeeper");
|
|
self.disk.lock().write(startpos, &b);
|
|
startpos += b.len() as u64;
|
|
}
|
|
NetEvent::Message(_) => unreachable!(),
|
|
}
|
|
}
|
|
|
|
debug!("recovery finished at {}", startpos);
|
|
|
|
startpos == endpos
|
|
}
|
|
|
|
fn conn_finish(&self, sk: &mut walproposer::bindings::Safekeeper) {
|
|
let mut conn = self.get_conn(sk);
|
|
debug!("conn_finish to {}", conn.node_id);
|
|
if let Some(socket) = conn.socket.as_mut() {
|
|
socket.close();
|
|
} else {
|
|
// connection is already closed
|
|
}
|
|
conn.socket = None;
|
|
}
|
|
|
|
fn conn_error_message(&self, _sk: &mut walproposer::bindings::Safekeeper) -> String {
|
|
"connection is closed, probably".into()
|
|
}
|
|
}
|