RG-Stats/database.cpp at master · timewall/RG-Stats · GitHub

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/*
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
*/

//test: test/database_main.cpp


#include "database.h"

std::string database::v_replace(std::string &s, std::string toReplace, std::string replaceWith) {
    return(s.replace(s.find(toReplace), toReplace.length(), replaceWith));
}

database::database(std::string pg_db_data){

     conn = tntdb::connect(pg_db_data);
}


tntdb::Connection database::get_conn(){

    return conn;
}

std::string database::get_rows(std::string ntable){

    std::string data;
    conn.selectValue("SELECT COUNT(*) FROM " + ntable).get(data);
    return data;
}

std::string database::get_table_size(std::string ntable){

    std::string data;
    conn.selectValue("SELECT pg_size_pretty(pg_relation_size('" + ntable + "'))").get(data);
    return data;
}

std::string database::get_db_size(){

    std::string data;
    conn.selectValue("SELECT pg_size_pretty(pg_database_size('rg-stats'))").get(data);

    return data;
}

std::string database::get_db_info(){

    std::string db_info =   "+---------------------------------------- \n"
                            "| Table: mp_all \n"
                            "+---------------------------------------- \n"
                            "| Number of Entries   : =v1 \n"
                            "| Table Size          : =v2 \n"
                            "+---------------------------------------- \n"
                            "+---------------------------------------- \n"
                            "| Table: mp_item_name \n"
                            "+---------------------------------------- \n"
                            "| Number of Entries   : =v3 \n"
                            "| Table Size          : =v4 \n"
                            "+---------------------------------------- \n"
                            "+---------------------------------------- \n"
                            "| Database Size       : =v5 \n"
                            "+----------------------------------------";


    db_info = v_replace(db_info, "=v1", get_rows("mp_all"));
    db_info = v_replace(db_info, "=v2", get_table_size("mp_all"));
    db_info = v_replace(db_info, "=v3", get_rows("mp_item_name"));
    db_info = v_replace(db_info, "=v4", get_table_size("mp_item_name"));
    db_info = v_replace(db_info, "=v5", get_db_size());

    return db_info;
}