File Coverage

blib/lib/DBIx/Replicate.pm

Criterion	Covered	Total	%
statement	24	52	46.1
branch	0	14	0.0
condition	0	15	0.0
subroutine	8	11	72.7
pod	1	3	33.3
total	33	95	34.7

# $Id: Replicate.pm 22870 2008-11-06 16:06:55Z kazuho $

package DBIx::Replicate;
use strict;
use warnings;
use Carp::Clan;
use DBI;
use DBIx::Replicate::Node;
use UNIVERSAL::require;
use Exporter 'import';
use base qw(Class::Accessor::Fast);

our %EXPORT_TAGS = (
    'all' => [ qw/dbix_replicate/ ],
);
our @EXPORT_OK = map { @{$EXPORT_TAGS{$_}} } qw/all/;
our $VERSION = '0.04';

__PACKAGE__->mk_accessors($_) for qw(src dest columns block extra_cond limit_cond strategy);

sub new
{
    my $class = shift;
    my $args  = shift || {};

    if (! $args->{strategy}) {
        $args->{strategy_class} ||= 'DBIx::Replicate::Strategy::PK';
    }

    if ( my $strategy_class = $args->{strategy_class}) {
        my $strategy_args = $args->{strategy_args} || {};
        $strategy_class->require or die;
        $args->{strategy} = $strategy_class->new($strategy_args);
    }

    foreach my $p (qw/src dest columns strategy/) {
        croak "required parameter $p is missing\n"
            unless $args->{$p};
    }
    $args->{block} ||= 1000;

    my $self  = $class->SUPER::new({
        strategy   => $args->{strategy},
        columns    => $args->{columns},
        block      => $args->{block},
        src        => $args->{src},
        dest       => $args->{dest},
        extra_cond => $args->{extra_cond},
        limit_cond => $args->{limit_cond},
    });

    return $self;
}

sub dbix_replicate {
    my $args = shift;

    $args = { %$args };

    foreach my $p (qw/src_table src_conn dest_table dest_conn columns/) {
        croak "required parameter $p is missing\n"
            unless $args->{$p};
    }

    my $src = DBIx::Replicate::Node->new( {
        table => delete $args->{src_table},
        conn  => delete $args->{src_conn},
    } );
    my $dest = DBIx::Replicate::Node->new( {
        table => delete $args->{dest_table},
        conn  => delete $args->{dest_conn}
    });

    if (! $args->{strategy} && ! $args->{strategy_class}) {
        if ($args->{copy_by}) {
            $args->{strategy_class} ||= 'DBIx::Replicate::Strategy::CopyBy';
        } else {
            $args->{strategy_class} ||= 'DBIx::Replicate::Strategy::PK';
        }
    }

    my %args = (
        src            => $src,
        dest           => $dest,
        columns        => delete $args->{columns},
        block          => delete $args->{block},
        extra_cond     => delete $args->{extra_cond},
        limit_cond     => delete $args->{limit_cond},
        strategy       => delete $args->{strategy},
        strategy_class => delete $args->{strategy_class},
        strategy_args  => delete $args->{strategy_args},
    );

    my $dr = DBIx::Replicate->new( \%args );
    $dr->replicate($args);
}

sub replicate
{
    my ($self, $args) = @_;

    $self->strategy->replicate( $self, $args );
}


1;
__END__