49
+ − 1 package IMPL::SQL::Schema::Traits;
+ − 2 use strict;
+ − 3 use base qw(IMPL::Object IMPL::Object::Autofill);
+ − 4 use IMPL::Class::Property;
+ − 5 use IMPL::Class::Property::Direct;
+ − 6
+ − 7 use constant {
+ − 8 STATE_NORMAL => 0,
+ − 9 STATE_UPDATED => 1,
+ − 10 STATE_CREATED => 2,
+ − 11 STATE_REMOVED => 3,
+ − 12 STATE_PENDING => 4
+ − 13 } ;
+ − 14
+ − 15 BEGIN {
+ − 16 public _direct property SrcSchema => prop_all;
+ − 17 public _direct property DstSchema => prop_all;
+ − 18 public _direct property PendingActions => prop_get;
+ − 19 public _direct property TableInfo => prop_get;
+ − 20 public _direct property Handler => prop_get;
+ − 21 public _direct property TableMap => prop_none;
+ − 22 public _direct property KeepTables => prop_all;
+ − 23 }
+ − 24
+ − 25 __PACKAGE__->PassThroughArgs;
+ − 26
+ − 27 sub CTOR {
+ − 28 my $this = shift;
+ − 29
+ − 30 $this->{$SrcSchema} or die new IMPL::InvalidArgumentException('A source schema is required');
+ − 31 $this->{$DstSchema} or die new IMPL::InvalidArgumentException('A destination schema is required');
+ − 32 $this->{$Handler} or die new IMPL::InvalidArgumentException('A handler is required to produce the update batch');
+ − 33
+ − 34 $this->{$TableInfo} = {};
+ − 35 $this->{$PendingActions} = [];
+ − 36
+ − 37 }
+ − 38
+ − 39 sub UpdateTable {
+ − 40 my ($this,$srcTable) = @_;
+ − 41
+ − 42 return 1 if $this->{$TableInfo}->{$srcTable->Name}->{'processed'};
+ − 43
+ − 44 my $dstTableName = $this->{$TableMap}->{$srcTable->Name} ? $this->{$TableMap}->{$srcTable->Name} : $srcTable->Name;
+ − 45 my $dstTable = $this->{$DstSchema}->Tables->{$dstTableName};
+ − 46
+ − 47 $this->{$TableInfo}->{$srcTable->Name}->{'processed'} = 1;
+ − 48
+ − 49 if (not $dstTable) {
+ − 50 $this->DropTable($srcTable) if not $this->{$KeepTables};
+ − 51 return 1;
+ − 52 }
+ − 53
+ − 54 if ( not grep {$srcTable->Column($_->Name)} @{$dstTable->Columns} ) {
+ − 55
+ − 56 $this->{$TableInfo}->{$srcTable->Name}->{'NewName'} = $dstTable->Name if $srcTable->Name ne $dstTable->Name;
+ − 57
+ − 58 $this->DropTable($srcTable);
+ − 59 $this->CreateTable($dstTable);
+ − 60
+ − 61 return 1;
+ − 62 }
+ − 63
+ − 64 if ($srcTable->Name ne $dstTableName) {
+ − 65 $this->RenameTable($srcTable,$dstTableName);
+ − 66 }
+ − 67
+ − 68 my %dstConstraints = %{$dstTable->Constraints};
+ − 69
+ − 70 foreach my $srcConstraint (values %{$srcTable->Constraints}) {
+ − 71 if (my $dstConstraint = delete $dstConstraints{$srcConstraint->Name}) {
+ − 72 $this->UpdateConstraint($srcConstraint,$dstConstraint);
+ − 73 } else {
+ − 74 $this->DropConstraint($srcConstraint);
+ − 75 }
+ − 76 }
+ − 77
+ − 78 my $i = 0;
+ − 79 my %dstColumns = map { $_->Name, $i++} @{$dstTable->Columns} ;
+ − 80
+ − 81 # ������� ������� �������
+ − 82 # ����� ��������� ����������� � �������� ������� � ������ �������
+ − 83
+ − 84 my @columnsToUpdate;
+ − 85
+ − 86 foreach my $srcColumn (@{$srcTable->Columns}) {
+ − 87 if (defined (my $dstColumnIndex = delete $dstColumns{$srcColumn->Name})) {
+ − 88 push @columnsToUpdate, { Action => 'update', ColumnSrc => $srcColumn, ColumnDst => $dstTable->ColumnAt($dstColumnIndex), NewPosition => $dstColumnIndex};
+ − 89 } else {
+ − 90 $this->DropColumn($srcTable,$srcColumn);
+ − 91 }
+ − 92 }
+ − 93 push @columnsToUpdate, map { {Action => 'add', ColumnDst => $dstTable->ColumnAt($_), NewPosition => $_} } values %dstColumns;
+ − 94
+ − 95 foreach my $action (sort {$a->{'NewPosition'} <=> $b->{'NewPosition'}} @columnsToUpdate ) {
+ − 96 if ($action->{'Action'} eq 'update') {
+ − 97 $this->UpdateColumn($srcTable,@$action{'ColumnSrc','ColumnDst'},$dstTable,$action->{'NewPosition'}); # change type and position
+ − 98 }elsif ($action->{'Action'} eq 'add') {
+ − 99 $this->AddColumn($srcTable,$action->{'ColumnDst'},$dstTable,$action->{'NewPosition'}); # add at specified position
+ − 100 }
+ − 101 }
+ − 102
+ − 103 foreach my $dstConstraint (values %dstConstraints) {
+ − 104 $this->AddConstraint($dstConstraint);
+ − 105 }
+ − 106
+ − 107 $this->{$TableInfo}{$srcTable->Name}{'State'} = STATE_UPDATED;
+ − 108 }
+ − 109
+ − 110 sub UpdateConstraint {
+ − 111 my ($this,$src,$dst) = @_;
+ − 112
+ − 113 if (not ConstraintEquals($src,$dst)) {
+ − 114 if (UNIVERSAL::isa($src,'IMPL::SQL::Schema::Constraint::PrimaryKey')) {
+ − 115 $this->UpdateTable($_->Table) foreach values %{$src->ConnectedFK};
+ − 116 }
+ − 117 $this->DropConstraint($src);
+ − 118 $this->AddConstraint($dst);
+ − 119 } else {
+ − 120 $this->{$TableInfo}->{$this->MapTableName($src->Table->Name)}->{'Constraints'}->{$src->Name} = STATE_UPDATED;
+ − 121 }
+ − 122 }
+ − 123
+ − 124 sub ConstraintEquals {
+ − 125 my ($src,$dst) = @_;
+ − 126
+ − 127 ref $src eq ref $dst or return 0;
+ − 128
+ − 129 my @dstColumns = @{$dst->Columns};
+ − 130 scalar(@{$src->Columns}) == scalar(@{$dst->Columns}) and not grep { my $column = shift @dstColumns; not $column->isSame($_) } @{$src->Columns} or return 0;
+ − 131
+ − 132 not UNIVERSAL::isa($src,'IMPL::SQL::Schema::Constraint::ForeignKey') or ConstraintEquals($src->ReferencedPrimaryKey,$dst->ReferencedPrimaryKey) or return 0;
+ − 133
+ − 134 1;
+ − 135 }
+ − 136
+ − 137 sub UpdateSchema {
+ − 138 my ($this) = @_;
+ − 139
+ − 140 my %Updated = map { $this->UpdateTable($_); $this->MapTableName($_->Name) , 1; } values %{$this->{$SrcSchema}->Tables ? $this->{$SrcSchema}->Tables : {} };
+ − 141
+ − 142 $this->CreateTable($_) foreach grep {not $Updated{$_->Name}} values %{$this->{$DstSchema}->Tables};
+ − 143
+ − 144 $this->ProcessPendingActions();
+ − 145 }
+ − 146
+ − 147 sub RenameTable {
+ − 148 my ($this,$tblSrc,$tblDstName) = @_;
+ − 149
+ − 150 $this->{$Handler}->AlterTableRename($tblSrc->Name,$tblDstName);
+ − 151 $this->{$TableInfo}->{$tblSrc->Name}->{'NewName'} = $tblDstName;
+ − 152 }
+ − 153
+ − 154 sub MapTableName {
+ − 155 my ($this,$srcName) = @_;
+ − 156
+ − 157 $this->{$TableInfo}->{$srcName}->{'NewName'} ? $this->{$TableInfo}->{$srcName}->{'NewName'} : $srcName;
+ − 158 }
+ − 159
+ − 160 sub DropTable {
+ − 161 my ($this,$tbl) = @_;
+ − 162
+ − 163 if ($tbl->PrimaryKey) {
+ − 164 $this->UpdateTable($_->Table) foreach values %{$tbl->PrimaryKey->ConnectedFK};
+ − 165 }
+ − 166
+ − 167 $this->{$Handler}->DropTable($this->MapTableName($tbl->Name));
+ − 168 $this->{$TableInfo}{$this->MapTableName($tbl->Name)}{'State'} = STATE_REMOVED;
+ − 169 $this->{$TableInfo}{$this->MapTableName($tbl->Name)}{'Constraints'} = {map {$_,STATE_REMOVED} keys %{$tbl->Constraints}};
+ − 170 $this->{$TableInfo}{$this->MapTableName($tbl->Name)}{'Columns'} = {map { $_->Name, STATE_REMOVED} @{$tbl->Columns}};
+ − 171
+ − 172 return 1;
+ − 173 }
+ − 174
+ − 175 sub CreateTable {
+ − 176 my ($this,$tbl) = @_;
+ − 177
+ − 178 # ������� �������, ����� ������� ������
+ − 179 $this->{$Handler}->CreateTable($tbl,skip_foreign_keys => 1);
+ − 180
+ − 181 $this->{$TableInfo}->{$tbl->Name}->{'State'} = STATE_CREATED;
+ − 182
+ − 183 $this->{$TableInfo}->{$tbl->Name}->{'Columns'} = {map { $_->Name, STATE_CREATED } @{$tbl->Columns}};
+ − 184 $this->{$TableInfo}->{$tbl->Name}->{'Constraints'} = {map {$_->Name, STATE_CREATED} grep { not UNIVERSAL::isa($_,'IMPL::SQL::Schema::Constraint::ForeignKey') } values %{$tbl->Constraints}};
+ − 185
+ − 186 $this->AddConstraint($_) foreach grep { UNIVERSAL::isa($_,'IMPL::SQL::Schema::Constraint::ForeignKey') } values %{$tbl->Constraints};
+ − 187
+ − 188 return 1;
+ − 189 }
+ − 190
+ − 191 sub AddColumn {
+ − 192 my ($this,$tblSrc,$column,$tblDst,$pos) = @_;
+ − 193
+ − 194 $this->{$Handler}->AlterTableAddColumn($this->MapTableName($tblSrc->Name),$column,$tblDst,$pos);
+ − 195 $this->{$TableInfo}->{$this->MapTableName($tblSrc->Name)}->{'Columns'}->{$column->Name} = STATE_CREATED;
+ − 196
+ − 197 return 1;
+ − 198 }
+ − 199
+ − 200 sub DropColumn {
+ − 201 my ($this,$tblSrc,$column) = @_;
+ − 202 $this->{$Handler}->AlterTableDropColumn($this->MapTableName($tblSrc->Name),$column->Name);
+ − 203 $this->{$TableInfo}->{$this->MapTableName($tblSrc->Name)}->{'Columns'}->{$column->Name} = STATE_REMOVED;
+ − 204
+ − 205 return 1;
+ − 206 }
+ − 207
+ − 208 sub UpdateColumn {
+ − 209 my ($this,$tblSrc,$srcColumn,$dstColumn,$tblDst,$pos) = @_;
+ − 210
+ − 211 if ($srcColumn->isSame($dstColumn) and $pos < @{$tblSrc->Columns} and $tblSrc->ColumnAt($pos) == $srcColumn) {
+ − 212 $this->{$TableInfo}->{$this->MapTableName($tblSrc->Name)}->{'Columns'}->{$dstColumn->Name} = STATE_UPDATED;
+ − 213 return 1;
+ − 214 }
+ − 215
+ − 216 $this->{$Handler}->AlterTableChangeColumn($this->MapTableName($tblSrc->Name),$dstColumn,$tblDst,$pos);
+ − 217 $this->{$TableInfo}->{$this->MapTableName($tblSrc->Name)}->{'Columns'}->{$dstColumn->Name} = STATE_UPDATED;
+ − 218
+ − 219 return 1;
+ − 220 }
+ − 221
+ − 222 sub DropConstraint {
+ − 223 my ($this,$constraint) = @_;
+ − 224
+ − 225 $this->{$Handler}->AlterTableDropConstraint($this->MapTableName($constraint->Table->Name),$constraint);
+ − 226 $this->{$TableInfo}->{$constraint->Table->Name}->{'Constraints'}->{$constraint->Name} = STATE_REMOVED;
+ − 227
+ − 228 return 1;
+ − 229 }
+ − 230
+ − 231 sub IfUndef {
+ − 232 my ($value,$default) = @_;
+ − 233
+ − 234 return defined $value ? $value : $default;
+ − 235 }
+ − 236
+ − 237 sub AddConstraint {
+ − 238 my ($this,$constraint) = @_;
+ − 239
+ − 240 # ����� ����������� ����������� ����� ��������� � ���, ��� ������� ��� ����������� ������� � �������������
+ − 241 # ����������� (�������� ��������� �����)
+ − 242
+ − 243 my $pending;
+ − 244
+ − 245 $pending = grep {
+ − 246 my $column = $_;
+ − 247 not grep {
+ − 248 IfUndef($this->{$TableInfo}{$constraint->Table->Name}{'Columns'}{$column->Name}, STATE_NORMAL) == $_
+ − 249 } (STATE_UPDATED, STATE_CREATED)
+ − 250 } @{$constraint->Columns};
+ − 251
+ − 252 if ($pending) {
+ − 253 push @{$this->{$PendingActions}},{Action => \&AddConstraint, Args => [$constraint]};
+ − 254 return 2;
+ − 255 } else {
+ − 256 if (UNIVERSAL::isa($constraint,'IMPL::SQL::Schema::Constraint::ForeignKey')) {
+ − 257 if (not grep { IfUndef($this->{$TableInfo}{$constraint->ReferencedPrimaryKey->Table->Name}{'Constraints'}{$constraint->ReferencedPrimaryKey->Name},STATE_NORMAL) == $_} (STATE_UPDATED, STATE_CREATED)) {
+ − 258 push @{$this->{$PendingActions}},{Action => \&AddConstraint, Args => [$constraint]};
+ − 259 return 2;
+ − 260 }
+ − 261 }
+ − 262 $this->{$Handler}->AlterTableAddConstraint($constraint->Table->Name,$constraint);
+ − 263 $this->{$TableInfo}->{$constraint->Table->Name}->{'Constraints'}->{$constraint->Name} = STATE_CREATED;
+ − 264 }
+ − 265 }
+ − 266
+ − 267 sub ProcessPendingActions {
+ − 268 my ($this) = @_;
+ − 269
+ − 270 while (my $action = shift @{$this->{$PendingActions}}) {
+ − 271 $action->{'Action'}->($this,@{$action->{'Args'}});
+ − 272 }
+ − 273 }
+ − 274
+ − 275 1;