netmito 0.6.8

A Unified Distributed Transport Evaluation Framework
Documentation
//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.15

use sea_orm::entity::prelude::*;

use super::state::TaskState;

#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "active_tasks")]
pub struct Model {
    #[sea_orm(primary_key)]
    pub id: i64,
    pub creator_id: i64,
    pub group_id: i64,
    pub task_id: i64,
    #[sea_orm(unique)]
    pub uuid: Uuid,
    pub tags: Vec<String>,
    pub labels: Vec<String>,
    pub created_at: TimeDateTimeWithTimeZone,
    pub updated_at: TimeDateTimeWithTimeZone,
    pub state: TaskState,
    pub assigned_worker: Option<i64>,
    pub timeout: i64,
    pub priority: i32,
    pub spec: Json,
    pub result: Option<Json>,
    pub upstream_task_uuid: Option<Uuid>,
    pub downstream_task_uuid: Option<Uuid>,
}

#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
    #[sea_orm(
        belongs_to = "super::groups::Entity",
        from = "Column::GroupId",
        to = "super::groups::Column::Id",
        on_update = "Cascade",
        on_delete = "Restrict"
    )]
    Groups,
    #[sea_orm(
        belongs_to = "super::users::Entity",
        from = "Column::CreatorId",
        to = "super::users::Column::Id",
        on_update = "Cascade",
        on_delete = "Restrict"
    )]
    Users,
}

impl Related<super::groups::Entity> for Entity {
    fn to() -> RelationDef {
        Relation::Groups.def()
    }
}

impl Related<super::users::Entity> for Entity {
    fn to() -> RelationDef {
        Relation::Users.def()
    }
}

impl ActiveModelBehavior for ActiveModel {}

/// This conversion is used when archiving an active task.
/// But do remember to set reporter_uuid separately after conversion
impl From<Model> for super::archived_tasks::Model {
    fn from(task: Model) -> super::archived_tasks::Model {
        Self {
            id: task.id,
            creator_id: task.creator_id,
            group_id: task.group_id,
            task_id: task.task_id,
            uuid: task.uuid,
            tags: task.tags,
            labels: task.labels,
            created_at: task.created_at,
            updated_at: task.updated_at,
            state: task.state,
            assigned_worker: task.assigned_worker,
            timeout: task.timeout,
            priority: task.priority,
            spec: task.spec,
            result: task.result,
            upstream_task_uuid: task.upstream_task_uuid,
            downstream_task_uuid: task.downstream_task_uuid,
            reporter_uuid: None,
        }
    }
}