reifydb_value/value/container/
row.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2026 ReifyDB
3
4use std::{
5	fmt::{self, Debug},
6	ops::Deref,
7	result::Result as StdResult,
8};
9
10use serde::{Deserialize, Deserializer, Serialize, Serializer};
11
12use crate::{
13	Result,
14	util::bitvec::BitVec,
15	value::{Value, row_number::RowNumber, value_type::ValueType},
16};
17
18pub struct RowNumberContainer {
19	data: Vec<RowNumber>,
20}
21
22impl Clone for RowNumberContainer {
23	fn clone(&self) -> Self {
24		Self {
25			data: self.data.clone(),
26		}
27	}
28}
29
30impl Debug for RowNumberContainer {
31	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
32		f.debug_struct("RowNumberContainer").field("data", &self.data).finish()
33	}
34}
35
36impl PartialEq for RowNumberContainer {
37	fn eq(&self, other: &Self) -> bool {
38		self.data == other.data
39	}
40}
41
42impl Serialize for RowNumberContainer {
43	fn serialize<Ser: Serializer>(&self, serializer: Ser) -> StdResult<Ser::Ok, Ser::Error> {
44		#[derive(Serialize)]
45		struct Helper<'a> {
46			data: &'a Vec<RowNumber>,
47		}
48
49		Helper {
50			data: &self.data,
51		}
52		.serialize(serializer)
53	}
54}
55
56impl<'de> Deserialize<'de> for RowNumberContainer {
57	fn deserialize<D: Deserializer<'de>>(deserializer: D) -> StdResult<Self, D::Error> {
58		#[derive(Deserialize)]
59		struct Helper {
60			data: Vec<RowNumber>,
61		}
62		let h = Helper::deserialize(deserializer)?;
63		Ok(RowNumberContainer {
64			data: h.data,
65		})
66	}
67}
68
69impl Deref for RowNumberContainer {
70	type Target = [RowNumber];
71
72	fn deref(&self) -> &Self::Target {
73		self.data.as_slice()
74	}
75}
76
77impl RowNumberContainer {
78	pub fn new(data: Vec<RowNumber>) -> Self {
79		Self {
80			data,
81		}
82	}
83
84	pub fn with_capacity(capacity: usize) -> Self {
85		Self {
86			data: Vec::with_capacity(capacity),
87		}
88	}
89
90	pub fn from_vec(data: Vec<RowNumber>) -> Self {
91		Self {
92			data,
93		}
94	}
95}
96
97impl RowNumberContainer {
98	pub fn from_parts(data: Vec<RowNumber>) -> Self {
99		Self {
100			data,
101		}
102	}
103
104	pub fn len(&self) -> usize {
105		self.data.len()
106	}
107
108	pub fn capacity(&self) -> usize {
109		self.data.capacity()
110	}
111
112	pub fn is_empty(&self) -> bool {
113		self.data.is_empty()
114	}
115
116	pub fn push(&mut self, value: RowNumber) {
117		self.data.push(value);
118	}
119
120	pub fn push_default(&mut self) {
121		self.data.push(RowNumber::default());
122	}
123
124	pub fn get(&self, index: usize) -> Option<&RowNumber> {
125		if index < self.len() {
126			self.data.get(index)
127		} else {
128			None
129		}
130	}
131
132	pub fn is_defined(&self, idx: usize) -> bool {
133		idx < self.len()
134	}
135
136	pub fn data(&self) -> &Vec<RowNumber> {
137		&self.data
138	}
139
140	pub fn data_mut(&mut self) -> &mut Vec<RowNumber> {
141		&mut self.data
142	}
143
144	pub fn as_string(&self, index: usize) -> String {
145		if index < self.len() {
146			self.data[index].to_string()
147		} else {
148			"none".to_string()
149		}
150	}
151
152	pub fn get_value(&self, index: usize) -> Value {
153		if index < self.len() {
154			Value::Uint8(self.data[index].value())
155		} else {
156			Value::none_of(ValueType::Uint8)
157		}
158	}
159
160	pub fn extend(&mut self, other: &Self) -> Result<()> {
161		self.data.extend(other.data.iter().cloned());
162		Ok(())
163	}
164
165	pub fn iter(&self) -> impl Iterator<Item = Option<RowNumber>> + '_ {
166		self.data.iter().map(|&v| Some(v))
167	}
168
169	pub fn slice(&self, start: usize, end: usize) -> Self {
170		let count = (end - start).min(self.len().saturating_sub(start));
171		let mut new_data = Vec::with_capacity(count);
172		for i in start..(start + count) {
173			new_data.push(self.data[i]);
174		}
175		Self {
176			data: new_data,
177		}
178	}
179
180	pub fn filter(&mut self, mask: &BitVec) {
181		let mut new_data = Vec::with_capacity(mask.count_ones());
182
183		for (i, keep) in mask.iter().enumerate() {
184			if keep && i < self.len() {
185				new_data.push(self.data[i]);
186			}
187		}
188
189		self.data = new_data;
190	}
191
192	pub fn reorder(&mut self, indices: &[usize]) {
193		let mut new_data = Vec::with_capacity(indices.len());
194
195		for &idx in indices {
196			if idx < self.len() {
197				new_data.push(self.data[idx]);
198			} else {
199				new_data.push(RowNumber::default());
200			}
201		}
202
203		self.data = new_data;
204	}
205
206	pub fn take(&self, num: usize) -> Self {
207		Self {
208			data: self.data[..num.min(self.data.len())].to_vec(),
209		}
210	}
211}
212
213impl Default for RowNumberContainer {
214	fn default() -> Self {
215		Self::with_capacity(0)
216	}
217}
reifydb_value/value/container/row.rs

reifydb_value/value/container/
row.rs
