starfish_core/entities/
relation.rs

1use sea_orm::entity::prelude::*;
2use serde::{Deserialize, Serialize};
3
4#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Serialize, Deserialize)]
5#[sea_orm(table_name = "relation")]
6pub struct Model {
7    #[sea_orm(primary_key)]
8    pub id: i32,
9    #[sea_orm(unique, indexed)]
10    pub name: String,
11    #[sea_orm(indexed)]
12    pub from_entity: String,
13    #[sea_orm(indexed)]
14    pub to_entity: String,
15    #[sea_orm(indexed)]
16    pub directed: bool,
17}
18
19#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
20pub enum Relation {
21    #[sea_orm(
22        belongs_to = "super::entity::Entity",
23        from = "Column::FromEntity",
24        to = "super::entity::Column::Name"
25    )]
26    FromEntity,
27    #[sea_orm(
28        belongs_to = "super::entity::Entity",
29        from = "Column::ToEntity",
30        to = "super::entity::Column::Name"
31    )]
32    ToEntity,
33}
34
35impl ActiveModelBehavior for ActiveModel {}