Add Materialzied views

Author: kurotych

puml-lib/db_ent.puml

@@ -18,6 +18,14 @@
   !return 'entity "**' + $name + '**"' + " as " + $name
 !endfunction
 
+!function view($name, $materialized=false)
+  !if ($materialized == false)
+    !return 'entity "**' + $name + ' **<color:SkyBlue>**(V)**</color>"' + " as " + $name
+  !else
+    !return 'entity "**' + $name + ' **<color:DarkBlue>**(MV)**</color>"' + " as " + $name
+  !endif
+!endfunction
+
 !procedure enum($name, $variants)
   !$list = %splitstr($variants, ",")
 
@@ -35,5 +43,7 @@
    |<color:#aaaaaa><&key></color>| Foreign Key |
    | &#8226; | Mandatory field (Not Null) |
    | <color:purple>**(E)**</color> | Enum |
+   | <color:SkyBlue>**(V)**</color> | View |
+   | <color:DarkBlue>**(MV)**</color> | Materialized View |
   endlegend
 !endprocedure

src/plantuml_generator.rs

@@ -20,7 +20,8 @@ static PUML_TEMPLATE: &str = "@startuml\n\n\
 
 static ENTITY_TEMPLATE: &str = "table({name}) \\{\n{pks}  ---\n{fks}{nns}{others}}\n";
 
-static VIEW_TEMPLATE: &str = "view({name}) \\{\n{pks}  ---\n{fks}{nns}{others}}\n";
+static VIEW_TEMPLATE: &str =
+    "view({name}{{ if materialized}}, $materialized=true{{ endif }}) \\{\n{columns}}\n";
 
 static COLUMN_TEMPLATE: &str = "  column({col.name}, \"{col.datatype}\"{{ if is_pk }}, $pk=true{{ endif }}{{ if is_fk }}, $fk=true{{ endif }}{{if is_nn}}, $nn=true{{ endif }})\n";
 
@@ -47,6 +48,13 @@ struct SColumn<'a> {
     is_nn_and_not_pk: bool,
 }
 
+#[derive(Serialize)]
+struct SView {
+    name: String,
+    columns: String,
+    materialized: bool,
+}
+
 #[derive(Serialize)]
 struct SEntity {
     name: String,
@@ -206,12 +214,10 @@ impl<'a> PlantUmlDefaultGenerator<'a> {
         };
         Ok(self.str_templates.render(
             "view",
-            &SEntity {
-                pks: String::default(),
-                fks: String::default(),
-                nns: String::default(),
-                others: columns_render(view.columns.clone())?,
+            &SView {
+                columns: columns_render(view.columns.clone())?,
                 name: view.name.clone(),
+                materialized: view.materizlied,
             },
         )?)
     }

src/psql_erd_loader.rs

@@ -25,6 +25,13 @@ WHERE table_schema = $1
 ORDER BY table_name;
 "#;
 
+static GET_MATERIALIZED_VIEWS: &str = r#"
+SELECT trim(both '"' from matviewname) AS matview_name
+FROM pg_matviews
+WHERE schemaname = $1
+ORDER BY matviewname;
+"#;
+
 /// https://www.postgresql.org/docs/current/catalog-pg-attribute.html
 static GET_COLUMNS_BASIC_INFO_QUERY: &str = r#"
 SELECT attname                                  AS col_name,
@@ -405,7 +412,34 @@ impl SqlERDataLoader for PostgreSqlERDLoader {
         let (tables_and_views, enums) = self.load_tables(table_names).await?;
         let foreign_keys = self.get_fks(&tables_and_views)?;
 
-        let mut views: Vec<Arc<View>> = vec![];
+        let mat_views_name: Vec<String> = self
+            .client
+            .query(GET_MATERIALIZED_VIEWS, &[&self.schema_name])
+            .await?
+            .iter()
+            .map(|row| row.get("matview_name"))
+            .collect();
+        let (mat_views, _) = self.load_tables(mat_views_name).await?;
+
+        // Collect table names and types as a vector of tuples
+        let table_names_with_types: Vec<(String, TableType)> = res
+            .iter()
+            .map(|row| (row.get("table_name"), row.get("table_type")))
+            .collect();
+
+        let mut views: Vec<Arc<View>> = mat_views
+            .into_iter()
+            .map(|v| {
+                let v = Arc::try_unwrap(v).unwrap();
+                View {
+                    materizlied: true,
+                    name: v.name,
+                    columns: v.columns,
+                }
+                .into()
+            })
+            .collect();
+
         let mut tables: Vec<Arc<Table>> = vec![];
 
         for entity in tables_and_views.into_iter() {