← Back to branch summary

~ubuntu-branches/ubuntu/karmic/apgdiff/karmic

« back to all changes in this revision

Viewing changes to src/main/java/cz/startnet/utils/pgdiff/PgDiffTriggers.java

  • Committer: Bazaar Package Importer
  • Author(s): Christoph Berg
  • Date: 2008-09-09 15:42:54 UTC
  • Revision ID: james.westby@ubuntu.com-20080909154254-458sv7ew1rczdal1
Tags: upstream-1.2
Import upstream version 1.2

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/main/java/cz/startnet/utils/pgdiff/PgDiffTriggers.java
 
1
/*
 
2
 * $Id: PgDiffTriggers.java 80 2007-09-01 20:25:45Z fordfrog $
 
3
 */
 
4
package cz.startnet.utils.pgdiff;
 
5
 
 
6
import cz.startnet.utils.pgdiff.schema.PgSchema;
 
7
import cz.startnet.utils.pgdiff.schema.PgTable;
 
8
import cz.startnet.utils.pgdiff.schema.PgTrigger;
 
9
 
 
10
import java.io.PrintWriter;
 
11
 
 
12
import java.util.ArrayList;
 
13
import java.util.List;
 
14
 
 
15
 
 
16
/**
 
17
 * Diffs triggers.
 
18
 *
 
19
 * @author fordfrog
 
20
 * @version $Id: PgDiffTriggers.java 80 2007-09-01 20:25:45Z fordfrog $
 
21
 */
 
22
public class PgDiffTriggers {
 
23
    /**
 
24
     * Creates a new instance of PgDiffTriggers.
 
25
     */
 
26
    private PgDiffTriggers() {
 
27
        super();
 
28
    }
 
29
 
 
30
    /**
 
31
     * Outputs commands for differences in triggers.
 
32
     *
 
33
     * @param writer writer the output should be written to
 
34
     * @param arguments object containing arguments settings
 
35
     * @param oldSchema original schema
 
36
     * @param newSchema new schema
 
37
     */
 
38
    public static void diffTriggers(
 
39
        final PrintWriter writer,
 
40
        final PgDiffArguments arguments,
 
41
        final PgSchema oldSchema,
 
42
        final PgSchema newSchema) {
 
43
        for (PgTable newTable : newSchema.getTables()) {
 
44
            final PgTable oldTable;
 
45
 
 
46
            if (oldSchema == null) {
 
47
                oldTable = null;
 
48
            } else {
 
49
                oldTable = oldSchema.getTable(newTable.getName());
 
50
            }
 
51
 
 
52
            // Drop triggers that no more exist or are modified
 
53
            for (PgTrigger trigger : getDropTriggers(oldTable, newTable)) {
 
54
                writer.println();
 
55
                writer.println(trigger.getDropSQL(arguments.isQuoteNames()));
 
56
            }
 
57
 
 
58
            // Add new triggers
 
59
            for (PgTrigger trigger : getNewTriggers(oldTable, newTable)) {
 
60
                writer.println();
 
61
                writer.println(
 
62
                        trigger.getCreationSQL(arguments.isQuoteNames()));
 
63
            }
 
64
        }
 
65
    }
 
66
 
 
67
    /**
 
68
     * Returns list of triggers that should be dropped.
 
69
     *
 
70
     * @param oldTable original table
 
71
     * @param newTable new table
 
72
     *
 
73
     * @return list of triggers that should be dropped
 
74
     */
 
75
    private static List<PgTrigger> getDropTriggers(
 
76
        final PgTable oldTable,
 
77
        final PgTable newTable) {
 
78
        final List<PgTrigger> list = new ArrayList<PgTrigger>();
 
79
 
 
80
        if ((newTable != null) && (oldTable != null)) {
 
81
            final List<PgTrigger> newTriggers = newTable.getTriggers();
 
82
 
 
83
            for (final PgTrigger oldTrigger : oldTable.getTriggers()) {
 
84
                if (!newTriggers.contains(oldTrigger)) {
 
85
                    list.add(oldTrigger);
 
86
                }
 
87
            }
 
88
        }
 
89
 
 
90
        return list;
 
91
    }
 
92
 
 
93
    /**
 
94
     * Returns list of triggers that should be added.
 
95
     *
 
96
     * @param oldTable original table
 
97
     * @param newTable new table
 
98
     *
 
99
     * @return list of triggers that should be added
 
100
     */
 
101
    private static List<PgTrigger> getNewTriggers(
 
102
        final PgTable oldTable,
 
103
        final PgTable newTable) {
 
104
        final List<PgTrigger> list = new ArrayList<PgTrigger>();
 
105
 
 
106
        if (newTable != null) {
 
107
            if (oldTable == null) {
 
108
                list.addAll(newTable.getTriggers());
 
109
            } else {
 
110
                for (PgTrigger newTrigger : newTable.getTriggers()) {
 
111
                    if (!oldTable.getTriggers().contains(newTrigger)) {
 
112
                        list.add(newTrigger);
 
113
                    }
 
114
                }
 
115
            }
 
116
        }
 
117
 
 
118
        return list;
 
119
    }
 
120
}