reifydb_core/value/column/data/
reorder.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2025 ReifyDB
3
4use reifydb_type::storage::DataBitVec;
5
6use crate::value::column::{ColumnData, data::with_container};
7
8impl ColumnData {
9	pub fn reorder(&mut self, indices: &[usize]) {
10		match self {
11			ColumnData::Option {
12				inner,
13				bitvec,
14			} => {
15				inner.reorder(indices);
16				let mut new_bitvec = DataBitVec::spawn(bitvec, indices.len());
17				for &idx in indices {
18					if idx < DataBitVec::len(bitvec) {
19						DataBitVec::push(&mut new_bitvec, DataBitVec::get(bitvec, idx));
20					} else {
21						DataBitVec::push(&mut new_bitvec, false);
22					}
23				}
24				*bitvec = new_bitvec;
25			}
26			_ => with_container!(self, |c| c.reorder(indices)),
27		}
28	}
29}
30
31#[cfg(test)]
32pub mod tests {
33	use reifydb_runtime::context::{
34		clock::{Clock, MockClock},
35		rng::Rng,
36	};
37	use reifydb_type::value::{Value, dictionary::DictionaryEntryId, identity::IdentityId, r#type::Type};
38
39	use crate::value::column::ColumnData;
40
41	fn test_clock_and_rng() -> (MockClock, Clock, Rng) {
42		let mock = MockClock::from_millis(1000);
43		let clock = Clock::Mock(mock.clone());
44		let rng = Rng::seeded(42);
45		(mock, clock, rng)
46	}
47
48	#[test]
49	fn test_reorder_bool() {
50		let mut col = ColumnData::bool([true, false, true]);
51		col.reorder(&[2, 0, 1]);
52
53		assert_eq!(col.len(), 3);
54		assert_eq!(col.get_value(0), Value::Boolean(true));
55		assert_eq!(col.get_value(1), Value::Boolean(true));
56		assert_eq!(col.get_value(2), Value::Boolean(false));
57	}
58
59	#[test]
60	fn test_reorder_float4() {
61		let mut col = ColumnData::float4([1.0, 2.0, 3.0]);
62		col.reorder(&[2, 0, 1]);
63
64		assert_eq!(col.len(), 3);
65		// Check data after reordering
66		match col.get_value(0) {
67			Value::Float4(v) => assert_eq!(v.value(), 3.0),
68			_ => panic!("Expected Float4"),
69		}
70		match col.get_value(1) {
71			Value::Float4(v) => assert_eq!(v.value(), 1.0),
72			_ => panic!("Expected Float4"),
73		}
74		match col.get_value(2) {
75			Value::Float4(v) => assert_eq!(v.value(), 2.0),
76			_ => panic!("Expected Float4"),
77		}
78	}
79
80	#[test]
81	fn test_reorder_int4() {
82		let mut col = ColumnData::int4([1, 2, 3]);
83		col.reorder(&[2, 0, 1]);
84
85		assert_eq!(col.len(), 3);
86		assert_eq!(col.get_value(0), Value::Int4(3));
87		assert_eq!(col.get_value(1), Value::Int4(1));
88		assert_eq!(col.get_value(2), Value::Int4(2));
89	}
90
91	#[test]
92	fn test_reorder_string() {
93		let mut col = ColumnData::utf8(["a".to_string(), "b".to_string(), "c".to_string()]);
94		col.reorder(&[2, 0, 1]);
95
96		assert_eq!(col.len(), 3);
97		assert_eq!(col.get_value(0), Value::Utf8("c".to_string()));
98		assert_eq!(col.get_value(1), Value::Utf8("a".to_string()));
99		assert_eq!(col.get_value(2), Value::Utf8("b".to_string()));
100	}
101
102	#[test]
103	fn test_reorder_none() {
104		let mut col = ColumnData::none_typed(Type::Boolean, 3);
105		col.reorder(&[2, 0, 1]);
106		assert_eq!(col.len(), 3);
107
108		col.reorder(&[1, 0]);
109		assert_eq!(col.len(), 2);
110	}
111
112	#[test]
113	fn test_reorder_identity_id() {
114		let (mock, clock, rng) = test_clock_and_rng();
115		let id1 = IdentityId::generate(&clock, &rng);
116		mock.advance_millis(1);
117		let id2 = IdentityId::generate(&clock, &rng);
118		mock.advance_millis(1);
119		let id3 = IdentityId::generate(&clock, &rng);
120
121		let mut col = ColumnData::identity_id([id1, id2, id3]);
122		col.reorder(&[2, 0, 1]);
123
124		assert_eq!(col.len(), 3);
125		assert_eq!(col.get_value(0), Value::IdentityId(id3));
126		assert_eq!(col.get_value(1), Value::IdentityId(id1));
127		assert_eq!(col.get_value(2), Value::IdentityId(id2));
128	}
129
130	#[test]
131	fn test_reorder_dictionary_id() {
132		let e1 = DictionaryEntryId::U4(10);
133		let e2 = DictionaryEntryId::U4(20);
134		let e3 = DictionaryEntryId::U4(30);
135
136		let mut col = ColumnData::dictionary_id([e1, e2, e3]);
137		col.reorder(&[2, 0, 1]);
138
139		assert_eq!(col.len(), 3);
140		assert_eq!(col.get_value(0), Value::DictionaryId(e3));
141		assert_eq!(col.get_value(1), Value::DictionaryId(e1));
142		assert_eq!(col.get_value(2), Value::DictionaryId(e2));
143	}
144}
reifydb_core/value/column/data/reorder.rs

reifydb_core/value/column/data/
reorder.rs
