use derive_db_execute::Execute; use diesel::expression::sql_literal::sql; use diesel::prelude::*; use jirs_data::{IssueId, IssuePriority, IssueStatusId, IssueType, ProjectId, UserId}; use crate::models::Issue; #[derive(Default, Execute)] #[db_exec( result = "Issue", schema = "issues", find = "issues.filter(id.eq(msg.issue_id)).distinct()" )] pub struct LoadIssue { pub issue_id: IssueId, } #[derive(Execute)] #[db_exec( result = "Issue", schema = "issues", load = "issues.filter(project_id.eq(msg.project_id)).distinct()" )] pub struct LoadProjectIssues { pub project_id: ProjectId, } #[derive(Default, Execute)] #[db_exec(result = "Issue", schema = "issues")] pub struct UpdateIssue { pub issue_id: jirs_data::IssueId, pub title: Option, pub issue_type: Option, pub priority: Option, pub list_position: Option, pub description: Option, pub description_text: Option, pub estimate: Option, pub time_spent: Option, pub time_remaining: Option, pub project_id: Option, pub user_ids: Option>, pub reporter_id: Option, pub issue_status_id: Option, pub epic_id: Option>, } impl UpdateIssue { fn execute(self, conn: &crate::DbPooledConn) -> Result { let msg = self; use crate::schema::issues::dsl::*; if let Some(user_ids) = msg.user_ids { crate::issue_assignees::DropIssueAssignees { issue_id: msg.issue_id, user_ids: user_ids.clone(), } .execute(conn)?; let existing: Vec = crate::issue_assignees::LoadAssigneesIds { issue_id: msg.issue_id, } .execute(conn)?; crate::issue_assignees::AsignMultiple { issue_id: msg.issue_id, user_ids: user_ids .into_iter() .filter(|u_id| !existing.contains(u_id)) .collect::>(), } .execute(conn)?; } diesel::update(issues.find(msg.issue_id)) .set(( msg.title.map(|v| title.eq(v)), msg.issue_type.map(|v| issue_type.eq(v)), msg.issue_status_id.map(|v| issue_status_id.eq(v)), msg.priority.map(|p| priority.eq(p)), msg.list_position.map(|pos| list_position.eq(pos)), msg.description.map(|desc| description.eq(desc)), msg.description_text.map(|t| description_text.eq(t)), msg.estimate.map(|v| estimate.eq(v)), msg.time_spent.map(|v| time_spent.eq(v)), msg.time_remaining.map(|v| time_remaining.eq(v)), msg.project_id.map(|v| project_id.eq(v)), msg.reporter_id.map(|v| reporter_id.eq(v)), msg.epic_id.map(|v| epic_id.eq(v)), updated_at.eq(chrono::Utc::now().naive_utc()), )) .get_result(conn) .map_err(|e| { log::debug!("{:?}", e); crate::DatabaseError::GenericFailure( crate::OperationError::Create, crate::ResourceKind::Issue, ) }) } } #[derive(Execute)] #[db_exec( result = "Issue", schema = "issues", destroy = r#"{ crate::issue_assignees::DeleteIssueAssignees { issue_id: msg.issue_id } .execute(conn)?; diesel::delete(issues.find(msg.issue_id)) }"# )] pub struct DeleteIssue { pub issue_id: IssueId, } mod inner { use derive_db_execute::Execute; use diesel::prelude::*; use jirs_data::{IssuePriority, IssueStatusId, IssueType}; use crate::models::Issue; #[derive(Default, Execute)] #[db_exec( result = "Issue", schema = "issues", create = r#" diesel::insert_into(issues) .values(( title.eq(msg.title), issue_type.eq(msg.issue_type), issue_status_id.eq(msg.issue_status_id), priority.eq(msg.priority), list_position.eq(msg.list_position), description.eq(msg.description), description_text.eq(msg.description_text), estimate.eq(msg.estimate), time_spent.eq(msg.time_spent), time_remaining.eq(msg.time_remaining), reporter_id.eq(msg.reporter_id), project_id.eq(msg.project_id), epic_id.eq(msg.epic_id) )) .on_conflict_do_nothing() "# )] pub struct CreateIssue { pub title: String, pub list_position: i32, pub issue_type: IssueType, pub issue_status_id: IssueStatusId, pub priority: IssuePriority, pub description: Option, pub description_text: Option, pub estimate: Option, pub time_spent: Option, pub time_remaining: Option, pub project_id: jirs_data::ProjectId, pub reporter_id: jirs_data::UserId, pub epic_id: Option, } } #[derive(Execute)] #[db_exec(result = "Issue", schema = "issues")] pub struct CreateIssue { pub title: String, pub issue_type: IssueType, pub issue_status_id: IssueStatusId, pub priority: IssuePriority, pub description: Option, pub description_text: Option, pub estimate: Option, pub time_spent: Option, pub time_remaining: Option, pub project_id: jirs_data::ProjectId, pub reporter_id: jirs_data::UserId, pub user_ids: Vec, pub epic_id: Option, } impl CreateIssue { fn execute(self, conn: &crate::DbPooledConn) -> Result { use crate::schema::issues::dsl::*; let msg = self; let pos = issues .select(sql("COALESCE(max(list_position), 0) + 1")) .get_result::(conn) .map_err(|e| { log::error!("resolve new issue position failed {}", e); crate::DatabaseError::Issue(crate::IssueError::BadListPosition) })?; let i_s_id: IssueStatusId = if msg.issue_status_id == 0 { crate::issue_statuses::LoadIssueStatuses { project_id: msg.project_id, } .execute(conn)? .first() .ok_or(crate::DatabaseError::Issue( crate::IssueError::NoIssueStatuses, ))? .id } else { msg.issue_status_id }; let assign_users = msg .user_ids .iter() .cloned() .filter(|u_id| *u_id != msg.reporter_id) .collect::>(); let issue = inner::CreateIssue { title: msg.title, list_position: pos, issue_type: msg.issue_type, issue_status_id: i_s_id, priority: msg.priority, description: msg.description, description_text: msg.description_text, estimate: msg.estimate, time_spent: msg.time_spent, time_remaining: msg.time_remaining, project_id: msg.project_id, reporter_id: msg.reporter_id, epic_id: msg.epic_id, } .execute(conn)?; crate::issue_assignees::AsignMultiple { issue_id: issue.id, user_ids: assign_users, } .execute(conn); issues.find(issue.id).get_result(conn).map_err(|e| { log::error!("{:?}", e); crate::DatabaseError::GenericFailure( crate::OperationError::Create, crate::ResourceKind::Issue, ) }) } }