use std::sync::Arc;
use reifydb_core::{
interface::catalog::vtable::VTable,
value::column::{ColumnWithName, buffer::ColumnBuffer, columns::Columns},
};
use reifydb_transaction::transaction::Transaction;
use reifydb_type::fragment::Fragment;
use crate::{
Result,
system::SystemCatalog,
vtable::{BaseVTable, Batch, VTableContext},
};
pub struct SystemCdcConsumers {
pub(crate) vtable: Arc<VTable>,
exhausted: bool,
}
impl Default for SystemCdcConsumers {
fn default() -> Self {
Self::new()
}
}
impl SystemCdcConsumers {
pub fn new() -> Self {
Self {
vtable: SystemCatalog::get_system_cdc_consumers_table().clone(),
exhausted: false,
}
}
}
impl BaseVTable for SystemCdcConsumers {
fn initialize(&mut self, _txn: &mut Transaction<'_>, _ctx: VTableContext) -> Result<()> {
self.exhausted = false;
Ok(())
}
fn next(&mut self, _txn: &mut Transaction<'_>) -> Result<Option<Batch>> {
if self.exhausted {
return Ok(None);
}
let columns = vec![
ColumnWithName::new(Fragment::internal("consumer_id"), ColumnBuffer::utf8_with_capacity(0)),
ColumnWithName::new(Fragment::internal("checkpoint"), ColumnBuffer::uint8_with_capacity(0)),
];
self.exhausted = true;
Ok(Some(Batch {
columns: Columns::new(columns),
}))
}
fn vtable(&self) -> &VTable {
&self.vtable
}
}