89 lines
2.0 KiB
Rust
89 lines
2.0 KiB
Rust
use crate as sea_orm;
|
|
use crate::entity::prelude::*;
|
|
|
|
#[derive(Copy, Clone, Default, Debug, DeriveEntity)]
|
|
#[sea_orm(table_name = "filling")]
|
|
pub struct Entity;
|
|
|
|
#[derive(Clone, Debug, PartialEq, Eq, DeriveModel, DeriveActiveModel)]
|
|
pub struct Model {
|
|
pub id: i32,
|
|
pub name: String,
|
|
pub vendor_id: Option<i32>,
|
|
#[sea_orm(ignore)]
|
|
pub ignored_attr: i32,
|
|
}
|
|
|
|
// If your column names are not in snake-case, derive `DeriveCustomColumn` here.
|
|
#[derive(Copy, Clone, Debug, EnumIter, DeriveCustomColumn)]
|
|
pub enum Column {
|
|
Id,
|
|
Name,
|
|
VendorId,
|
|
}
|
|
|
|
// Then, customize each column names here.
|
|
impl IdenStatic for Column {
|
|
fn as_str(&self) -> &str {
|
|
match self {
|
|
// Override column names
|
|
Self::Id => "id",
|
|
// Leave all other columns using default snake-case values
|
|
_ => self.default_as_str(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter, DerivePrimaryKey)]
|
|
pub enum PrimaryKey {
|
|
Id,
|
|
}
|
|
|
|
impl PrimaryKeyTrait for PrimaryKey {
|
|
type ValueType = i32;
|
|
|
|
fn auto_increment() -> bool {
|
|
true
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter)]
|
|
pub enum Relation {
|
|
Vendor,
|
|
}
|
|
|
|
impl ColumnTrait for Column {
|
|
type EntityName = Entity;
|
|
|
|
fn def(&self) -> ColumnDef {
|
|
match self {
|
|
Self::Id => ColumnType::Integer.def(),
|
|
Self::Name => ColumnType::String(None).def(),
|
|
Self::VendorId => ColumnType::Integer.def().nullable(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl RelationTrait for Relation {
|
|
fn def(&self) -> RelationDef {
|
|
match self {
|
|
Self::Vendor => Entity::belongs_to(super::vendor::Entity)
|
|
.from(Column::VendorId)
|
|
.to(super::vendor::Column::Id)
|
|
.into(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Related<super::cake::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
super::cake_filling::Relation::Cake.def()
|
|
}
|
|
|
|
fn via() -> Option<RelationDef> {
|
|
Some(super::cake_filling::Relation::Filling.def().rev())
|
|
}
|
|
}
|
|
|
|
impl ActiveModelBehavior for ActiveModel {}
|