diff --git a/counit-server/src/model/archguard_model.rs b/counit-server/src/model/archguard_model.rs index e9fca5a..4c31b42 100644 --- a/counit-server/src/model/archguard_model.rs +++ b/counit-server/src/model/archguard_model.rs @@ -55,9 +55,18 @@ pub struct CodeDatabaseRelation { tables: Vec, #[serde(default)] sqls: Vec, + #[serde(default)] + implementations: Vec, + #[serde(default)] relations: Vec } +impl std::fmt::Display for CodeDatabaseRelation { + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { + write!(f, "{}.{}.{}", self.package_name, self.class_name, self.function_name) + } +} + #[derive(Serialize, Deserialize)] #[serde(rename_all = "camelCase")] pub struct NodeRelation {