Method write_datum

crates/paimon/src/spec/binary_row.rs:566–623

Write a Datum value at the given position, dispatching by type.

(&mut self, pos: usize, datum: &Datum, data_type: &DataType)

    /// Write a Datum value at the given position, dispatching by type.
    pub fn write_datum(&mut self, pos: usize, datum: &Datum, data_type: &DataType) {
        match datum {
            Datum::Bool(v) => self.write_boolean(pos, *v),
            Datum::TinyInt(v) => self.write_byte(pos, *v),
            Datum::SmallInt(v) => self.write_short(pos, *v),
            Datum::Int(v) | Datum::Date(v) | Datum::Time(v) => self.write_int(pos, *v),
            Datum::Long(v) => self.write_long(pos, *v),
            Datum::Float(v) => self.write_float(pos, *v),
            Datum::Double(v) => self.write_double(pos, *v),
            Datum::Timestamp { millis, nanos } => {
                let precision = match data_type {
                    DataType::Timestamp(ts) => ts.precision(),
                    _ => 3,
                };
                if precision <= 3 {
                    self.write_timestamp_compact(pos, *millis);
                } else {
                    self.write_timestamp_non_compact(pos, *millis, *nanos);
                }
            }
            Datum::LocalZonedTimestamp { millis, nanos } => {
                let precision = match data_type {
                    DataType::LocalZonedTimestamp(ts) => ts.precision(),
                    _ => 3,
                };
                if precision <= 3 {
                    self.write_timestamp_compact(pos, *millis);
                } else {
                    self.write_timestamp_non_compact(pos, *millis, *nanos);
                }
            }
            Datum::Decimal {
                unscaled,
                precision,
                ..
            } => {
                if *precision <= 18 {
                    self.write_decimal_compact(pos, *unscaled as i64);
                } else {
                    self.write_decimal_var_len(pos, *unscaled);
                }
            }
            Datum::String(s) => {
                if s.len() <= 7 {
                    self.write_string_inline(pos, s);
                } else {
                    self.write_string(pos, s);
                }
            }
            Datum::Bytes(b) => {
                if b.len() <= 7 {
                    self.write_binary_inline(pos, b);
                } else {
                    self.write_binary(pos, b);
                }
            }
        }
    }