43a5cca8cbbe2d7308289424f7946fb526f5c6f4
[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 If you have a multi-statement read only transaction you can force it to select
72 a random server in the pool by:
73
74   my $rs = $schema->resultset('Source')->search( undef,
75     { force_pool => $db->storage->read_handler->next_storage }
76   );
77
78 =head1 DESCRIPTION
79
80 Warning: This class is marked BETA.  This has been running a production
81 website using MySQL native replication as its backend and we have some decent
82 test coverage but the code hasn't yet been stressed by a variety of databases.
83 Individual DBs may have quirks we are not aware of.  Please use this in first
84 development and pass along your experiences/bug fixes.
85
86 This class implements replicated data store for DBI. Currently you can define
87 one master and numerous slave database connections. All write-type queries
88 (INSERT, UPDATE, DELETE and even LAST_INSERT_ID) are routed to master
89 database, all read-type queries (SELECTs) go to the slave database.
90
91 Basically, any method request that L<DBIx::Class::Storage::DBI> would normally
92 handle gets delegated to one of the two attributes: L</read_handler> or to
93 L</write_handler>.  Additionally, some methods need to be distributed
94 to all existing storages.  This way our storage class is a drop in replacement
95 for L<DBIx::Class::Storage::DBI>.
96
97 Read traffic is spread across the replicants (slaves) occurring to a user
98 selected algorithm.  The default algorithm is random weighted.
99
100 =head1 NOTES
101
102 The consistency between master and replicants is database specific.  The Pool
103 gives you a method to validate its replicants, removing and replacing them
104 when they fail/pass predefined criteria.  Please make careful use of the ways
105 to force a query to run against Master when needed.
106
107 =head1 REQUIREMENTS
108
109 Replicated Storage has additional requirements not currently part of
110 L<DBIx::Class>. See L<DBIx::Class::Optional::Dependencies> for more details.
111
112 =head1 ATTRIBUTES
113
114 This class defines the following attributes.
115
116 =head2 schema
117
118 The underlying L<DBIx::Class::Schema> object this storage is attaching
119
120 =cut
121
122 has 'schema' => (
123     is=>'rw',
124     isa=>DBICSchema,
125     weak_ref=>1,
126     required=>1,
127 );
128
129 =head2 pool_type
130
131 Contains the classname which will instantiate the L</pool> object.  Defaults
132 to: L<DBIx::Class::Storage::DBI::Replicated::Pool>.
133
134 =cut
135
136 has 'pool_type' => (
137   is=>'rw',
138   isa=>ClassName,
139   default=>'DBIx::Class::Storage::DBI::Replicated::Pool',
140   handles=>{
141     'create_pool' => 'new',
142   },
143 );
144
145 =head2 pool_args
146
147 Contains a hashref of initialized information to pass to the Balancer object.
148 See L<DBIx::Class::Storage::DBI::Replicated::Pool> for available arguments.
149
150 =cut
151
152 has 'pool_args' => (
153   is=>'rw',
154   isa=>HashRef,
155   lazy=>1,
156   default=>sub { {} },
157 );
158
159
160 =head2 balancer_type
161
162 The replication pool requires a balance class to provider the methods for
163 choose how to spread the query load across each replicant in the pool.
164
165 =cut
166
167 has 'balancer_type' => (
168   is=>'rw',
169   isa=>BalancerClassNamePart,
170   coerce=>1,
171   required=>1,
172   default=> 'DBIx::Class::Storage::DBI::Replicated::Balancer::First',
173   handles=>{
174     'create_balancer' => 'new',
175   },
176 );
177
178 =head2 balancer_args
179
180 Contains a hashref of initialized information to pass to the Balancer object.
181 See L<DBIx::Class::Storage::DBI::Replicated::Balancer> for available arguments.
182
183 =cut
184
185 has 'balancer_args' => (
186   is=>'rw',
187   isa=>HashRef,
188   lazy=>1,
189   required=>1,
190   default=>sub { {} },
191 );
192
193 =head2 pool
194
195 Is a L<DBIx::Class::Storage::DBI::Replicated::Pool> or derived class.  This is a
196 container class for one or more replicated databases.
197
198 =cut
199
200 has 'pool' => (
201   is=>'ro',
202   isa=>'DBIx::Class::Storage::DBI::Replicated::Pool',
203   lazy_build=>1,
204   handles=>[qw/
205     connect_replicants
206     replicants
207     has_replicants
208   /],
209 );
210
211 =head2 balancer
212
213 Is a L<DBIx::Class::Storage::DBI::Replicated::Balancer> or derived class.  This
214 is a class that takes a pool (L<DBIx::Class::Storage::DBI::Replicated::Pool>)
215
216 =cut
217
218 has 'balancer' => (
219   is=>'rw',
220   isa=>'DBIx::Class::Storage::DBI::Replicated::Balancer',
221   lazy_build=>1,
222   handles=>[qw/auto_validate_every/],
223 );
224
225 =head2 master
226
227 The master defines the canonical state for a pool of connected databases.  All
228 the replicants are expected to match this databases state.  Thus, in a classic
229 Master / Slaves distributed system, all the slaves are expected to replicate
230 the Master's state as quick as possible.  This is the only database in the
231 pool of databases that is allowed to handle write traffic.
232
233 =cut
234
235 has 'master' => (
236   is=> 'ro',
237   isa=>DBICStorageDBI,
238   lazy_build=>1,
239 );
240
241 =head1 ATTRIBUTES IMPLEMENTING THE DBIx::Storage::DBI INTERFACE
242
243 The following methods are delegated all the methods required for the
244 L<DBIx::Class::Storage::DBI> interface.
245
246 =cut
247
248 my $method_dispatch = {
249   writer => [qw/
250     on_connect_do
251     on_disconnect_do
252     on_connect_call
253     on_disconnect_call
254     connect_info
255     _connect_info
256     throw_exception
257     sql_maker
258     sqlt_type
259     create_ddl_dir
260     deployment_statements
261     datetime_parser
262     datetime_parser_type
263     build_datetime_parser
264     last_insert_id
265     insert
266     insert_bulk
267     update
268     delete
269     dbh
270     txn_begin
271     txn_do
272     txn_commit
273     txn_rollback
274     txn_scope_guard
275     _exec_txn_rollback
276     _exec_txn_begin
277     _exec_txn_commit
278     deploy
279     with_deferred_fk_checks
280     dbh_do
281     _prep_for_execute
282     is_datatype_numeric
283     _count_select
284     _subq_update_delete
285     svp_rollback
286     svp_begin
287     svp_release
288     relname_to_table_alias
289     _dbh_last_insert_id
290     _default_dbi_connect_attributes
291     _dbi_connect_info
292     _dbic_connect_attributes
293     auto_savepoint
294     _query_start
295     _query_end
296     _format_for_trace
297     _dbi_attrs_for_bind
298     bind_attribute_by_data_type
299     transaction_depth
300     _dbh
301     _select_args
302     _dbh_execute_array
303     _sql_maker
304     _per_row_update_delete
305     _dbh_execute_inserts_with_no_binds
306     _select_args_to_query
307     _gen_sql_bind
308     _svp_generate_name
309     _multipk_update_delete
310     _normalize_connect_info
311     _parse_connect_do
312     _execute_array
313     savepoints
314     _sql_maker_opts
315     _conn_pid
316     _dbh_autocommit
317     _native_data_type
318     _get_dbh
319     sql_maker_class
320     _execute
321     _do_query
322     _sth
323     _dbh_sth
324     _dbh_execute
325   /, Class::MOP::Class->initialize('DBIx::Class::Storage::DBIHacks')->get_method_list ],
326   reader => [qw/
327     select
328     select_single
329     columns_info_for
330     _dbh_columns_info_for
331     _select
332   /],
333   unimplemented => [qw/
334     _arm_global_destructor
335     _verify_pid
336
337     source_bind_attributes
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     _prefetch_autovalues
351
352     _resolve_bindattrs
353
354     _max_column_bytesize
355     _is_lob_type
356     _is_binary_lob_type
357     _is_text_lob_type
358
359     sth
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;