LCOV - code coverage report
Current view: top level - libs/walproposer/src - api_bindings.rs (source / functions) Coverage Total Hit
Test: bf6f7808c062fc948868e31e1c34d9d09c9893fb.info Lines: 93.2 % 368 343
Test Date: 2025-07-31 15:49:28 Functions: 94.9 % 39 37

            Line data    Source code
       1              : //! A C-Rust shim: defines implementation of C walproposer API, assuming wp
       2              : //! callback_data stores Box to some Rust implementation.
       3              : 
       4              : #![allow(dead_code)]
       5              : 
       6              : use std::ffi::{CStr, CString};
       7              : 
       8              : use crate::bindings::{
       9              :     NeonWALReadResult, PGAsyncReadResult, PGAsyncWriteResult, Safekeeper, Size, StringInfoData,
      10              :     TimestampTz, WalProposer, WalProposerConnStatusType, WalProposerConnectPollStatusType,
      11              :     WalProposerExecStatusType, WalproposerShmemState, XLogRecPtr, uint32, walproposer_api,
      12              : };
      13              : use crate::walproposer::{ApiImpl, StreamingCallback, WaitResult};
      14              : 
      15          950 : extern "C" fn get_shmem_state(wp: *mut WalProposer) -> *mut WalproposerShmemState {
      16              :     unsafe {
      17          950 :         let callback_data = (*(*wp).config).callback_data;
      18          950 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      19          950 :         (*api).get_shmem_state()
      20              :     }
      21          950 : }
      22              : 
      23          147 : extern "C-unwind" fn start_streaming(wp: *mut WalProposer, startpos: XLogRecPtr) {
      24          147 :     unsafe {
      25          147 :         let callback_data = (*(*wp).config).callback_data;
      26          147 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      27          147 :         let callback = StreamingCallback::new(wp);
      28          147 :         (*api).start_streaming(startpos, &callback);
      29          147 :     }
      30          147 : }
      31              : 
      32          115 : extern "C" fn get_flush_rec_ptr(wp: *mut WalProposer) -> XLogRecPtr {
      33              :     unsafe {
      34          115 :         let callback_data = (*(*wp).config).callback_data;
      35          115 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      36          115 :         (*api).get_flush_rec_ptr()
      37              :     }
      38          115 : }
      39              : 
      40          892 : extern "C" fn update_donor(wp: *mut WalProposer, donor: *mut Safekeeper, donor_lsn: XLogRecPtr) {
      41              :     unsafe {
      42          892 :         let callback_data = (*(*wp).config).callback_data;
      43          892 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      44          892 :         (*api).update_donor(&mut (*donor), donor_lsn)
      45              :     }
      46          892 : }
      47              : 
      48       330378 : extern "C" fn get_current_timestamp(wp: *mut WalProposer) -> TimestampTz {
      49              :     unsafe {
      50       330378 :         let callback_data = (*(*wp).config).callback_data;
      51       330378 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      52       330378 :         (*api).get_current_timestamp()
      53              :     }
      54       330378 : }
      55              : 
      56        12365 : extern "C" fn conn_error_message(sk: *mut Safekeeper) -> *mut ::std::os::raw::c_char {
      57              :     unsafe {
      58        12365 :         let callback_data = (*(*(*sk).wp).config).callback_data;
      59        12365 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      60        12365 :         let msg = (*api).conn_error_message(&mut (*sk));
      61        12365 :         let msg = CString::new(msg).unwrap();
      62              :         // TODO: fix leaking error message
      63        12365 :         msg.into_raw()
      64              :     }
      65        12365 : }
      66              : 
      67        34959 : extern "C" fn conn_status(sk: *mut Safekeeper) -> WalProposerConnStatusType {
      68              :     unsafe {
      69        34959 :         let callback_data = (*(*(*sk).wp).config).callback_data;
      70        34959 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      71        34959 :         (*api).conn_status(&mut (*sk))
      72              :     }
      73        34959 : }
      74              : 
      75        34959 : extern "C" fn conn_connect_start(sk: *mut Safekeeper) {
      76              :     unsafe {
      77        34959 :         let callback_data = (*(*(*sk).wp).config).callback_data;
      78        34959 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      79        34959 :         (*api).conn_connect_start(&mut (*sk))
      80              :     }
      81        34959 : }
      82              : 
      83        31394 : extern "C" fn conn_connect_poll(sk: *mut Safekeeper) -> WalProposerConnectPollStatusType {
      84              :     unsafe {
      85        31394 :         let callback_data = (*(*(*sk).wp).config).callback_data;
      86        31394 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      87        31394 :         (*api).conn_connect_poll(&mut (*sk))
      88              :     }
      89        31394 : }
      90              : 
      91        31394 : extern "C" fn conn_send_query(sk: *mut Safekeeper, query: *mut ::std::os::raw::c_char) -> bool {
      92        31394 :     let query = unsafe { CStr::from_ptr(query) };
      93        31394 :     let query = query.to_str().unwrap();
      94              : 
      95              :     unsafe {
      96        31394 :         let callback_data = (*(*(*sk).wp).config).callback_data;
      97        31394 :         let api = callback_data as *mut Box<dyn ApiImpl>;
      98        31394 :         (*api).conn_send_query(&mut (*sk), query)
      99              :     }
     100        31394 : }
     101              : 
     102        31394 : extern "C" fn conn_get_query_result(sk: *mut Safekeeper) -> WalProposerExecStatusType {
     103              :     unsafe {
     104        31394 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     105        31394 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     106        31394 :         (*api).conn_get_query_result(&mut (*sk))
     107              :     }
     108        31394 : }
     109              : 
     110            0 : extern "C" fn conn_flush(sk: *mut Safekeeper) -> ::std::os::raw::c_int {
     111              :     unsafe {
     112            0 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     113            0 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     114            0 :         (*api).conn_flush(&mut (*sk))
     115              :     }
     116            0 : }
     117              : 
     118        12661 : extern "C" fn conn_finish(sk: *mut Safekeeper) {
     119              :     unsafe {
     120        12661 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     121        12661 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     122        12661 :         (*api).conn_finish(&mut (*sk))
     123              :     }
     124        12661 : }
     125              : 
     126        19112 : extern "C" fn conn_async_read(
     127        19112 :     sk: *mut Safekeeper,
     128        19112 :     buf: *mut *mut ::std::os::raw::c_char,
     129        19112 :     amount: *mut ::std::os::raw::c_int,
     130        19112 : ) -> PGAsyncReadResult {
     131              :     unsafe {
     132        19112 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     133        19112 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     134              : 
     135              :         // This function has guarantee that returned buf will be valid until
     136              :         // the next call. So we can store a Vec in each Safekeeper and reuse
     137              :         // it on the next call.
     138        19112 :         let mut inbuf = take_vec_u8(&mut (*sk).inbuf).unwrap_or_default();
     139        19112 :         inbuf.clear();
     140              : 
     141        19112 :         let result = (*api).conn_async_read(&mut (*sk), &mut inbuf);
     142              : 
     143              :         // Put a Vec back to sk->inbuf and return data ptr.
     144        19112 :         *amount = inbuf.len() as i32;
     145        19112 :         *buf = store_vec_u8(&mut (*sk).inbuf, inbuf);
     146              : 
     147        19112 :         result
     148              :     }
     149        19112 : }
     150              : 
     151         4879 : extern "C" fn conn_async_write(
     152         4879 :     sk: *mut Safekeeper,
     153         4879 :     buf: *const ::std::os::raw::c_void,
     154         4879 :     size: usize,
     155         4879 : ) -> PGAsyncWriteResult {
     156              :     unsafe {
     157         4879 :         let buf = std::slice::from_raw_parts(buf as *const u8, size);
     158         4879 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     159         4879 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     160         4879 :         (*api).conn_async_write(&mut (*sk), buf)
     161              :     }
     162         4879 : }
     163              : 
     164        34755 : extern "C" fn conn_blocking_write(
     165        34755 :     sk: *mut Safekeeper,
     166        34755 :     buf: *const ::std::os::raw::c_void,
     167        34755 :     size: usize,
     168        34755 : ) -> bool {
     169              :     unsafe {
     170        34755 :         let buf = std::slice::from_raw_parts(buf as *const u8, size);
     171        34755 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     172        34755 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     173        34755 :         (*api).conn_blocking_write(&mut (*sk), buf)
     174              :     }
     175        34755 : }
     176              : 
     177          474 : extern "C-unwind" fn recovery_download(wp: *mut WalProposer, sk: *mut Safekeeper) -> bool {
     178              :     unsafe {
     179          474 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     180          474 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     181              : 
     182              :         // currently `recovery_download` is always called right after election
     183          474 :         (*api).after_election(&mut (*wp));
     184              : 
     185          474 :         (*api).recovery_download(&mut (*wp), &mut (*sk))
     186              :     }
     187          474 : }
     188              : 
     189          903 : extern "C" fn wal_reader_allocate(sk: *mut Safekeeper) {
     190          903 :     unsafe {
     191          903 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     192          903 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     193          903 :         (*api).wal_reader_allocate(&mut (*sk));
     194          903 :     }
     195          903 : }
     196              : 
     197              : #[allow(clippy::unnecessary_cast)]
     198         1111 : extern "C" fn wal_read(
     199         1111 :     sk: *mut Safekeeper,
     200         1111 :     buf: *mut ::std::os::raw::c_char,
     201         1111 :     startptr: XLogRecPtr,
     202         1111 :     count: Size,
     203         1111 :     _errmsg: *mut *mut ::std::os::raw::c_char,
     204         1111 : ) -> NeonWALReadResult {
     205              :     unsafe {
     206         1111 :         let buf = std::slice::from_raw_parts_mut(buf as *mut u8, count);
     207         1111 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     208         1111 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     209              :         // TODO: errmsg is not forwarded
     210         1111 :         (*api).wal_read(&mut (*sk), buf, startptr)
     211              :     }
     212         1111 : }
     213              : 
     214        18831 : extern "C" fn wal_reader_events(sk: *mut Safekeeper) -> uint32 {
     215              :     unsafe {
     216        18831 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     217        18831 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     218        18831 :         (*api).wal_reader_events(&mut (*sk))
     219              :     }
     220        18831 : }
     221              : 
     222         9344 : extern "C" fn init_event_set(wp: *mut WalProposer) {
     223         9344 :     unsafe {
     224         9344 :         let callback_data = (*(*wp).config).callback_data;
     225         9344 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     226         9344 :         (*api).init_event_set(&mut (*wp));
     227         9344 :     }
     228         9344 : }
     229              : 
     230        67902 : extern "C" fn update_event_set(sk: *mut Safekeeper, events: uint32) {
     231        67902 :     unsafe {
     232        67902 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     233        67902 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     234        67902 :         (*api).update_event_set(&mut (*sk), events);
     235        67902 :     }
     236        67902 : }
     237              : 
     238         6126 : extern "C" fn active_state_update_event_set(sk: *mut Safekeeper) {
     239         6126 :     unsafe {
     240         6126 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     241         6126 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     242         6126 :         (*api).active_state_update_event_set(&mut (*sk));
     243         6126 :     }
     244         6126 : }
     245              : 
     246        62788 : extern "C" fn add_safekeeper_event_set(sk: *mut Safekeeper, events: uint32) {
     247        62788 :     unsafe {
     248        62788 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     249        62788 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     250        62788 :         (*api).add_safekeeper_event_set(&mut (*sk), events);
     251        62788 :     }
     252        62788 : }
     253              : 
     254        40490 : extern "C" fn rm_safekeeper_event_set(sk: *mut Safekeeper) {
     255        40490 :     unsafe {
     256        40490 :         let callback_data = (*(*(*sk).wp).config).callback_data;
     257        40490 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     258        40490 :         (*api).rm_safekeeper_event_set(&mut (*sk));
     259        40490 :     }
     260        40490 : }
     261              : 
     262        92222 : extern "C-unwind" fn wait_event_set(
     263        92222 :     wp: *mut WalProposer,
     264        92222 :     timeout: ::std::os::raw::c_long,
     265        92222 :     event_sk: *mut *mut Safekeeper,
     266        92222 :     events: *mut uint32,
     267        92222 : ) -> ::std::os::raw::c_int {
     268              :     unsafe {
     269        92222 :         let callback_data = (*(*wp).config).callback_data;
     270        92222 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     271        92222 :         let result = (*api).wait_event_set(&mut (*wp), timeout);
     272        92222 :         match result {
     273              :             WaitResult::Latch => {
     274         9431 :                 *event_sk = std::ptr::null_mut();
     275         9431 :                 *events = crate::bindings::WL_LATCH_SET;
     276         9431 :                 1
     277              :             }
     278              :             WaitResult::Timeout => {
     279         2894 :                 *event_sk = std::ptr::null_mut();
     280              :                 // WaitEventSetWait returns 0 for timeout.
     281         2894 :                 *events = 0;
     282         2894 :                 0
     283              :             }
     284        79897 :             WaitResult::Network(sk, event_mask) => {
     285        79897 :                 *event_sk = sk;
     286        79897 :                 *events = event_mask;
     287        79897 :                 1
     288              :             }
     289              :         }
     290              :     }
     291        92222 : }
     292              : 
     293            0 : extern "C" fn strong_random(
     294            0 :     wp: *mut WalProposer,
     295            0 :     buf: *mut ::std::os::raw::c_void,
     296            0 :     len: usize,
     297            0 : ) -> bool {
     298              :     unsafe {
     299            0 :         let buf = std::slice::from_raw_parts_mut(buf as *mut u8, len);
     300            0 :         let callback_data = (*(*wp).config).callback_data;
     301            0 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     302            0 :         (*api).strong_random(buf)
     303              :     }
     304            0 : }
     305              : 
     306          202 : extern "C" fn get_redo_start_lsn(wp: *mut WalProposer) -> XLogRecPtr {
     307              :     unsafe {
     308          202 :         let callback_data = (*(*wp).config).callback_data;
     309          202 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     310          202 :         (*api).get_redo_start_lsn()
     311              :     }
     312          202 : }
     313              : 
     314          328 : unsafe extern "C-unwind" fn finish_sync_safekeepers(wp: *mut WalProposer, lsn: XLogRecPtr) -> ! {
     315              :     unsafe {
     316          328 :         let callback_data = (*(*wp).config).callback_data;
     317          328 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     318          328 :         (*api).finish_sync_safekeepers(lsn)
     319              :     }
     320              : }
     321              : 
     322         2351 : extern "C" fn process_safekeeper_feedback(wp: *mut WalProposer, sk: *mut Safekeeper) {
     323         2351 :     unsafe {
     324         2351 :         let callback_data = (*(*wp).config).callback_data;
     325         2351 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     326         2351 :         (*api).process_safekeeper_feedback(&mut (*wp), &mut (*sk));
     327         2351 :     }
     328         2351 : }
     329              : 
     330       155495 : extern "C-unwind" fn log_internal(
     331       155495 :     wp: *mut WalProposer,
     332       155495 :     level: ::std::os::raw::c_int,
     333       155495 :     line: *const ::std::os::raw::c_char,
     334       155495 : ) {
     335              :     unsafe {
     336       155495 :         let callback_data = (*(*wp).config).callback_data;
     337       155495 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     338       155495 :         let line = CStr::from_ptr(line);
     339       155495 :         let line = line.to_str().unwrap();
     340       155495 :         (*api).log_internal(&mut (*wp), Level::from(level as u32), line)
     341              :     }
     342       155495 : }
     343              : 
     344              : /* BEGIN_HADRON */
     345         9344 : extern "C" fn reset_safekeeper_statuses_for_metrics(wp: *mut WalProposer, num_safekeepers: u32) {
     346              :     unsafe {
     347         9344 :         let callback_data = (*(*wp).config).callback_data;
     348         9344 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     349         9344 :         if api.is_null() {
     350            0 :             return;
     351         9344 :         }
     352         9344 :         (*api).reset_safekeeper_statuses_for_metrics(&mut (*wp), num_safekeepers);
     353              :     }
     354         9344 : }
     355              : 
     356         9999 : extern "C" fn update_safekeeper_status_for_metrics(
     357         9999 :     wp: *mut WalProposer,
     358         9999 :     sk_index: u32,
     359         9999 :     status: u8,
     360         9999 : ) {
     361              :     unsafe {
     362         9999 :         let callback_data = (*(*wp).config).callback_data;
     363         9999 :         let api = callback_data as *mut Box<dyn ApiImpl>;
     364         9999 :         if api.is_null() {
     365            0 :             return;
     366         9999 :         }
     367         9999 :         (*api).update_safekeeper_status_for_metrics(&mut (*wp), sk_index, status);
     368              :     }
     369         9999 : }
     370              : /* END_HADRON */
     371              : 
     372              : #[derive(Debug, PartialEq)]
     373              : pub enum Level {
     374              :     Debug5,
     375              :     Debug4,
     376              :     Debug3,
     377              :     Debug2,
     378              :     Debug1,
     379              :     Log,
     380              :     Info,
     381              :     Notice,
     382              :     Warning,
     383              :     Error,
     384              :     Fatal,
     385              :     Panic,
     386              :     WPEvent,
     387              : }
     388              : 
     389              : impl Level {
     390       155495 :     pub fn from(elevel: u32) -> Level {
     391              :         use crate::bindings::*;
     392              : 
     393       155495 :         match elevel {
     394         3976 :             DEBUG5 => Level::Debug5,
     395            0 :             DEBUG4 => Level::Debug4,
     396            0 :             DEBUG3 => Level::Debug3,
     397         2351 :             DEBUG2 => Level::Debug2,
     398            0 :             DEBUG1 => Level::Debug1,
     399       135081 :             LOG => Level::Log,
     400            0 :             INFO => Level::Info,
     401            0 :             NOTICE => Level::Notice,
     402        14018 :             WARNING => Level::Warning,
     403            0 :             ERROR => Level::Error,
     404           68 :             FATAL => Level::Fatal,
     405            1 :             PANIC => Level::Panic,
     406            0 :             WPEVENT => Level::WPEvent,
     407            0 :             _ => panic!("unknown log level {elevel}"),
     408              :         }
     409       155495 :     }
     410              : }
     411              : 
     412         9344 : pub(crate) fn create_api() -> walproposer_api {
     413         9344 :     walproposer_api {
     414         9344 :         get_shmem_state: Some(get_shmem_state),
     415         9344 :         start_streaming: Some(start_streaming),
     416         9344 :         get_flush_rec_ptr: Some(get_flush_rec_ptr),
     417         9344 :         update_donor: Some(update_donor),
     418         9344 :         get_current_timestamp: Some(get_current_timestamp),
     419         9344 :         conn_error_message: Some(conn_error_message),
     420         9344 :         conn_status: Some(conn_status),
     421         9344 :         conn_connect_start: Some(conn_connect_start),
     422         9344 :         conn_connect_poll: Some(conn_connect_poll),
     423         9344 :         conn_send_query: Some(conn_send_query),
     424         9344 :         conn_get_query_result: Some(conn_get_query_result),
     425         9344 :         conn_flush: Some(conn_flush),
     426         9344 :         conn_finish: Some(conn_finish),
     427         9344 :         conn_async_read: Some(conn_async_read),
     428         9344 :         conn_async_write: Some(conn_async_write),
     429         9344 :         conn_blocking_write: Some(conn_blocking_write),
     430         9344 :         recovery_download: Some(recovery_download),
     431         9344 :         wal_reader_allocate: Some(wal_reader_allocate),
     432         9344 :         wal_read: Some(wal_read),
     433         9344 :         wal_reader_events: Some(wal_reader_events),
     434         9344 :         init_event_set: Some(init_event_set),
     435         9344 :         update_event_set: Some(update_event_set),
     436         9344 :         active_state_update_event_set: Some(active_state_update_event_set),
     437         9344 :         add_safekeeper_event_set: Some(add_safekeeper_event_set),
     438         9344 :         rm_safekeeper_event_set: Some(rm_safekeeper_event_set),
     439         9344 :         wait_event_set: Some(wait_event_set),
     440         9344 :         strong_random: Some(strong_random),
     441         9344 :         get_redo_start_lsn: Some(get_redo_start_lsn),
     442         9344 :         finish_sync_safekeepers: Some(finish_sync_safekeepers),
     443         9344 :         process_safekeeper_feedback: Some(process_safekeeper_feedback),
     444         9344 :         log_internal: Some(log_internal),
     445         9344 :         /* BEGIN_HADRON */
     446         9344 :         reset_safekeeper_statuses_for_metrics: Some(reset_safekeeper_statuses_for_metrics),
     447         9344 :         update_safekeeper_status_for_metrics: Some(update_safekeeper_status_for_metrics),
     448         9344 :         /* END_HADRON */
     449         9344 :     }
     450         9344 : }
     451              : 
     452         9344 : pub fn empty_shmem() -> crate::bindings::WalproposerShmemState {
     453         9344 :     let empty_feedback = crate::bindings::PageserverFeedback {
     454         9344 :         present: false,
     455         9344 :         currentClusterSize: 0,
     456         9344 :         last_received_lsn: 0,
     457         9344 :         disk_consistent_lsn: 0,
     458         9344 :         remote_consistent_lsn: 0,
     459         9344 :         replytime: 0,
     460         9344 :         shard_number: 0,
     461         9344 :         corruption_detected: false,
     462         9344 :     };
     463              : 
     464         9344 :     let empty_wal_rate_limiter = crate::bindings::WalRateLimiter {
     465         9344 :         effective_max_wal_bytes_per_second: crate::bindings::pg_atomic_uint32 { value: 0 },
     466         9344 :         should_limit: crate::bindings::pg_atomic_uint32 { value: 0 },
     467         9344 :         sent_bytes: 0,
     468         9344 :         batch_start_time_us: crate::bindings::pg_atomic_uint64 { value: 0 },
     469         9344 :         batch_end_time_us: crate::bindings::pg_atomic_uint64 { value: 0 },
     470         9344 :     };
     471              : 
     472         9344 :     crate::bindings::WalproposerShmemState {
     473         9344 :         propEpochStartLsn: crate::bindings::pg_atomic_uint64 { value: 0 },
     474         9344 :         donor_name: [0; 64],
     475         9344 :         donor_conninfo: [0; 1024],
     476         9344 :         donor_lsn: 0,
     477         9344 :         mutex: 0,
     478         9344 :         mineLastElectedTerm: crate::bindings::pg_atomic_uint64 { value: 0 },
     479         9344 :         backpressureThrottlingTime: crate::bindings::pg_atomic_uint64 { value: 0 },
     480         9344 :         currentClusterSize: crate::bindings::pg_atomic_uint64 { value: 0 },
     481         9344 :         shard_ps_feedback: [empty_feedback; 128],
     482         9344 :         num_shards: 0,
     483         9344 :         replica_promote: false,
     484         9344 :         min_ps_feedback: empty_feedback,
     485         9344 :         wal_rate_limiter: empty_wal_rate_limiter,
     486         9344 :         num_safekeepers: 0,
     487         9344 :         safekeeper_status: [0; 32],
     488         9344 :         safekeeper_commit_lsn: [0; 32],
     489         9344 :     }
     490         9344 : }
     491              : 
     492              : impl std::fmt::Display for Level {
     493          693 :     fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
     494          693 :         write!(f, "{self:?}")
     495          693 :     }
     496              : }
     497              : 
     498              : /// Take ownership of `Vec<u8>` from StringInfoData.
     499              : #[allow(clippy::unnecessary_cast)]
     500        47136 : pub(crate) fn take_vec_u8(pg: &mut StringInfoData) -> Option<Vec<u8>> {
     501        47136 :     if pg.data.is_null() {
     502        28030 :         return None;
     503        19106 :     }
     504              : 
     505        19106 :     let ptr = pg.data as *mut u8;
     506        19106 :     let length = pg.len as usize;
     507        19106 :     let capacity = pg.maxlen as usize;
     508              : 
     509        19106 :     pg.data = std::ptr::null_mut();
     510        19106 :     pg.len = 0;
     511        19106 :     pg.maxlen = 0;
     512              : 
     513        19106 :     unsafe { Some(Vec::from_raw_parts(ptr, length, capacity)) }
     514        47136 : }
     515              : 
     516              : /// Store `Vec<u8>` in StringInfoData.
     517        19112 : fn store_vec_u8(pg: &mut StringInfoData, vec: Vec<u8>) -> *mut ::std::os::raw::c_char {
     518        19112 :     let ptr = vec.as_ptr() as *mut ::std::os::raw::c_char;
     519        19112 :     let length = vec.len();
     520        19112 :     let capacity = vec.capacity();
     521              : 
     522        19112 :     assert!(pg.data.is_null());
     523              : 
     524        19112 :     pg.data = ptr;
     525        19112 :     pg.len = length as i32;
     526        19112 :     pg.maxlen = capacity as i32;
     527              : 
     528        19112 :     std::mem::forget(vec);
     529              : 
     530        19112 :     ptr
     531        19112 : }
        

Generated by: LCOV version 2.1-beta