1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
use crate::queries::shared::Query;
use sqlx::postgres::PgRow;
use sqlx::Row;

#[derive(Debug, Clone)]
pub struct DuplicateIndexes {
    pub size: String,
    pub idx1: String,
    pub idx2: String,
    pub idx3: String,
    pub idx4: String,
}

impl Query for DuplicateIndexes {
    fn new(row: &PgRow) -> Self {
        Self {
            size: row.try_get("size").unwrap_or_default(),
            idx1: row.try_get("idx1").unwrap_or_default(),
            idx2: row.try_get("idx2").unwrap_or_default(),
            idx3: row.try_get("idx3").unwrap_or_default(),
            idx4: row.try_get("idx4").unwrap_or_default(),
        }
    }

    fn to_row(&self) -> prettytable::Row {
        row![self.size, self.idx1, self.idx2, self.idx3, self.idx4]
    }

    fn headers() -> prettytable::Row {
        row!["size", "idx1", "idx2", "idx3", "idx4"]
    }

    fn read_file() -> String {
        include_str!("../sql/duplicate_indexes.sql").to_string()
    }
}