Massive rewrite of bind handling, and overall simplification of ::Storage::DBI
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Storage / DBI / Replicated.pm
1 package DBIx::Class::Storage::DBI::Replicated;
2
3 BEGIN {
4   use DBIx::Class;
5   die('The following modules are required for Replication ' . DBIx::Class::Optional::Dependencies->req_missing_for ('replicated') . "\n" )
6     unless DBIx::Class::Optional::Dependencies->req_ok_for ('replicated');
7 }
8
9 use Moose;
10 use DBIx::Class::Storage::DBI;
11 use DBIx::Class::Storage::DBI::Replicated::Pool;
12 use DBIx::Class::Storage::DBI::Replicated::Balancer;
13 use DBIx::Class::Storage::DBI::Replicated::Types qw/BalancerClassNamePart DBICSchema DBICStorageDBI/;
14 use MooseX::Types::Moose qw/ClassName HashRef Object/;
15 use Scalar::Util 'reftype';
16 use Hash::Merge;
17 use List::Util qw/min max reduce/;
18 use Try::Tiny;
19 use namespace::clean;
20
21 use namespace::clean -except => 'meta';
22
23 =head1 NAME
24
25 DBIx::Class::Storage::DBI::Replicated - BETA Replicated database support
26
27 =head1 SYNOPSIS
28
29 The Following example shows how to change an existing $schema to a replicated
30 storage type, add some replicated (read-only) databases, and perform reporting
31 tasks.
32
33 You should set the 'storage_type attribute to a replicated type.  You should
34 also define your arguments, such as which balancer you want and any arguments
35 that the Pool object should get.
36
37   my $schema = Schema::Class->clone;
38   $schema->storage_type( ['::DBI::Replicated', {balancer=>'::Random'}] );
39   $schema->connection(...);
40
41 Next, you need to add in the Replicants.  Basically this is an array of
42 arrayrefs, where each arrayref is database connect information.  Think of these
43 arguments as what you'd pass to the 'normal' $schema->connect method.
44
45   $schema->storage->connect_replicants(
46     [$dsn1, $user, $pass, \%opts],
47     [$dsn2, $user, $pass, \%opts],
48     [$dsn3, $user, $pass, \%opts],
49   );
50
51 Now, just use the $schema as you normally would.  Automatically all reads will
52 be delegated to the replicants, while writes to the master.
53
54   $schema->resultset('Source')->search({name=>'etc'});
55
56 You can force a given query to use a particular storage using the search
57 attribute 'force_pool'.  For example:
58
59   my $RS = $schema->resultset('Source')->search(undef, {force_pool=>'master'});
60
61 Now $RS will force everything (both reads and writes) to use whatever was setup
62 as the master storage.  'master' is hardcoded to always point to the Master,
63 but you can also use any Replicant name.  Please see:
64 L<DBIx::Class::Storage::DBI::Replicated::Pool> and the replicants attribute for more.
65
66 Also see transactions and L</execute_reliably> for alternative ways to
67 force read traffic to the master.  In general, you should wrap your statements
68 in a transaction when you are reading and writing to the same tables at the
69 same time, since your replicants will often lag a bit behind the master.
70
71 See L<DBIx::Class::Storage::DBI::Replicated::Instructions> for more help and
72 walkthroughs.
73
74 =head1 DESCRIPTION
75
76 Warning: This class is marked BETA.  This has been running a production
77 website using MySQL native replication as its backend and we have some decent
78 test coverage but the code hasn't yet been stressed by a variety of databases.
79 Individual DBs may have quirks we are not aware of.  Please use this in first
80 development and pass along your experiences/bug fixes.
81
82 This class implements replicated data store for DBI. Currently you can define
83 one master and numerous slave database connections. All write-type queries
84 (INSERT, UPDATE, DELETE and even LAST_INSERT_ID) are routed to master
85 database, all read-type queries (SELECTs) go to the slave database.
86
87 Basically, any method request that L<DBIx::Class::Storage::DBI> would normally
88 handle gets delegated to one of the two attributes: L</read_handler> or to
89 L</write_handler>.  Additionally, some methods need to be distributed
90 to all existing storages.  This way our storage class is a drop in replacement
91 for L<DBIx::Class::Storage::DBI>.
92
93 Read traffic is spread across the replicants (slaves) occurring to a user
94 selected algorithm.  The default algorithm is random weighted.
95
96 =head1 NOTES
97
98 The consistency between master and replicants is database specific.  The Pool
99 gives you a method to validate its replicants, removing and replacing them
100 when they fail/pass predefined criteria.  Please make careful use of the ways
101 to force a query to run against Master when needed.
102
103 =head1 REQUIREMENTS
104
105 Replicated Storage has additional requirements not currently part of
106 L<DBIx::Class>. See L<DBIx::Class::Optional::Dependencies> for more details.
107
108 =head1 ATTRIBUTES
109
110 This class defines the following attributes.
111
112 =head2 schema
113
114 The underlying L<DBIx::Class::Schema> object this storage is attaching
115
116 =cut
117
118 has 'schema' => (
119     is=>'rw',
120     isa=>DBICSchema,
121     weak_ref=>1,
122     required=>1,
123 );
124
125 =head2 pool_type
126
127 Contains the classname which will instantiate the L</pool> object.  Defaults
128 to: L<DBIx::Class::Storage::DBI::Replicated::Pool>.
129
130 =cut
131
132 has 'pool_type' => (
133   is=>'rw',
134   isa=>ClassName,
135   default=>'DBIx::Class::Storage::DBI::Replicated::Pool',
136   handles=>{
137     'create_pool' => 'new',
138   },
139 );
140
141 =head2 pool_args
142
143 Contains a hashref of initialized information to pass to the Balancer object.
144 See L<DBIx::Class::Storage::DBI::Replicated::Pool> for available arguments.
145
146 =cut
147
148 has 'pool_args' => (
149   is=>'rw',
150   isa=>HashRef,
151   lazy=>1,
152   default=>sub { {} },
153 );
154
155
156 =head2 balancer_type
157
158 The replication pool requires a balance class to provider the methods for
159 choose how to spread the query load across each replicant in the pool.
160
161 =cut
162
163 has 'balancer_type' => (
164   is=>'rw',
165   isa=>BalancerClassNamePart,
166   coerce=>1,
167   required=>1,
168   default=> 'DBIx::Class::Storage::DBI::Replicated::Balancer::First',
169   handles=>{
170     'create_balancer' => 'new',
171   },
172 );
173
174 =head2 balancer_args
175
176 Contains a hashref of initialized information to pass to the Balancer object.
177 See L<DBIx::Class::Storage::DBI::Replicated::Balancer> for available arguments.
178
179 =cut
180
181 has 'balancer_args' => (
182   is=>'rw',
183   isa=>HashRef,
184   lazy=>1,
185   required=>1,
186   default=>sub { {} },
187 );
188
189 =head2 pool
190
191 Is a L<DBIx::Class::Storage::DBI::Replicated::Pool> or derived class.  This is a
192 container class for one or more replicated databases.
193
194 =cut
195
196 has 'pool' => (
197   is=>'ro',
198   isa=>'DBIx::Class::Storage::DBI::Replicated::Pool',
199   lazy_build=>1,
200   handles=>[qw/
201     connect_replicants
202     replicants
203     has_replicants
204   /],
205 );
206
207 =head2 balancer
208
209 Is a L<DBIx::Class::Storage::DBI::Replicated::Balancer> or derived class.  This
210 is a class that takes a pool (L<DBIx::Class::Storage::DBI::Replicated::Pool>)
211
212 =cut
213
214 has 'balancer' => (
215   is=>'rw',
216   isa=>'DBIx::Class::Storage::DBI::Replicated::Balancer',
217   lazy_build=>1,
218   handles=>[qw/auto_validate_every/],
219 );
220
221 =head2 master
222
223 The master defines the canonical state for a pool of connected databases.  All
224 the replicants are expected to match this databases state.  Thus, in a classic
225 Master / Slaves distributed system, all the slaves are expected to replicate
226 the Master's state as quick as possible.  This is the only database in the
227 pool of databases that is allowed to handle write traffic.
228
229 =cut
230
231 has 'master' => (
232   is=> 'ro',
233   isa=>DBICStorageDBI,
234   lazy_build=>1,
235 );
236
237 =head1 ATTRIBUTES IMPLEMENTING THE DBIx::Storage::DBI INTERFACE
238
239 The following methods are delegated all the methods required for the
240 L<DBIx::Class::Storage::DBI> interface.
241
242 =cut
243
244 my $method_dispatch = {
245   writer => [qw/
246     on_connect_do
247     on_disconnect_do
248     on_connect_call
249     on_disconnect_call
250     connect_info
251     _connect_info
252     throw_exception
253     sql_maker
254     sqlt_type
255     create_ddl_dir
256     deployment_statements
257     datetime_parser
258     datetime_parser_type
259     build_datetime_parser
260     last_insert_id
261     insert
262     insert_bulk
263     update
264     delete
265     dbh
266     txn_begin
267     txn_do
268     txn_commit
269     txn_rollback
270     txn_scope_guard
271     deploy
272     with_deferred_fk_checks
273     dbh_do
274     _prep_for_execute
275     is_datatype_numeric
276     _count_select
277     _subq_update_delete
278     svp_rollback
279     svp_begin
280     svp_release
281     relname_to_table_alias
282     _dbh_last_insert_id
283     _default_dbi_connect_attributes
284     _dbi_connect_info
285     _dbic_connect_attributes
286     auto_savepoint
287     _query_start
288     _query_end
289     _format_for_trace
290     _dbi_attrs_for_bind
291     bind_attribute_by_data_type
292     transaction_depth
293     _dbh
294     _select_args
295     _dbh_execute_array
296     _sql_maker
297     _per_row_update_delete
298     _dbh_begin_work
299     _dbh_execute_inserts_with_no_binds
300     _select_args_to_query
301     _svp_generate_name
302     _multipk_update_delete
303     _normalize_connect_info
304     _parse_connect_do
305     _dbh_commit
306     _execute_array
307     savepoints
308     _sql_maker_opts
309     _conn_pid
310     _dbh_autocommit
311     _native_data_type
312     _get_dbh
313     sql_maker_class
314     _dbh_rollback
315     _adjust_select_args_for_complex_prefetch
316     _resolve_ident_sources
317     _resolve_column_info
318     _prune_unused_joins
319     _strip_cond_qualifiers
320     _strip_cond_qualifiers_from_array
321     _resolve_aliastypes_from_select_args
322     _execute
323     _do_query
324     _sth
325     _dbh_sth
326     _dbh_execute
327   /],
328   reader => [qw/
329     select
330     select_single
331     columns_info_for
332     _dbh_columns_info_for
333     _select
334   /],
335   unimplemented => [qw/
336     _arm_global_destructor
337     _verify_pid
338
339     source_bind_attributes
340
341     get_use_dbms_capability
342     set_use_dbms_capability
343     get_dbms_capability
344     set_dbms_capability
345     _dbh_details
346     _dbh_get_info
347
348     sql_limit_dialect
349     sql_quote_char
350     sql_name_sep
351
352     _inner_join_to_node
353     _group_over_selection
354     _extract_order_criteria
355
356     _prefetch_autovalues
357
358     _max_column_bytesize
359     _is_lob_type
360     _is_binary_lob_type
361     _is_text_lob_type
362
363     sth
364   /,(
365     # the capability framework
366     # not sure if CMOP->initialize does evil things to DBIC::S::DBI, fix if a problem
367     grep
368       { $_ =~ /^ _ (?: use | supports | determine_supports ) _ /x }
369       ( Class::MOP::Class->initialize('DBIx::Class::Storage::DBI')->get_all_method_names )
370   )],
371 };
372
373 if (DBIx::Class::_ENV_::DBICTEST) {
374
375   my $seen;
376   for my $type (keys %$method_dispatch) {
377     for (@{$method_dispatch->{$type}}) {
378       push @{$seen->{$_}}, $type;
379     }
380   }
381
382   if (my @dupes = grep { @{$seen->{$_}} > 1 } keys %$seen) {
383     die(join "\n", '',
384       'The following methods show up multiple times in ::Storage::DBI::Replicated handlers:',
385       (map { "$_: " . (join ', ', @{$seen->{$_}}) } sort @dupes),
386       '',
387     );
388   }
389
390   if (my @cant = grep { ! DBIx::Class::Storage::DBI->can($_) } keys %$seen) {
391     die(join "\n", '',
392       '::Storage::DBI::Replicated specifies handling of the following *NON EXISTING* ::Storage::DBI methods:',
393       @cant,
394       '',
395     );
396   }
397 }
398
399 for my $method (@{$method_dispatch->{unimplemented}}) {
400   __PACKAGE__->meta->add_method($method, sub {
401     my $self = shift;
402     $self->throw_exception("$method must not be called on ".(blessed $self).' objects');
403   });
404 }
405
406 =head2 read_handler
407
408 Defines an object that implements the read side of L<BIx::Class::Storage::DBI>.
409
410 =cut
411
412 has 'read_handler' => (
413   is=>'rw',
414   isa=>Object,
415   lazy_build=>1,
416   handles=>$method_dispatch->{reader},
417 );
418
419 =head2 write_handler
420
421 Defines an object that implements the write side of L<BIx::Class::Storage::DBI>,
422 as well as methods that don't write or read that can be called on only one
423 storage, methods that return a C<$dbh>, and any methods that don't make sense to
424 run on a replicant.
425
426 =cut
427
428 has 'write_handler' => (
429   is=>'ro',
430   isa=>Object,
431   lazy_build=>1,
432   handles=>$method_dispatch->{writer},
433 );
434
435
436
437 has _master_connect_info_opts =>
438   (is => 'rw', isa => HashRef, default => sub { {} });
439
440 =head2 around: connect_info
441
442 Preserves master's C<connect_info> options (for merging with replicants.)
443 Also sets any Replicated-related options from connect_info, such as
444 C<pool_type>, C<pool_args>, C<balancer_type> and C<balancer_args>.
445
446 =cut
447
448 around connect_info => sub {
449   my ($next, $self, $info, @extra) = @_;
450
451   my $merge = Hash::Merge->new('LEFT_PRECEDENT');
452
453   my %opts;
454   for my $arg (@$info) {
455     next unless (reftype($arg)||'') eq 'HASH';
456     %opts = %{ $merge->merge($arg, \%opts) };
457   }
458   delete $opts{dsn};
459
460   if (@opts{qw/pool_type pool_args/}) {
461     $self->pool_type(delete $opts{pool_type})
462       if $opts{pool_type};
463
464     $self->pool_args(
465       $merge->merge((delete $opts{pool_args} || {}), $self->pool_args)
466     );
467
468     ## Since we possibly changed the pool_args, we need to clear the current
469     ## pool object so that next time it is used it will be rebuilt.
470     $self->clear_pool;
471   }
472
473   if (@opts{qw/balancer_type balancer_args/}) {
474     $self->balancer_type(delete $opts{balancer_type})
475       if $opts{balancer_type};
476
477     $self->balancer_args(
478       $merge->merge((delete $opts{balancer_args} || {}), $self->balancer_args)
479     );
480
481     $self->balancer($self->_build_balancer)
482       if $self->balancer;
483   }
484
485   $self->_master_connect_info_opts(\%opts);
486
487   my @res;
488   if (wantarray) {
489     @res = $self->$next($info, @extra);
490   } else {
491     $res[0] = $self->$next($info, @extra);
492   }
493
494   # Make sure master is blessed into the correct class and apply role to it.
495   my $master = $self->master;
496   $master->_determine_driver;
497   Moose::Meta::Class->initialize(ref $master);
498
499   DBIx::Class::Storage::DBI::Replicated::WithDSN->meta->apply($master);
500
501   # link pool back to master
502   $self->pool->master($master);
503
504   wantarray ? @res : $res[0];
505 };
506
507 =head1 METHODS
508
509 This class defines the following methods.
510
511 =head2 BUILDARGS
512
513 L<DBIx::Class::Schema> when instantiating its storage passed itself as the
514 first argument.  So we need to massage the arguments a bit so that all the
515 bits get put into the correct places.
516
517 =cut
518
519 sub BUILDARGS {
520   my ($class, $schema, $storage_type_args, @args) = @_;
521
522   return {
523     schema=>$schema,
524     %$storage_type_args,
525     @args
526   }
527 }
528
529 =head2 _build_master
530
531 Lazy builder for the L</master> attribute.
532
533 =cut
534
535 sub _build_master {
536   my $self = shift @_;
537   my $master = DBIx::Class::Storage::DBI->new($self->schema);
538   $master
539 }
540
541 =head2 _build_pool
542
543 Lazy builder for the L</pool> attribute.
544
545 =cut
546
547 sub _build_pool {
548   my $self = shift @_;
549   $self->create_pool(%{$self->pool_args});
550 }
551
552 =head2 _build_balancer
553
554 Lazy builder for the L</balancer> attribute.  This takes a Pool object so that
555 the balancer knows which pool it's balancing.
556
557 =cut
558
559 sub _build_balancer {
560   my $self = shift @_;
561   $self->create_balancer(
562     pool=>$self->pool,
563     master=>$self->master,
564     %{$self->balancer_args},
565   );
566 }
567
568 =head2 _build_write_handler
569
570 Lazy builder for the L</write_handler> attribute.  The default is to set this to
571 the L</master>.
572
573 =cut
574
575 sub _build_write_handler {
576   return shift->master;
577 }
578
579 =head2 _build_read_handler
580
581 Lazy builder for the L</read_handler> attribute.  The default is to set this to
582 the L</balancer>.
583
584 =cut
585
586 sub _build_read_handler {
587   return shift->balancer;
588 }
589
590 =head2 around: connect_replicants
591
592 All calls to connect_replicants needs to have an existing $schema tacked onto
593 top of the args, since L<DBIx::Storage::DBI> needs it, and any C<connect_info>
594 options merged with the master, with replicant opts having higher priority.
595
596 =cut
597
598 around connect_replicants => sub {
599   my ($next, $self, @args) = @_;
600
601   for my $r (@args) {
602     $r = [ $r ] unless reftype $r eq 'ARRAY';
603
604     $self->throw_exception('coderef replicant connect_info not supported')
605       if ref $r->[0] && reftype $r->[0] eq 'CODE';
606
607 # any connect_info options?
608     my $i = 0;
609     $i++ while $i < @$r && (reftype($r->[$i])||'') ne 'HASH';
610
611 # make one if none
612     $r->[$i] = {} unless $r->[$i];
613
614 # merge if two hashes
615     my @hashes = @$r[$i .. $#{$r}];
616
617     $self->throw_exception('invalid connect_info options')
618       if (grep { reftype($_) eq 'HASH' } @hashes) != @hashes;
619
620     $self->throw_exception('too many hashrefs in connect_info')
621       if @hashes > 2;
622
623     my $merge = Hash::Merge->new('LEFT_PRECEDENT');
624     my %opts = %{ $merge->merge(reverse @hashes) };
625
626 # delete them
627     splice @$r, $i+1, ($#{$r} - $i), ();
628
629 # make sure master/replicants opts don't clash
630     my %master_opts = %{ $self->_master_connect_info_opts };
631     if (exists $opts{dbh_maker}) {
632         delete @master_opts{qw/dsn user password/};
633     }
634     delete $master_opts{dbh_maker};
635
636 # merge with master
637     %opts = %{ $merge->merge(\%opts, \%master_opts) };
638
639 # update
640     $r->[$i] = \%opts;
641   }
642
643   $self->$next($self->schema, @args);
644 };
645
646 =head2 all_storages
647
648 Returns an array of of all the connected storage backends.  The first element
649 in the returned array is the master, and the remainings are each of the
650 replicants.
651
652 =cut
653
654 sub all_storages {
655   my $self = shift @_;
656   return grep {defined $_ && blessed $_} (
657      $self->master,
658      values %{ $self->replicants },
659   );
660 }
661
662 =head2 execute_reliably ($coderef, ?@args)
663
664 Given a coderef, saves the current state of the L</read_handler>, forces it to
665 use reliable storage (e.g. sets it to the master), executes a coderef and then
666 restores the original state.
667
668 Example:
669
670   my $reliably = sub {
671     my $name = shift @_;
672     $schema->resultset('User')->create({name=>$name});
673     my $user_rs = $schema->resultset('User')->find({name=>$name});
674     return $user_rs;
675   };
676
677   my $user_rs = $schema->storage->execute_reliably($reliably, 'John');
678
679 Use this when you must be certain of your database state, such as when you just
680 inserted something and need to get a resultset including it, etc.
681
682 =cut
683
684 sub execute_reliably {
685   my ($self, $coderef, @args) = @_;
686
687   unless( ref $coderef eq 'CODE') {
688     $self->throw_exception('Second argument must be a coderef');
689   }
690
691   ##Get copy of master storage
692   my $master = $self->master;
693
694   ##Get whatever the current read hander is
695   my $current = $self->read_handler;
696
697   ##Set the read handler to master
698   $self->read_handler($master);
699
700   ## do whatever the caller needs
701   my @result;
702   my $want_array = wantarray;
703
704   try {
705     if($want_array) {
706       @result = $coderef->(@args);
707     } elsif(defined $want_array) {
708       ($result[0]) = ($coderef->(@args));
709     } else {
710       $coderef->(@args);
711     }
712   } catch {
713     $self->throw_exception("coderef returned an error: $_");
714   } finally {
715     ##Reset to the original state
716     $self->read_handler($current);
717   };
718
719   return wantarray ? @result : $result[0];
720 }
721
722 =head2 set_reliable_storage
723
724 Sets the current $schema to be 'reliable', that is all queries, both read and
725 write are sent to the master
726
727 =cut
728
729 sub set_reliable_storage {
730   my $self = shift @_;
731   my $schema = $self->schema;
732   my $write_handler = $self->schema->storage->write_handler;
733
734   $schema->storage->read_handler($write_handler);
735 }
736
737 =head2 set_balanced_storage
738
739 Sets the current $schema to be use the </balancer> for all reads, while all
740 writes are sent to the master only
741
742 =cut
743
744 sub set_balanced_storage {
745   my $self = shift @_;
746   my $schema = $self->schema;
747   my $balanced_handler = $self->schema->storage->balancer;
748
749   $schema->storage->read_handler($balanced_handler);
750 }
751
752 =head2 connected
753
754 Check that the master and at least one of the replicants is connected.
755
756 =cut
757
758 sub connected {
759   my $self = shift @_;
760   return
761     $self->master->connected &&
762     $self->pool->connected_replicants;
763 }
764
765 =head2 ensure_connected
766
767 Make sure all the storages are connected.
768
769 =cut
770
771 sub ensure_connected {
772   my $self = shift @_;
773   foreach my $source ($self->all_storages) {
774     $source->ensure_connected(@_);
775   }
776 }
777
778 =head2 limit_dialect
779
780 Set the limit_dialect for all existing storages
781
782 =cut
783
784 sub limit_dialect {
785   my $self = shift @_;
786   foreach my $source ($self->all_storages) {
787     $source->limit_dialect(@_);
788   }
789   return $self->master->limit_dialect;
790 }
791
792 =head2 quote_char
793
794 Set the quote_char for all existing storages
795
796 =cut
797
798 sub quote_char {
799   my $self = shift @_;
800   foreach my $source ($self->all_storages) {
801     $source->quote_char(@_);
802   }
803   return $self->master->quote_char;
804 }
805
806 =head2 name_sep
807
808 Set the name_sep for all existing storages
809
810 =cut
811
812 sub name_sep {
813   my $self = shift @_;
814   foreach my $source ($self->all_storages) {
815     $source->name_sep(@_);
816   }
817   return $self->master->name_sep;
818 }
819
820 =head2 set_schema
821
822 Set the schema object for all existing storages
823
824 =cut
825
826 sub set_schema {
827   my $self = shift @_;
828   foreach my $source ($self->all_storages) {
829     $source->set_schema(@_);
830   }
831 }
832
833 =head2 debug
834
835 set a debug flag across all storages
836
837 =cut
838
839 sub debug {
840   my $self = shift @_;
841   if(@_) {
842     foreach my $source ($self->all_storages) {
843       $source->debug(@_);
844     }
845   }
846   return $self->master->debug;
847 }
848
849 =head2 debugobj
850
851 set a debug object
852
853 =cut
854
855 sub debugobj {
856   my $self = shift @_;
857   return $self->master->debugobj(@_);
858 }
859
860 =head2 debugfh
861
862 set a debugfh object
863
864 =cut
865
866 sub debugfh {
867   my $self = shift @_;
868   return $self->master->debugfh(@_);
869 }
870
871 =head2 debugcb
872
873 set a debug callback
874
875 =cut
876
877 sub debugcb {
878   my $self = shift @_;
879   return $self->master->debugcb(@_);
880 }
881
882 =head2 disconnect
883
884 disconnect everything
885
886 =cut
887
888 sub disconnect {
889   my $self = shift @_;
890   foreach my $source ($self->all_storages) {
891     $source->disconnect(@_);
892   }
893 }
894
895 =head2 cursor_class
896
897 set cursor class on all storages, or return master's
898
899 =cut
900
901 sub cursor_class {
902   my ($self, $cursor_class) = @_;
903
904   if ($cursor_class) {
905     $_->cursor_class($cursor_class) for $self->all_storages;
906   }
907   $self->master->cursor_class;
908 }
909
910 =head2 cursor
911
912 set cursor class on all storages, or return master's, alias for L</cursor_class>
913 above.
914
915 =cut
916
917 sub cursor {
918   my ($self, $cursor_class) = @_;
919
920   if ($cursor_class) {
921     $_->cursor($cursor_class) for $self->all_storages;
922   }
923   $self->master->cursor;
924 }
925
926 =head2 unsafe
927
928 sets the L<DBIx::Class::Storage::DBI/unsafe> option on all storages or returns
929 master's current setting
930
931 =cut
932
933 sub unsafe {
934   my $self = shift;
935
936   if (@_) {
937     $_->unsafe(@_) for $self->all_storages;
938   }
939
940   return $self->master->unsafe;
941 }
942
943 =head2 disable_sth_caching
944
945 sets the L<DBIx::Class::Storage::DBI/disable_sth_caching> option on all storages
946 or returns master's current setting
947
948 =cut
949
950 sub disable_sth_caching {
951   my $self = shift;
952
953   if (@_) {
954     $_->disable_sth_caching(@_) for $self->all_storages;
955   }
956
957   return $self->master->disable_sth_caching;
958 }
959
960 =head2 lag_behind_master
961
962 returns the highest Replicant L<DBIx::Class::Storage::DBI/lag_behind_master>
963 setting
964
965 =cut
966
967 sub lag_behind_master {
968   my $self = shift;
969
970   return max map $_->lag_behind_master, $self->replicants;
971 }
972
973 =head2 is_replicating
974
975 returns true if all replicants return true for
976 L<DBIx::Class::Storage::DBI/is_replicating>
977
978 =cut
979
980 sub is_replicating {
981   my $self = shift;
982
983   return (grep $_->is_replicating, $self->replicants) == ($self->replicants);
984 }
985
986 =head2 connect_call_datetime_setup
987
988 calls L<DBIx::Class::Storage::DBI/connect_call_datetime_setup> for all storages
989
990 =cut
991
992 sub connect_call_datetime_setup {
993   my $self = shift;
994   $_->connect_call_datetime_setup for $self->all_storages;
995 }
996
997 sub _populate_dbh {
998   my $self = shift;
999   $_->_populate_dbh for $self->all_storages;
1000 }
1001
1002 sub _connect {
1003   my $self = shift;
1004   $_->_connect for $self->all_storages;
1005 }
1006
1007 sub _rebless {
1008   my $self = shift;
1009   $_->_rebless for $self->all_storages;
1010 }
1011
1012 sub _determine_driver {
1013   my $self = shift;
1014   $_->_determine_driver for $self->all_storages;
1015 }
1016
1017 sub _driver_determined {
1018   my $self = shift;
1019
1020   if (@_) {
1021     $_->_driver_determined(@_) for $self->all_storages;
1022   }
1023
1024   return $self->master->_driver_determined;
1025 }
1026
1027 sub _init {
1028   my $self = shift;
1029
1030   $_->_init for $self->all_storages;
1031 }
1032
1033 sub _run_connection_actions {
1034   my $self = shift;
1035
1036   $_->_run_connection_actions for $self->all_storages;
1037 }
1038
1039 sub _do_connection_actions {
1040   my $self = shift;
1041
1042   if (@_) {
1043     $_->_do_connection_actions(@_) for $self->all_storages;
1044   }
1045 }
1046
1047 sub connect_call_do_sql {
1048   my $self = shift;
1049   $_->connect_call_do_sql(@_) for $self->all_storages;
1050 }
1051
1052 sub disconnect_call_do_sql {
1053   my $self = shift;
1054   $_->disconnect_call_do_sql(@_) for $self->all_storages;
1055 }
1056
1057 sub _seems_connected {
1058   my $self = shift;
1059
1060   return min map $_->_seems_connected, $self->all_storages;
1061 }
1062
1063 sub _ping {
1064   my $self = shift;
1065
1066   return min map $_->_ping, $self->all_storages;
1067 }
1068
1069 # not using the normalized_version, because we want to preserve
1070 # version numbers much longer than the conventional xxx.yyyzzz
1071 my $numify_ver = sub {
1072   my $ver = shift;
1073   my @numparts = split /\D+/, $ver;
1074   my $format = '%d.' . (join '', ('%06d') x (@numparts - 1));
1075
1076   return sprintf $format, @numparts;
1077 };
1078 sub _server_info {
1079   my $self = shift;
1080
1081   if (not $self->_dbh_details->{info}) {
1082     $self->_dbh_details->{info} = (
1083       reduce { $a->[0] < $b->[0] ? $a : $b }
1084       map [ $numify_ver->($_->{dbms_version}), $_ ],
1085       map $_->_server_info, $self->all_storages
1086     )->[1];
1087   }
1088
1089   return $self->next::method;
1090 }
1091
1092 sub _get_server_version {
1093   my $self = shift;
1094
1095   return $self->_server_info->{dbms_version};
1096 }
1097
1098 =head1 GOTCHAS
1099
1100 Due to the fact that replicants can lag behind a master, you must take care to
1101 make sure you use one of the methods to force read queries to a master should
1102 you need realtime data integrity.  For example, if you insert a row, and then
1103 immediately re-read it from the database (say, by doing $row->discard_changes)
1104 or you insert a row and then immediately build a query that expects that row
1105 to be an item, you should force the master to handle reads.  Otherwise, due to
1106 the lag, there is no certainty your data will be in the expected state.
1107
1108 For data integrity, all transactions automatically use the master storage for
1109 all read and write queries.  Using a transaction is the preferred and recommended
1110 method to force the master to handle all read queries.
1111
1112 Otherwise, you can force a single query to use the master with the 'force_pool'
1113 attribute:
1114
1115   my $row = $resultset->search(undef, {force_pool=>'master'})->find($pk);
1116
1117 This attribute will safely be ignore by non replicated storages, so you can use
1118 the same code for both types of systems.
1119
1120 Lastly, you can use the L</execute_reliably> method, which works very much like
1121 a transaction.
1122
1123 For debugging, you can turn replication on/off with the methods L</set_reliable_storage>
1124 and L</set_balanced_storage>, however this operates at a global level and is not
1125 suitable if you have a shared Schema object being used by multiple processes,
1126 such as on a web application server.  You can get around this limitation by
1127 using the Schema clone method.
1128
1129   my $new_schema = $schema->clone;
1130   $new_schema->set_reliable_storage;
1131
1132   ## $new_schema will use only the Master storage for all reads/writes while
1133   ## the $schema object will use replicated storage.
1134
1135 =head1 AUTHOR
1136
1137   John Napiorkowski <john.napiorkowski@takkle.com>
1138
1139 Based on code originated by:
1140
1141   Norbert Csongrádi <bert@cpan.org>
1142   Peter Siklósi <einon@einon.hu>
1143
1144 =head1 LICENSE
1145
1146 You may distribute this code under the same terms as Perl itself.
1147
1148 =cut
1149
1150 __PACKAGE__->meta->make_immutable;
1151
1152 1;