lib/A3C/SQL.pm

package A3C::SQL;

use strict;
use warnings;

use base qw(Jifty::Object Class::Accessor::Fast);
__PACKAGE__->mk_accessors( qw(query arguments dbh encoding) );

use Data::Dump qw/dump/;

=head1 NAME

A3C::SQL

=head1 DESCRIPTION

Issue SQL queries

=head1 METHODS

=head2 new

  my $sql = A3C::SQL->new({ query => 'select now()' });

As a alternative, if you don't want to use Jifty's database handle,
specify it like this:

  my $sql = A3C::SQL->new({
        query => 'select now()',
        dbh => $my_dbh,
        encoding => 'UTF-8',
  });

Mungling with C<encoding> is rearly needed, especially if using recent C<DBD::Pg> as driver.

=head2 sth

Execute query and return statement handle. Ususally you don't have to call this manually.

=cut

sub sth {
        my $self = shift;
        if ( ! $self->{_sth} ) {
                my $dbh = $self->dbh || Jifty->handle->dbh;
                my $sth = $dbh->prepare( $self->query ) or $dbh->errstr;
                if ( $self->arguments ) {
                        Jifty->log->debug( $self->sql . ' arguments: ' . dump( $self->arguments ) );
                        $sth->execute( $self->arguments ) or die $dbh->errstr;
                } else {
                        $sth->execute or die $dbh->errstr;
                }
                $self->{_sth} = $sth;
        }

        return $self->{_sth};
}

=head2 next

  while ( my $row = $sql->next ) {
        print $row->now;
  }

=cut

sub next {
        my $self = shift;
        my $row = $self->sth->fetchrow_hashref;
        return unless defined $row;
        return A3C::SQL::row->new( $row, $self->encoding );
}

=head2 count

  print $sql->count;

=cut

sub count {
        my $self = shift;
        return $self->sth->rows;
}

=head1 HELPERS

This helpers are accessor to L<DBI>

=head2 _column_names

  my @columns = $sql->_column_names;

=cut

sub _column_names {
        my $self = shift;
        return @{ $self->sth->{NAME} };
}

package A3C::SQL::row;

use Encode qw/decode/;
use Data::Dump qw/dump/;
use base qw/Jifty::Object/;

our $AUTOLOAD;

sub new {
        my $that  = shift;
        my $class = ref($that) || $that;
        my $self = shift;
        bless $self, $class;
        $self->{__encoding} = shift || 'UTF-8';
        return $self;
}  

sub AUTOLOAD {
        my $self = shift;
        my $type = ref($self) or die "$self is not an object";
        my $name = $AUTOLOAD;
        $name =~ s/.*://;
        my $v = $self->{$name};
        Jifty->log->error("SQL: $name doesn't exist") unless defined $v;
        if ( ! Encode::is_utf8( $v ) ) {
                eval { $v = decode( $self->{__encoding}, $self->{$name} ) };
                if ( $@ ) {
                        warn "## column $name can't decode ",dump( $self->{$name} );
                        $v = $self->{$name};
                }
        }
        return $v;
}

sub DESTROY {}

1;
1	dpavlin	78	package A3C::SQL;
2
3			use strict;
4			use warnings;
5
6	dpavlin	145	use base qw(Jifty::Object Class::Accessor::Fast);
7	dpavlin	162	__PACKAGE__->mk_accessors( qw(query arguments dbh encoding) );
8	dpavlin	145
9			use Data::Dump qw/dump/;
10
11	dpavlin	78	=head1 NAME
12
13			A3C::SQL
14
15			=head1 DESCRIPTION
16
17			Issue SQL queries
18
19			=head1 METHODS
20
21			=head2 new
22
23			my $sql = A3C::SQL->new({ query => 'select now()' });
24
25	dpavlin	145	As a alternative, if you don't want to use Jifty's database handle,
26			specify it like this:
27	dpavlin	78
28	dpavlin	145	my $sql = A3C::SQL->new({
29			query => 'select now()',
30			dbh => $my_dbh,
31	dpavlin	162	encoding => 'UTF-8',
32	dpavlin	145	});
33	dpavlin	78
34	dpavlin	218	Mungling with C<encoding> is rearly needed, especially if using recent C<DBD::Pg> as driver.
35
36	dpavlin	78	=head2 sth
37
38			Execute query and return statement handle. Ususally you don't have to call this manually.
39
40			=cut
41
42			sub sth {
43			my $self = shift;
44			if ( ! $self->{_sth} ) {
45	dpavlin	145	my $dbh = $self->dbh \|\| Jifty->handle->dbh;
46	dpavlin	78	my $sth = $dbh->prepare( $self->query ) or $dbh->errstr;
47			if ( $self->arguments ) {
48			Jifty->log->debug( $self->sql . ' arguments: ' . dump( $self->arguments ) );
49	dpavlin	160	$sth->execute( $self->arguments ) or die $dbh->errstr;
50	dpavlin	78	} else {
51	dpavlin	160	$sth->execute or die $dbh->errstr;
52	dpavlin	78	}
53			$self->{_sth} = $sth;
54			}
55
56			return $self->{_sth};
57			}
58
59			=head2 next
60
61			while ( my $row = $sql->next ) {
62			print $row->now;
63			}
64
65			=cut
66
67			sub next {
68			my $self = shift;
69			my $row = $self->sth->fetchrow_hashref;
70			return unless defined $row;
71	dpavlin	162	return A3C::SQL::row->new( $row, $self->encoding );
72	dpavlin	78	}
73
74			=head2 count
75
76			print $sql->count;
77
78			=cut
79
80			sub count {
81			my $self = shift;
82			return $self->sth->rows;
83			}
84
85	dpavlin	159	=head1 HELPERS
86
87			This helpers are accessor to L<DBI>
88
89			=head2 _column_names
90
91			my @columns = $sql->_column_names;
92
93			=cut
94
95			sub _column_names {
96			my $self = shift;
97			return @{ $self->sth->{NAME} };
98			}
99
100	dpavlin	78	package A3C::SQL::row;
101
102			use Encode qw/decode/;
103			use Data::Dump qw/dump/;
104	dpavlin	159	use base qw/Jifty::Object/;
105	dpavlin	78
106			our $AUTOLOAD;
107
108			sub new {
109			my $that = shift;
110			my $class = ref($that) \|\| $that;
111			my $self = shift;
112			bless $self, $class;
113	dpavlin	162	$self->{__encoding} = shift \|\| 'UTF-8';
114	dpavlin	78	return $self;
115			}
116
117			sub AUTOLOAD {
118			my $self = shift;
119			my $type = ref($self) or die "$self is not an object";
120			my $name = $AUTOLOAD;
121			$name =~ s/.*://;
122	dpavlin	218	my $v = $self->{$name};
123			Jifty->log->error("SQL: $name doesn't exist") unless defined $v;
124			if ( ! Encode::is_utf8( $v ) ) {
125			eval { $v = decode( $self->{__encoding}, $self->{$name} ) };
126			if ( $@ ) {
127			warn "## column $name can't decode ",dump( $self->{$name} );
128			$v = $self->{$name};
129			}
130			}
131			return $v;
132	dpavlin	78	}
133
134	dpavlin	82	sub DESTROY {}
135
136	dpavlin	78	1;