simdjson_rust/ondemand/
document.rs

1use std::{marker::PhantomData, ptr::NonNull};
2
3use simdjson_sys as ffi;
4
5use super::{array::Array, number::Number, object::Object, parser::Parser, value::Value, JsonType};
6use crate::{
7    error::Result,
8    macros::{impl_drop, map_result},
9    utils::string_view_to_str,
10};
11
12pub struct Document<'p, 's> {
13    ptr: NonNull<ffi::SJ_OD_document>,
14    _parser: PhantomData<&'p mut Parser>,
15    _padded_string: PhantomData<&'s String>,
16}
17impl<'p, 's> Document<'p, 's> {
18    pub fn new(ptr: NonNull<ffi::SJ_OD_document>) -> Self {
19        Self {
20            ptr,
21            _parser: PhantomData,
22            _padded_string: PhantomData,
23        }
24    }
25
26    pub fn get_uint64(&mut self) -> Result<u64> {
27        map_result!(
28            primitive,
29            ffi::SJ_OD_document_get_uint64(self.ptr.as_mut()),
30            ffi::uint64_t_result_error,
31            ffi::uint64_t_result_value_unsafe
32        )
33    }
34
35    pub fn get_int64(&mut self) -> Result<i64> {
36        map_result!(
37            primitive,
38            ffi::SJ_OD_document_get_int64(self.ptr.as_mut()),
39            ffi::int64_t_result_error,
40            ffi::int64_t_result_value_unsafe
41        )
42    }
43
44    pub fn get_bool(&mut self) -> Result<bool> {
45        map_result!(
46            primitive,
47            ffi::SJ_OD_document_get_bool(self.ptr.as_mut()),
48            ffi::bool_result_error,
49            ffi::bool_result_value_unsafe
50        )
51    }
52
53    pub fn get_double(&mut self) -> Result<f64> {
54        map_result!(
55            primitive,
56            ffi::SJ_OD_document_get_double(self.ptr.as_mut()),
57            ffi::double_result_error,
58            ffi::double_result_value_unsafe
59        )
60    }
61
62    pub fn get_value<'a>(&mut self) -> Result<Value<'a>> {
63        map_result!(
64            ffi::SJ_OD_document_get_value(self.ptr.as_mut()),
65            ffi::SJ_OD_value_result_error,
66            ffi::SJ_OD_value_result_value_unsafe
67        )
68        .map(Value::new)
69    }
70
71    pub fn get_array<'a>(&mut self) -> Result<Array<'a>> {
72        map_result!(
73            ffi::SJ_OD_document_get_array(self.ptr.as_mut()),
74            ffi::SJ_OD_array_result_error,
75            ffi::SJ_OD_array_result_value_unsafe
76        )
77        .map(Array::new)
78    }
79
80    pub fn get_object<'a>(&mut self) -> Result<Object<'a>> {
81        map_result!(
82            ffi::SJ_OD_document_get_object(self.ptr.as_mut()),
83            ffi::SJ_OD_object_result_error,
84            ffi::SJ_OD_object_result_value_unsafe
85        )
86        .map(Object::new)
87    }
88
89    pub fn get_wobbly_string<'a>(&mut self) -> Result<&'a str> {
90        let sv = map_result!(
91            ffi::SJ_OD_document_get_wobbly_string(self.ptr.as_mut()),
92            ffi::STD_string_view_result_error,
93            ffi::STD_string_view_result_value_unsafe
94        )?;
95        Ok(string_view_to_str(sv))
96    }
97
98    pub fn get_string<'a>(&mut self) -> Result<&'a str> {
99        let sv = map_result!(
100            ffi::SJ_OD_document_get_string(self.ptr.as_mut()),
101            ffi::STD_string_view_result_error,
102            ffi::STD_string_view_result_value_unsafe
103        )?;
104        Ok(string_view_to_str(sv))
105    }
106
107    pub fn at_pointer<'a>(&mut self, json_pointer: &str) -> Result<Value<'a>> {
108        map_result!(
109            ffi::SJ_OD_document_at_pointer(
110                self.ptr.as_mut(),
111                json_pointer.as_ptr().cast(),
112                json_pointer.len()
113            ),
114            ffi::SJ_OD_value_result_error,
115            ffi::SJ_OD_value_result_value_unsafe
116        )
117        .map(Value::new)
118    }
119
120    pub fn get_number<'a>(&mut self) -> Result<Number<'a>> {
121        map_result!(
122            ffi::SJ_OD_document_get_number(self.ptr.as_mut()),
123            ffi::SJ_OD_number_result_error,
124            ffi::SJ_OD_number_result_value_unsafe
125        )
126        .map(Number::new)
127    }
128
129    pub fn is_null(&mut self) -> Result<bool> {
130        map_result!(
131            primitive,
132            ffi::SJ_OD_document_is_null(self.ptr.as_mut()),
133            ffi::bool_result_error,
134            ffi::bool_result_value_unsafe
135        )
136    }
137
138    pub fn json_type(&mut self) -> Result<JsonType> {
139        let json_type = map_result!(
140            primitive,
141            ffi::SJ_OD_document_type(self.ptr.as_mut()),
142            ffi::int_result_error,
143            ffi::int_result_value_unsafe
144        )?;
145        Ok(JsonType::from(json_type))
146    }
147}
148
149impl_drop!(Document<'p, 's>, ffi::SJ_OD_document_free);
150
151#[cfg(test)]
152mod tests {
153    use crate::{ondemand, prelude::*};
154
155    #[test]
156    fn get_bool() {
157        let mut parser = ondemand::Parser::default();
158
159        {
160            let json = "true".to_padded_string();
161            let mut doc = parser.iterate(&json).unwrap();
162            assert!(doc.get_bool().unwrap());
163        }
164        {
165            let json = "false".to_padded_string();
166            let mut doc = parser.iterate(&json).unwrap();
167            assert!(!doc.get_bool().unwrap());
168        }
169        {
170            let json = "1".to_padded_string();
171            let mut doc = parser.iterate(&json).unwrap();
172            assert!(doc.get_bool().is_err());
173        }
174    }
175}
simdjson_rust/ondemand/document.rs

simdjson_rust/ondemand/
document.rs
