File Coverage

blib/lib/App/DBI/Loader.pm

Criterion	Covered	Total	%
statement	57	57	100.0
branch	16	24	66.6
condition	9	14	64.2
subroutine	8	8	100.0
pod	1	1	100.0
total	91	104	87.5

package App::DBI::Loader;

use strict;
use warnings;

# ABSTRACT: A tiny script to load CSV/TSV contents into a database table via DBI
our $VERSION = 'v0.0.2'; # VERSION

use Getopt::Std;
use Getopt::Config::FromPod;
use Pod::Usage;

use DBI;
use String::Unescape;

sub run
{
    shift if @_ && eval { $_[0]->isa(__PACKAGE__) };
    local (@ARGV) = @_;

    my %opts;
    getopts(Getopt::Config::FromPod->string, \%opts);
    pod2usage(-verbose => 2) if exists $opts{h};
    pod2usage(-msg => 'At least 2 arguments MUST be specified', -verbose => 0, -exitval => 1) if @ARGV < 2;
    push @ARGV, '-' if @ARGV == 2;

    $opts{t} ||= '';
    my $sep = String::Unescape->unescape($opts{t}) || ',';

    my $dbstr = shift @ARGV;
    my $table = shift @ARGV;

    my $dbh = DBI->connect($dbstr, $opts{u} || '', $opts{p} || '') or die;
    my $has_transaction = 1;
    eval { $dbh->{AutoCommit} = 0 };
    $has_transaction = 0 if $@;
    if($ARGV[0] =~ /\(.*\)/) {
        my $schema = shift @ARGV;
        $dbh->do("DROP TABLE IF EXISTS $table");
        $dbh->do("CREATE TABLE $table $schema");
    }
    if(exists $opts{c}) {
        $dbh->do("DELETE FROM $table");
    }
    my $sth;

    while(my $file = shift @ARGV) {
        my $fh;
        if($file eq '-') {
            $fh = \*STDIN;
        } else {
            open $fh, '<', $file or die;
        }
        while(<$fh>) {
            s/[\r\n]+$//;
            my (@t) = $sep ? split /$sep/ : $_;
            $sth ||= $dbh->prepare('INSERT INTO '.$table.' VALUES ('.join(',', ('?')x @t).')');
            $sth->execute(@t);
        }
        close $fh;
    }
    $dbh->commit if $has_transaction;
}

1;

__END__