mirror of
https://github.com/neondatabase/neon.git
synced 2026-05-31 12:00:42 +00:00
10dae79c6d
Closes https://github.com/neondatabase/neon/issues/3114 Adds more typization into errors that appear during protocol messages (`FeMessage`), postgres and walreceiver connections. Socket IO errors are now better detected and logged with lesser (INFO, DEBUG) error level, without traces that they were logged before, when they were wrapped in anyhow context.
63 lines
2.1 KiB
Rust
63 lines
2.1 KiB
Rust
//!
|
|
//! WAL service listens for client connections and
|
|
//! receive WAL from wal_proposer and send it to WAL receivers
|
|
//!
|
|
use regex::Regex;
|
|
use std::net::{TcpListener, TcpStream};
|
|
use std::thread;
|
|
use tracing::*;
|
|
use utils::postgres_backend_async::QueryError;
|
|
|
|
use crate::handler::SafekeeperPostgresHandler;
|
|
use crate::SafeKeeperConf;
|
|
use utils::postgres_backend::{AuthType, PostgresBackend};
|
|
|
|
/// Accept incoming TCP connections and spawn them into a background thread.
|
|
pub fn thread_main(conf: SafeKeeperConf, listener: TcpListener) -> ! {
|
|
loop {
|
|
match listener.accept() {
|
|
Ok((socket, peer_addr)) => {
|
|
debug!("accepted connection from {}", peer_addr);
|
|
let conf = conf.clone();
|
|
|
|
let _ = thread::Builder::new()
|
|
.name("WAL service thread".into())
|
|
.spawn(move || {
|
|
if let Err(err) = handle_socket(socket, conf) {
|
|
error!("connection handler exited: {}", err);
|
|
}
|
|
})
|
|
.unwrap();
|
|
}
|
|
Err(e) => error!("Failed to accept connection: {}", e),
|
|
}
|
|
}
|
|
}
|
|
|
|
// Get unique thread id (Rust internal), with ThreadId removed for shorter printing
|
|
fn get_tid() -> u64 {
|
|
let tids = format!("{:?}", thread::current().id());
|
|
let r = Regex::new(r"ThreadId\((\d+)\)").unwrap();
|
|
let caps = r.captures(&tids).unwrap();
|
|
caps.get(1).unwrap().as_str().parse().unwrap()
|
|
}
|
|
|
|
/// This is run by `thread_main` above, inside a background thread.
|
|
///
|
|
fn handle_socket(socket: TcpStream, conf: SafeKeeperConf) -> Result<(), QueryError> {
|
|
let _enter = info_span!("", tid = ?get_tid()).entered();
|
|
|
|
socket.set_nodelay(true)?;
|
|
|
|
let auth_type = match conf.auth {
|
|
None => AuthType::Trust,
|
|
Some(_) => AuthType::NeonJWT,
|
|
};
|
|
let mut conn_handler = SafekeeperPostgresHandler::new(conf);
|
|
let pgbackend = PostgresBackend::new(socket, auth_type, None, false)?;
|
|
// libpq replication protocol between safekeeper and replicas/pagers
|
|
pgbackend.run(&mut conn_handler)?;
|
|
|
|
Ok(())
|
|
}
|