reifydb_core/encoded/
u128.rs

1// SPDX-License-Identifier: AGPL-3.0-or-later
2// Copyright (c) 2025 ReifyDB
3
4use std::ptr;
5
6use reifydb_type::value::r#type::Type;
7
8use crate::encoded::{encoded::EncodedValues, schema::Schema};
9
10impl Schema {
11	pub fn set_u128(&self, row: &mut EncodedValues, index: usize, value: impl Into<u128>) {
12		let field = &self.fields()[index];
13		debug_assert!(row.len() >= self.total_static_size());
14		debug_assert_eq!(*field.constraint.get_type().inner_type(), Type::Uint16);
15		row.set_valid(index, true);
16		unsafe {
17			ptr::write_unaligned(
18				row.make_mut().as_mut_ptr().add(field.offset as usize) as *mut u128,
19				value.into(),
20			)
21		}
22	}
23
24	pub fn get_u128(&self, row: &EncodedValues, index: usize) -> u128 {
25		let field = &self.fields()[index];
26		debug_assert!(row.len() >= self.total_static_size());
27		debug_assert_eq!(*field.constraint.get_type().inner_type(), Type::Uint16);
28		unsafe { (row.as_ptr().add(field.offset as usize) as *const u128).read_unaligned() }
29	}
30
31	pub fn try_get_u128(&self, row: &EncodedValues, index: usize) -> Option<u128> {
32		if row.is_defined(index) && self.fields()[index].constraint.get_type() == Type::Uint16 {
33			Some(self.get_u128(row, index))
34		} else {
35			None
36		}
37	}
38}
39
40#[cfg(test)]
41pub mod tests {
42	use reifydb_type::value::r#type::Type;
43
44	use crate::encoded::schema::Schema;
45
46	#[test]
47	fn test_set_get_u128() {
48		let schema = Schema::testing(&[Type::Uint16]);
49		let mut row = schema.allocate();
50		schema.set_u128(&mut row, 0, 340282366920938463463374607431768211455u128);
51		assert_eq!(schema.get_u128(&row, 0), 340282366920938463463374607431768211455u128);
52	}
53
54	#[test]
55	fn test_try_get_u128() {
56		let schema = Schema::testing(&[Type::Uint16]);
57		let mut row = schema.allocate();
58
59		assert_eq!(schema.try_get_u128(&row, 0), None);
60
61		schema.set_u128(&mut row, 0, 340282366920938463463374607431768211455u128);
62		assert_eq!(schema.try_get_u128(&row, 0), Some(340282366920938463463374607431768211455u128));
63	}
64
65	#[test]
66	fn test_extremes() {
67		let schema = Schema::testing(&[Type::Uint16]);
68		let mut row = schema.allocate();
69
70		schema.set_u128(&mut row, 0, u128::MAX);
71		assert_eq!(schema.get_u128(&row, 0), u128::MAX);
72
73		let mut row2 = schema.allocate();
74		schema.set_u128(&mut row2, 0, u128::MIN);
75		assert_eq!(schema.get_u128(&row2, 0), u128::MIN);
76
77		let mut row3 = schema.allocate();
78		schema.set_u128(&mut row3, 0, 0u128);
79		assert_eq!(schema.get_u128(&row3, 0), 0u128);
80	}
81
82	#[test]
83	fn test_very_large_values() {
84		let schema = Schema::testing(&[Type::Uint16]);
85
86		let test_values = [
87			0u128,
88			1u128,
89			99999999999999999999999999999999999999u128,
90			170141183460469231731687303715884105727u128, // i128::MAX as u128
91			170141183460469231731687303715884105728u128, // i128::MAX + 1
92			300000000000000000000000000000000000000u128,
93			340282366920938463463374607431768211454u128,
94			340282366920938463463374607431768211455u128, // u128::MAX
95		];
96
97		for value in test_values {
98			let mut row = schema.allocate();
99			schema.set_u128(&mut row, 0, value);
100			assert_eq!(schema.get_u128(&row, 0), value);
101		}
102	}
103
104	#[test]
105	fn test_powers_of_two() {
106		let schema = Schema::testing(&[Type::Uint16]);
107
108		let powers = [
109			1u128, 2u128, 4u128, 8u128, 16u128, 32u128, 64u128, 128u128, 256u128, 512u128, 1024u128,
110			2048u128, 4096u128, 8192u128, 16384u128, 32768u128, 65536u128,
111		];
112
113		for power in powers {
114			let mut row = schema.allocate();
115			schema.set_u128(&mut row, 0, power);
116			assert_eq!(schema.get_u128(&row, 0), power);
117		}
118	}
119
120	#[test]
121	fn test_ipv6_addresses() {
122		let schema = Schema::testing(&[Type::Uint16]);
123
124		// Test values representing IPv6 addresses as u128
125		let ipv6_values = [
126			0u128,                                       // ::0
127			1u128,                                       // ::1 (loopback)
128			281470681743360u128,                         // ::ffff:0:0 (IPv4-mapped prefix)
129			338953138925153547590470800371487866880u128, // Example IPv6
130		];
131
132		for ipv6 in ipv6_values {
133			let mut row = schema.allocate();
134			schema.set_u128(&mut row, 0, ipv6);
135			assert_eq!(schema.get_u128(&row, 0), ipv6);
136		}
137	}
138
139	#[test]
140	fn test_uuid_values() {
141		let schema = Schema::testing(&[Type::Uint16]);
142
143		// Test values that could represent UUIDs as u128
144		let uuid_values = [
145			123456789012345678901234567890123456789u128,
146			123456789012345678901234567890123456789u128,
147			111111111111111111111111111111111111111u128,
148		];
149
150		for uuid_val in uuid_values {
151			let mut row = schema.allocate();
152			schema.set_u128(&mut row, 0, uuid_val);
153			assert_eq!(schema.get_u128(&row, 0), uuid_val);
154		}
155	}
156
157	#[test]
158	fn test_mixed_with_other_types() {
159		let schema = Schema::testing(&[Type::Uint16, Type::Boolean, Type::Uint16]);
160		let mut row = schema.allocate();
161
162		let large_value1 = 200000000000000000000000000000000000000u128;
163		let large_value2 = 150000000000000000000000000000000000000u128;
164
165		schema.set_u128(&mut row, 0, large_value1);
166		schema.set_bool(&mut row, 1, true);
167		schema.set_u128(&mut row, 2, large_value2);
168
169		assert_eq!(schema.get_u128(&row, 0), large_value1);
170		assert_eq!(schema.get_bool(&row, 1), true);
171		assert_eq!(schema.get_u128(&row, 2), large_value2);
172	}
173
174	#[test]
175	fn test_undefined_handling() {
176		let schema = Schema::testing(&[Type::Uint16, Type::Uint16]);
177		let mut row = schema.allocate();
178
179		let value = 340282366920938463463374607431768211455u128;
180		schema.set_u128(&mut row, 0, value);
181
182		assert_eq!(schema.try_get_u128(&row, 0), Some(value));
183		assert_eq!(schema.try_get_u128(&row, 1), None);
184
185		schema.set_undefined(&mut row, 0);
186		assert_eq!(schema.try_get_u128(&row, 0), None);
187	}
188
189	#[test]
190	fn test_try_get_u128_wrong_type() {
191		let schema = Schema::testing(&[Type::Boolean]);
192		let mut row = schema.allocate();
193
194		schema.set_bool(&mut row, 0, true);
195
196		assert_eq!(schema.try_get_u128(&row, 0), None);
197	}
198}
reifydb_core/encoded/u128.rs

reifydb_core/encoded/
u128.rs
