Commit | Line | Data |
45d0d9d5 |
1 | package DBIx::Class::DeploymentHandler::DeployMethod::SQL::Translator; |
334bced5 |
2 | use Moose; |
9af9d0b2 |
3 | |
9a3a62f1 |
4 | # ABSTRACT: Manage your SQL and Perl migrations in nicely laid out directories |
5 | |
9af9d0b2 |
6 | use autodie; |
7 | use Carp qw( carp croak ); |
8465e767 |
8 | use DBIx::Class::DeploymentHandler::Logger; |
c4f51462 |
9 | use Log::Contextual qw(:log :dlog), -package_logger => |
8465e767 |
10 | DBIx::Class::DeploymentHandler::Logger->new({ |
11 | env_prefix => 'DBICDH' |
12 | }); |
9af9d0b2 |
13 | |
2e68a8e1 |
14 | use Method::Signatures::Simple; |
7f50d101 |
15 | use Try::Tiny; |
9af9d0b2 |
16 | |
d23c7c77 |
17 | use SQL::Translator; |
18 | require SQL::Translator::Diff; |
9af9d0b2 |
19 | |
d23c7c77 |
20 | require DBIx::Class::Storage; # loaded for type constraint |
41863428 |
21 | use DBIx::Class::DeploymentHandler::Types; |
22 | |
9af9d0b2 |
23 | use File::Path 'mkpath'; |
24 | use File::Spec::Functions; |
2e68a8e1 |
25 | |
7521a845 |
26 | with 'DBIx::Class::DeploymentHandler::HandlesDeploy'; |
3c1b5ee8 |
27 | |
93460690 |
28 | has ignore_ddl => ( |
29 | isa => 'Bool', |
30 | is => 'ro', |
31 | default => undef, |
32 | ); |
33 | |
92624ee5 |
34 | has force_overwrite => ( |
35 | isa => 'Bool', |
36 | is => 'ro', |
37 | default => undef, |
38 | ); |
39 | |
d54b8d69 |
40 | has schema => ( |
41 | isa => 'DBIx::Class::Schema', |
42 | is => 'ro', |
43 | required => 1, |
d54b8d69 |
44 | ); |
45 | |
334bced5 |
46 | has storage => ( |
47 | isa => 'DBIx::Class::Storage', |
48 | is => 'ro', |
49 | lazy_build => 1, |
50 | ); |
51 | |
2eaf903b |
52 | method _build_storage { |
53 | my $s = $self->schema->storage; |
54 | $s->_determine_driver; |
55 | $s |
56 | } |
57 | |
02a7b8ac |
58 | has sql_translator_args => ( |
334bced5 |
59 | isa => 'HashRef', |
60 | is => 'ro', |
61 | default => sub { {} }, |
62 | ); |
91adde75 |
63 | has script_directory => ( |
334bced5 |
64 | isa => 'Str', |
65 | is => 'ro', |
66 | required => 1, |
67 | default => 'sql', |
68 | ); |
69 | |
334bced5 |
70 | has databases => ( |
71 | coerce => 1, |
72 | isa => 'DBIx::Class::DeploymentHandler::Databases', |
73 | is => 'ro', |
74 | default => sub { [qw( MySQL SQLite PostgreSQL )] }, |
75 | ); |
76 | |
a7d53deb |
77 | has txn_wrap => ( |
78 | is => 'ro', |
79 | isa => 'Bool', |
80 | default => 1, |
81 | ); |
82 | |
73caa630 |
83 | has schema_version => ( |
84 | is => 'ro', |
e86c0c07 |
85 | isa => 'Str', |
73caa630 |
86 | lazy_build => 1, |
87 | ); |
88 | |
6df6dcb9 |
89 | # this will probably never get called as the DBICDH |
90 | # will be passing down a schema_version normally, which |
91 | # is built the same way, but we leave this in place |
73caa630 |
92 | method _build_schema_version { $self->schema->schema_version } |
93 | |
76d311e7 |
94 | method __ddl_consume_with_prefix($type, $versions, $prefix) { |
91adde75 |
95 | my $base_dir = $self->script_directory; |
262166c1 |
96 | |
76d08d08 |
97 | my $main = catfile( $base_dir, $type ); |
76d08d08 |
98 | my $common = |
99 | catfile( $base_dir, '_common', $prefix, join q(-), @{$versions} ); |
262166c1 |
100 | |
101 | my $dir; |
102 | if (-d $main) { |
76d08d08 |
103 | $dir = catfile($main, $prefix, join q(-), @{$versions}) |
262166c1 |
104 | } else { |
38857f30 |
105 | if ($self->ignore_ddl) { |
106 | return [] |
107 | } else { |
108 | croak "$main does not exist; please write/generate some SQL" |
109 | } |
262166c1 |
110 | } |
111 | |
ef44838b |
112 | my %files; |
113 | try { |
114 | opendir my($dh), $dir; |
115 | %files = |
116 | map { $_ => "$dir/$_" } |
117 | grep { /\.(?:sql|pl|sql-\w+)$/ && -f "$dir/$_" } |
118 | readdir $dh; |
119 | closedir $dh; |
120 | } catch { |
121 | die $_ unless $self->ignore_ddl; |
122 | }; |
262166c1 |
123 | if (-d $common) { |
124 | opendir my($dh), $common; |
41219a5d |
125 | for my $filename (grep { /\.(?:sql|pl)$/ && -f catfile($common,$_) } readdir $dh) { |
262166c1 |
126 | unless ($files{$filename}) { |
9af9d0b2 |
127 | $files{$filename} = catfile($common,$filename); |
262166c1 |
128 | } |
129 | } |
130 | closedir $dh; |
131 | } |
132 | |
133 | return [@files{sort keys %files}] |
134 | } |
3c1b5ee8 |
135 | |
ff40cb1f |
136 | method _ddl_initialize_consume_filenames($type, $version) { |
137 | $self->__ddl_consume_with_prefix($type, [ $version ], 'initialize') |
fc4b7602 |
138 | } |
139 | |
76d311e7 |
140 | method _ddl_schema_consume_filenames($type, $version) { |
58eb99c3 |
141 | $self->__ddl_consume_with_prefix($type, [ $version ], 'deploy') |
3c1b5ee8 |
142 | } |
143 | |
c72832d7 |
144 | method _ddl_protoschema_deploy_consume_filenames($version) { |
145 | my $base_dir = $self->script_directory; |
146 | |
147 | my $dir = catfile( $base_dir, '_source', 'deploy', $version); |
148 | return [] unless -d $dir; |
149 | |
150 | opendir my($dh), $dir; |
151 | my %files = map { $_ => "$dir/$_" } grep { /\.yml$/ && -f "$dir/$_" } readdir $dh; |
152 | closedir $dh; |
153 | |
154 | return [@files{sort keys %files}] |
155 | } |
156 | |
58eb99c3 |
157 | method _ddl_protoschema_upgrade_consume_filenames($versions) { |
f9c6ab50 |
158 | my $base_dir = $self->script_directory; |
159 | |
58eb99c3 |
160 | my $dir = catfile( $base_dir, '_preprocess_schema', 'upgrade', join q(-), @{$versions}); |
f9c6ab50 |
161 | |
162 | return [] unless -d $dir; |
163 | |
164 | opendir my($dh), $dir; |
165 | my %files = map { $_ => "$dir/$_" } grep { /\.pl$/ && -f "$dir/$_" } readdir $dh; |
166 | closedir $dh; |
167 | |
168 | return [@files{sort keys %files}] |
169 | } |
170 | |
58eb99c3 |
171 | method _ddl_protoschema_downgrade_consume_filenames($versions) { |
f9c6ab50 |
172 | my $base_dir = $self->script_directory; |
173 | |
58eb99c3 |
174 | my $dir = catfile( $base_dir, '_preprocess_schema', 'downgrade', join q(-), @{$versions}); |
f9c6ab50 |
175 | |
176 | return [] unless -d $dir; |
177 | |
178 | opendir my($dh), $dir; |
179 | my %files = map { $_ => "$dir/$_" } grep { /\.pl$/ && -f "$dir/$_" } readdir $dh; |
180 | closedir $dh; |
181 | |
182 | return [@files{sort keys %files}] |
183 | } |
184 | |
7e08eddd |
185 | method _ddl_protoschema_produce_filename($version) { |
58eb99c3 |
186 | my $dirname = catfile( $self->script_directory, '_source', 'deploy', $version ); |
7e08eddd |
187 | mkpath($dirname) unless -d $dirname; |
188 | |
189 | return catfile( $dirname, '001-auto.yml' ); |
190 | } |
191 | |
76d311e7 |
192 | method _ddl_schema_produce_filename($type, $version) { |
58eb99c3 |
193 | my $dirname = catfile( $self->script_directory, $type, 'deploy', $version ); |
76d08d08 |
194 | mkpath($dirname) unless -d $dirname; |
d54b8d69 |
195 | |
09bc35e3 |
196 | return catfile( $dirname, '001-auto.sql' ); |
d54b8d69 |
197 | } |
198 | |
58eb99c3 |
199 | method _ddl_schema_upgrade_consume_filenames($type, $versions) { |
200 | $self->__ddl_consume_with_prefix($type, $versions, 'upgrade') |
3c1b5ee8 |
201 | } |
202 | |
58eb99c3 |
203 | method _ddl_schema_downgrade_consume_filenames($type, $versions) { |
204 | $self->__ddl_consume_with_prefix($type, $versions, 'downgrade') |
a41a04e5 |
205 | } |
206 | |
58eb99c3 |
207 | method _ddl_schema_upgrade_produce_filename($type, $versions) { |
91adde75 |
208 | my $dir = $self->script_directory; |
76d311e7 |
209 | |
58eb99c3 |
210 | my $dirname = catfile( $dir, $type, 'upgrade', join q(-), @{$versions}); |
76d08d08 |
211 | mkpath($dirname) unless -d $dirname; |
a41a04e5 |
212 | |
e62add58 |
213 | return catfile( $dirname, '001-auto.sql' ); |
a41a04e5 |
214 | } |
215 | |
58eb99c3 |
216 | method _ddl_schema_downgrade_produce_filename($type, $versions, $dir) { |
217 | my $dirname = catfile( $dir, $type, 'downgrade', join q(-), @{$versions} ); |
76d08d08 |
218 | mkpath($dirname) unless -d $dirname; |
24f4524b |
219 | |
09bc35e3 |
220 | return catfile( $dirname, '001-auto.sql'); |
24f4524b |
221 | } |
222 | |
f36afe83 |
223 | method _run_sql_array($sql) { |
41219a5d |
224 | my $storage = $self->storage; |
5d7b27cf |
225 | |
1f0d0633 |
226 | $sql = [grep { |
227 | $_ && # remove blank lines |
228 | !/^(BEGIN|BEGIN TRANSACTION|COMMIT)/ # strip txn's |
229 | } map { |
230 | s/^\s+//; s/\s+$//; # trim whitespace |
231 | join '', grep { !/^--/ } split /\n/ # remove comments |
232 | } @$sql]; |
233 | |
f4075791 |
234 | Dlog_trace { "Running SQL $_" } $sql; |
f36afe83 |
235 | foreach my $line (@{$sql}) { |
5d7b27cf |
236 | $storage->_query_start($line); |
10a62c3d |
237 | # the whole reason we do this is so that we can see the line that was run |
5d7b27cf |
238 | try { |
5d7b27cf |
239 | $storage->dbh_do (sub { $_[1]->do($line) }); |
240 | } |
241 | catch { |
10a62c3d |
242 | die "$_ (running line '$line')" |
60e09fce |
243 | }; |
5d7b27cf |
244 | $storage->_query_end($line); |
245 | } |
4d09f712 |
246 | return join "\n", @$sql |
f36afe83 |
247 | } |
248 | |
249 | method _run_sql($filename) { |
f4075791 |
250 | log_debug { "Running SQL from $filename" }; |
f36afe83 |
251 | return $self->_run_sql_array($self->_read_sql_file($filename)); |
5d7b27cf |
252 | } |
2e68a8e1 |
253 | |
5d7b27cf |
254 | method _run_perl($filename) { |
f4075791 |
255 | log_debug { "Running Perl from $filename" }; |
5d7b27cf |
256 | my $filedata = do { local( @ARGV, $/ ) = $filename; <> }; |
c8a2f7bd |
257 | |
5d7b27cf |
258 | no warnings 'redefine'; |
259 | my $fn = eval "$filedata"; |
260 | use warnings; |
f4075791 |
261 | Dlog_trace { "Running Perl $_" } $fn; |
5d7b27cf |
262 | |
263 | if ($@) { |
264 | carp "$filename failed to compile: $@"; |
265 | } elsif (ref $fn eq 'CODE') { |
266 | $fn->($self->schema) |
267 | } else { |
268 | carp "$filename should define an anonymouse sub that takes a schema but it didn't!"; |
269 | } |
270 | } |
5d7b27cf |
271 | |
ef44838b |
272 | method _run_sql_and_perl($filenames, $sql_to_run) { |
5d7b27cf |
273 | my @files = @{$filenames}; |
274 | my $guard = $self->schema->txn_scope_guard if $self->txn_wrap; |
275 | |
ef44838b |
276 | $self->_run_sql_array($sql_to_run) if $self->ignore_ddl; |
277 | |
278 | my $sql = ($sql_to_run)?join ";\n", @$sql_to_run:''; |
279 | FILENAME: |
41219a5d |
280 | for my $filename (@files) { |
ef44838b |
281 | if ($self->ignore_ddl && $filename =~ /^[^_]*-auto.*\.sql$/) { |
282 | next FILENAME |
283 | } elsif ($filename =~ /\.sql$/) { |
5d7b27cf |
284 | $sql .= $self->_run_sql($filename) |
398b1385 |
285 | } elsif ( $filename =~ /\.pl$/ ) { |
5d7b27cf |
286 | $self->_run_perl($filename) |
41219a5d |
287 | } else { |
fc4b7602 |
288 | croak "A file ($filename) got to deploy that wasn't sql or perl!"; |
2e68a8e1 |
289 | } |
2e68a8e1 |
290 | } |
a7d53deb |
291 | |
292 | $guard->commit if $self->txn_wrap; |
41219a5d |
293 | |
294 | return $sql; |
295 | } |
296 | |
297 | sub deploy { |
298 | my $self = shift; |
be140a5f |
299 | my $version = (shift @_ || {})->{version} || $self->schema_version; |
f4075791 |
300 | log_info { "deploying version $version" }; |
ef44838b |
301 | my $sqlt_type = $self->storage->sqlt_type; |
302 | my $sql; |
303 | if ($self->ignore_ddl) { |
304 | $sql = $self->_sql_from_yaml({}, |
4ea5caf5 |
305 | '_ddl_protoschema_deploy_consume_filenames', $sqlt_type |
ef44838b |
306 | ); |
307 | } |
308 | return $self->_run_sql_and_perl($self->_ddl_schema_consume_filenames( |
309 | $sqlt_type, |
310 | $version, |
311 | ), $sql); |
2e68a8e1 |
312 | } |
313 | |
ff40cb1f |
314 | sub initialize { |
9faec51a |
315 | my $self = shift; |
316 | my $args = shift; |
317 | my $version = $args->{version} || $self->schema_version; |
ff40cb1f |
318 | log_info { "initializing version $version" }; |
9faec51a |
319 | my $storage_type = $args->{storage_type} || $self->storage->sqlt_type; |
fc4b7602 |
320 | |
ff40cb1f |
321 | my @files = @{$self->_ddl_initialize_consume_filenames( |
9faec51a |
322 | $storage_type, |
fc4b7602 |
323 | $version, |
324 | )}; |
325 | |
326 | for my $filename (@files) { |
327 | # We ignore sql for now (till I figure out what to do with it) |
328 | if ( $filename =~ /^(.+)\.pl$/ ) { |
fc4b7602 |
329 | my $filedata = do { local( @ARGV, $/ ) = $filename; <> }; |
fc4b7602 |
330 | |
9faec51a |
331 | no warnings 'redefine'; |
5b5defbc |
332 | my $fn = eval "$filedata"; |
fc4b7602 |
333 | use warnings; |
5b5defbc |
334 | |
9faec51a |
335 | if ($@) { |
3fa64c79 |
336 | carp "$filename failed to compile: $@"; |
9faec51a |
337 | } elsif (ref $fn eq 'CODE') { |
fc4b7602 |
338 | $fn->() |
339 | } else { |
5b5defbc |
340 | carp "$filename should define an anonymous sub but it didn't!"; |
fc4b7602 |
341 | } |
342 | } else { |
ff40cb1f |
343 | croak "A file ($filename) got to initialize_scripts that wasn't sql or perl!"; |
fc4b7602 |
344 | } |
345 | } |
346 | } |
347 | |
f9c6ab50 |
348 | method _sqldiff_from_yaml($from_version, $to_version, $db, $direction) { |
91adde75 |
349 | my $dir = $self->script_directory; |
28563f97 |
350 | my $sqltargs = { |
351 | add_drop_table => 1, |
352 | ignore_constraint_names => 1, |
353 | ignore_index_names => 1, |
354 | %{$self->sql_translator_args} |
355 | }; |
d54b8d69 |
356 | |
28563f97 |
357 | my $source_schema; |
358 | { |
359 | my $prefilename = $self->_ddl_protoschema_produce_filename($from_version, $dir); |
360 | |
361 | # should probably be a croak |
362 | carp("No previous schema file found ($prefilename)") |
363 | unless -e $prefilename; |
364 | |
365 | my $t = SQL::Translator->new({ |
366 | %{$sqltargs}, |
367 | debug => 0, |
368 | trace => 0, |
369 | parser => 'SQL::Translator::Parser::YAML', |
370 | }); |
371 | |
372 | my $out = $t->translate( $prefilename ) |
373 | or croak($t->error); |
374 | |
375 | $source_schema = $t->schema; |
376 | |
377 | $source_schema->name( $prefilename ) |
378 | unless $source_schema->name; |
379 | } |
380 | |
381 | my $dest_schema; |
382 | { |
383 | my $filename = $self->_ddl_protoschema_produce_filename($to_version, $dir); |
384 | |
385 | # should probably be a croak |
386 | carp("No next schema file found ($filename)") |
387 | unless -e $filename; |
388 | |
389 | my $t = SQL::Translator->new({ |
390 | %{$sqltargs}, |
391 | debug => 0, |
392 | trace => 0, |
393 | parser => 'SQL::Translator::Parser::YAML', |
394 | }); |
395 | |
396 | my $out = $t->translate( $filename ) |
397 | or croak($t->error); |
398 | |
399 | $dest_schema = $t->schema; |
400 | |
401 | $dest_schema->name( $filename ) |
402 | unless $dest_schema->name; |
403 | } |
f9c6ab50 |
404 | |
405 | my $transform_files_method = "_ddl_protoschema_${direction}_consume_filenames"; |
406 | my $transforms = $self->_coderefs_per_files( |
407 | $self->$transform_files_method([$from_version, $to_version]) |
408 | ); |
409 | $_->($source_schema, $dest_schema) for @$transforms; |
410 | |
28563f97 |
411 | return [SQL::Translator::Diff::schema_diff( |
412 | $source_schema, $db, |
413 | $dest_schema, $db, |
414 | $sqltargs |
415 | )]; |
416 | } |
417 | |
418 | method _sql_from_yaml($sqltargs, $from_file, $db) { |
419 | my $schema = $self->schema; |
420 | my $version = $self->schema_version; |
93460690 |
421 | |
4ea5caf5 |
422 | my @sql; |
2e68a8e1 |
423 | |
4ea5caf5 |
424 | my $actual_file = $self->$from_file($version); |
425 | for my $yaml_filename (@{ |
426 | DlogS_trace { "generating SQL from Serialized SQL Files: $_" } |
427 | (ref $actual_file?$actual_file:[$actual_file]) |
428 | }) { |
429 | my $sqlt = SQL::Translator->new({ |
430 | add_drop_table => 0, |
431 | parser => 'SQL::Translator::Parser::YAML', |
432 | %{$sqltargs}, |
433 | producer => $db, |
434 | }); |
435 | |
436 | push @sql, $sqlt->translate($yaml_filename); |
437 | if(!@sql) { |
438 | carp("Failed to translate to $db, skipping. (" . $sqlt->error . ")"); |
439 | return undef; |
440 | } |
28563f97 |
441 | } |
442 | return \@sql; |
443 | } |
444 | |
445 | sub _prepare_install { |
446 | my $self = shift; |
447 | my $sqltargs = { %{$self->sql_translator_args}, %{shift @_} }; |
448 | my $from_file = shift; |
449 | my $to_file = shift; |
450 | my $dir = $self->script_directory; |
451 | my $databases = $self->databases; |
452 | my $version = $self->schema_version; |
453 | |
2e68a8e1 |
454 | foreach my $db (@$databases) { |
28563f97 |
455 | my $sql = $self->_sql_from_yaml($sqltargs, $from_file, $db ) or next; |
2e68a8e1 |
456 | |
c8a2f7bd |
457 | my $filename = $self->$to_file($db, $version, $dir); |
9600776d |
458 | if (-e $filename ) { |
92624ee5 |
459 | if ($self->force_overwrite) { |
460 | carp "Overwriting existing DDL file - $filename"; |
461 | unlink $filename; |
462 | } else { |
463 | die "Cannot overwrite '$filename', either enable force_overwrite or delete it" |
464 | } |
2e68a8e1 |
465 | } |
387b11d2 |
466 | open my $file, q(>), $filename; |
28563f97 |
467 | print {$file} join ";\n", @$sql; |
2e68a8e1 |
468 | close $file; |
469 | } |
470 | } |
471 | |
c8a2f7bd |
472 | sub _resultsource_install_filename { |
473 | my ($self, $source_name) = @_; |
474 | return sub { |
475 | my ($self, $type, $version) = @_; |
58eb99c3 |
476 | my $dirname = catfile( $self->script_directory, $type, 'deploy', $version ); |
c8a2f7bd |
477 | mkpath($dirname) unless -d $dirname; |
478 | |
09bc35e3 |
479 | return catfile( $dirname, "001-auto-$source_name.sql" ); |
c8a2f7bd |
480 | } |
481 | } |
482 | |
e62add58 |
483 | sub _resultsource_protoschema_filename { |
484 | my ($self, $source_name) = @_; |
485 | return sub { |
486 | my ($self, $version) = @_; |
d3d6512c |
487 | my $dirname = catfile( $self->script_directory, '_source', 'deploy', $version ); |
e62add58 |
488 | mkpath($dirname) unless -d $dirname; |
489 | |
490 | return catfile( $dirname, "001-auto-$source_name.yml" ); |
491 | } |
492 | } |
493 | |
c8a2f7bd |
494 | sub install_resultsource { |
be140a5f |
495 | my ($self, $args) = @_; |
ba99ba44 |
496 | my $source = $args->{result_source} |
497 | or die 'result_source must be passed to install_resultsource'; |
498 | my $version = $args->{version} |
499 | or die 'version must be passed to install_resultsource'; |
f4075791 |
500 | log_info { 'installing_resultsource ' . $source->source_name . ", version $version" }; |
c8a2f7bd |
501 | my $rs_install_file = |
502 | $self->_resultsource_install_filename($source->source_name); |
503 | |
504 | my $files = [ |
505 | $self->$rs_install_file( |
506 | $self->storage->sqlt_type, |
507 | $version, |
508 | ) |
509 | ]; |
510 | $self->_run_sql_and_perl($files); |
511 | } |
512 | |
513 | sub prepare_resultsource_install { |
514 | my $self = shift; |
be140a5f |
515 | my $source = (shift @_)->{result_source}; |
f4075791 |
516 | log_info { 'preparing install for resultsource ' . $source->source_name }; |
c8a2f7bd |
517 | |
e62add58 |
518 | my $install_filename = $self->_resultsource_install_filename($source->source_name); |
519 | my $proto_filename = $self->_resultsource_protoschema_filename($source->source_name); |
6cae2f56 |
520 | $self->prepare_protoschema({ |
c8a2f7bd |
521 | parser_args => { sources => [$source->source_name], } |
e62add58 |
522 | }, $proto_filename); |
523 | $self->_prepare_install({}, $proto_filename, $install_filename); |
c8a2f7bd |
524 | } |
525 | |
91557c90 |
526 | sub prepare_deploy { |
f4075791 |
527 | log_info { 'preparing deploy' }; |
c8a2f7bd |
528 | my $self = shift; |
6776a6d4 |
529 | $self->prepare_protoschema({ |
530 | # Exclude __VERSION so that it gets installed separately |
531 | parser_args => { sources => [grep { $_ ne '__VERSION' } $self->schema->sources], } |
532 | }, '_ddl_protoschema_produce_filename'); |
e62add58 |
533 | $self->_prepare_install({}, '_ddl_protoschema_produce_filename', '_ddl_schema_produce_filename'); |
c8a2f7bd |
534 | } |
535 | |
a41a04e5 |
536 | sub prepare_upgrade { |
be140a5f |
537 | my ($self, $args) = @_; |
0df68524 |
538 | log_info { |
f4075791 |
539 | "preparing upgrade from $args->{from_version} to $args->{to_version}" |
0df68524 |
540 | }; |
be140a5f |
541 | $self->_prepare_changegrade( |
58eb99c3 |
542 | $args->{from_version}, $args->{to_version}, $args->{version_set}, 'upgrade' |
be140a5f |
543 | ); |
76d311e7 |
544 | } |
545 | |
546 | sub prepare_downgrade { |
be140a5f |
547 | my ($self, $args) = @_; |
0df68524 |
548 | log_info { |
f4075791 |
549 | "preparing downgrade from $args->{from_version} to $args->{to_version}" |
0df68524 |
550 | }; |
be140a5f |
551 | $self->_prepare_changegrade( |
58eb99c3 |
552 | $args->{from_version}, $args->{to_version}, $args->{version_set}, 'downgrade' |
be140a5f |
553 | ); |
76d311e7 |
554 | } |
555 | |
f9c6ab50 |
556 | method _coderefs_per_files($files) { |
557 | no warnings 'redefine'; |
558 | [map eval do { local( @ARGV, $/ ) = $_; <> }, @$files] |
559 | } |
560 | |
76d311e7 |
561 | method _prepare_changegrade($from_version, $to_version, $version_set, $direction) { |
2e68a8e1 |
562 | my $schema = $self->schema; |
563 | my $databases = $self->databases; |
91adde75 |
564 | my $dir = $self->script_directory; |
2e68a8e1 |
565 | |
73caa630 |
566 | my $schema_version = $self->schema_version; |
e62add58 |
567 | my $diff_file_method = "_ddl_schema_${direction}_produce_filename"; |
e62add58 |
568 | foreach my $db (@$databases) { |
569 | my $diff_file = $self->$diff_file_method($db, $version_set, $dir ); |
570 | if(-e $diff_file) { |
92624ee5 |
571 | if ($self->force_overwrite) { |
572 | carp("Overwriting existing $direction-diff file - $diff_file"); |
573 | unlink $diff_file; |
574 | } else { |
575 | die "Cannot overwrite '$diff_file', either enable force_overwrite or delete it" |
576 | } |
2e68a8e1 |
577 | } |
578 | |
387b11d2 |
579 | open my $file, q(>), $diff_file; |
f9c6ab50 |
580 | print {$file} join ";\n", @{$self->_sqldiff_from_yaml($from_version, $to_version, $db, $direction)}; |
2e68a8e1 |
581 | close $file; |
582 | } |
583 | } |
584 | |
334bced5 |
585 | method _read_sql_file($file) { |
586 | return unless $file; |
587 | |
aabd4237 |
588 | open my $fh, '<', $file; |
0d19af1d |
589 | my @data = split /;\n/, join '', <$fh>; |
334bced5 |
590 | close $fh; |
591 | |
09bc35e3 |
592 | @data = grep { |
593 | $_ && # remove blank lines |
594 | !/^(BEGIN|BEGIN TRANSACTION|COMMIT)/ # strip txn's |
595 | } map { |
596 | s/^\s+//; s/\s+$//; # trim whitespace |
597 | join '', grep { !/^--/ } split /\n/ # remove comments |
598 | } @data; |
334bced5 |
599 | |
09bc35e3 |
600 | return \@data; |
1f0d0633 |
601 | } |
602 | |
7d2a6974 |
603 | sub downgrade_single_step { |
76d311e7 |
604 | my $self = shift; |
be140a5f |
605 | my $version_set = (shift @_)->{version_set}; |
f4075791 |
606 | Dlog_info { "downgrade_single_step'ing $_" } $version_set; |
41219a5d |
607 | |
ef44838b |
608 | my $sqlt_type = $self->storage->sqlt_type; |
609 | my $sql_to_run; |
610 | if ($self->ignore_ddl) { |
611 | $sql_to_run = $self->_sqldiff_from_yaml( |
58eb99c3 |
612 | $version_set->[0], $version_set->[1], $sqlt_type, 'downgrade', |
ef44838b |
613 | ); |
614 | } |
58eb99c3 |
615 | my $sql = $self->_run_sql_and_perl($self->_ddl_schema_downgrade_consume_filenames( |
ef44838b |
616 | $sqlt_type, |
627581cd |
617 | $version_set, |
ef44838b |
618 | ), $sql_to_run); |
3249629f |
619 | |
41219a5d |
620 | return ['', $sql]; |
76d311e7 |
621 | } |
622 | |
7d2a6974 |
623 | sub upgrade_single_step { |
7521a845 |
624 | my $self = shift; |
be140a5f |
625 | my $version_set = (shift @_)->{version_set}; |
f4075791 |
626 | Dlog_info { "upgrade_single_step'ing $_" } $version_set; |
41219a5d |
627 | |
ef44838b |
628 | my $sqlt_type = $self->storage->sqlt_type; |
629 | my $sql_to_run; |
630 | if ($self->ignore_ddl) { |
631 | $sql_to_run = $self->_sqldiff_from_yaml( |
58eb99c3 |
632 | $version_set->[0], $version_set->[1], $sqlt_type, 'upgrade', |
ef44838b |
633 | ); |
634 | } |
58eb99c3 |
635 | my $sql = $self->_run_sql_and_perl($self->_ddl_schema_upgrade_consume_filenames( |
ef44838b |
636 | $sqlt_type, |
627581cd |
637 | $version_set, |
ef44838b |
638 | ), $sql_to_run); |
41219a5d |
639 | return ['', $sql]; |
334bced5 |
640 | } |
641 | |
6cae2f56 |
642 | sub prepare_protoschema { |
7e08eddd |
643 | my $self = shift; |
e62add58 |
644 | my $sqltargs = { %{$self->sql_translator_args}, %{shift @_} }; |
645 | my $to_file = shift; |
7e08eddd |
646 | my $filename |
e62add58 |
647 | = $self->$to_file($self->schema_version); |
7e08eddd |
648 | |
e62add58 |
649 | # we do this because the code that uses this sets parser args, |
650 | # so we just need to merge in the package |
651 | $sqltargs->{parser_args}{package} = $self->schema; |
7e08eddd |
652 | my $sqlt = SQL::Translator->new({ |
653 | parser => 'SQL::Translator::Parser::DBIx::Class', |
654 | producer => 'SQL::Translator::Producer::YAML', |
e62add58 |
655 | %{ $sqltargs }, |
7e08eddd |
656 | }); |
657 | |
658 | my $yml = $sqlt->translate; |
659 | |
660 | croak("Failed to translate to YAML: " . $sqlt->error) |
661 | unless $yml; |
662 | |
663 | if (-e $filename ) { |
92624ee5 |
664 | if ($self->force_overwrite) { |
665 | carp "Overwriting existing DDL-YML file - $filename"; |
666 | unlink $filename; |
667 | } else { |
668 | die "Cannot overwrite '$filename', either enable force_overwrite or delete it" |
669 | } |
7e08eddd |
670 | } |
671 | |
672 | open my $file, q(>), $filename; |
673 | print {$file} $yml; |
674 | close $file; |
675 | } |
676 | |
aabd4237 |
677 | __PACKAGE__->meta->make_immutable; |
678 | |
2e68a8e1 |
679 | 1; |
e051bb00 |
680 | |
e52174e3 |
681 | # vim: ts=2 sw=2 expandtab |
682 | |
e051bb00 |
683 | __END__ |
684 | |
bcc72297 |
685 | =head1 DESCRIPTION |
686 | |
e62add58 |
687 | This class is the meat of L<DBIx::Class::DeploymentHandler>. It takes care |
688 | of generating serialized schemata as well as sql files to move from one |
689 | version of a schema to the rest. One of the hallmark features of this class |
690 | is that it allows for multiple sql files for deploy and upgrade, allowing |
691 | developers to fine tune deployment. In addition it also allows for perl |
692 | files to be run at any stage of the process. |
bcc72297 |
693 | |
694 | For basic usage see L<DBIx::Class::DeploymentHandler::HandlesDeploy>. What's |
695 | documented here is extra fun stuff or private methods. |
696 | |
697 | =head1 DIRECTORY LAYOUT |
698 | |
39c88a9a |
699 | Arguably this is the best feature of L<DBIx::Class::DeploymentHandler>. |
700 | It's spiritually based upon L<DBIx::Migration::Directories>, but has a |
701 | lot of extensions and modifications, so even if you are familiar with it, |
702 | please read this. I feel like the best way to describe the layout is with |
703 | the following example: |
92c34cab |
704 | |
705 | $sql_migration_dir |
58eb99c3 |
706 | |- _source |
707 | | |- deploy |
03882cab |
708 | | |- 1 |
709 | | | `- 001-auto.yml |
710 | | |- 2 |
711 | | | `- 001-auto.yml |
712 | | `- 3 |
713 | | `- 001-auto.yml |
92c34cab |
714 | |- SQLite |
58eb99c3 |
715 | | |- downgrade |
4f85efc6 |
716 | | | `- 2-1 |
e62add58 |
717 | | | `- 001-auto.sql |
58eb99c3 |
718 | | |- deploy |
92c34cab |
719 | | | `- 1 |
e62add58 |
720 | | | `- 001-auto.sql |
58eb99c3 |
721 | | `- upgrade |
92c34cab |
722 | | |- 1-2 |
e62add58 |
723 | | | `- 001-auto.sql |
92c34cab |
724 | | `- 2-3 |
e62add58 |
725 | | `- 001-auto.sql |
92c34cab |
726 | |- _common |
58eb99c3 |
727 | | |- downgrade |
4f85efc6 |
728 | | | `- 2-1 |
92c34cab |
729 | | | `- 002-remove-customers.pl |
58eb99c3 |
730 | | `- upgrade |
92c34cab |
731 | | `- 1-2 |
732 | | `- 002-generate-customers.pl |
92c34cab |
733 | `- MySQL |
58eb99c3 |
734 | |- downgrade |
4f85efc6 |
735 | | `- 2-1 |
e62add58 |
736 | | `- 001-auto.sql |
ff40cb1f |
737 | |- initialize |
80ff6f6d |
738 | | `- 1 |
739 | | |- 001-create_database.pl |
740 | | `- 002-create_users_and_permissions.pl |
58eb99c3 |
741 | |- deploy |
92c34cab |
742 | | `- 1 |
e62add58 |
743 | | `- 001-auto.sql |
58eb99c3 |
744 | `- upgrade |
92c34cab |
745 | `- 1-2 |
e62add58 |
746 | `- 001-auto.sql |
92c34cab |
747 | |
748 | So basically, the code |
749 | |
750 | $dm->deploy(1) |
751 | |
752 | on an C<SQLite> database that would simply run |
58eb99c3 |
753 | C<$sql_migration_dir/SQLite/deploy/1/001-auto.sql>. Next, |
92c34cab |
754 | |
755 | $dm->upgrade_single_step([1,2]) |
756 | |
58eb99c3 |
757 | would run C<$sql_migration_dir/SQLite/upgrade/1-2/001-auto.sql> followed by |
758 | C<$sql_migration_dir/_common/upgrade/1-2/002-generate-customers.pl>. |
92c34cab |
759 | |
0824f31f |
760 | C<.pl> files don't have to be in the C<_common> directory, but most of the time |
39c88a9a |
761 | they should be, because perl scripts are generally database independent. |
92c34cab |
762 | |
ff40cb1f |
763 | Note that unlike most steps in the process, C<initialize> will not run SQL, as |
764 | there may not even be an database at initialize time. It will run perl scripts |
80ff6f6d |
765 | just like the other steps in the process, but nothing is passed to them. |
766 | Until people have used this more it will remain freeform, but a recommended use |
ff40cb1f |
767 | of initialize is to have it prompt for username and password, and then call the |
80ff6f6d |
768 | appropriate C<< CREATE DATABASE >> commands etc. |
769 | |
03882cab |
770 | =head2 Directory Specification |
771 | |
772 | The following subdirectories are recognized by this DeployMethod: |
773 | |
774 | =over 2 |
775 | |
58eb99c3 |
776 | =item C<_source> This directory can contain the following directories: |
03882cab |
777 | |
778 | =over 2 |
779 | |
39c88a9a |
780 | =item C<deploy> This directory merely contains directories named after schema |
781 | versions, which in turn contain C<yaml> files that are serialized versions |
782 | of the schema at that version. These files are not for editing by hand. |
783 | |
784 | =back |
785 | |
786 | =item C<_preprocess_schema> This directory can contain the following |
787 | directories: |
788 | |
789 | =over 2 |
790 | |
58eb99c3 |
791 | =item C<downgrade> This directory merely contains directories named after |
03882cab |
792 | migrations, which are of the form C<$from_version-$to_version>. Inside of |
793 | these directories you may put Perl scripts which are to return a subref |
794 | that takes the arguments C<< $from_schema, $to_schema >>, which are |
795 | L<SQL::Translator::Schema> objects. |
796 | |
58eb99c3 |
797 | =item C<upgrade> This directory merely contains directories named after |
03882cab |
798 | migrations, which are of the form C<$from_version-$to_version>. Inside of |
799 | these directories you may put Perl scripts which are to return a subref |
800 | that takes the arguments C<< $from_schema, $to_schema >>, which are |
801 | L<SQL::Translator::Schema> objects. |
802 | |
03882cab |
803 | =back |
804 | |
5b766a24 |
805 | =item C<$storage_type> This is a set of scripts that gets run depending on what |
806 | your storage type is. If you are not sure what your storage type is, take a |
807 | look at the producers listed for L<SQL::Translator>. Also note, C<_common> |
808 | is a special case. C<_common> will get merged into whatever other files you |
809 | already have. This directory can containt the following directories itself: |
71d00500 |
810 | |
811 | =over 2 |
812 | |
ff40cb1f |
813 | =item C<initialize> Gets run before the C<deploy> is C<deploy>ed. Has the |
58eb99c3 |
814 | same structure as the C<deploy> subdirectory as well; that is, it has a |
815 | directory for each schema version. Unlike C<deploy>, C<upgrade>, and C<downgrade> |
71d00500 |
816 | though, it can only run C<.pl> files, and the coderef in the perl files get |
817 | no arguments passed to them. |
818 | |
58eb99c3 |
819 | =item C<deploy> Gets run when the schema is C<deploy>ed. Structure is a |
71d00500 |
820 | directory per schema version, and then files are merged with C<_common> and run |
821 | in filename order. C<.sql> files are merely run, as expected. C<.pl> files are |
822 | run according to L</PERL SCRIPTS>. |
823 | |
58eb99c3 |
824 | =item C<upgrade> Gets run when the schema is C<upgrade>d. Structure is a directory |
71d00500 |
825 | per upgrade step, (for example, C<1-2> for upgrading from version 1 to version |
826 | 2,) and then files are merged with C<_common> and run in filename order. |
827 | C<.sql> files are merely run, as expected. C<.pl> files are run according |
828 | to L</PERL SCRIPTS>. |
829 | |
58eb99c3 |
830 | =item C<downgrade> Gets run when the schema is C<downgrade>d. Structure is a directory |
71d00500 |
831 | per downgrade step, (for example, C<2-1> for downgrading from version 2 to version |
832 | 1,) and then files are merged with C<_common> and run in filename order. |
833 | C<.sql> files are merely run, as expected. C<.pl> files are run according |
834 | to L</PERL SCRIPTS>. |
835 | |
836 | |
837 | =back |
838 | |
03882cab |
839 | =back |
840 | |
92c34cab |
841 | =head1 PERL SCRIPTS |
842 | |
7d0b0f2b |
843 | A perl script for this tool is very simple. It merely needs to contain an |
844 | anonymous sub that takes a L<DBIx::Class::Schema> as it's only argument. |
92c34cab |
845 | A very basic perl script might look like: |
846 | |
847 | #!perl |
848 | |
849 | use strict; |
850 | use warnings; |
851 | |
7d0b0f2b |
852 | sub { |
92c34cab |
853 | my $schema = shift; |
854 | |
855 | $schema->resultset('Users')->create({ |
856 | name => 'root', |
857 | password => 'root', |
858 | }) |
859 | } |
bcc72297 |
860 | |
39c88a9a |
861 | =attr ignore_ddl |
862 | |
863 | This attribute will, when set to true (default is false), cause the DM to use |
864 | L<SQL::Translator> to use the C<_source>'s serialized SQL::Translator::Schema |
865 | instead of any pregenerated SQL. If you have a development server this is |
866 | probably the best plan of action as you will not be putting as many generated |
867 | files in your version control. Goes well with with C<databases> of C<[]>. |
868 | |
92624ee5 |
869 | =attr force_overwrite |
870 | |
871 | When this attribute is true generated files will be overwritten when the |
872 | methods which create such files are run again. The default is false, in which |
873 | case the program will die with a message saying which file needs to be deleted. |
874 | |
eb28403b |
875 | =attr schema |
a65184c8 |
876 | |
bcc72297 |
877 | The L<DBIx::Class::Schema> (B<required>) that is used to talk to the database |
878 | and generate the DDL. |
879 | |
eb28403b |
880 | =attr storage |
a65184c8 |
881 | |
bcc72297 |
882 | The L<DBIx::Class::Storage> that is I<actually> used to talk to the database |
883 | and generate the DDL. This is automatically created with L</_build_storage>. |
884 | |
02a7b8ac |
885 | =attr sql_translator_args |
cfc9edf9 |
886 | |
02a7b8ac |
887 | The arguments that get passed to L<SQL::Translator> when it's used. |
a65184c8 |
888 | |
91adde75 |
889 | =attr script_directory |
cfc9edf9 |
890 | |
91adde75 |
891 | The directory (default C<'sql'>) that scripts are stored in |
cfc9edf9 |
892 | |
eb28403b |
893 | =attr databases |
cfc9edf9 |
894 | |
895 | The types of databases (default C<< [qw( MySQL SQLite PostgreSQL )] >>) to |
896 | generate files for |
897 | |
eb28403b |
898 | =attr txn_wrap |
899 | |
bcc72297 |
900 | Set to true (which is the default) to wrap all upgrades and deploys in a single |
901 | transaction. |
902 | |
73caa630 |
903 | =attr schema_version |
904 | |
905 | The version the schema on your harddrive is at. Defaults to |
906 | C<< $self->schema->schema_version >>. |