1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
package # hide from PAUSE
DBIx::Class::CDBICompat::ImaDBI;
use strict;
use warnings;
use DBIx::ContextualFetch;
use base qw/DBIx::Class/;
__PACKAGE__->mk_classdata('_transform_sql_handler_order'
=> [ qw/TABLE ESSENTIAL JOIN/ ] );
__PACKAGE__->mk_classdata('_transform_sql_handlers' =>
{
'TABLE' =>
sub {
my ($self, $class, $data) = @_;
return $class->result_source_instance->name unless $data;
my ($f_class, $alias) = split(/=/, $data);
$f_class ||= $class;
$self->{_classes}{$alias} = $f_class;
return $f_class->result_source_instance->name." ${alias}";
},
'ESSENTIAL' =>
sub {
my ($self, $class, $data) = @_;
return join(' ', $class->columns('Essential')) unless $data;
return join(' ', $self->{_classes}{$data}->columns('Essential'));
},
'JOIN' =>
sub {
my ($self, $class, $data) = @_;
my ($from, $to) = split(/ /, $data);
my ($from_class, $to_class) = @{$self->{_classes}}{$from, $to};
my ($rel_obj) = grep { $_->{class} && $_->{class} eq $to_class }
map { $from_class->relationship_info($_) }
$from_class->relationships;
unless ($rel_obj) {
($from, $to) = ($to, $from);
($from_class, $to_class) = ($to_class, $from_class);
($rel_obj) = grep { $_->{class} && $_->{class} eq $to_class }
map { $from_class->relationship_info($_) }
$from_class->relationships;
}
$self->throw_exception( "No relationship to JOIN from ${from_class} to ${to_class}" )
unless $rel_obj;
my $join = $from_class->storage->sql_maker->_join_condition(
$from_class->result_source_instance->resolve_condition(
$rel_obj->{cond}, $to, $from) );
return $join;
}
} );
sub db_Main {
return $_[0]->storage->dbh;
}
sub connection {
my ($class, @info) = @_;
$info[3] = { %{ $info[3] || {}} };
$info[3]->{RootClass} = 'DBIx::ContextualFetch';
return $class->next::method(@info);
}
sub __driver {
return $_[0]->storage->dbh->{Driver}->{Name};
}
sub set_sql {
my ($class, $name, $sql) = @_;
no strict 'refs';
*{"${class}::sql_${name}"} =
sub {
my $sql = $sql;
my $class = shift;
return $class->storage->sth($class->transform_sql($sql, @_));
};
if ($sql =~ /select/i) {
my $meth = "sql_${name}";
*{"${class}::search_${name}"} =
sub {
my ($class, @args) = @_;
my $sth = $class->$meth;
$sth->execute(@args);
return $class->sth_to_objects($sth);
};
}
}
sub sth_to_objects {
my ($class, $sth) = @_;
my @ret;
while (my $row = $sth->fetchrow_hashref) {
push(@ret, $class->inflate_result($class->result_source_instance, $row));
}
return @ret;
}
sub transform_sql {
my ($class, $sql, @args) = @_;
my $attrs = { };
foreach my $key (@{$class->_transform_sql_handler_order}) {
my $h = $class->_transform_sql_handlers->{$key};
$sql =~ s/__$key(?:\(([^\)]+)\))?__/$h->($attrs, $class, $1)/eg;
}
#warn $sql;
return sprintf($sql, @args);
}
package
DBIx::ContextualFetch::st; # HIDE FROM PAUSE THIS IS NOT OUR CLASS
no warnings 'redefine';
sub _untaint_execute {
my $sth = shift;
my $old_value = $sth->{Taint};
$sth->{Taint} = 0;
my $ret;
{
no warnings 'uninitialized';
$ret = $sth->SUPER::execute(@_);
}
$sth->{Taint} = $old_value;
return $ret;
}
1;
|
