134 lines
2.8 KiB
Rust
134 lines
2.8 KiB
Rust
use crate::sql::comment::shouldbespace;
|
|
use crate::sql::error::IResult;
|
|
use crate::sql::ident::ident_raw;
|
|
use derive::Store;
|
|
use nom::branch::alt;
|
|
use nom::bytes::complete::tag_no_case;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::fmt;
|
|
|
|
#[derive(Clone, Debug, Default, Eq, PartialEq, Serialize, Deserialize, Store, Hash)]
|
|
pub struct UseStatement {
|
|
pub ns: Option<String>,
|
|
pub db: Option<String>,
|
|
}
|
|
|
|
impl fmt::Display for UseStatement {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
f.write_str("USE")?;
|
|
if let Some(ref ns) = self.ns {
|
|
write!(f, " NS {ns}")?;
|
|
}
|
|
if let Some(ref db) = self.db {
|
|
write!(f, " DB {db}")?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub fn yuse(i: &str) -> IResult<&str, UseStatement> {
|
|
alt((both, ns, db))(i)
|
|
}
|
|
|
|
fn both(i: &str) -> IResult<&str, UseStatement> {
|
|
let (i, _) = tag_no_case("USE")(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, _) = alt((tag_no_case("NAMESPACE"), tag_no_case("NS")))(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, ns) = ident_raw(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, _) = alt((tag_no_case("DATABASE"), tag_no_case("DB")))(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, db) = ident_raw(i)?;
|
|
Ok((
|
|
i,
|
|
UseStatement {
|
|
ns: Some(ns),
|
|
db: Some(db),
|
|
},
|
|
))
|
|
}
|
|
|
|
fn ns(i: &str) -> IResult<&str, UseStatement> {
|
|
let (i, _) = tag_no_case("USE")(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, _) = alt((tag_no_case("NAMESPACE"), tag_no_case("NS")))(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, ns) = ident_raw(i)?;
|
|
Ok((
|
|
i,
|
|
UseStatement {
|
|
ns: Some(ns),
|
|
db: None,
|
|
},
|
|
))
|
|
}
|
|
|
|
fn db(i: &str) -> IResult<&str, UseStatement> {
|
|
let (i, _) = tag_no_case("USE")(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, _) = alt((tag_no_case("DATABASE"), tag_no_case("DB")))(i)?;
|
|
let (i, _) = shouldbespace(i)?;
|
|
let (i, db) = ident_raw(i)?;
|
|
Ok((
|
|
i,
|
|
UseStatement {
|
|
ns: None,
|
|
db: Some(db),
|
|
},
|
|
))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn use_query_ns() {
|
|
let sql = "USE NS test";
|
|
let res = yuse(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!(
|
|
out,
|
|
UseStatement {
|
|
ns: Some(String::from("test")),
|
|
db: None,
|
|
}
|
|
);
|
|
assert_eq!("USE NS test", format!("{}", out));
|
|
}
|
|
|
|
#[test]
|
|
fn use_query_db() {
|
|
let sql = "USE DB test";
|
|
let res = yuse(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!(
|
|
out,
|
|
UseStatement {
|
|
ns: None,
|
|
db: Some(String::from("test")),
|
|
}
|
|
);
|
|
assert_eq!("USE DB test", format!("{}", out));
|
|
}
|
|
|
|
#[test]
|
|
fn use_query_both() {
|
|
let sql = "USE NS test DB test";
|
|
let res = yuse(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!(
|
|
out,
|
|
UseStatement {
|
|
ns: Some(String::from("test")),
|
|
db: Some(String::from("test")),
|
|
}
|
|
);
|
|
assert_eq!("USE NS test DB test", format!("{}", out));
|
|
}
|
|
}
|