clickhouse_arrow/native/values/
ip.rs

1use std::fmt;
2use std::net::{Ipv4Addr, Ipv6Addr};
3use std::ops::Deref;
4
5use super::*;
6
7/// Wrapper type for `ClickHouse` `IPv4` type.
8#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
9#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
10pub struct Ipv4(pub Ipv4Addr);
11
12impl fmt::Display for Ipv4 {
13    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { self.0.fmt(f) }
14}
15
16impl Deref for Ipv4 {
17    type Target = Ipv4Addr;
18
19    fn deref(&self) -> &Self::Target { &self.0 }
20}
21
22impl From<Ipv4> for Ipv4Addr {
23    fn from(i: Ipv4) -> Self { i.0 }
24}
25
26impl From<Ipv4Addr> for Ipv4 {
27    fn from(x: Ipv4Addr) -> Self { Self(x) }
28}
29
30impl Default for Ipv4 {
31    fn default() -> Self { Self(Ipv4Addr::UNSPECIFIED) }
32}
33
34/// Wrapper type for `ClickHouse` `IPv6` type.
35#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
36#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
37pub struct Ipv6(pub Ipv6Addr);
38
39impl fmt::Display for Ipv6 {
40    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { self.0.fmt(f) }
41}
42
43impl Deref for Ipv6 {
44    type Target = Ipv6Addr;
45
46    fn deref(&self) -> &Self::Target { &self.0 }
47}
48
49impl From<Ipv6> for Ipv6Addr {
50    fn from(i: Ipv6) -> Self { i.0 }
51}
52
53impl From<Ipv6Addr> for Ipv6 {
54    fn from(x: Ipv6Addr) -> Self { Self(x) }
55}
56
57impl Default for Ipv6 {
58    fn default() -> Self { Self(Ipv6Addr::UNSPECIFIED) }
59}
60
61impl ToSql for Ipv4 {
62    fn to_sql(self, _type_hint: Option<&Type>) -> Result<Value> { Ok(Value::Ipv4(self)) }
63}
64
65impl FromSql for Ipv4 {
66    fn from_sql(type_: &Type, value: Value) -> Result<Self> {
67        if !matches!(type_, Type::Ipv4) {
68            return Err(unexpected_type(type_));
69        }
70        match value {
71            Value::Ipv4(x) => Ok(x),
72            _ => unimplemented!(),
73        }
74    }
75}
76
77impl ToSql for Ipv6 {
78    fn to_sql(self, _type_hint: Option<&Type>) -> Result<Value> { Ok(Value::Ipv6(self)) }
79}
80
81impl FromSql for Ipv6 {
82    fn from_sql(type_: &Type, value: Value) -> Result<Self> {
83        if !matches!(type_, Type::Ipv6) {
84            return Err(unexpected_type(type_));
85        }
86        match value {
87            Value::Ipv6(x) => Ok(x),
88            _ => unimplemented!(),
89        }
90    }
91}