59 lines
1.2 KiB
Rust
59 lines
1.2 KiB
Rust
|
use crate::sql::common::val_char;
|
||
|
|
use nom::bytes::complete::tag;
|
||
|
|
use nom::bytes::complete::take_while1;
|
||
|
|
use nom::IResult;
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
use std::fmt;
|
||
|
|
use std::str;
|
||
|
|
|
||
|
|
#[derive(Clone, Debug, Default, Eq, Hash, PartialEq, Serialize, Deserialize)]
|
||
|
|
pub struct Param {
|
||
|
|
pub name: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'a> From<&'a str> for Param {
|
||
|
|
fn from(p: &str) -> Param {
|
||
|
|
Param {
|
||
|
|
name: String::from(p),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl fmt::Display for Param {
|
||
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||
|
|
write!(f, "${}", &self.name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn param(i: &str) -> IResult<&str, Param> {
|
||
|
|
let (i, _) = tag("$")(i)?;
|
||
|
|
let (i, v) = take_while1(val_char)(i)?;
|
||
|
|
Ok((i, Param::from(v)))
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
|
||
|
|
use super::*;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn param_normal() {
|
||
|
|
let sql = "$test";
|
||
|
|
let res = param(sql);
|
||
|
|
assert!(res.is_ok());
|
||
|
|
let out = res.unwrap().1;
|
||
|
|
assert_eq!("$test", format!("{}", out));
|
||
|
|
assert_eq!(out, Param::from("test"));
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn param_longer() {
|
||
|
|
let sql = "$test_and_deliver";
|
||
|
|
let res = param(sql);
|
||
|
|
assert!(res.is_ok());
|
||
|
|
let out = res.unwrap().1;
|
||
|
|
assert_eq!("$test_and_deliver", format!("{}", out));
|
||
|
|
assert_eq!(out, Param::from("test_and_deliver"));
|
||
|
|
}
|
||
|
|
}
|