psychologylib/enums/
feature.rs
use serde::Deserialize;
use serde::Serialize;
use std::str;
use strum_macros::EnumString;
#[cfg(feature = "diesel")]
use {
diesel::backend::Backend,
diesel::deserialize::FromSql,
diesel::pg::Pg,
diesel::serialize::{IsNull, Output, ToSql},
diesel::types::Varchar,
diesel::{deserialize, not_none, serialize},
diesel::{AsExpression, FromSqlRow},
std::io::Write,
};
#[derive(Debug, PartialEq, Clone, Serialize, Deserialize, EnumString)]
#[cfg_attr(feature = "diesel", derive(AsExpression, FromSqlRow))]
#[cfg_attr(feature = "diesel", sql_type = "Varchar")]
pub enum Feature {
Friendliness,
Interestingness,
Clarity,
}
#[cfg(feature = "diesel")]
impl ToSql<Varchar, Pg> for Feature {
fn to_sql<W: Write>(&self, out: &mut Output<W, Pg>) -> serialize::Result {
match *self {
Feature::Friendliness => out.write_all(b"Friendliness")?,
Feature::Interestingness => out.write_all(b"Interestingness")?,
Feature::Clarity => out.write_all(b"Clarity")?,
}
Ok(IsNull::No)
}
}
#[cfg(feature = "diesel")]
impl FromSql<Varchar, Pg> for Feature {
fn from_sql(bytes: Option<&<Pg as Backend>::RawValue>) -> deserialize::Result<Self> {
match not_none!(bytes) {
b"Friendliness" => Ok(Feature::Friendliness),
b"Interestingness" => Ok(Feature::Interestingness),
b"Clarity" => Ok(Feature::Clarity),
_ => Err("Unrecognized enum variant".into()),
}
}
}