1 package DBIx::Class::Table;
6 use base qw/Class::Data::Inheritable DBIx::Class::SQL/;
8 __PACKAGE__->mk_classdata('_columns' => {});
10 __PACKAGE__->mk_classdata('_table_name');
13 my ($class, $attrs) = @_;
14 $class = ref $class if ref $class;
15 my $new = bless({ _column_data => { } }, $class);
17 die "attrs must be a hashref" unless ref($attrs) eq 'HASH';
18 while (my ($k, $v) = each %{$attrs}) {
19 $new->store_column($k => $v);
27 return if $self->{_in_database};
28 my $sth = $self->_get_sth('insert', [ keys %{$self->{_column_data}} ],
29 $self->_table_name, undef);
30 $sth->execute(values %{$self->{_column_data}});
31 $self->{_in_database} = 1;
32 $self->{_dirty_columns} = {};
37 my ($class, $attrs) = @_;
38 die "create needs a hashref" unless ref $attrs eq 'HASH';
39 return $class->new($attrs)->insert;
44 die "Not in database" unless $self->{_in_database};
45 my @to_update = keys %{$self->{_dirty_columns} || {}};
46 return -1 unless @to_update;
47 my $sth = $self->_get_sth('update', \@to_update,
48 $self->_table_name, $self->_ident_cond);
49 my $rows = $sth->execute( (map { $self->{_column_data}{$_} } @to_update),
50 $self->_ident_values );
52 die "Can't update $self: row not found";
54 die "Can't update $self: updated more than one row";
56 $self->{_dirty_columns} = {};
63 #warn $self->_ident_cond.' '.join(', ', $self->_ident_values);
64 my $sth = $self->_get_sth('delete', undef,
65 $self->_table_name, $self->_ident_cond);
66 $sth->execute($self->_ident_values);
68 delete $self->{_in_database};
71 if (@_ > 1 && ref $_[$#_] eq 'HASH') {
72 $attrs = { %{ pop(@_) } };
74 my $query = (ref $_[0] eq 'HASH' ? $_[0] : {@_});
75 my ($cond, @param) = $self->_cond_resolve($query, $attrs);
76 my $sth = $self->_get_sth('delete', undef, $self->_table_name, $cond);
77 $sth->execute(@param);
84 my ($self, $column) = @_;
85 die "Can't fetch data as class method" unless ref $self;
86 die "No such column '${column}'" unless $self->_columns->{$column};
87 return $self->{_column_data}{$column} if $self->_columns->{$column};
93 my $ret = $self->store_column(@_);
94 $self->{_dirty_columns}{$column} = 1;
99 my ($self, $column, $value) = @_;
100 die "No such column '${column}'" unless $self->_columns->{$column};
101 die "set_column called for ${column} without value" if @_ < 3;
102 return $self->{_column_data}{$column} = $value;
105 sub _register_columns {
106 my ($class, @cols) = @_;
107 my $names = { %{$class->_columns} };
108 $names->{$_} ||= {} for @cols;
109 $class->_columns($names);
112 sub _mk_column_accessors {
113 my ($class, @cols) = @_;
114 $class->mk_group_accessors('column' => @cols);
118 my ($class, @cols) = @_;
119 $class->_register_columns(@cols);
120 $class->_mk_column_accessors(@cols);
123 sub retrieve_from_sql {
124 my ($class, $cond, @vals) = @_;
125 $cond =~ s/^\s*WHERE//i;
126 my @cols = $class->_select_columns;
127 my $sth = $class->_get_sth( 'select', \@cols, $class->_table_name, $cond);
129 return $class->sth_to_objects($sth, \@vals, \@cols);
133 my ($class, $sth, $args, $cols) = @_;
134 my @cols = ((ref $cols eq 'ARRAY') ? @$cols : @{$sth->{NAME_lc}} );
135 $sth->execute(@$args);
137 while (my @row = $sth->fetchrow_array) {
138 my $new = $class->new;
139 $new->store_column($_, shift @row) for @cols;
140 $new->{_in_database} = 1;
149 if (@_ > 1 && ref $_[$#_] eq 'HASH') {
150 $attrs = { %{ pop(@_) } };
152 my $query = ref $_[0] eq "HASH" ? shift: {@_};
153 my ($cond, @param) = $class->_cond_resolve($query, $attrs);
154 return $class->retrieve_from_sql($cond, @param);
160 if (@_ > 1 && ref $_[$#_] eq 'HASH') {
163 return $class->search(@_, { %$attrs, cmp => 'LIKE' });
166 sub _select_columns {
167 return keys %{$_[0]->_columns};
171 my ($self, $changes) = @_;
172 my $new = bless({ _column_data => { %{$self->{_column_data}}} }, ref $self);
173 $new->set_column($_ => $changes->{$_}) for keys %$changes;
178 my ($self, $query, $attrs) = @_;
179 my $op = $attrs->{'cmp'} || '=';
180 my $cond = join(' AND ',
181 map { (defined $query->{$_}
183 : (do { delete $query->{$_}; "$_ IS NULL"; }));
185 return ($cond, values %$query);
189 shift->_table_name(@_);