source: trunk/packages/xvm-munin-config/master/usr/share/xvm-munin-master-config/plugins/postgres_space_ @ 1761

Last change on this file since 1761 was 1761, checked in by broder, 16 years ago

Initial checkin of xvm-munin-config package

  • Property svn:executable set to *
File size: 6.0 KB
RevLine 
[1761]1#!/usr/bin/perl -w
2# -*- perl -*-
3
4# Written by Bjrn Ruberg (bjorn@linpro.no) 2006
5# Rewritten by Moses Moore 2006-04-08  moc.iazom@sesom
6# Licenced under GPL
7
8# Magic markers
9#%# family=auto
10#%# capabilities=autoconf suggest
11
12use strict;
13use DBI;
14use vars qw ( $debug $suggest $configure $dbh );
15
16# Package maintainers should provide an environment
17# file for the /etc/munin/plugin-conf.d/ directory
18# to override these values if necessary.
19# NOTE: The plugin (also when auto configured) should
20# be run by the postgresql user account.
21
22# Need these variables at an early stage to enable
23# autoconf and suggest
24my $dbhost = $ENV{'dbhost'} || ''; # Connect to localhost by default
25my $dbport = $ENV{'dbport'} || '';
26my $dbuser = $ENV{'dbuser'} || 'postgres';
27my $dbpass = $ENV{'dbpass'} || '';
28
29if (exists $ARGV[0]) {
30    if ($ARGV[0] eq 'autoconf') {
31        # Check for DBD::Pg
32        if (! eval "require DBD::Pg;") {
33            print "no (DBD::Pg not found)";
34            exit 1;
35        }
36        # Then we try to detect Postgres presence by connecting to
37        # 'template1'.
38        my $dsn = "dbi:Pg:dbname=template1";
39        $dsn .= ";host=$dbhost" if $dbhost;
40        $dsn .= ";port=$dbport" if $dbport;
41        my $tempdbh = DBI->connect ($dsn, $dbuser, $dbpass);
42        if ($tempdbh) {
43            print "yes\n";
44            exit 0;
45        } else {
46            print "no (Can't connect to given host, please check environment settings)\n";
47            exit 1;
48        }
49    } elsif ($ARGV[0] and $ARGV[0] eq 'debug') {
50        # Set config flag
51        $debug = 1;
52    } elsif ($ARGV[0] and $ARGV[0] eq 'config') {
53        # Set config flag
54        $configure = 1;
55    } elsif ($ARGV[0] eq 'suggest') {
56        # doesn't always work
57        my @datasources = DBI->data_sources ('Pg');
58        foreach my $dsn (grep !/\=template\d$/, @datasources) {
59            (my $db = $dsn) =~ s/^.*=//;
60            print "$db\n";
61        }
62        exit 0;
63    }
64}
65
66# Must do this here, after checking for autoconf/suggest/etc, because the
67# plugin must be able to run before it is linked to the databases.
68my (undef, undef, $dbname) = split (/_/, $0, 3);
69die "No dbname configured (did you make the proper symlink?)" unless $dbname;
70
71my @datasources = DBI->data_sources ('Pg')
72    or die ("Can't read any possible data sources: $?");
73
74my $dsn = "DBI:Pg:dbname=$dbname";
75$dsn .= ";host=$dbhost" if $dbhost;
76$dsn .= ";port=$dbport" if $dbport;
77print "#$dsn\n" if $debug;
78my $dbh = DBI->connect ($dsn, $dbuser, $dbpass, {RaiseError =>1});
79unless($dbh) {
80    die("Database $dbname\@$dbhost (". $DBI::errstr .")\n");
81}
82
83if ($configure) {
84    print <<_EOM;
85graph_title Postgres database $dbname
86graph_args -l 0 --base 1024
87graph_vlabel bytes
88graph_category Postgresql
89graph_info Size
90size.label Database size (bytes)
91size.info Database size
92size.type GAUGE
93size.draw AREA
94indexsize.label Index size (bytes)
95indexsize.info Index size
96indexsize.type GAUGE
97indexsize.draw STACK
98metasize.label Meta database size (bytes)
99metasize.info Meta database size
100metasize.type GAUGE
101metasize.draw STACK
102metaindexsize.label Meta index size (bytes)
103metaindexsize.info Meta index size
104metaindexsize.type GAUGE
105metaindexsize.draw STACK
106_EOM
107} else {
108    my $database_pages = 0;
109    my $database_indexes = 0;
110    my $metadatabase_pages = 0;
111    my $metadatabase_indexes = 0;
112    my @names = $dbh->tables;
113   
114    # Find relfilenode and relpages from the given table
115    my $q_ind = "SELECT relkind, relfilenode, relpages FROM pg_class
116                     WHERE relname = ?
117                     UNION
118                     SELECT relkind, relfilenode, relpages FROM pg_class
119                     WHERE relfilenode IN (SELECT indexrelid FROM pg_index
120                     WHERE indrelid IN (SELECT relfilenode FROM pg_class
121                     WHERE relname = ?))";
122    my $sth = $dbh->prepare ($q_ind) or die $dbh->errstr;
123   
124    # Iterate over the tables in the database
125    foreach my $table (@names) {
126        my $meta = 1;
127        print "#TABLE: $table\n" if $debug;
128        my $table_pages = 0;
129        my $table_indexes = 0;
130        my $metatable_pages = 0;
131        my $metatable_indexes = 0;
132        # "public" tables are the user data
133        $meta = 0 if $table =~ /^public\./;
134        $table =~ s/^.*\.//;
135       
136        # Call the query with $table twice for each side of the UNION
137        $sth->execute ($table, $table) or die $dbh->errstr;
138        while (my ($relkind, $relfilenode, $relpages) = $sth->fetchrow_array) {
139            if ($relkind eq 'r') {
140                $table_pages     += $relpages if $meta == 0;
141                $metatable_pages += $relpages if $meta == 1;
142            } elsif ($relkind eq 'i') {
143                $table_indexes     += $relpages if $meta == 0;
144                $metatable_indexes += $relpages if $meta == 1;
145            }
146            # Define the query
147            my $q2 = "SELECT SUM(relpages)
148                      FROM pg_class
149                      WHERE relname IN (?, ?)";
150            my $sth2 = $dbh->prepare ($q2);
151            $sth2->execute ("pg_toast_${relfilenode}",
152                            "pg_toast_${relfilenode}_index");
153            my $relpages = $sth2->fetchrow_array;
154            if ($relpages) {
155                if ($relkind eq 'r') {
156                    $table_pages     += $relpages if $meta == 0;
157                    $metatable_pages += $relpages if $meta == 1;
158                } elsif ($relkind eq 'i') {
159                    $table_indexes     += $relpages if $meta == 0;
160                    $metatable_indexes += $relpages if $meta == 1;
161                }
162            }
163            print "#\tR:$relfilenode\tP:$table_pages\tI:$table_indexes\n" if $debug;
164        }
165        $database_pages       += $table_pages;
166        $database_indexes     += $table_indexes;
167        $metadatabase_pages   += $metatable_pages;
168        $metadatabase_indexes += $metatable_indexes;
169    }   
170    $sth->finish;
171    $dbh->disconnect;
172    print "size\.value " . $database_pages * 8192 . "\n";
173    print "indexsize\.value " . $database_indexes * 8192 . "\n";
174    print "metasize\.value " . $metadatabase_pages * 8192 . "\n";
175    print "metaindexsize\.value " . $metadatabase_indexes * 8192 . "\n";
176}
Note: See TracBrowser for help on using the repository browser.