risingwave_common/types/
serial.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::error::Error;
16use std::hash::Hash;
17
18use bytes::BytesMut;
19use postgres_types::{IsNull, ToSql, Type, accepts, to_sql_checked};
20use risingwave_common_estimate_size::ZeroHeapSize;
21use serde::{Serialize, Serializer};
22
23use crate::util::row_id::RowId;
24
25// Serial is an alias for i64
26#[derive(Debug, Copy, Clone, PartialEq, Eq, Ord, PartialOrd, Default, Hash)]
27pub struct Serial(pub(crate) i64);
28
29impl From<Serial> for i64 {
30    fn from(value: Serial) -> i64 {
31        value.0
32    }
33}
34
35impl From<i64> for Serial {
36    fn from(value: i64) -> Self {
37        Self(value)
38    }
39}
40
41impl ZeroHeapSize for Serial {}
42
43impl Serial {
44    #[inline]
45    pub fn into_inner(self) -> i64 {
46        self.0
47    }
48
49    #[inline]
50    pub fn as_row_id(self) -> RowId {
51        self.0 as RowId
52    }
53}
54
55impl Serialize for Serial {
56    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
57    where
58        S: Serializer,
59    {
60        serializer.serialize_i64(self.0)
61    }
62}
63
64impl crate::types::to_text::ToText for Serial {
65    fn write<W: std::fmt::Write>(&self, f: &mut W) -> std::fmt::Result {
66        write!(f, "{}", self.0)
67    }
68
69    fn write_with_type<W: std::fmt::Write>(
70        &self,
71        _ty: &crate::types::DataType,
72        f: &mut W,
73    ) -> std::fmt::Result {
74        self.write(f)
75    }
76}
77
78impl ToSql for Serial {
79    accepts!(INT8);
80
81    to_sql_checked!();
82
83    fn to_sql(&self, ty: &Type, out: &mut BytesMut) -> Result<IsNull, Box<dyn Error + Sync + Send>>
84    where
85        Self: Sized,
86    {
87        self.0.to_sql(ty, out)
88    }
89}
risingwave_common/types/serial.rs

risingwave_common/types/
serial.rs
