169 lines
3.6 KiB
Rust
169 lines
3.6 KiB
Rust
use crate::sql::common::is_hex;
|
|
use crate::sql::error::IResult;
|
|
use crate::sql::escape::quote_str;
|
|
use crate::sql::strand::Strand;
|
|
use nom::branch::alt;
|
|
use nom::bytes::complete::take_while_m_n;
|
|
use nom::character::complete::char;
|
|
use nom::combinator::recognize;
|
|
use nom::sequence::delimited;
|
|
use nom::sequence::tuple;
|
|
use revision::revisioned;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::fmt::{self, Display, Formatter};
|
|
use std::ops::Deref;
|
|
use std::str;
|
|
use std::str::FromStr;
|
|
|
|
pub(crate) const TOKEN: &str = "$surrealdb::private::sql::Uuid";
|
|
|
|
#[derive(Clone, Debug, Default, Eq, Ord, PartialEq, PartialOrd, Serialize, Deserialize, Hash)]
|
|
#[serde(rename = "$surrealdb::private::sql::Uuid")]
|
|
#[revisioned(revision = 1)]
|
|
pub struct Uuid(pub uuid::Uuid);
|
|
|
|
impl From<uuid::Uuid> for Uuid {
|
|
fn from(v: uuid::Uuid) -> Self {
|
|
Uuid(v)
|
|
}
|
|
}
|
|
|
|
impl From<Uuid> for uuid::Uuid {
|
|
fn from(s: Uuid) -> Self {
|
|
s.0
|
|
}
|
|
}
|
|
|
|
impl FromStr for Uuid {
|
|
type Err = ();
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
Self::try_from(s)
|
|
}
|
|
}
|
|
|
|
impl TryFrom<String> for Uuid {
|
|
type Error = ();
|
|
fn try_from(v: String) -> Result<Self, Self::Error> {
|
|
Self::try_from(v.as_str())
|
|
}
|
|
}
|
|
|
|
impl TryFrom<Strand> for Uuid {
|
|
type Error = ();
|
|
fn try_from(v: Strand) -> Result<Self, Self::Error> {
|
|
Self::try_from(v.as_str())
|
|
}
|
|
}
|
|
|
|
impl TryFrom<&str> for Uuid {
|
|
type Error = ();
|
|
fn try_from(v: &str) -> Result<Self, Self::Error> {
|
|
match uuid::Uuid::try_parse(v) {
|
|
Ok(v) => Ok(Self(v)),
|
|
Err(_) => Err(()),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Deref for Uuid {
|
|
type Target = uuid::Uuid;
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.0
|
|
}
|
|
}
|
|
|
|
impl Uuid {
|
|
/// Generate a new V4 UUID
|
|
pub fn new() -> Self {
|
|
#[cfg(uuid_unstable)]
|
|
{
|
|
Self(uuid::Uuid::now_v7())
|
|
}
|
|
#[cfg(not(uuid_unstable))]
|
|
{
|
|
Self(uuid::Uuid::new_v4())
|
|
}
|
|
}
|
|
/// Generate a new V4 UUID
|
|
pub fn new_v4() -> Self {
|
|
Self(uuid::Uuid::new_v4())
|
|
}
|
|
/// Generate a new V7 UUID
|
|
#[cfg(uuid_unstable)]
|
|
pub fn new_v7() -> Self {
|
|
Self(uuid::Uuid::now_v7())
|
|
}
|
|
/// Convert the Uuid to a raw String
|
|
pub fn to_raw(&self) -> String {
|
|
self.0.to_string()
|
|
}
|
|
}
|
|
|
|
impl Display for Uuid {
|
|
fn fmt(&self, f: &mut Formatter) -> fmt::Result {
|
|
Display::fmt("e_str(&self.0.to_string()), f)
|
|
}
|
|
}
|
|
|
|
pub fn uuid(i: &str) -> IResult<&str, Uuid> {
|
|
alt((uuid_single, uuid_double))(i)
|
|
}
|
|
|
|
fn uuid_single(i: &str) -> IResult<&str, Uuid> {
|
|
delimited(char('\''), uuid_raw, char('\''))(i)
|
|
}
|
|
|
|
fn uuid_double(i: &str) -> IResult<&str, Uuid> {
|
|
delimited(char('\"'), uuid_raw, char('\"'))(i)
|
|
}
|
|
|
|
fn uuid_raw(i: &str) -> IResult<&str, Uuid> {
|
|
let (i, v) = recognize(tuple((
|
|
take_while_m_n(8, 8, is_hex),
|
|
char('-'),
|
|
take_while_m_n(4, 4, is_hex),
|
|
char('-'),
|
|
alt((
|
|
char('1'),
|
|
char('2'),
|
|
char('3'),
|
|
char('4'),
|
|
char('5'),
|
|
char('6'),
|
|
char('7'),
|
|
char('8'),
|
|
)),
|
|
take_while_m_n(3, 3, is_hex),
|
|
char('-'),
|
|
take_while_m_n(4, 4, is_hex),
|
|
char('-'),
|
|
take_while_m_n(12, 12, is_hex),
|
|
)))(i)?;
|
|
Ok((i, Uuid::try_from(v).unwrap()))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn uuid_v1() {
|
|
let sql = "e72bee20-f49b-11ec-b939-0242ac120002";
|
|
let res = uuid_raw(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!("'e72bee20-f49b-11ec-b939-0242ac120002'", format!("{}", out));
|
|
assert_eq!(out, Uuid::try_from("e72bee20-f49b-11ec-b939-0242ac120002").unwrap());
|
|
}
|
|
|
|
#[test]
|
|
fn uuid_v4() {
|
|
let sql = "b19bc00b-aa98-486c-ae37-c8e1c54295b1";
|
|
let res = uuid_raw(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!("'b19bc00b-aa98-486c-ae37-c8e1c54295b1'", format!("{}", out));
|
|
assert_eq!(out, Uuid::try_from("b19bc00b-aa98-486c-ae37-c8e1c54295b1").unwrap());
|
|
}
|
|
}
|