Ditch Carp::Clan for our own thing
[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     sth
272     deploy
273     with_deferred_fk_checks
274     dbh_do
275     _prep_for_execute
276     is_datatype_numeric
277     _count_select
278     _subq_update_delete
279     svp_rollback
280     svp_begin
281     svp_release
282     relname_to_table_alias
283     _dbh_last_insert_id
284     _fix_bind_params
285     _default_dbi_connect_attributes
286     _dbi_connect_info
287     _dbic_connect_attributes
288     auto_savepoint
289     _query_end
290     bind_attribute_by_data_type
291     transaction_depth
292     _dbh
293     _select_args
294     _dbh_execute_array
295     _sql_maker
296     _query_start
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     source_bind_attributes
304     _normalize_connect_info
305     _parse_connect_do
306     _dbh_commit
307     _execute_array
308     savepoints
309     _sql_maker_opts
310     _conn_pid
311     _dbh_autocommit
312     _native_data_type
313     _get_dbh
314     sql_maker_class
315     _dbh_rollback
316     _adjust_select_args_for_complex_prefetch
317     _resolve_ident_sources
318     _resolve_column_info
319     _prune_unused_joins
320     _strip_cond_qualifiers
321     _strip_cond_qualifiers_from_array
322     _resolve_aliastypes_from_select_args
323     _execute
324     _do_query
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     get_use_dbms_capability
340     set_use_dbms_capability
341     get_dbms_capability
342     set_dbms_capability
343     _dbh_details
344     _dbh_get_info
345
346     sql_limit_dialect
347     sql_quote_char
348     sql_name_sep
349
350     _inner_join_to_node
351     _group_over_selection
352     _extract_order_criteria
353
354     _prefetch_autovalues
355
356     _max_column_bytesize
357     _is_lob_type
358     _is_binary_lob_type
359     _is_text_lob_type
360   /,(
361     # the capability framework
362     # not sure if CMOP->initialize does evil things to DBIC::S::DBI, fix if a problem
363     grep
364       { $_ =~ /^ _ (?: use | supports | determine_supports ) _ /x }
365       ( Class::MOP::Class->initialize('DBIx::Class::Storage::DBI')->get_all_method_names )
366   )],
367 };
368
369 if (DBIx::Class::_ENV_::DBICTEST) {
370
371   my $seen;
372   for my $type (keys %$method_dispatch) {
373     for (@{$method_dispatch->{$type}}) {
374       push @{$seen->{$_}}, $type;
375     }
376   }
377
378   if (my @dupes = grep { @{$seen->{$_}} > 1 } keys %$seen) {
379     die(join "\n", '',
380       'The following methods show up multiple times in ::Storage::DBI::Replicated handlers:',
381       (map { "$_: " . (join ', ', @{$seen->{$_}}) } sort @dupes),
382       '',
383     );
384   }
385
386   if (my @cant = grep { ! DBIx::Class::Storage::DBI->can($_) } keys %$seen) {
387     die(join "\n", '',
388       '::Storage::DBI::Replicated specifies handling of the following *NON EXISTING* ::Storage::DBI methods:',
389       @cant,
390       '',
391     );
392   }
393 }
394
395 for my $method (@{$method_dispatch->{unimplemented}}) {
396   __PACKAGE__->meta->add_method($method, sub {
397     my $self = shift;
398     $self->throw_exception("$method must not be called on ".(blessed $self).' objects');
399   });
400 }
401
402 =head2 read_handler
403
404 Defines an object that implements the read side of L<BIx::Class::Storage::DBI>.
405
406 =cut
407
408 has 'read_handler' => (
409   is=>'rw',
410   isa=>Object,
411   lazy_build=>1,
412   handles=>$method_dispatch->{reader},
413 );
414
415 =head2 write_handler
416
417 Defines an object that implements the write side of L<BIx::Class::Storage::DBI>,
418 as well as methods that don't write or read that can be called on only one
419 storage, methods that return a C<$dbh>, and any methods that don't make sense to
420 run on a replicant.
421
422 =cut
423
424 has 'write_handler' => (
425   is=>'ro',
426   isa=>Object,
427   lazy_build=>1,
428   handles=>$method_dispatch->{writer},
429 );
430
431
432
433 has _master_connect_info_opts =>
434   (is => 'rw', isa => HashRef, default => sub { {} });
435
436 =head2 around: connect_info
437
438 Preserves master's C<connect_info> options (for merging with replicants.)
439 Also sets any Replicated-related options from connect_info, such as
440 C<pool_type>, C<pool_args>, C<balancer_type> and C<balancer_args>.
441
442 =cut
443
444 around connect_info => sub {
445   my ($next, $self, $info, @extra) = @_;
446
447   my $merge = Hash::Merge->new('LEFT_PRECEDENT');
448
449   my %opts;
450   for my $arg (@$info) {
451     next unless (reftype($arg)||'') eq 'HASH';
452     %opts = %{ $merge->merge($arg, \%opts) };
453   }
454   delete $opts{dsn};
455
456   if (@opts{qw/pool_type pool_args/}) {
457     $self->pool_type(delete $opts{pool_type})
458       if $opts{pool_type};
459
460     $self->pool_args(
461       $merge->merge((delete $opts{pool_args} || {}), $self->pool_args)
462     );
463
464     ## Since we possibly changed the pool_args, we need to clear the current
465     ## pool object so that next time it is used it will be rebuilt.
466     $self->clear_pool;
467   }
468
469   if (@opts{qw/balancer_type balancer_args/}) {
470     $self->balancer_type(delete $opts{balancer_type})
471       if $opts{balancer_type};
472
473     $self->balancer_args(
474       $merge->merge((delete $opts{balancer_args} || {}), $self->balancer_args)
475     );
476
477     $self->balancer($self->_build_balancer)
478       if $self->balancer;
479   }
480
481   $self->_master_connect_info_opts(\%opts);
482
483   my @res;
484   if (wantarray) {
485     @res = $self->$next($info, @extra);
486   } else {
487     $res[0] = $self->$next($info, @extra);
488   }
489
490   # Make sure master is blessed into the correct class and apply role to it.
491   my $master = $self->master;
492   $master->_determine_driver;
493   Moose::Meta::Class->initialize(ref $master);
494
495   DBIx::Class::Storage::DBI::Replicated::WithDSN->meta->apply($master);
496
497   # link pool back to master
498   $self->pool->master($master);
499
500   wantarray ? @res : $res[0];
501 };
502
503 =head1 METHODS
504
505 This class defines the following methods.
506
507 =head2 BUILDARGS
508
509 L<DBIx::Class::Schema> when instantiating its storage passed itself as the
510 first argument.  So we need to massage the arguments a bit so that all the
511 bits get put into the correct places.
512
513 =cut
514
515 sub BUILDARGS {
516   my ($class, $schema, $storage_type_args, @args) = @_;
517
518   return {
519     schema=>$schema,
520     %$storage_type_args,
521     @args
522   }
523 }
524
525 =head2 _build_master
526
527 Lazy builder for the L</master> attribute.
528
529 =cut
530
531 sub _build_master {
532   my $self = shift @_;
533   my $master = DBIx::Class::Storage::DBI->new($self->schema);
534   $master
535 }
536
537 =head2 _build_pool
538
539 Lazy builder for the L</pool> attribute.
540
541 =cut
542
543 sub _build_pool {
544   my $self = shift @_;
545   $self->create_pool(%{$self->pool_args});
546 }
547
548 =head2 _build_balancer
549
550 Lazy builder for the L</balancer> attribute.  This takes a Pool object so that
551 the balancer knows which pool it's balancing.
552
553 =cut
554
555 sub _build_balancer {
556   my $self = shift @_;
557   $self->create_balancer(
558     pool=>$self->pool,
559     master=>$self->master,
560     %{$self->balancer_args},
561   );
562 }
563
564 =head2 _build_write_handler
565
566 Lazy builder for the L</write_handler> attribute.  The default is to set this to
567 the L</master>.
568
569 =cut
570
571 sub _build_write_handler {
572   return shift->master;
573 }
574
575 =head2 _build_read_handler
576
577 Lazy builder for the L</read_handler> attribute.  The default is to set this to
578 the L</balancer>.
579
580 =cut
581
582 sub _build_read_handler {
583   return shift->balancer;
584 }
585
586 =head2 around: connect_replicants
587
588 All calls to connect_replicants needs to have an existing $schema tacked onto
589 top of the args, since L<DBIx::Storage::DBI> needs it, and any C<connect_info>
590 options merged with the master, with replicant opts having higher priority.
591
592 =cut
593
594 around connect_replicants => sub {
595   my ($next, $self, @args) = @_;
596
597   for my $r (@args) {
598     $r = [ $r ] unless reftype $r eq 'ARRAY';
599
600     $self->throw_exception('coderef replicant connect_info not supported')
601       if ref $r->[0] && reftype $r->[0] eq 'CODE';
602
603 # any connect_info options?
604     my $i = 0;
605     $i++ while $i < @$r && (reftype($r->[$i])||'') ne 'HASH';
606
607 # make one if none
608     $r->[$i] = {} unless $r->[$i];
609
610 # merge if two hashes
611     my @hashes = @$r[$i .. $#{$r}];
612
613     $self->throw_exception('invalid connect_info options')
614       if (grep { reftype($_) eq 'HASH' } @hashes) != @hashes;
615
616     $self->throw_exception('too many hashrefs in connect_info')
617       if @hashes > 2;
618
619     my $merge = Hash::Merge->new('LEFT_PRECEDENT');
620     my %opts = %{ $merge->merge(reverse @hashes) };
621
622 # delete them
623     splice @$r, $i+1, ($#{$r} - $i), ();
624
625 # make sure master/replicants opts don't clash
626     my %master_opts = %{ $self->_master_connect_info_opts };
627     if (exists $opts{dbh_maker}) {
628         delete @master_opts{qw/dsn user password/};
629     }
630     delete $master_opts{dbh_maker};
631
632 # merge with master
633     %opts = %{ $merge->merge(\%opts, \%master_opts) };
634
635 # update
636     $r->[$i] = \%opts;
637   }
638
639   $self->$next($self->schema, @args);
640 };
641
642 =head2 all_storages
643
644 Returns an array of of all the connected storage backends.  The first element
645 in the returned array is the master, and the remainings are each of the
646 replicants.
647
648 =cut
649
650 sub all_storages {
651   my $self = shift @_;
652   return grep {defined $_ && blessed $_} (
653      $self->master,
654      values %{ $self->replicants },
655   );
656 }
657
658 =head2 execute_reliably ($coderef, ?@args)
659
660 Given a coderef, saves the current state of the L</read_handler>, forces it to
661 use reliable storage (e.g. sets it to the master), executes a coderef and then
662 restores the original state.
663
664 Example:
665
666   my $reliably = sub {
667     my $name = shift @_;
668     $schema->resultset('User')->create({name=>$name});
669     my $user_rs = $schema->resultset('User')->find({name=>$name});
670     return $user_rs;
671   };
672
673   my $user_rs = $schema->storage->execute_reliably($reliably, 'John');
674
675 Use this when you must be certain of your database state, such as when you just
676 inserted something and need to get a resultset including it, etc.
677
678 =cut
679
680 sub execute_reliably {
681   my ($self, $coderef, @args) = @_;
682
683   unless( ref $coderef eq 'CODE') {
684     $self->throw_exception('Second argument must be a coderef');
685   }
686
687   ##Get copy of master storage
688   my $master = $self->master;
689
690   ##Get whatever the current read hander is
691   my $current = $self->read_handler;
692
693   ##Set the read handler to master
694   $self->read_handler($master);
695
696   ## do whatever the caller needs
697   my @result;
698   my $want_array = wantarray;
699
700   try {
701     if($want_array) {
702       @result = $coderef->(@args);
703     } elsif(defined $want_array) {
704       ($result[0]) = ($coderef->(@args));
705     } else {
706       $coderef->(@args);
707     }
708   } catch {
709     $self->throw_exception("coderef returned an error: $_");
710   } finally {
711     ##Reset to the original state
712     $self->read_handler($current);
713   };
714
715   return wantarray ? @result : $result[0];
716 }
717
718 =head2 set_reliable_storage
719
720 Sets the current $schema to be 'reliable', that is all queries, both read and
721 write are sent to the master
722
723 =cut
724
725 sub set_reliable_storage {
726   my $self = shift @_;
727   my $schema = $self->schema;
728   my $write_handler = $self->schema->storage->write_handler;
729
730   $schema->storage->read_handler($write_handler);
731 }
732
733 =head2 set_balanced_storage
734
735 Sets the current $schema to be use the </balancer> for all reads, while all
736 writes are sent to the master only
737
738 =cut
739
740 sub set_balanced_storage {
741   my $self = shift @_;
742   my $schema = $self->schema;
743   my $balanced_handler = $self->schema->storage->balancer;
744
745   $schema->storage->read_handler($balanced_handler);
746 }
747
748 =head2 connected
749
750 Check that the master and at least one of the replicants is connected.
751
752 =cut
753
754 sub connected {
755   my $self = shift @_;
756   return
757     $self->master->connected &&
758     $self->pool->connected_replicants;
759 }
760
761 =head2 ensure_connected
762
763 Make sure all the storages are connected.
764
765 =cut
766
767 sub ensure_connected {
768   my $self = shift @_;
769   foreach my $source ($self->all_storages) {
770     $source->ensure_connected(@_);
771   }
772 }
773
774 =head2 limit_dialect
775
776 Set the limit_dialect for all existing storages
777
778 =cut
779
780 sub limit_dialect {
781   my $self = shift @_;
782   foreach my $source ($self->all_storages) {
783     $source->limit_dialect(@_);
784   }
785   return $self->master->limit_dialect;
786 }
787
788 =head2 quote_char
789
790 Set the quote_char for all existing storages
791
792 =cut
793
794 sub quote_char {
795   my $self = shift @_;
796   foreach my $source ($self->all_storages) {
797     $source->quote_char(@_);
798   }
799   return $self->master->quote_char;
800 }
801
802 =head2 name_sep
803
804 Set the name_sep for all existing storages
805
806 =cut
807
808 sub name_sep {
809   my $self = shift @_;
810   foreach my $source ($self->all_storages) {
811     $source->name_sep(@_);
812   }
813   return $self->master->name_sep;
814 }
815
816 =head2 set_schema
817
818 Set the schema object for all existing storages
819
820 =cut
821
822 sub set_schema {
823   my $self = shift @_;
824   foreach my $source ($self->all_storages) {
825     $source->set_schema(@_);
826   }
827 }
828
829 =head2 debug
830
831 set a debug flag across all storages
832
833 =cut
834
835 sub debug {
836   my $self = shift @_;
837   if(@_) {
838     foreach my $source ($self->all_storages) {
839       $source->debug(@_);
840     }
841   }
842   return $self->master->debug;
843 }
844
845 =head2 debugobj
846
847 set a debug object
848
849 =cut
850
851 sub debugobj {
852   my $self = shift @_;
853   return $self->master->debugobj(@_);
854 }
855
856 =head2 debugfh
857
858 set a debugfh object
859
860 =cut
861
862 sub debugfh {
863   my $self = shift @_;
864   return $self->master->debugfh(@_);
865 }
866
867 =head2 debugcb
868
869 set a debug callback
870
871 =cut
872
873 sub debugcb {
874   my $self = shift @_;
875   return $self->master->debugcb(@_);
876 }
877
878 =head2 disconnect
879
880 disconnect everything
881
882 =cut
883
884 sub disconnect {
885   my $self = shift @_;
886   foreach my $source ($self->all_storages) {
887     $source->disconnect(@_);
888   }
889 }
890
891 =head2 cursor_class
892
893 set cursor class on all storages, or return master's
894
895 =cut
896
897 sub cursor_class {
898   my ($self, $cursor_class) = @_;
899
900   if ($cursor_class) {
901     $_->cursor_class($cursor_class) for $self->all_storages;
902   }
903   $self->master->cursor_class;
904 }
905
906 =head2 cursor
907
908 set cursor class on all storages, or return master's, alias for L</cursor_class>
909 above.
910
911 =cut
912
913 sub cursor {
914   my ($self, $cursor_class) = @_;
915
916   if ($cursor_class) {
917     $_->cursor($cursor_class) for $self->all_storages;
918   }
919   $self->master->cursor;
920 }
921
922 =head2 unsafe
923
924 sets the L<DBIx::Class::Storage::DBI/unsafe> option on all storages or returns
925 master's current setting
926
927 =cut
928
929 sub unsafe {
930   my $self = shift;
931
932   if (@_) {
933     $_->unsafe(@_) for $self->all_storages;
934   }
935
936   return $self->master->unsafe;
937 }
938
939 =head2 disable_sth_caching
940
941 sets the L<DBIx::Class::Storage::DBI/disable_sth_caching> option on all storages
942 or returns master's current setting
943
944 =cut
945
946 sub disable_sth_caching {
947   my $self = shift;
948
949   if (@_) {
950     $_->disable_sth_caching(@_) for $self->all_storages;
951   }
952
953   return $self->master->disable_sth_caching;
954 }
955
956 =head2 lag_behind_master
957
958 returns the highest Replicant L<DBIx::Class::Storage::DBI/lag_behind_master>
959 setting
960
961 =cut
962
963 sub lag_behind_master {
964   my $self = shift;
965
966   return max map $_->lag_behind_master, $self->replicants;
967 }
968
969 =head2 is_replicating
970
971 returns true if all replicants return true for
972 L<DBIx::Class::Storage::DBI/is_replicating>
973
974 =cut
975
976 sub is_replicating {
977   my $self = shift;
978
979   return (grep $_->is_replicating, $self->replicants) == ($self->replicants);
980 }
981
982 =head2 connect_call_datetime_setup
983
984 calls L<DBIx::Class::Storage::DBI/connect_call_datetime_setup> for all storages
985
986 =cut
987
988 sub connect_call_datetime_setup {
989   my $self = shift;
990   $_->connect_call_datetime_setup for $self->all_storages;
991 }
992
993 sub _populate_dbh {
994   my $self = shift;
995   $_->_populate_dbh for $self->all_storages;
996 }
997
998 sub _connect {
999   my $self = shift;
1000   $_->_connect for $self->all_storages;
1001 }
1002
1003 sub _rebless {
1004   my $self = shift;
1005   $_->_rebless for $self->all_storages;
1006 }
1007
1008 sub _determine_driver {
1009   my $self = shift;
1010   $_->_determine_driver for $self->all_storages;
1011 }
1012
1013 sub _driver_determined {
1014   my $self = shift;
1015
1016   if (@_) {
1017     $_->_driver_determined(@_) for $self->all_storages;
1018   }
1019
1020   return $self->master->_driver_determined;
1021 }
1022
1023 sub _init {
1024   my $self = shift;
1025
1026   $_->_init for $self->all_storages;
1027 }
1028
1029 sub _run_connection_actions {
1030   my $self = shift;
1031
1032   $_->_run_connection_actions for $self->all_storages;
1033 }
1034
1035 sub _do_connection_actions {
1036   my $self = shift;
1037
1038   if (@_) {
1039     $_->_do_connection_actions(@_) for $self->all_storages;
1040   }
1041 }
1042
1043 sub connect_call_do_sql {
1044   my $self = shift;
1045   $_->connect_call_do_sql(@_) for $self->all_storages;
1046 }
1047
1048 sub disconnect_call_do_sql {
1049   my $self = shift;
1050   $_->disconnect_call_do_sql(@_) for $self->all_storages;
1051 }
1052
1053 sub _seems_connected {
1054   my $self = shift;
1055
1056   return min map $_->_seems_connected, $self->all_storages;
1057 }
1058
1059 sub _ping {
1060   my $self = shift;
1061
1062   return min map $_->_ping, $self->all_storages;
1063 }
1064
1065 # not using the normalized_version, because we want to preserve
1066 # version numbers much longer than the conventional xxx.yyyzzz
1067 my $numify_ver = sub {
1068   my $ver = shift;
1069   my @numparts = split /\D+/, $ver;
1070   my $format = '%d.' . (join '', ('%06d') x (@numparts - 1));
1071
1072   return sprintf $format, @numparts;
1073 };
1074 sub _server_info {
1075   my $self = shift;
1076
1077   if (not $self->_dbh_details->{info}) {
1078     $self->_dbh_details->{info} = (
1079       reduce { $a->[0] < $b->[0] ? $a : $b }
1080       map [ $numify_ver->($_->{dbms_version}), $_ ],
1081       map $_->_server_info, $self->all_storages
1082     )->[1];
1083   }
1084
1085   return $self->next::method;
1086 }
1087
1088 sub _get_server_version {
1089   my $self = shift;
1090
1091   return $self->_server_info->{dbms_version};
1092 }
1093
1094 =head1 GOTCHAS
1095
1096 Due to the fact that replicants can lag behind a master, you must take care to
1097 make sure you use one of the methods to force read queries to a master should
1098 you need realtime data integrity.  For example, if you insert a row, and then
1099 immediately re-read it from the database (say, by doing $row->discard_changes)
1100 or you insert a row and then immediately build a query that expects that row
1101 to be an item, you should force the master to handle reads.  Otherwise, due to
1102 the lag, there is no certainty your data will be in the expected state.
1103
1104 For data integrity, all transactions automatically use the master storage for
1105 all read and write queries.  Using a transaction is the preferred and recommended
1106 method to force the master to handle all read queries.
1107
1108 Otherwise, you can force a single query to use the master with the 'force_pool'
1109 attribute:
1110
1111   my $row = $resultset->search(undef, {force_pool=>'master'})->find($pk);
1112
1113 This attribute will safely be ignore by non replicated storages, so you can use
1114 the same code for both types of systems.
1115
1116 Lastly, you can use the L</execute_reliably> method, which works very much like
1117 a transaction.
1118
1119 For debugging, you can turn replication on/off with the methods L</set_reliable_storage>
1120 and L</set_balanced_storage>, however this operates at a global level and is not
1121 suitable if you have a shared Schema object being used by multiple processes,
1122 such as on a web application server.  You can get around this limitation by
1123 using the Schema clone method.
1124
1125   my $new_schema = $schema->clone;
1126   $new_schema->set_reliable_storage;
1127
1128   ## $new_schema will use only the Master storage for all reads/writes while
1129   ## the $schema object will use replicated storage.
1130
1131 =head1 AUTHOR
1132
1133   John Napiorkowski <john.napiorkowski@takkle.com>
1134
1135 Based on code originated by:
1136
1137   Norbert Csongrádi <bert@cpan.org>
1138   Peter Siklósi <einon@einon.hu>
1139
1140 =head1 LICENSE
1141
1142 You may distribute this code under the same terms as Perl itself.
1143
1144 =cut
1145
1146 __PACKAGE__->meta->make_immutable;
1147
1148 1;