pg.pm

bugzilla 项目缺陷管理工具源码

# -*- Mode: perl; indent-tabs-mode: nil -*-
#
# The contents of this file are subject to the Mozilla Public
# License Version 1.1 (the "License"); you may not use this file
# except in compliance with the License. You may obtain a copy of
# the License at http://www.mozilla.org/MPL/
#
# Software distributed under the License is distributed on an "AS
# IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
# implied. See the License for the specific language governing
# rights and limitations under the License.
#
# The Original Code is the Bugzilla Bug Tracking System.
#
# The Initial Developer of the Original Code is Netscape Communications
# Corporation. Portions created by Netscape are
# Copyright (C) 1998 Netscape Communications Corporation. All
# Rights Reserved.
#
# Contributor(s): Andrew Dunstan <[e-mail]>,
#                 Edward J. Sabol <[e-mail]>
#                 Max Kanat-Alexander <[e-mail]>

package Bugzilla::DB::Schema::Pg;

###############################################################################
#
# DB::Schema implementation for PostgreSQL
#
###############################################################################

use strict;

use base qw(Bugzilla::DB::Schema);

#------------------------------------------------------------------------------
sub _initialize {

    my $self = shift;

    $self = $self->SUPER::_initialize(@_);

    # Remove FULLTEXT index types from the schemas.
    foreach my $table (keys %{ $self->{schema} }) {
        if ($self->{schema}{$table}{INDEXES}) {
            foreach my $index (@{ $self->{schema}{$table}{INDEXES} }) {
                if (ref($index) eq 'HASH') {
                    delete($index->{TYPE}) if (exists $index->{TYPE} 
                        && $index->{TYPE} eq 'FULLTEXT');
                }
            }
            foreach my $index (@{ $self->{abstract_schema}{$table}{INDEXES} }) {
                if (ref($index) eq 'HASH') {
                    delete($index->{TYPE}) if (exists $index->{TYPE} 
                        && $index->{TYPE} eq 'FULLTEXT');
                }
            }
        }
    }

    $self->{db_specific} = {

        BOOLEAN =>      'smallint',
        FALSE =>        '0', 
        TRUE =>         '1',

        INT1 =>         'integer',
        INT2 =>         'integer',
        INT3 =>         'integer',
        INT4 =>         'integer',

        SMALLSERIAL =>  'serial unique',
        MEDIUMSERIAL => 'serial unique',
        INTSERIAL =>    'serial unique',

        TINYTEXT =>     'varchar(255)',
        MEDIUMTEXT =>   'text',
        TEXT =>         'text',

        LONGBLOB =>     'bytea',

        DATETIME =>     'timestamp(0) without time zone',

    };

    $self->_adjust_schema;

    return $self;

} #eosub--_initialize
#--------------------------------------------------------------------

# Overridden because Pg has such weird ALTER TABLE problems.
sub get_add_column_ddl {
    my ($self, $table, $column, $definition, $init_value) = @_;

    my @statements;
    my $specific = $self->{db_specific};

    my $type = $definition->{TYPE};
    $type = $specific->{$type} if exists $specific->{$type};
    push(@statements, "ALTER TABLE $table ADD COLUMN $column $type");

    my $default = $definition->{DEFAULT};
    if (defined $default) {
        # Replace any abstract default value (such as 'TRUE' or 'FALSE')
        # with its database-specific implementation.
        $default = $specific->{$default} if exists $specific->{$default};
        push(@statements, "ALTER TABLE $table ALTER COLUMN $column "
                         . " SET DEFAULT $default");
    }

    if (defined $init_value) {
        push(@statements, "UPDATE $table SET $column = $init_value");
    }

    if ($definition->{NOTNULL}) {
        # Handle rows that were NULL when we added the column.
        # We *must* have a DEFAULT. This check is usually handled
        # at a higher level than this code, but I figure it can't
        # hurt to have it here.
        die "NOT NULL columns must have a DEFAULT or an init_value."
            unless (exists $definition->{DEFAULT} || defined $init_value);
        push(@statements, "UPDATE $table SET $column = $default");
        push(@statements, "ALTER TABLE $table ALTER COLUMN $column "
                         . " SET NOT NULL");
    }

    if ($definition->{PRIMARYKEY}) {
         push(@statements, "ALTER TABLE $table ALTER COLUMN "
                         . " ADD PRIMARY KEY ($column)");
    }

    return @statements;
}

sub get_rename_column_ddl {
    my ($self, $table, $old_name, $new_name) = @_;

    return ("ALTER TABLE $table RENAME COLUMN $old_name TO $new_name");
}

1;