reifydb_routine/function/rql/
fingerprint.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2025 ReifyDB
3
4use bumpalo::Bump;
5use reifydb_core::value::column::data::ColumnData;
6use reifydb_rql::{
7	ast::parse_str,
8	fingerprint::{request::fingerprint_request, statement::fingerprint_statement},
9};
10use reifydb_type::value::r#type::Type;
11
12use crate::function::{
13	ScalarFunction, ScalarFunctionContext,
14	error::{ScalarFunctionError, ScalarFunctionResult},
15	propagate_options,
16};
17
18pub struct RqlFingerprint;
19
20impl RqlFingerprint {
21	pub fn new() -> Self {
22		Self
23	}
24}
25
26impl ScalarFunction for RqlFingerprint {
27	fn scalar(&self, ctx: ScalarFunctionContext) -> ScalarFunctionResult<ColumnData> {
28		if let Some(result) = propagate_options(self, &ctx) {
29			return result;
30		}
31
32		let columns = ctx.columns;
33		let row_count = ctx.row_count;
34
35		if columns.len() != 1 {
36			return Err(ScalarFunctionError::ArityMismatch {
37				function: ctx.fragment.clone(),
38				expected: 1,
39				actual: columns.len(),
40			});
41		}
42
43		let column = columns.get(0).unwrap();
44
45		match &column.data() {
46			ColumnData::Utf8 {
47				container,
48				..
49			} => {
50				let mut result_data = Vec::with_capacity(row_count);
51				let mut result_bitvec = Vec::with_capacity(row_count);
52
53				for i in 0..row_count {
54					if container.is_defined(i) {
55						let query = &container[i];
56						let bump = Bump::new();
57						let stmts = parse_str(&bump, query).map_err(|e| {
58							ScalarFunctionError::ExecutionFailed {
59								function: ctx.fragment.clone(),
60								reason: format!("{e}"),
61							}
62						})?;
63						let fps: Vec<_> =
64							stmts.iter().map(|s| fingerprint_statement(s)).collect();
65						let req = fingerprint_request(&fps);
66						result_data.push(req.to_hex());
67						result_bitvec.push(true);
68					} else {
69						result_data.push(String::new());
70						result_bitvec.push(false);
71					}
72				}
73
74				Ok(ColumnData::utf8_with_bitvec(result_data, result_bitvec))
75			}
76			other => Err(ScalarFunctionError::InvalidArgumentType {
77				function: ctx.fragment.clone(),
78				argument_index: 0,
79				expected: vec![Type::Utf8],
80				actual: other.get_type(),
81			}),
82		}
83	}
84
85	fn return_type(&self, _input_types: &[Type]) -> Type {
86		Type::Utf8
87	}
88}
reifydb_routine/function/rql/fingerprint.rs

reifydb_routine/function/rql/
fingerprint.rs
