use std::borrow::Cow;
use exprable::Escape;
use std::any::Any;
use std::marker::PhantomData;
use expr::{Unescaped, Escaped};
use groupable::{Fragmentable, Query};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct SuffixExpr<'a, S: Any = Unescaped> {
column: Cow<'a, str>,
target: Cow<'a, str>,
_marker: PhantomData<S>,
}
impl<'a> SuffixExpr<'a, Unescaped> {
pub fn new<T>(column: T, target: T) -> SuffixExpr<'a, Unescaped>
where T: Into<Cow<'a, str>>
{
SuffixExpr {
column: column.into(),
target: target.into(),
_marker: PhantomData,
}
}
pub fn prepare(self) -> SuffixExpr<'a, Escaped> {
SuffixExpr {
column: self.column,
target: format!("{}", Escape(&*self.target.into_owned())).into(),
_marker: PhantomData,
}
}
}
impl<'a> SuffixExpr<'a, Escaped> {
pub fn build(self) -> String {
format!("{}:${}", self.column.into_owned(), self.target.into_owned())
}
}
impl<'a> Fragmentable for SuffixExpr<'a, Escaped> {
fn to_fragment(self) -> Query {
vec![self.build()]
}
}
operators!(SuffixExpr<'a, Escaped>);
#[cfg(test)]
mod tests {
use super::*;
use std::borrow::Cow;
use std::marker::PhantomData;
use expr::{Unescaped, Escaped};
#[test]
fn test_new() {
let expr = SuffixExpr::new("key", "Rust");
let expected = SuffixExpr::<Unescaped> {
column: Cow::Borrowed("key"),
target: Cow::Borrowed("Rust"),
_marker: PhantomData,
};
assert_eq!(expected, expr)
}
#[test]
fn test_prepare() {
let expr = SuffixExpr::new("key", "Rust").prepare();
let expected = SuffixExpr::<Escaped> {
column: Cow::Borrowed("key"),
target: Cow::Borrowed("Rust"),
_marker: PhantomData,
};
assert_eq!(expected, expr)
}
#[test]
fn test_build() {
let expr = SuffixExpr::new("key", "Rust").prepare().build();
let expected = "key:$Rust";
assert_eq!(expected, expr)
}
}