1 package DBIx::Class::Schema;
6 use Carp::Clan qw/^DBIx::Class/;
7 use Scalar::Util qw/weaken/;
10 use base qw/DBIx::Class/;
12 __PACKAGE__->mk_classdata('class_mappings' => {});
13 __PACKAGE__->mk_classdata('source_registrations' => {});
14 __PACKAGE__->mk_classdata('storage_type' => '::DBI');
15 __PACKAGE__->mk_classdata('storage');
16 __PACKAGE__->mk_classdata('exception_action');
20 DBIx::Class::Schema - composable schemas
24 package Library::Schema;
25 use base qw/DBIx::Class::Schema/;
27 # load Library::Schema::CD, Library::Schema::Book, Library::Schema::DVD
28 __PACKAGE__->load_classes(qw/CD Book DVD/);
30 package Library::Schema::CD;
31 use base qw/DBIx::Class/;
32 __PACKAGE__->load_components(qw/PK::Auto Core/); # for example
33 __PACKAGE__->table('cd');
35 # Elsewhere in your code:
36 my $schema1 = Library::Schema->connect(
43 my $schema2 = Library::Schema->connect($coderef_returning_dbh);
45 # fetch objects using Library::Schema::DVD
46 my $resultset = $schema1->resultset('DVD')->search( ... );
47 my @dvd_objects = $schema2->resultset('DVD')->search( ... );
51 Creates database classes based on a schema. This is the recommended way to
52 use L<DBIx::Class> and allows you to use more than one concurrent connection
55 NB: If you're used to L<Class::DBI> it's worth reading the L</SYNOPSIS>
56 carefully, as DBIx::Class does things a little differently. Note in
57 particular which module inherits off which.
65 =item Arguments: $moniker, $component_class
69 Registers a class which isa DBIx::Class::ResultSourceProxy. Equivalent to
72 $schema->register_source($moniker, $component_class->result_source_instance);
77 my ($self, $moniker, $to_register) = @_;
78 $self->register_source($moniker => $to_register->result_source_instance);
81 =head2 register_source
85 =item Arguments: $moniker, $result_source
89 Registers the L<DBIx::Class::ResultSource> in the schema with the given
95 my ($self, $moniker, $source) = @_;
96 my %reg = %{$self->source_registrations};
97 $reg{$moniker} = $source;
98 $self->source_registrations(\%reg);
99 $source->schema($self);
100 weaken($source->{schema}) if ref($self);
101 if ($source->result_class) {
102 my %map = %{$self->class_mappings};
103 $map{$source->result_class} = $moniker;
104 $self->class_mappings(\%map);
112 =item Arguments: $moniker
114 =item Return Value: $classname
118 Retrieves the result class name for the given moniker. For example:
120 my $class = $schema->class('CD');
125 my ($self, $moniker) = @_;
126 return $self->source($moniker)->result_class;
133 =item Arguments: $moniker
135 =item Return Value: $result_source
139 my $source = $schema->source('Book');
141 Returns the L<DBIx::Class::ResultSource> object for the registered moniker.
146 my ($self, $moniker) = @_;
147 my $sreg = $self->source_registrations;
148 return $sreg->{$moniker} if exists $sreg->{$moniker};
150 # if we got here, they probably passed a full class name
151 my $mapped = $self->class_mappings->{$moniker};
152 $self->throw_exception("Can't find source for ${moniker}")
153 unless $mapped && exists $sreg->{$mapped};
154 return $sreg->{$mapped};
161 =item Return Value: @source_monikers
165 Returns the source monikers of all source registrations on this schema.
168 my @source_monikers = $schema->sources;
172 sub sources { return keys %{shift->source_registrations}; }
178 =item Arguments: $moniker
180 =item Return Value: $result_set
184 my $rs = $schema->resultset('DVD');
186 Returns the L<DBIx::Class::ResultSet> object for the registered moniker.
191 my ($self, $moniker) = @_;
192 return $self->source($moniker)->resultset;
199 =item Arguments: @classes?, { $namespace => [ @classes ] }+
203 With no arguments, this method uses L<Module::Find> to find all classes under
204 the schema's namespace. Otherwise, this method loads the classes you specify
205 (using L<use>), and registers them (using L</"register_class">).
207 It is possible to comment out classes with a leading C<#>, but note that perl
208 will think it's a mistake (trying to use a comment in a qw list), so you'll
209 need to add C<no warnings 'qw';> before your load_classes call.
213 My::Schema->load_classes(); # loads My::Schema::CD, My::Schema::Artist,
214 # etc. (anything under the My::Schema namespace)
216 # loads My::Schema::CD, My::Schema::Artist, Other::Namespace::Producer but
217 # not Other::Namespace::LinerNotes nor My::Schema::Track
218 My::Schema->load_classes(qw/ CD Artist #Track /, {
219 Other::Namespace => [qw/ Producer #LinerNotes /],
225 my ($class, @params) = @_;
230 foreach my $param (@params) {
231 if (ref $param eq 'ARRAY') {
232 # filter out commented entries
233 my @modules = grep { $_ !~ /^#/ } @$param;
235 push (@{$comps_for{$class}}, @modules);
237 elsif (ref $param eq 'HASH') {
238 # more than one namespace possible
239 for my $comp ( keys %$param ) {
240 # filter out commented entries
241 my @modules = grep { $_ !~ /^#/ } @{$param->{$comp}};
243 push (@{$comps_for{$comp}}, @modules);
247 # filter out commented entries
248 push (@{$comps_for{$class}}, $param) if $param !~ /^#/;
252 my @comp = map { substr $_, length "${class}::" }
253 Module::Find::findallmod($class);
254 $comps_for{$class} = \@comp;
259 no warnings qw/redefine/;
260 local *Class::C3::reinitialize = sub { };
261 foreach my $prefix (keys %comps_for) {
262 foreach my $comp (@{$comps_for{$prefix}||[]}) {
263 my $comp_class = "${prefix}::${comp}";
264 $class->ensure_class_loaded($comp_class);
265 $comp_class->source_name($comp) unless $comp_class->source_name;
267 push(@to_register, [ $comp_class->source_name, $comp_class ]);
271 Class::C3->reinitialize;
273 foreach my $to (@to_register) {
274 $class->register_class(@$to);
275 # if $class->can('result_source_instance');
279 =head2 load_namespaces
283 =item Arguments: %options?
287 This is an alternative to L</load_classes> above which assumes an alternative
288 layout for automatic class loading. It assumes that all source-definition
289 classes are underneath a sub-namespace of the schema called C<Source>, any
290 corresponding ResultSet classes are underneath a sub-namespace of the schema
293 Both of the sub-namespaces are configurable if you don't like the defaults,
294 via the options C<source_namespace> and C<resultset_namespace>.
296 If (and only if) you specify the option C<default_resultset_class>, any found
297 source-definition classes for which we do not find a corresponding
298 ResultSet class will have their C<resultset_class> set to
299 C<default_resultset_class>.
301 C<load_namespaces> takes care of calling C<resultset_class> for you where
302 neccessary if you didn't do it for yourself.
304 All of the namespace and classname options to this method are relative to
305 the schema classname by default. To specify a fully-qualified name, prefix
306 it with a literal C<+>.
310 # load My::Schema::Source::CD, My::Schema::Source::Artist,
311 # My::Schema::ResultSet::CD, etc...
312 My::Schema->load_namespaces;
314 # Override everything...
315 My::Schema->load_namespaces(
316 source_namespace => 'Srcs',
317 resultset_namespace => 'RSets',
318 default_resultset_class => 'RSetBase',
320 # In the above, if there is a My::Schema::Srcs::Foo, but no matching
321 # My::Schema::RSets::Foo, then the Foo source will have its
322 # resultset_class set to My::Schema::RSetBase
324 # Put things in other namespaces
325 My::Schema->load_namespaces(
326 source_namespace => '+Some::Place::Sources',
327 resultset_namespace => '+Another::Place::RSets',
330 If you'd like to use multiple namespaces of each type, simply use an arrayref
331 of namespaces for that option. In the case that the same source-definition
332 (or resultset) class exists in multiple namespaces, the latter entries in
333 your list of namespaces will override earlier ones.
335 My::Schema->load_namespaces(
336 # My::Schema::Sources_C::Foo takes precedence over My::Schema::Sources_B::Foo :
337 source_namespace => [ 'Sources_A', 'Sources_B', 'Sources_C' ],
338 resultset_namespace => [ '+Some::Place::RSets', 'RSets' ],
343 # Pre-pends our classname to the given relative classname or
344 # class namespace, unless there is a '+' prefix, which will
346 sub _expand_relative_name {
347 my ($class, $name) = @_;
349 $name = $class . '::' . $name if ! ($name =~ s/^\+//);
353 # returns a hash of $shortname => $fullname for every package
354 # found in the given namespaces ($shortname is with the $fullname's
355 # namespace stripped off)
356 sub _map_namespaces {
357 my ($class, @namespaces) = @_;
360 foreach my $namespace (@namespaces) {
363 map { (substr($_, length "${namespace}::"), $_) }
364 Module::Find::findallmod($namespace)
371 sub load_namespaces {
372 my ($class, %args) = @_;
374 my $source_namespace = delete $args{source_namespace} || 'Source';
375 my $resultset_namespace = delete $args{resultset_namespace} || 'ResultSet';
376 my $default_resultset_class = delete $args{default_resultset_class};
378 $class->throw_exception('load_namespaces: unknown option(s): '
379 . join(q{,}, map { qq{'$_'} } keys %args))
380 if scalar keys %args;
382 $default_resultset_class
383 = $class->_expand_relative_name($default_resultset_class);
385 for my $arg ($source_namespace, $resultset_namespace) {
386 $arg = [ $arg ] if !ref($arg) && $arg;
388 $class->throw_exception('load_namespaces: namespace arguments must be '
389 . 'a simple string or an arrayref')
390 if ref($arg) ne 'ARRAY';
392 $_ = $class->_expand_relative_name($_) for (@$arg);
395 my %sources = $class->_map_namespaces(@$source_namespace);
396 my %resultsets = $class->_map_namespaces(@$resultset_namespace);
400 no warnings 'redefine';
401 local *Class::C3::reinitialize = sub { };
402 use warnings 'redefine';
404 foreach my $source (keys %sources) {
405 my $source_class = $sources{$source};
406 $class->ensure_class_loaded($source_class);
407 $source_class->source_name($source) unless $source_class->source_name;
409 my $rs_class = delete $resultsets{$source};
410 my $rs_set = $source_class->resultset_class;
411 if($rs_set && $rs_set ne 'DBIx::Class::ResultSet') {
412 if($rs_class && $rs_class ne $rs_set) {
413 warn "We found ResultSet class '$rs_class' for '$source', but it seems "
414 . "that you had already set '$source' to use '$rs_set' instead";
417 elsif($rs_class ||= $default_resultset_class) {
418 $class->ensure_class_loaded($rs_class);
419 $source_class->resultset_class($rs_class);
422 push(@to_register, [ $source_class->source_name, $source_class ]);
426 foreach (sort keys %resultsets) {
427 warn "load_namespaces found ResultSet class $_ with no "
428 . 'corresponding source-definition class';
431 Class::C3->reinitialize;
432 $class->register_class(@$_) for (@to_register);
437 =head2 compose_connection
441 =item Arguments: $target_namespace, @db_info
443 =item Return Value: $new_schema
447 Calls L<DBIx::Class::Schema/"compose_namespace"> to the target namespace,
448 calls L<DBIx::Class::Schema/connection> with @db_info on the new schema,
449 then injects the L<DBix::Class::ResultSetProxy> component and a
450 resultset_instance classdata entry on all the new classes, in order to support
451 $target_namespaces::$class->search(...) method calls.
453 This is primarily useful when you have a specific need for class method access
454 to a connection. In normal usage it is preferred to call
455 L<DBIx::Class::Schema/connect> and use the resulting schema object to operate
456 on L<DBIx::Class::ResultSet> objects with L<DBIx::Class::Schema/resultset> for
461 sub compose_connection {
462 my ($self, $target, @info) = @_;
463 my $base = 'DBIx::Class::ResultSetProxy';
464 eval "require ${base};";
465 $self->throw_exception
466 ("No arguments to load_classes and couldn't load ${base} ($@)")
469 if ($self eq $target) {
470 # Pathological case, largely caused by the docs on early C::M::DBIC::Plain
471 foreach my $moniker ($self->sources) {
472 my $source = $self->source($moniker);
473 my $class = $source->result_class;
474 $self->inject_base($class, $base);
475 $class->mk_classdata(resultset_instance => $source->resultset);
476 $class->mk_classdata(class_resolver => $self);
478 $self->connection(@info);
482 my $schema = $self->compose_namespace($target, $base);
485 *{"${target}::schema"} = sub { $schema };
488 $schema->connection(@info);
489 foreach my $moniker ($schema->sources) {
490 my $source = $schema->source($moniker);
491 my $class = $source->result_class;
492 #warn "$moniker $class $source ".$source->storage;
493 $class->mk_classdata(result_source_instance => $source);
494 $class->mk_classdata(resultset_instance => $source->resultset);
495 $class->mk_classdata(class_resolver => $schema);
500 =head2 compose_namespace
504 =item Arguments: $target_namespace, $additional_base_class?
506 =item Return Value: $new_schema
510 For each L<DBIx::Class::ResultSource> in the schema, this method creates a
511 class in the target namespace (e.g. $target_namespace::CD,
512 $target_namespace::Artist) that inherits from the corresponding classes
513 attached to the current schema.
515 It also attaches a corresponding L<DBIx::Class::ResultSource> object to the
516 new $schema object. If C<$additional_base_class> is given, the new composed
517 classes will inherit from first the corresponding classe from the current
518 schema then the base class.
520 For example, for a schema with My::Schema::CD and My::Schema::Artist classes,
522 $schema->compose_namespace('My::DB', 'Base::Class');
523 print join (', ', @My::DB::CD::ISA) . "\n";
524 print join (', ', @My::DB::Artist::ISA) ."\n";
526 will produce the output
528 My::Schema::CD, Base::Class
529 My::Schema::Artist, Base::Class
533 sub compose_namespace {
534 my ($self, $target, $base) = @_;
535 my %reg = %{ $self->source_registrations };
538 my $schema = $self->clone;
540 no warnings qw/redefine/;
541 local *Class::C3::reinitialize = sub { };
542 foreach my $moniker ($schema->sources) {
543 my $source = $schema->source($moniker);
544 my $target_class = "${target}::${moniker}";
546 $target_class => $source->result_class, ($base ? $base : ())
548 $source->result_class($target_class);
549 $target_class->result_source_instance($source)
550 if $target_class->can('result_source_instance');
553 Class::C3->reinitialize();
556 foreach my $meth (qw/class source resultset/) {
557 *{"${target}::${meth}"} =
558 sub { shift->schema->$meth(@_) };
564 =head2 setup_connection_class
568 =item Arguments: $target, @info
572 Sets up a database connection class to inject between the schema and the
573 subclasses that the schema creates.
577 sub setup_connection_class {
578 my ($class, $target, @info) = @_;
579 $class->inject_base($target => 'DBIx::Class::DB');
580 #$target->load_components('DB');
581 $target->connection(@info);
588 =item Arguments: @args
590 =item Return Value: $new_schema
594 Instantiates a new Storage object of type
595 L<DBIx::Class::Schema/"storage_type"> and passes the arguments to
596 $storage->connect_info. Sets the connection in-place on the schema. See
597 L<DBIx::Class::Storage::DBI/"connect_info"> for more information.
602 my ($self, @info) = @_;
603 return $self if !@info && $self->storage;
604 my $storage_class = $self->storage_type;
605 $storage_class = 'DBIx::Class::Storage'.$storage_class
606 if $storage_class =~ m/^::/;
607 eval "require ${storage_class};";
608 $self->throw_exception(
609 "No arguments to load_classes and couldn't load ${storage_class} ($@)"
611 my $storage = $storage_class->new($self);
612 $storage->connect_info(\@info);
613 $self->storage($storage);
621 =item Arguments: @info
623 =item Return Value: $new_schema
627 This is a convenience method. It is equivalent to calling
628 $schema->clone->connection(@info). See L</connection> and L</clone> for more
633 sub connect { shift->clone->connection(@_) }
637 Begins a transaction (does nothing if AutoCommit is off). Equivalent to
638 calling $schema->storage->txn_begin. See
639 L<DBIx::Class::Storage::DBI/"txn_begin"> for more information.
643 sub txn_begin { shift->storage->txn_begin }
647 Commits the current transaction. Equivalent to calling
648 $schema->storage->txn_commit. See L<DBIx::Class::Storage::DBI/"txn_commit">
649 for more information.
653 sub txn_commit { shift->storage->txn_commit }
657 Rolls back the current transaction. Equivalent to calling
658 $schema->storage->txn_rollback. See
659 L<DBIx::Class::Storage::DBI/"txn_rollback"> for more information.
663 sub txn_rollback { shift->storage->txn_rollback }
669 =item Arguments: C<$coderef>, @coderef_args?
671 =item Return Value: The return value of $coderef
675 Executes C<$coderef> with (optional) arguments C<@coderef_args> atomically,
676 returning its result (if any). If an exception is caught, a rollback is issued
677 and the exception is rethrown. If the rollback fails, (i.e. throws an
678 exception) an exception is thrown that includes a "Rollback failed" message.
682 my $author_rs = $schema->resultset('Author')->find(1);
683 my @titles = qw/Night Day It/;
686 # If any one of these fails, the entire transaction fails
687 $author_rs->create_related('books', {
689 }) foreach (@titles);
691 return $author->books;
696 $rs = $schema->txn_do($coderef);
699 if ($@) { # Transaction failed
700 die "something terrible has happened!" #
701 if ($@ =~ /Rollback failed/); # Rollback failed
703 deal_with_failed_transaction();
706 In a nested transaction (calling txn_do() from within a txn_do() coderef) only
707 the outermost transaction will issue a L<DBIx::Class::Schema/"txn_commit"> on
708 the Schema's storage, and txn_do() can be called in void, scalar and list
709 context and it will behave as expected.
714 my ($self, $coderef, @args) = @_;
716 $self->storage or $self->throw_exception
717 ('txn_do called on $schema without storage');
718 ref $coderef eq 'CODE' or $self->throw_exception
719 ('$coderef must be a CODE reference');
721 my (@return_values, $return_value);
723 $self->txn_begin; # If this throws an exception, no rollback is needed
725 my $wantarray = wantarray; # Need to save this since the context
726 # inside the eval{} block is independent
727 # of the context that called txn_do()
730 # Need to differentiate between scalar/list context to allow for
731 # returning a list in scalar context to get the size of the list
734 @return_values = $coderef->(@args);
735 } elsif (defined $wantarray) {
737 $return_value = $coderef->(@args);
753 my $rollback_error = $@;
754 my $exception_class = "DBIx::Class::Storage::NESTED_ROLLBACK_EXCEPTION";
755 $self->throw_exception($error) # propagate nested rollback
756 if $rollback_error =~ /$exception_class/;
758 $self->throw_exception(
759 "Transaction aborted: $error. Rollback failed: ${rollback_error}"
762 $self->throw_exception($error); # txn failed but rollback succeeded
766 return $wantarray ? @return_values : $return_value;
773 =item Return Value: $new_schema
777 Clones the schema and its associated result_source objects and returns the
784 my $clone = bless({ (ref $self ? %$self : ()) }, ref $self || $self);
785 foreach my $moniker ($self->sources) {
786 my $source = $self->source($moniker);
787 my $new = $source->new($source);
788 $clone->register_source($moniker => $new);
790 $clone->storage->set_schema($clone) if $clone->storage;
798 =item Arguments: $moniker, \@data;
802 Populates the source registered with the given moniker with the supplied data.
803 @data should be a list of listrefs -- the first containing column names, the
804 second matching values.
808 $schema->populate('Artist', [
809 [ qw/artistid name/ ],
810 [ 1, 'Popular Band' ],
818 my ($self, $name, $data) = @_;
819 my $rs = $self->resultset($name);
820 my @names = @{shift(@$data)};
822 foreach my $item (@$data) {
824 @create{@names} = @$item;
825 push(@created, $rs->create(\%create));
830 =head2 exception_action
834 =item Arguments: $code_reference
838 If C<exception_action> is set for this class/object, L</throw_exception>
839 will prefer to call this code reference with the exception as an argument,
840 rather than its normal <croak> action.
842 Your subroutine should probably just wrap the error in the exception
843 object/class of your choosing and rethrow. If, against all sage advice,
844 you'd like your C<exception_action> to suppress a particular exception
845 completely, simply have it return true.
850 use base qw/DBIx::Class::Schema/;
851 use My::ExceptionClass;
852 __PACKAGE__->exception_action(sub { My::ExceptionClass->throw(@_) });
853 __PACKAGE__->load_classes;
856 my $schema_obj = My::Schema->connect( .... );
857 $schema_obj->exception_action(sub { My::ExceptionClass->throw(@_) });
859 # suppress all exceptions, like a moron:
860 $schema_obj->exception_action(sub { 1 });
862 =head2 throw_exception
866 =item Arguments: $message
870 Throws an exception. Defaults to using L<Carp::Clan> to report errors from
871 user's perspective. See L</exception_action> for details on overriding
872 this method's behavior.
876 sub throw_exception {
878 croak @_ if !$self->exception_action || !$self->exception_action->(@_);
881 =head2 deploy (EXPERIMENTAL)
885 =item Arguments: $sqlt_args
889 Attempts to deploy the schema to the current storage using L<SQL::Translator>.
891 Note that this feature is currently EXPERIMENTAL and may not work correctly
892 across all databases, or fully handle complex relationships.
894 See L<SQL::Translator/METHODS> for a list of values for C<$sqlt_args>. The most
895 common value for this would be C<< { add_drop_table => 1, } >> to have the SQL
896 produced include a DROP TABLE statement for each table created.
901 my ($self, $sqltargs) = @_;
902 $self->throw_exception("Can't deploy without storage") unless $self->storage;
903 $self->storage->deploy($self, undef, $sqltargs);
906 =head2 create_ddl_dir (EXPERIMENTAL)
910 =item Arguments: \@databases, $version, $directory, $sqlt_args
914 Creates an SQL file based on the Schema, for each of the specified
915 database types, in the given directory.
917 Note that this feature is currently EXPERIMENTAL and may not work correctly
918 across all databases, or fully handle complex relationships.
926 $self->throw_exception("Can't create_ddl_dir without storage") unless $self->storage;
927 $self->storage->create_ddl_dir($self, @_);
930 =head2 ddl_filename (EXPERIMENTAL)
932 my $filename = $table->ddl_filename($type, $dir, $version)
934 Creates a filename for a SQL file based on the table class name. Not
935 intended for direct end user use.
941 my ($self, $type, $dir, $version) = @_;
943 my $filename = ref($self);
944 $filename =~ s/::/-/;
945 $filename = "$dir$filename-$version-$type.sql";
954 Matt S. Trout <mst@shadowcatsystems.co.uk>
958 You may distribute this code under the same terms as Perl itself.