mirror of
https://github.com/neondatabase/neon.git
synced 2026-06-04 14:00:38 +00:00
9bc12f7444
Re-export only things that are used by other modules. In the future, I'm imagining that we run bindgen twice, for Postgres v14 and v15. The two sets of bindings would go into separate 'bindings_v14' and 'bindings_v15' modules. Rearrange postgres_ffi modules. Move function, to avoid Postgres version dependency in timelines.rs Move function to generate a logical-message WAL record to postgres_ffi.
84 lines
3.0 KiB
Rust
84 lines
3.0 KiB
Rust
//!
|
|
//! Common utilities for dealing with PostgreSQL non-relation files.
|
|
//!
|
|
use crate::transaction_id_precedes;
|
|
use super::pg_constants;
|
|
use bytes::BytesMut;
|
|
use log::*;
|
|
|
|
use super::bindings::MultiXactId;
|
|
|
|
pub fn transaction_id_set_status(xid: u32, status: u8, page: &mut BytesMut) {
|
|
trace!(
|
|
"handle_apply_request for RM_XACT_ID-{} (1-commit, 2-abort, 3-sub_commit)",
|
|
status
|
|
);
|
|
|
|
let byteno: usize = ((xid as u32 % pg_constants::CLOG_XACTS_PER_PAGE as u32)
|
|
/ pg_constants::CLOG_XACTS_PER_BYTE) as usize;
|
|
|
|
let bshift: u8 =
|
|
((xid % pg_constants::CLOG_XACTS_PER_BYTE) * pg_constants::CLOG_BITS_PER_XACT as u32) as u8;
|
|
|
|
page[byteno] =
|
|
(page[byteno] & !(pg_constants::CLOG_XACT_BITMASK << bshift)) | (status << bshift);
|
|
}
|
|
|
|
pub fn transaction_id_get_status(xid: u32, page: &[u8]) -> u8 {
|
|
let byteno: usize = ((xid as u32 % pg_constants::CLOG_XACTS_PER_PAGE as u32)
|
|
/ pg_constants::CLOG_XACTS_PER_BYTE) as usize;
|
|
|
|
let bshift: u8 =
|
|
((xid % pg_constants::CLOG_XACTS_PER_BYTE) * pg_constants::CLOG_BITS_PER_XACT as u32) as u8;
|
|
|
|
((page[byteno] >> bshift) & pg_constants::CLOG_XACT_BITMASK) as u8
|
|
}
|
|
|
|
// See CLOGPagePrecedes in clog.c
|
|
pub const fn clogpage_precedes(page1: u32, page2: u32) -> bool {
|
|
let mut xid1 = page1 * pg_constants::CLOG_XACTS_PER_PAGE;
|
|
xid1 += pg_constants::FIRST_NORMAL_TRANSACTION_ID + 1;
|
|
let mut xid2 = page2 * pg_constants::CLOG_XACTS_PER_PAGE;
|
|
xid2 += pg_constants::FIRST_NORMAL_TRANSACTION_ID + 1;
|
|
|
|
transaction_id_precedes(xid1, xid2)
|
|
&& transaction_id_precedes(xid1, xid2 + pg_constants::CLOG_XACTS_PER_PAGE - 1)
|
|
}
|
|
|
|
// See SlruMayDeleteSegment() in slru.c
|
|
pub fn slru_may_delete_clogsegment(segpage: u32, cutoff_page: u32) -> bool {
|
|
let seg_last_page = segpage + pg_constants::SLRU_PAGES_PER_SEGMENT - 1;
|
|
|
|
assert_eq!(segpage % pg_constants::SLRU_PAGES_PER_SEGMENT, 0);
|
|
|
|
clogpage_precedes(segpage, cutoff_page) && clogpage_precedes(seg_last_page, cutoff_page)
|
|
}
|
|
|
|
// Multixact utils
|
|
|
|
pub fn mx_offset_to_flags_offset(xid: MultiXactId) -> usize {
|
|
((xid / pg_constants::MULTIXACT_MEMBERS_PER_MEMBERGROUP as u32) as u16
|
|
% pg_constants::MULTIXACT_MEMBERGROUPS_PER_PAGE
|
|
* pg_constants::MULTIXACT_MEMBERGROUP_SIZE) as usize
|
|
}
|
|
|
|
pub fn mx_offset_to_flags_bitshift(xid: MultiXactId) -> u16 {
|
|
(xid as u16) % pg_constants::MULTIXACT_MEMBERS_PER_MEMBERGROUP
|
|
* pg_constants::MXACT_MEMBER_BITS_PER_XACT
|
|
}
|
|
|
|
/* Location (byte offset within page) of TransactionId of given member */
|
|
pub fn mx_offset_to_member_offset(xid: MultiXactId) -> usize {
|
|
mx_offset_to_flags_offset(xid)
|
|
+ (pg_constants::MULTIXACT_FLAGBYTES_PER_GROUP
|
|
+ (xid as u16 % pg_constants::MULTIXACT_MEMBERS_PER_MEMBERGROUP) * 4) as usize
|
|
}
|
|
|
|
fn mx_offset_to_member_page(xid: u32) -> u32 {
|
|
xid / pg_constants::MULTIXACT_MEMBERS_PER_PAGE as u32
|
|
}
|
|
|
|
pub fn mx_offset_to_member_segment(xid: u32) -> i32 {
|
|
(mx_offset_to_member_page(xid) / pg_constants::SLRU_PAGES_PER_SEGMENT) as i32
|
|
}
|