Frey/DBI/Sponge.pm

package Frey::DBI::Sponge;
use Moose;

extends 'Frey';

#use Frey::Types;

has sponge => (
        is => 'rw',
        isa => 'Sponge',
        required => 1,
);

has 'dsn' => (
        is => 'ro',
        isa => 'dsn',
        default => 'dbi:Pg:dbname=frey',
        required => 1,
);

sub to_database_as_markup {
        my ($self) = @_;

        my $table = $self->sponge->{table} || die "no table in sponge ", $self->dump( $self->sponge );

        warn "# dsn: ", $self->dsn;
        my $dbh = DBI->connect( $self->dsn, '', '', { RaiseError => 1 } ) || die $DBI::errstr;
        $dbh->do( qq{ set client_encoding='utf-8' } );

        my @columns = @{ $self->sponge->{NAME} };


        my $create_sql = qq{
                create table $table (
        } . join( ",\n", map { s/%//; "$_ text" } @columns ) . qq{
                );
        };
        eval { $dbh->do( $create_sql ); };
        if ( $@ ) {
                die "$create_sql\n$@" if $@ !~ m{$table.*already exists};
                warn "INFO: using existing table $table because of $@";
        } else {
                warn "INFO: created $table using\n$create_sql";
        }

        my $sth = $dbh->prepare( qq{
                insert into $table values (
        } . join( ",", map { '?' } @columns ) . qq{
                );
        });

        my $sponge_rows = $#{ $self->sponge->{rows} };
        warn "# convert ", $sponge_rows + 1 ," rows from spunge into $table in ", $self->dsn;

        foreach my $row_nr ( 0 .. $sponge_rows ) {
                $sth->execute( @{ $self->sponge->{rows}->[$row_nr] } );
        }

        $sponge_rows++;

        my $html = 
                        ( $sponge_rows
                                ? qq|Inserted <b>$sponge_rows</b> rows|
                                : qq|No rows inserted into $table|
                        )
                        . qq| to table <tt>$table</tt> at <tt>| . $self->dsn . qq|</tt>|
                        ;

        return $html;
}

1;
1	package Frey::DBI::Sponge;
2	use Moose;
3
4	extends 'Frey';
5
6	#use Frey::Types;
7
8	has sponge => (
9	is => 'rw',
10	isa => 'Sponge',
11	required => 1,
12	);
13
14	has 'dsn' => (
15	is => 'ro',
16	isa => 'dsn',
17	default => 'dbi:Pg:dbname=frey',
18	required => 1,
19	);
20
21	sub to_database_as_markup {
22	my ($self) = @_;
23
24	my $table = $self->sponge->{table} \|\| die "no table in sponge ", $self->dump( $self->sponge );
25
26	warn "# dsn: ", $self->dsn;
27	my $dbh = DBI->connect( $self->dsn, '', '', { RaiseError => 1 } ) \|\| die $DBI::errstr;
28	$dbh->do( qq{ set client_encoding='utf-8' } );
29
30	my @columns = @{ $self->sponge->{NAME} };
31
32
33	my $create_sql = qq{
34	create table $table (
35	} . join( ",\n", map { s/%//; "$_ text" } @columns ) . qq{
36	);
37	};
38	eval { $dbh->do( $create_sql ); };
39	if ( $@ ) {
40	die "$create_sql\n$@" if $@ !~ m{$table.*already exists};
41	warn "INFO: using existing table $table because of $@";
42	} else {
43	warn "INFO: created $table using\n$create_sql";
44	}
45
46	my $sth = $dbh->prepare( qq{
47	insert into $table values (
48	} . join( ",", map { '?' } @columns ) . qq{
49	);
50	});
51
52	my $sponge_rows = $#{ $self->sponge->{rows} };
53	warn "# convert ", $sponge_rows + 1 ," rows from spunge into $table in ", $self->dsn;
54
55	foreach my $row_nr ( 0 .. $sponge_rows ) {
56	$sth->execute( @{ $self->sponge->{rows}->[$row_nr] } );
57	}
58
59	$sponge_rows++;
60
61	my $html =
62	( $sponge_rows
63	? qq\|Inserted <b>$sponge_rows</b> rows\|
64	: qq\|No rows inserted into $table\|
65	)
66	. qq\| to table <tt>$table</tt> at <tt>\| . $self->dsn . qq\|</tt>\|
67	;
68
69	return $html;
70	}
71
72	1;