polars_core/chunked_array/builder/
string.rs

Help
1use super::*;
2
3pub struct BinViewChunkedBuilder<T: ViewType + ?Sized> {
4    pub(crate) chunk_builder: MutableBinaryViewArray<T>,
5    pub(crate) field: FieldRef,
6}
7
8impl<T: ViewType + ?Sized> Clone for BinViewChunkedBuilder<T> {
9    fn clone(&self) -> Self {
10        Self {
11            chunk_builder: self.chunk_builder.clone(),
12            field: self.field.clone(),
13        }
14    }
15}
16
17pub type StringChunkedBuilder = BinViewChunkedBuilder<str>;
18pub type BinaryChunkedBuilder = BinViewChunkedBuilder<[u8]>;
19
20impl<T: ViewType + ?Sized> BinViewChunkedBuilder<T> {
21    /// Create a new BinViewChunkedBuilder
22    ///
23    /// # Arguments
24    ///
25    /// * `capacity` - Number of string elements in the final array.
26    pub fn new(name: PlSmallStr, capacity: usize) -> Self {
27        Self {
28            chunk_builder: MutableBinaryViewArray::with_capacity(capacity),
29            field: Arc::new(Field::new(name, DataType::from_arrow_dtype(&T::DATA_TYPE))),
30        }
31    }
32
33    /// Appends a value of type `T` into the builder
34    #[inline]
35    pub fn append_value<S: AsRef<T>>(&mut self, v: S) {
36        self.chunk_builder.push_value(v.as_ref());
37    }
38
39    /// Appends a null slot into the builder
40    #[inline]
41    pub fn append_null(&mut self) {
42        self.chunk_builder.push_null()
43    }
44
45    #[inline]
46    pub fn append_option<S: AsRef<T>>(&mut self, opt: Option<S>) {
47        self.chunk_builder.push(opt);
48    }
49}
50
51impl StringChunkedBuilder {
52    pub fn finish(mut self) -> StringChunked {
53        let arr = self.chunk_builder.as_box();
54        ChunkedArray::new_with_compute_len(self.field, vec![arr])
55    }
56}
57impl BinaryChunkedBuilder {
58    pub fn finish(mut self) -> BinaryChunked {
59        let arr = self.chunk_builder.as_box();
60        ChunkedArray::new_with_compute_len(self.field, vec![arr])
61    }
62}
polars_core/chunked_array/builder/string.rs

polars_core/chunked_array/builder/
string.rs
