3162d8112550ba1e7f4ee3697ef5da1e52ba6605
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Storage / DBI / Replicated.pm
1 package DBIx::Class::Storage::DBI::Replicated;
2
3 BEGIN {
4   use Carp::Clan qw/^DBIx::Class/;
5   use DBIx::Class;
6   croak('The following modules are required for Replication ' . DBIx::Class::Optional::Dependencies->req_missing_for ('replicated') )
7     unless DBIx::Class::Optional::Dependencies->req_ok_for ('replicated');
8 }
9
10 use Moose;
11 use DBIx::Class::Storage::DBI;
12 use DBIx::Class::Storage::DBI::Replicated::Pool;
13 use DBIx::Class::Storage::DBI::Replicated::Balancer;
14 use DBIx::Class::Storage::DBI::Replicated::Types qw/BalancerClassNamePart DBICSchema DBICStorageDBI/;
15 use MooseX::Types::Moose qw/ClassName HashRef Object/;
16 use Scalar::Util 'reftype';
17 use Hash::Merge;
18 use List::Util qw/min max reduce/;
19 use Try::Tiny;
20 use namespace::clean;
21
22 use namespace::clean -except => 'meta';
23
24 =head1 NAME
25
26 DBIx::Class::Storage::DBI::Replicated - BETA Replicated database support
27
28 =head1 SYNOPSIS
29
30 The Following example shows how to change an existing $schema to a replicated
31 storage type, add some replicated (read-only) databases, and perform reporting
32 tasks.
33
34 You should set the 'storage_type attribute to a replicated type.  You should
35 also define your arguments, such as which balancer you want and any arguments
36 that the Pool object should get.
37
38   my $schema = Schema::Class->clone;
39   $schema->storage_type( ['::DBI::Replicated', {balancer=>'::Random'}] );
40   $schema->connection(...);
41
42 Next, you need to add in the Replicants.  Basically this is an array of
43 arrayrefs, where each arrayref is database connect information.  Think of these
44 arguments as what you'd pass to the 'normal' $schema->connect method.
45
46   $schema->storage->connect_replicants(
47     [$dsn1, $user, $pass, \%opts],
48     [$dsn2, $user, $pass, \%opts],
49     [$dsn3, $user, $pass, \%opts],
50   );
51
52 Now, just use the $schema as you normally would.  Automatically all reads will
53 be delegated to the replicants, while writes to the master.
54
55   $schema->resultset('Source')->search({name=>'etc'});
56
57 You can force a given query to use a particular storage using the search
58 attribute 'force_pool'.  For example:
59
60   my $RS = $schema->resultset('Source')->search(undef, {force_pool=>'master'});
61
62 Now $RS will force everything (both reads and writes) to use whatever was setup
63 as the master storage.  'master' is hardcoded to always point to the Master,
64 but you can also use any Replicant name.  Please see:
65 L<DBIx::Class::Storage::DBI::Replicated::Pool> and the replicants attribute for more.
66
67 Also see transactions and L</execute_reliably> for alternative ways to
68 force read traffic to the master.  In general, you should wrap your statements
69 in a transaction when you are reading and writing to the same tables at the
70 same time, since your replicants will often lag a bit behind the master.
71
72 See L<DBIx::Class::Storage::DBI::Replicated::Instructions> for more help and
73 walkthroughs.
74
75 =head1 DESCRIPTION
76
77 Warning: This class is marked BETA.  This has been running a production
78 website using MySQL native replication as its backend and we have some decent
79 test coverage but the code hasn't yet been stressed by a variety of databases.
80 Individual DBs may have quirks we are not aware of.  Please use this in first
81 development and pass along your experiences/bug fixes.
82
83 This class implements replicated data store for DBI. Currently you can define
84 one master and numerous slave database connections. All write-type queries
85 (INSERT, UPDATE, DELETE and even LAST_INSERT_ID) are routed to master
86 database, all read-type queries (SELECTs) go to the slave database.
87
88 Basically, any method request that L<DBIx::Class::Storage::DBI> would normally
89 handle gets delegated to one of the two attributes: L</read_handler> or to
90 L</write_handler>.  Additionally, some methods need to be distributed
91 to all existing storages.  This way our storage class is a drop in replacement
92 for L<DBIx::Class::Storage::DBI>.
93
94 Read traffic is spread across the replicants (slaves) occurring to a user
95 selected algorithm.  The default algorithm is random weighted.
96
97 =head1 NOTES
98
99 The consistency between master and replicants is database specific.  The Pool
100 gives you a method to validate its replicants, removing and replacing them
101 when they fail/pass predefined criteria.  Please make careful use of the ways
102 to force a query to run against Master when needed.
103
104 =head1 REQUIREMENTS
105
106 Replicated Storage has additional requirements not currently part of
107 L<DBIx::Class>. See L<DBIx::Class::Optional::Dependencies> for more details.
108
109 =head1 ATTRIBUTES
110
111 This class defines the following attributes.
112
113 =head2 schema
114
115 The underlying L<DBIx::Class::Schema> object this storage is attaching
116
117 =cut
118
119 has 'schema' => (
120     is=>'rw',
121     isa=>DBICSchema,
122     weak_ref=>1,
123     required=>1,
124 );
125
126 =head2 pool_type
127
128 Contains the classname which will instantiate the L</pool> object.  Defaults
129 to: L<DBIx::Class::Storage::DBI::Replicated::Pool>.
130
131 =cut
132
133 has 'pool_type' => (
134   is=>'rw',
135   isa=>ClassName,
136   default=>'DBIx::Class::Storage::DBI::Replicated::Pool',
137   handles=>{
138     'create_pool' => 'new',
139   },
140 );
141
142 =head2 pool_args
143
144 Contains a hashref of initialized information to pass to the Balancer object.
145 See L<DBIx::Class::Storage::DBI::Replicated::Pool> for available arguments.
146
147 =cut
148
149 has 'pool_args' => (
150   is=>'rw',
151   isa=>HashRef,
152   lazy=>1,
153   default=>sub { {} },
154 );
155
156
157 =head2 balancer_type
158
159 The replication pool requires a balance class to provider the methods for
160 choose how to spread the query load across each replicant in the pool.
161
162 =cut
163
164 has 'balancer_type' => (
165   is=>'rw',
166   isa=>BalancerClassNamePart,
167   coerce=>1,
168   required=>1,
169   default=> 'DBIx::Class::Storage::DBI::Replicated::Balancer::First',
170   handles=>{
171     'create_balancer' => 'new',
172   },
173 );
174
175 =head2 balancer_args
176
177 Contains a hashref of initialized information to pass to the Balancer object.
178 See L<DBIx::Class::Storage::DBI::Replicated::Balancer> for available arguments.
179
180 =cut
181
182 has 'balancer_args' => (
183   is=>'rw',
184   isa=>HashRef,
185   lazy=>1,
186   required=>1,
187   default=>sub { {} },
188 );
189
190 =head2 pool
191
192 Is a L<DBIx::Class::Storage::DBI::Replicated::Pool> or derived class.  This is a
193 container class for one or more replicated databases.
194
195 =cut
196
197 has 'pool' => (
198   is=>'ro',
199   isa=>'DBIx::Class::Storage::DBI::Replicated::Pool',
200   lazy_build=>1,
201   handles=>[qw/
202     connect_replicants
203     replicants
204     has_replicants
205   /],
206 );
207
208 =head2 balancer
209
210 Is a L<DBIx::Class::Storage::DBI::Replicated::Balancer> or derived class.  This
211 is a class that takes a pool (L<DBIx::Class::Storage::DBI::Replicated::Pool>)
212
213 =cut
214
215 has 'balancer' => (
216   is=>'rw',
217   isa=>'DBIx::Class::Storage::DBI::Replicated::Balancer',
218   lazy_build=>1,
219   handles=>[qw/auto_validate_every/],
220 );
221
222 =head2 master
223
224 The master defines the canonical state for a pool of connected databases.  All
225 the replicants are expected to match this databases state.  Thus, in a classic
226 Master / Slaves distributed system, all the slaves are expected to replicate
227 the Master's state as quick as possible.  This is the only database in the
228 pool of databases that is allowed to handle write traffic.
229
230 =cut
231
232 has 'master' => (
233   is=> 'ro',
234   isa=>DBICStorageDBI,
235   lazy_build=>1,
236 );
237
238 =head1 ATTRIBUTES IMPLEMENTING THE DBIx::Storage::DBI INTERFACE
239
240 The following methods are delegated all the methods required for the
241 L<DBIx::Class::Storage::DBI> interface.
242
243 =cut
244
245 my $method_dispatch = {
246   writer => [qw/
247     on_connect_do
248     on_disconnect_do
249     on_connect_call
250     on_disconnect_call
251     connect_info
252     _connect_info
253     throw_exception
254     sql_maker
255     sqlt_type
256     create_ddl_dir
257     deployment_statements
258     datetime_parser
259     datetime_parser_type
260     build_datetime_parser
261     last_insert_id
262     insert
263     insert_bulk
264     update
265     delete
266     dbh
267     txn_begin
268     txn_do
269     txn_commit
270     txn_rollback
271     txn_scope_guard
272     sth
273     deploy
274     with_deferred_fk_checks
275     dbh_do
276     _prep_for_execute
277     is_datatype_numeric
278     _count_select
279     _subq_update_delete
280     svp_rollback
281     svp_begin
282     svp_release
283     relname_to_table_alias
284     _dbh_last_insert_id
285     _fix_bind_params
286     _default_dbi_connect_attributes
287     _dbi_connect_info
288     _dbic_connect_attributes
289     auto_savepoint
290     _query_end
291     bind_attribute_by_data_type
292     transaction_depth
293     _dbh
294     _select_args
295     _dbh_execute_array
296     _sql_maker
297     _query_start
298     _per_row_update_delete
299     _dbh_begin_work
300     _dbh_execute_inserts_with_no_binds
301     _select_args_to_query
302     _svp_generate_name
303     _multipk_update_delete
304     source_bind_attributes
305     _normalize_connect_info
306     _parse_connect_do
307     _dbh_commit
308     _execute_array
309     savepoints
310     _sql_maker_opts
311     _conn_pid
312     _dbh_autocommit
313     _native_data_type
314     _get_dbh
315     sql_maker_class
316     _dbh_rollback
317     _adjust_select_args_for_complex_prefetch
318     _resolve_ident_sources
319     _resolve_column_info
320     _prune_unused_joins
321     _strip_cond_qualifiers
322     _strip_cond_qualifiers_from_array
323     _resolve_aliastypes_from_select_args
324     _execute
325     _do_query
326     _dbh_sth
327     _dbh_execute
328   /],
329   reader => [qw/
330     select
331     select_single
332     columns_info_for
333     _dbh_columns_info_for
334     _select
335   /],
336   unimplemented => [qw/
337     _arm_global_destructor
338     _verify_pid
339
340     get_use_dbms_capability
341     set_use_dbms_capability
342     get_dbms_capability
343     set_dbms_capability
344     _dbh_details
345     _dbh_get_info
346
347     sql_limit_dialect
348     sql_quote_char
349     sql_name_sep
350
351     _inner_join_to_node
352     _group_over_selection
353     _extract_order_criteria
354
355     _prefetch_autovalues
356
357     _max_column_bytesize
358     _is_lob_type
359     _is_binary_lob_type
360     _is_text_lob_type
361   /,(
362     # the capability framework
363     # not sure if CMOP->initialize does evil things to DBIC::S::DBI, fix if a problem
364     grep
365       { $_ =~ /^ _ (?: use | supports | determine_supports ) _ /x }
366       ( Class::MOP::Class->initialize('DBIx::Class::Storage::DBI')->get_all_method_names )
367   )],
368 };
369
370 if (DBIx::Class::_ENV_::DBICTEST) {
371
372   my $seen;
373   for my $type (keys %$method_dispatch) {
374     for (@{$method_dispatch->{$type}}) {
375       push @{$seen->{$_}}, $type;
376     }
377   }
378
379   if (my @dupes = grep { @{$seen->{$_}} > 1 } keys %$seen) {
380     die(join "\n", '',
381       'The following methods show up multiple times in ::Storage::DBI::Replicated handlers:',
382       (map { "$_: " . (join ', ', @{$seen->{$_}}) } sort @dupes),
383       '',
384     );
385   }
386
387   if (my @cant = grep { ! DBIx::Class::Storage::DBI->can($_) } keys %$seen) {
388     die(join "\n", '',
389       '::Storage::DBI::Replicated specifies handling of the following *NON EXISTING* ::Storage::DBI methods:',
390       @cant,
391       '',
392     );
393   }
394 }
395
396 for my $method (@{$method_dispatch->{unimplemented}}) {
397   __PACKAGE__->meta->add_method($method, sub {
398     croak "$method must not be called on ".(blessed shift).' 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;