1 //! 任务日志
2 //!
3 //! DADK在执行任务时，会把一些日志记录到任务的文件夹下。
4
5 use chrono::{DateTime, Utc};
6 use log::warn;
7 use serde::{Deserialize, Deserializer, Serialize};
8
9 /// 任务日志（输出到任务构建日志目录下的）
10 #[derive(Debug, Clone, Serialize, Deserialize)]
11 pub struct TaskLog {
12 /// 任务执行完成时间
13 #[serde(
14 deserialize_with = "ok_or_default",
15 skip_serializing_if = "Option::is_none"
16 )]
17 build_timestamp: Option<DateTime<Utc>>,
18 install_timestamp: Option<DateTime<Utc>>,
19 /// 任务构建状态
20 build_status: Option<BuildStatus>,
21 /// 任务安装状态
22 install_status: Option<InstallStatus>,
23 }
24
ok_or_default<'a, T, D>(deserializer: D) -> Result<T, D::Error> where T: Deserialize<'a> + Default, D: Deserializer<'a>,25 fn ok_or_default<'a, T, D>(deserializer: D) -> Result<T, D::Error>
26 where
27 T: Deserialize<'a> + Default,
28 D: Deserializer<'a>,
29 {
30 let r = Option::deserialize(deserializer).map(|x: Option<T>| x.unwrap_or_default());
31
32 Ok(r.unwrap_or_default())
33 }
34
35 impl TaskLog {
new() -> Self36 pub fn new() -> Self {
37 Self {
38 build_timestamp: None,
39 build_status: None,
40 install_timestamp: None,
41 install_status: None,
42 }
43 }
44
45 #[allow(dead_code)]
set_build_time(&mut self, time: DateTime<Utc>)46 pub fn set_build_time(&mut self, time: DateTime<Utc>) {
47 self.build_timestamp = Some(time);
48 }
49
build_time(&self) -> Option<&DateTime<Utc>>50 pub fn build_time(&self) -> Option<&DateTime<Utc>> {
51 self.build_timestamp.as_ref()
52 }
53
set_build_time_now(&mut self)54 pub fn set_build_time_now(&mut self) {
55 self.build_timestamp = Some(Utc::now());
56 }
57
install_time(&self) -> Option<&DateTime<Utc>>58 pub fn install_time(&self) -> Option<&DateTime<Utc>> {
59 self.install_timestamp.as_ref()
60 }
61
set_install_time_now(&mut self)62 pub fn set_install_time_now(&mut self) {
63 self.install_timestamp = Some(Utc::now());
64 }
65
set_build_status(&mut self, status: BuildStatus)66 pub fn set_build_status(&mut self, status: BuildStatus) {
67 self.build_status = Some(status);
68 }
69
clean_build_status(&mut self)70 pub fn clean_build_status(&mut self) {
71 self.build_status = None;
72 }
73
build_status(&self) -> Option<&BuildStatus>74 pub fn build_status(&self) -> Option<&BuildStatus> {
75 self.build_status.as_ref()
76 }
77
install_status(&self) -> Option<&InstallStatus>78 pub fn install_status(&self) -> Option<&InstallStatus> {
79 self.install_status.as_ref()
80 }
81
set_install_status(&mut self, status: InstallStatus)82 pub fn set_install_status(&mut self, status: InstallStatus) {
83 self.install_status = Some(status);
84 }
85
clean_install_status(&mut self)86 pub fn clean_install_status(&mut self) {
87 self.install_status = None;
88 }
89 }
90
91 /// 任务构建状态
92 #[derive(Debug, Clone, Serialize, PartialEq)]
93 pub enum BuildStatus {
94 #[serde(rename = "success")]
95 Success,
96 #[serde(rename = "failed")]
97 Failed,
98 }
99
100 impl<'de> Deserialize<'de> for BuildStatus {
deserialize<D>(deserializer: D) -> Result<BuildStatus, D::Error> where D: Deserializer<'de>,101 fn deserialize<D>(deserializer: D) -> Result<BuildStatus, D::Error>
102 where
103 D: Deserializer<'de>,
104 {
105 let s = String::deserialize(deserializer)?.to_ascii_lowercase();
106 match s.as_str() {
107 "success" => Ok(BuildStatus::Success),
108 "failed" => Ok(BuildStatus::Failed),
109 _ => {
110 warn!("invalid build status: {}", s);
111 Ok(BuildStatus::Failed)
112 }
113 }
114 }
115 }
116
117 /// 任务安装状态
118 #[derive(Debug, Clone, Serialize, PartialEq)]
119 pub enum InstallStatus {
120 #[serde(rename = "success")]
121 Success,
122 #[serde(rename = "failed")]
123 Failed,
124 }
125
126 impl<'de> Deserialize<'de> for InstallStatus {
deserialize<D>(deserializer: D) -> Result<InstallStatus, D::Error> where D: Deserializer<'de>,127 fn deserialize<D>(deserializer: D) -> Result<InstallStatus, D::Error>
128 where
129 D: Deserializer<'de>,
130 {
131 let s = String::deserialize(deserializer)?.to_ascii_lowercase();
132 match s.as_str() {
133 "success" => Ok(InstallStatus::Success),
134 "failed" => Ok(InstallStatus::Failed),
135 _ => {
136 warn!("invalid install status: {}", s);
137 Ok(InstallStatus::Failed)
138 }
139 }
140 }
141 }
142