~ubuntuone/filesync-server/trunk

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
# Copyright 2008-2015 Canonical
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# For further info, check  http://launchpad.net/filesync-server

"""
Add a function to help on getting formatted stats on running transactions.
"""

SQL = [
    """
    SET search_path = public, pg_catalog;

    CREATE FUNCTION activity() RETURNS SETOF pg_stat_activity
        LANGUAGE sql SECURITY DEFINER
        SET search_path TO public
        AS $$
        SELECT
            datid, datname, procpid, usesysid, usename,
            application_name, client_addr, client_hostname, client_port,
            backend_start, xact_start, query_start, waiting,
            CASE
                WHEN current_query LIKE '<IDLE>%'
                    OR current_query LIKE 'autovacuum:%'
                    THEN current_query
                ELSE
                    '<HIDDEN>'
            END AS current_query
        FROM pg_catalog.pg_stat_activity;
    $$;

    COMMENT ON FUNCTION activity() IS
        'SECURITY DEFINER wrapper around pg_stat_activity allowing \
         unprivileged users to access most of its information.';
    """,
]


def apply(store):
    """Apply the patch."""
    for sql in SQL:
        store.execute(sql)