← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to database/schema/patch-2208-87-1.sql

  • Committer: Stuart Bishop
  • Date: 2011-09-28 12:49:24 UTC
  • mfrom: (9893.10.1 trivial)
  • mto: This revision was merged to the branch mainline in revision 14178.
  • Revision ID: stuart.bishop@canonical.com-20110928124924-m5a22fymqghw6c5i
Merged trivial into distinct-db-users.

Show diffs side-by-side

added added

removed removed

Lines of Context:
database/schema/patch-2208-87-1.sql
 
1
-- Copyright 2011 Canonical Ltd.  This software is licensed under the
 
2
-- GNU Affero General Public License version 3 (see the file LICENSE).
 
3
 
 
4
SET client_min_messages=ERROR;
 
5
 
 
6
CREATE OR REPLACE FUNCTION update_transitively_private(
 
7
    start_branch int,
 
8
    _root_branch int = NULL,
 
9
    _root_transitively_private boolean = NULL) RETURNS VOID
 
10
LANGUAGE plpgsql VOLATILE SECURITY DEFINER SET search_path TO PUBLIC AS
 
11
$$
 
12
DECLARE
 
13
    root_transitively_private boolean := _root_transitively_private;
 
14
    root_branch int := _root_branch;
 
15
BEGIN
 
16
    IF root_transitively_private IS NULL THEN
 
17
        -- We can't just trust the transitively_private flag of the
 
18
        -- branch we are stacked on, as if we are updating multiple
 
19
        -- records they will be updated in an indeterminate order.
 
20
        -- We need a recursive query.
 
21
        UPDATE Branch SET transitively_private = (
 
22
            WITH RECURSIVE stacked_branches AS (
 
23
                SELECT
 
24
                    top_branch.id, top_branch.stacked_on, top_branch.private
 
25
                FROM Branch AS top_branch
 
26
                WHERE top_branch.id = start_branch
 
27
                UNION ALL
 
28
                SELECT
 
29
                    sub_branch.id, sub_branch.stacked_on, sub_branch.private
 
30
                FROM stacked_branches, Branch AS sub_branch
 
31
                WHERE
 
32
                    stacked_branches.stacked_on = sub_branch.id
 
33
                    AND stacked_branches.stacked_on != start_branch
 
34
                    -- Shortcircuit. No need to recurse if already private.
 
35
                    AND stacked_branches.private IS FALSE
 
36
                    )
 
37
            SELECT COUNT(*) > 0
 
38
            FROM stacked_branches
 
39
            WHERE private IS TRUE)
 
40
        WHERE Branch.id = start_branch
 
41
        RETURNING transitively_private INTO root_transitively_private;
 
42
        root_branch := start_branch;
 
43
    ELSE
 
44
        -- Now we have calculated the correct transitively_private flag
 
45
        -- we can trust it.
 
46
        UPDATE Branch SET
 
47
            transitively_private = GREATEST(private, root_transitively_private)
 
48
        WHERE id = root_branch;
 
49
    END IF;
 
50
 
 
51
    -- Recurse to branches stacked on this one.
 
52
    PERFORM update_transitively_private(
 
53
        start_branch, id, GREATEST(private, root_transitively_private))
 
54
    FROM Branch WHERE stacked_on = root_branch AND id != start_branch;
 
55
END;
 
56
$$;
 
57
 
 
58
COMMENT ON FUNCTION update_transitively_private(int, int, boolean) IS
 
59
'A branch is transitively private if it is private or is stacked on any transitively private branches.';
 
60
 
 
61
CREATE OR REPLACE FUNCTION maintain_transitively_private() RETURNS TRIGGER
 
62
LANGUAGE plpgsql VOLATILE AS
 
63
$$
 
64
BEGIN
 
65
    IF TG_OP = 'UPDATE' THEN
 
66
        IF (NEW.stacked_on IS NOT DISTINCT FROM OLD.stacked_on
 
67
            AND NEW.private IS NOT DISTINCT FROM OLD.private) THEN
 
68
            RETURN NULL;
 
69
        END IF;
 
70
    END IF;
 
71
    PERFORM update_transitively_private(NEW.id);
 
72
    RETURN NULL;
 
73
END;
 
74
$$;
 
75
 
 
76
COMMENT ON FUNCTION maintain_transitively_private() IS
 
77
    'Trigger maintaining the Branch transitively_private column';
 
78
 
 
79
CREATE TRIGGER maintain_branch_transitive_privacy_t
 
80
    AFTER INSERT OR UPDATE ON Branch
 
81
    FOR EACH ROW
 
82
    EXECUTE PROCEDURE maintain_transitively_private();
 
83
 
 
84
INSERT INTO LaunchpadDatabaseRevision VALUES (2208, 87, 1);