1 package DBIx::Class::DeploymentHandler::DeployMethod::SQL::Translator;
4 # ABSTRACT: Manage your SQL and Perl migrations in nicely laid out directories
7 use Carp qw( carp croak );
8 use DBIx::Class::DeploymentHandler::Logger;
9 use Log::Contextual qw(:log :dlog), -package_logger =>
10 DBIx::Class::DeploymentHandler::Logger->new({
11 env_prefix => 'DBICDH'
17 require SQL::Translator::Diff;
19 require DBIx::Class::Storage; # loaded for type constraint
20 use DBIx::Class::DeploymentHandler::Types;
22 use File::Path 'mkpath';
23 use File::Spec::Functions;
25 with 'DBIx::Class::DeploymentHandler::HandlesDeploy';
33 has force_overwrite => (
40 isa => 'DBIx::Class::Schema',
46 isa => 'DBIx::Class::Storage',
53 my $s = $self->schema->storage;
54 $s->_determine_driver;
58 has sql_translator_args => (
61 default => sub { {} },
63 has script_directory => (
72 isa => 'DBIx::Class::DeploymentHandler::Databases',
74 default => sub { [qw( MySQL SQLite PostgreSQL )] },
83 has schema_version => (
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
92 sub _build_schema_version {
94 $self->schema->schema_version
97 sub __ddl_consume_with_prefix {
98 my ($self, $type, $versions, $prefix) = @_;
99 my $base_dir = $self->script_directory;
101 my $main = catfile( $base_dir, $type );
103 catfile( $base_dir, '_common', $prefix, join q(-), @{$versions} );
106 catfile( $base_dir, '_common', $prefix, '_any' );
110 $dir = catfile($main, $prefix, join q(-), @{$versions})
112 if ($self->ignore_ddl) {
115 croak "$main does not exist; please write/generate some SQL"
118 my $dir_any = catfile($main, $prefix, '_any');
122 opendir my($dh), $dir;
124 map { $_ => "$dir/$_" }
125 grep { /\.(?:sql|pl|sql-\w+)$/ && -f "$dir/$_" }
129 die $_ unless $self->ignore_ddl;
131 for my $dirname (grep { -d $_ } $common, $common_any, $dir_any) {
132 opendir my($dh), $dirname;
133 for my $filename (grep { /\.(?:sql|pl)$/ && -f catfile($dirname,$_) } readdir $dh) {
134 unless ($files{$filename}) {
135 $files{$filename} = catfile($dirname,$filename);
141 return [@files{sort keys %files}]
144 sub _ddl_initialize_consume_filenames {
145 my ($self, $type, $version) = @_;
146 $self->__ddl_consume_with_prefix($type, [ $version ], 'initialize')
149 sub _ddl_schema_consume_filenames {
150 my ($self, $type, $version) = @_;
151 $self->__ddl_consume_with_prefix($type, [ $version ], 'deploy')
154 sub _ddl_protoschema_deploy_consume_filenames {
155 my ($self, $version) = @_;
156 my $base_dir = $self->script_directory;
158 my $dir = catfile( $base_dir, '_source', 'deploy', $version);
159 return [] unless -d $dir;
161 opendir my($dh), $dir;
162 my %files = map { $_ => "$dir/$_" } grep { /\.yml$/ && -f "$dir/$_" } readdir $dh;
165 return [@files{sort keys %files}]
168 sub _ddl_protoschema_upgrade_consume_filenames {
169 my ($self, $versions) = @_;
170 my $base_dir = $self->script_directory;
172 my $dir = catfile( $base_dir, '_preprocess_schema', 'upgrade', join q(-), @{$versions});
174 return [] unless -d $dir;
176 opendir my($dh), $dir;
177 my %files = map { $_ => "$dir/$_" } grep { /\.pl$/ && -f "$dir/$_" } readdir $dh;
180 return [@files{sort keys %files}]
183 sub _ddl_protoschema_downgrade_consume_filenames {
184 my ($self, $versions) = @_;
185 my $base_dir = $self->script_directory;
187 my $dir = catfile( $base_dir, '_preprocess_schema', 'downgrade', join q(-), @{$versions});
189 return [] unless -d $dir;
191 opendir my($dh), $dir;
192 my %files = map { $_ => "$dir/$_" } grep { /\.pl$/ && -f "$dir/$_" } readdir $dh;
195 return [@files{sort keys %files}]
198 sub _ddl_protoschema_produce_filename {
199 my ($self, $version) = @_;
200 my $dirname = catfile( $self->script_directory, '_source', 'deploy', $version );
201 mkpath($dirname) unless -d $dirname;
203 return catfile( $dirname, '001-auto.yml' );
206 sub _ddl_schema_produce_filename {
207 my ($self, $type, $version) = @_;
208 my $dirname = catfile( $self->script_directory, $type, 'deploy', $version );
209 mkpath($dirname) unless -d $dirname;
211 return catfile( $dirname, '001-auto.sql' );
214 sub _ddl_schema_upgrade_consume_filenames {
215 my ($self, $type, $versions) = @_;
216 $self->__ddl_consume_with_prefix($type, $versions, 'upgrade')
219 sub _ddl_schema_downgrade_consume_filenames {
220 my ($self, $type, $versions) = @_;
221 $self->__ddl_consume_with_prefix($type, $versions, 'downgrade')
224 sub _ddl_schema_upgrade_produce_filename {
225 my ($self, $type, $versions) = @_;
226 my $dir = $self->script_directory;
228 my $dirname = catfile( $dir, $type, 'upgrade', join q(-), @{$versions});
229 mkpath($dirname) unless -d $dirname;
231 return catfile( $dirname, '001-auto.sql' );
234 sub _ddl_schema_downgrade_produce_filename {
235 my ($self, $type, $versions, $dir) = @_;
236 my $dirname = catfile( $dir, $type, 'downgrade', join q(-), @{$versions} );
237 mkpath($dirname) unless -d $dirname;
239 return catfile( $dirname, '001-auto.sql');
243 my ($self, $sql) = @_;
244 my $storage = $self->storage;
247 $_ && # remove blank lines
248 !/^(BEGIN|BEGIN TRANSACTION|COMMIT)/ # strip txn's
250 s/^\s+//; s/\s+$//; # trim whitespace
251 join '', grep { !/^--/ } split /\n/ # remove comments
254 Dlog_trace { "Running SQL $_" } $sql;
255 foreach my $line (@{$sql}) {
256 $storage->_query_start($line);
257 # the whole reason we do this is so that we can see the line that was run
259 $storage->dbh_do (sub { $_[1]->do($line) });
262 die "$_ (running line '$line')"
264 $storage->_query_end($line);
266 return join "\n", @$sql
270 my ($self, $filename) = @_;
271 log_debug { "Running SQL from $filename" };
272 return $self->_run_sql_array($self->_read_sql_file($filename));
276 my ($self, $filename, $versions) = @_;
277 log_debug { "Running Perl from $filename" };
278 my $filedata = do { local( @ARGV, $/ ) = $filename; <> };
280 no warnings 'redefine';
281 my $fn = eval "$filedata";
283 Dlog_trace { "Running Perl $_" } $fn;
286 croak "$filename failed to compile: $@";
287 } elsif (ref $fn eq 'CODE') {
288 $fn->($self->schema, $versions)
290 croak "$filename should define an anonymouse sub that takes a schema but it didn't!";
294 sub _run_sql_and_perl {
295 my ($self, $filenames, $sql_to_run, $versions) = @_;
296 my @files = @{$filenames};
297 my $guard = $self->schema->txn_scope_guard if $self->txn_wrap;
299 $self->_run_sql_array($sql_to_run) if $self->ignore_ddl;
301 my $sql = ($sql_to_run)?join ";\n", @$sql_to_run:'';
303 for my $filename (@files) {
304 if ($self->ignore_ddl && $filename =~ /^[^_]*-auto.*\.sql$/) {
306 } elsif ($filename =~ /\.sql$/) {
307 $sql .= $self->_run_sql($filename)
308 } elsif ( $filename =~ /\.pl$/ ) {
309 $self->_run_perl($filename, $versions)
311 croak "A file ($filename) got to deploy that wasn't sql or perl!";
315 $guard->commit if $self->txn_wrap;
322 my $version = (shift @_ || {})->{version} || $self->schema_version;
323 log_info { "deploying version $version" };
324 my $sqlt_type = $self->storage->sqlt_type;
326 if ($self->ignore_ddl) {
327 $sql = $self->_sql_from_yaml({},
328 '_ddl_protoschema_deploy_consume_filenames', $sqlt_type
331 return $self->_run_sql_and_perl($self->_ddl_schema_consume_filenames(
334 ), $sql, [$version]);
340 my $version = $args->{version} || $self->schema_version;
341 log_info { "initializing version $version" };
342 my $storage_type = $args->{storage_type} || $self->storage->sqlt_type;
344 my @files = @{$self->_ddl_initialize_consume_filenames(
349 for my $filename (@files) {
350 # We ignore sql for now (till I figure out what to do with it)
351 if ( $filename =~ /^(.+)\.pl$/ ) {
352 my $filedata = do { local( @ARGV, $/ ) = $filename; <> };
354 no warnings 'redefine';
355 my $fn = eval "$filedata";
359 croak "$filename failed to compile: $@";
360 } elsif (ref $fn eq 'CODE') {
363 croak "$filename should define an anonymous sub but it didn't!";
366 croak "A file ($filename) got to initialize_scripts that wasn't sql or perl!";
371 sub _sqldiff_from_yaml {
372 my ($self, $from_version, $to_version, $db, $direction) = @_;
373 my $dir = $self->script_directory;
376 ignore_constraint_names => 1,
377 ignore_index_names => 1,
378 %{$self->sql_translator_args}
383 my $prefilename = $self->_ddl_protoschema_produce_filename($from_version, $dir);
385 # should probably be a croak
386 carp("No previous schema file found ($prefilename)")
387 unless -e $prefilename;
389 my $t = SQL::Translator->new({
393 parser => 'SQL::Translator::Parser::YAML',
396 my $out = $t->translate( $prefilename )
399 $source_schema = $t->schema;
401 $source_schema->name( $prefilename )
402 unless $source_schema->name;
407 my $filename = $self->_ddl_protoschema_produce_filename($to_version, $dir);
409 # should probably be a croak
410 carp("No next schema file found ($filename)")
413 my $t = SQL::Translator->new({
417 parser => 'SQL::Translator::Parser::YAML',
420 my $out = $t->translate( $filename )
423 $dest_schema = $t->schema;
425 $dest_schema->name( $filename )
426 unless $dest_schema->name;
429 my $transform_files_method = "_ddl_protoschema_${direction}_consume_filenames";
430 my $transforms = $self->_coderefs_per_files(
431 $self->$transform_files_method([$from_version, $to_version])
433 $_->($source_schema, $dest_schema) for @$transforms;
435 return [SQL::Translator::Diff::schema_diff(
443 my ($self, $sqltargs, $from_file, $db) = @_;
444 my $schema = $self->schema;
445 my $version = $self->schema_version;
449 my $actual_file = $self->$from_file($version);
450 for my $yaml_filename (@{
451 DlogS_trace { "generating SQL from Serialized SQL Files: $_" }
452 (ref $actual_file?$actual_file:[$actual_file])
454 my $sqlt = SQL::Translator->new({
456 parser => 'SQL::Translator::Parser::YAML',
461 push @sql, $sqlt->translate($yaml_filename);
463 carp("Failed to translate to $db, skipping. (" . $sqlt->error . ")");
470 sub _prepare_install {
472 my $sqltargs = { %{$self->sql_translator_args}, %{shift @_} };
473 my $from_file = shift;
475 my $dir = $self->script_directory;
476 my $databases = $self->databases;
477 my $version = $self->schema_version;
479 foreach my $db (@$databases) {
480 my $sql = $self->_sql_from_yaml($sqltargs, $from_file, $db ) or next;
482 my $filename = $self->$to_file($db, $version, $dir);
484 if ($self->force_overwrite) {
485 carp "Overwriting existing DDL file - $filename";
488 die "Cannot overwrite '$filename', either enable force_overwrite or delete it"
491 open my $file, q(>), $filename;
492 print {$file} join ";\n", @$sql;
497 sub _resultsource_install_filename {
498 my ($self, $source_name) = @_;
500 my ($self, $type, $version) = @_;
501 my $dirname = catfile( $self->script_directory, $type, 'deploy', $version );
502 mkpath($dirname) unless -d $dirname;
504 return catfile( $dirname, "001-auto-$source_name.sql" );
508 sub _resultsource_protoschema_filename {
509 my ($self, $source_name) = @_;
511 my ($self, $version) = @_;
512 my $dirname = catfile( $self->script_directory, '_source', 'deploy', $version );
513 mkpath($dirname) unless -d $dirname;
515 return catfile( $dirname, "001-auto-$source_name.yml" );
519 sub install_resultsource {
520 my ($self, $args) = @_;
521 my $source = $args->{result_source}
522 or die 'result_source must be passed to install_resultsource';
523 my $version = $args->{version}
524 or die 'version must be passed to install_resultsource';
525 log_info { 'installing_resultsource ' . $source->source_name . ", version $version" };
526 my $rs_install_file =
527 $self->_resultsource_install_filename($source->source_name);
530 $self->$rs_install_file(
531 $self->storage->sqlt_type,
535 $self->_run_sql_and_perl($files, '', [$version]);
538 sub prepare_resultsource_install {
540 my $source = (shift @_)->{result_source};
541 log_info { 'preparing install for resultsource ' . $source->source_name };
543 my $install_filename = $self->_resultsource_install_filename($source->source_name);
544 my $proto_filename = $self->_resultsource_protoschema_filename($source->source_name);
545 $self->prepare_protoschema({
546 parser_args => { sources => [$source->source_name], }
548 $self->_prepare_install({}, $proto_filename, $install_filename);
552 log_info { 'preparing deploy' };
554 $self->prepare_protoschema({
555 # Exclude __VERSION so that it gets installed separately
556 parser_args => { sources => [grep { $_ ne '__VERSION' } $self->schema->sources], }
557 }, '_ddl_protoschema_produce_filename');
558 $self->_prepare_install({}, '_ddl_protoschema_produce_filename', '_ddl_schema_produce_filename');
561 sub prepare_upgrade {
562 my ($self, $args) = @_;
564 "preparing upgrade from $args->{from_version} to $args->{to_version}"
566 $self->_prepare_changegrade(
567 $args->{from_version}, $args->{to_version}, $args->{version_set}, 'upgrade'
571 sub prepare_downgrade {
572 my ($self, $args) = @_;
574 "preparing downgrade from $args->{from_version} to $args->{to_version}"
576 $self->_prepare_changegrade(
577 $args->{from_version}, $args->{to_version}, $args->{version_set}, 'downgrade'
581 sub _coderefs_per_files {
582 my ($self, $files) = @_;
583 no warnings 'redefine';
584 [map eval do { local( @ARGV, $/ ) = $_; <> }, @$files]
587 sub _prepare_changegrade {
588 my ($self, $from_version, $to_version, $version_set, $direction) = @_;
589 my $schema = $self->schema;
590 my $databases = $self->databases;
591 my $dir = $self->script_directory;
593 my $schema_version = $self->schema_version;
594 my $diff_file_method = "_ddl_schema_${direction}_produce_filename";
595 foreach my $db (@$databases) {
596 my $diff_file = $self->$diff_file_method($db, $version_set, $dir );
598 if ($self->force_overwrite) {
599 carp("Overwriting existing $direction-diff file - $diff_file");
602 die "Cannot overwrite '$diff_file', either enable force_overwrite or delete it"
606 open my $file, q(>), $diff_file;
607 print {$file} join ";\n", @{$self->_sqldiff_from_yaml($from_version, $to_version, $db, $direction)};
613 my ($self, $file) = @_;
616 open my $fh, '<', $file;
617 my @data = split /;\n/, join '', <$fh>;
621 $_ && # remove blank lines
622 !/^(BEGIN|BEGIN TRANSACTION|COMMIT)/ # strip txn's
624 s/^\s+//; s/\s+$//; # trim whitespace
625 join '', grep { !/^--/ } split /\n/ # remove comments
631 sub downgrade_single_step {
633 my $version_set = (shift @_)->{version_set};
634 Dlog_info { "downgrade_single_step'ing $_" } $version_set;
636 my $sqlt_type = $self->storage->sqlt_type;
638 if ($self->ignore_ddl) {
639 $sql_to_run = $self->_sqldiff_from_yaml(
640 $version_set->[0], $version_set->[1], $sqlt_type, 'downgrade',
643 my $sql = $self->_run_sql_and_perl($self->_ddl_schema_downgrade_consume_filenames(
646 ), $sql_to_run, $version_set);
651 sub upgrade_single_step {
653 my $version_set = (shift @_)->{version_set};
654 Dlog_info { "upgrade_single_step'ing $_" } $version_set;
656 my $sqlt_type = $self->storage->sqlt_type;
658 if ($self->ignore_ddl) {
659 $sql_to_run = $self->_sqldiff_from_yaml(
660 $version_set->[0], $version_set->[1], $sqlt_type, 'upgrade',
663 my $sql = $self->_run_sql_and_perl($self->_ddl_schema_upgrade_consume_filenames(
666 ), $sql_to_run, $version_set);
670 sub prepare_protoschema {
672 my $sqltargs = { %{$self->sql_translator_args}, %{shift @_} };
675 = $self->$to_file($self->schema_version);
677 # we do this because the code that uses this sets parser args,
678 # so we just need to merge in the package
679 $sqltargs->{parser_args}{package} = $self->schema;
680 my $sqlt = SQL::Translator->new({
681 parser => 'SQL::Translator::Parser::DBIx::Class',
682 producer => 'SQL::Translator::Producer::YAML',
686 my $yml = $sqlt->translate;
688 croak("Failed to translate to YAML: " . $sqlt->error)
692 if ($self->force_overwrite) {
693 carp "Overwriting existing DDL-YML file - $filename";
696 die "Cannot overwrite '$filename', either enable force_overwrite or delete it"
700 open my $file, q(>), $filename;
705 __PACKAGE__->meta->make_immutable;
709 # vim: ts=2 sw=2 expandtab
715 This class is the meat of L<DBIx::Class::DeploymentHandler>. It takes care
716 of generating serialized schemata as well as sql files to move from one
717 version of a schema to the rest. One of the hallmark features of this class
718 is that it allows for multiple sql files for deploy and upgrade, allowing
719 developers to fine tune deployment. In addition it also allows for perl
720 files to be run at any stage of the process.
722 For basic usage see L<DBIx::Class::DeploymentHandler::HandlesDeploy>. What's
723 documented here is extra fun stuff or private methods.
725 =head1 DIRECTORY LAYOUT
727 Arguably this is the best feature of L<DBIx::Class::DeploymentHandler>.
728 It's spiritually based upon L<DBIx::Migration::Directories>, but has a
729 lot of extensions and modifications, so even if you are familiar with it,
730 please read this. I feel like the best way to describe the layout is with
731 the following example:
757 | | `- 002-remove-customers.pl
760 | | `- 002-generate-customers.pl
762 | `- 999-bump-action.pl
769 | |- 001-create_database.pl
770 | `- 002-create_users_and_permissions.pl
778 So basically, the code
782 on an C<SQLite> database that would simply run
783 C<$sql_migration_dir/SQLite/deploy/1/001-auto.sql>. Next,
785 $dm->upgrade_single_step([1,2])
787 would run C<$sql_migration_dir/SQLite/upgrade/1-2/001-auto.sql> followed by
788 C<$sql_migration_dir/_common/upgrade/1-2/002-generate-customers.pl>, and
789 finally punctuated by
790 C<$sql_migration_dir/_common/upgrade/_any/999-bump-action.pl>.
792 C<.pl> files don't have to be in the C<_common> directory, but most of the time
793 they should be, because perl scripts are generally database independent.
795 Note that unlike most steps in the process, C<initialize> will not run SQL, as
796 there may not even be an database at initialize time. It will run perl scripts
797 just like the other steps in the process, but nothing is passed to them.
798 Until people have used this more it will remain freeform, but a recommended use
799 of initialize is to have it prompt for username and password, and then call the
800 appropriate C<< CREATE DATABASE >> commands etc.
802 =head2 Directory Specification
804 The following subdirectories are recognized by this DeployMethod:
808 =item C<_source> This directory can contain the following directories:
812 =item C<deploy> This directory merely contains directories named after schema
813 versions, which in turn contain C<yaml> files that are serialized versions
814 of the schema at that version. These files are not for editing by hand.
818 =item C<_preprocess_schema> This directory can contain the following
823 =item C<downgrade> This directory merely contains directories named after
824 migrations, which are of the form C<$from_version-$to_version>. Inside of
825 these directories you may put Perl scripts which are to return a subref
826 that takes the arguments C<< $from_schema, $to_schema >>, which are
827 L<SQL::Translator::Schema> objects.
829 =item C<upgrade> This directory merely contains directories named after
830 migrations, which are of the form C<$from_version-$to_version>. Inside of
831 these directories you may put Perl scripts which are to return a subref
832 that takes the arguments C<< $from_schema, $to_schema >>, which are
833 L<SQL::Translator::Schema> objects.
837 =item C<$storage_type> This is a set of scripts that gets run depending on what
838 your storage type is. If you are not sure what your storage type is, take a
839 look at the producers listed for L<SQL::Translator>. Also note, C<_common>
840 is a special case. C<_common> will get merged into whatever other files you
841 already have. This directory can contain the following directories itself:
845 =item C<initialize> Gets run before the C<deploy> is C<deploy>ed. Has the
846 same structure as the C<deploy> subdirectory as well; that is, it has a
847 directory for each schema version. Unlike C<deploy>, C<upgrade>, and C<downgrade>
848 though, it can only run C<.pl> files, and the coderef in the perl files get
849 no arguments passed to them.
851 =item C<deploy> Gets run when the schema is C<deploy>ed. Structure is a
852 directory per schema version, and then files are merged with C<_common> and run
853 in filename order. C<.sql> files are merely run, as expected. C<.pl> files are
854 run according to L</PERL SCRIPTS>.
856 =item C<upgrade> Gets run when the schema is C<upgrade>d. Structure is a directory
857 per upgrade step, (for example, C<1-2> for upgrading from version 1 to version
858 2,) and then files are merged with C<_common> and run in filename order.
859 C<.sql> files are merely run, as expected. C<.pl> files are run according
862 =item C<downgrade> Gets run when the schema is C<downgrade>d. Structure is a directory
863 per downgrade step, (for example, C<2-1> for downgrading from version 2 to version
864 1,) and then files are merged with C<_common> and run in filename order.
865 C<.sql> files are merely run, as expected. C<.pl> files are run according
873 Note that there can be an C<_any> in the place of any of the versions (like
874 C<1-2> or C<1>), which means those scripts will be run B<every> time. So if
875 you have an C<_any> in C<_common/upgrade>, that script will get run for every
880 A perl script for this tool is very simple. It merely needs to contain an
881 anonymous sub that takes a L<DBIx::Class::Schema> and the version set as it's
884 A very basic perl script might look like:
894 # [1] for deploy, [1,2] for upgrade or downgrade, probably used with _any
895 my $versions = shift;
897 $schema->resultset('Users')->create({
905 This attribute will, when set to true (default is false), cause the DM to use
906 L<SQL::Translator> to use the C<_source>'s serialized SQL::Translator::Schema
907 instead of any pregenerated SQL. If you have a development server this is
908 probably the best plan of action as you will not be putting as many generated
909 files in your version control. Goes well with with C<databases> of C<[]>.
911 =attr force_overwrite
913 When this attribute is true generated files will be overwritten when the
914 methods which create such files are run again. The default is false, in which
915 case the program will die with a message saying which file needs to be deleted.
919 The L<DBIx::Class::Schema> (B<required>) that is used to talk to the database
920 and generate the DDL.
924 The L<DBIx::Class::Storage> that is I<actually> used to talk to the database
925 and generate the DDL. This is automatically created with L</_build_storage>.
927 =attr sql_translator_args
929 The arguments that get passed to L<SQL::Translator> when it's used.
931 =attr script_directory
933 The directory (default C<'sql'>) that scripts are stored in
937 The types of databases (default C<< [qw( MySQL SQLite PostgreSQL )] >>) to
942 Set to true (which is the default) to wrap all upgrades and deploys in a single
947 The version the schema on your harddrive is at. Defaults to
948 C<< $self->schema->schema_version >>.