lib/DBIx/Class/SQLMaker/DateOps.pm

   1 package DBIx::Class::SQLMaker::DateOps;
   2
   3 use base qw/
   4   Class::Accessor::Grouped
   5 /;
   6 __PACKAGE__->mk_group_accessors (simple => qw/datetime_parser/);
   7 use Carp::Clan qw/^DBIx::Class|^SQL::Abstract|^Try::Tiny/;
   8 use Sub::Name 'subname';
   9
  10 sub _where_op_CONVERT_DATETIME {
  11   my $self = shift;
  12   my ($op, $rhs) = splice @_, -2;
  13   croak "-$op takes a DateTime only" unless ref $rhs  && $rhs->isa('DateTime');
  14
  15   # in case we are called as a top level special op (no '=')
  16   my $lhs = shift;
  17
  18   $rhs = $self->datetime_parser->format_datetime($rhs);
  19
  20   my @bind = [
  21     ($lhs || $self->{_nested_func_lhs} || undef),
  22     $rhs
  23   ];
  24
  25   return $lhs
  26     ? (
  27       $self->_convert($self->_quote($lhs)) . ' = ' . $self->_convert('?'),
  28       @bind
  29     )
  30     : (
  31       $self->_convert('?'),
  32       @bind
  33     )
  34   ;
  35 }
  36
  37 sub _unsupported_date_extraction {
  38    "date part extraction not supported for part \"$_[1]\" with database \"$_[2]\""
  39 }
  40
  41 sub _unsupported_date_adding {
  42    "date part adding not supported for part \"$_[1]\" with database \"$_[2]\""
  43 }
  44
  45 sub _unsupported_date_diff {
  46    "date diff not supported for part \"$_[1]\" with database \"$_[2]\""
  47 }
  48
  49 sub _datetime_sql { die 'date part extraction not implemented for this database' }
  50
  51 sub _datetime_diff_sql { die 'date diffing not implemented for this database' }
  52 sub _datetime_add_sql { die 'date adding not implemented for this database' }
  53
  54 sub _where_op_GET_DATETIME {
  55   my ($self) = @_;
  56
  57   my ($k, $op, $vals);
  58
  59   if (@_ == 3) {
  60      $op = $_[1];
  61      $vals = $_[2];
  62      $k = '';
  63   } elsif (@_ == 4) {
  64      $k = $_[1];
  65      $op = $_[2];
  66      $vals = $_[3];
  67   }
  68
  69   croak 'args to -dt_get must be an arrayref' unless ref $vals eq 'ARRAY';
  70   croak 'first arg to -dt_get must be a scalar' unless !ref $vals->[0];
  71
  72   my $part = $vals->[0];
  73   my $val  = $vals->[1];
  74
  75   my ($sql, @bind) = $self->_SWITCH_refkind($val, {
  76      SCALAR => sub {
  77        return ($self->_convert('?'), $self->_bindtype($k, $val) );
  78      },
  79      SCALARREF => sub {
  80        return $$val;
  81      },
  82      ARRAYREFREF => sub {
  83        my ($sql, @bind) = @$$val;
  84        $self->_assert_bindval_matches_bindtype(@bind);
  85        return ($sql, @bind);
  86      },
  87      HASHREF => sub {
  88        my $method = $self->_METHOD_FOR_refkind("_where_hashpair", $val);
  89        $self->$method('', $val);
  90      }
  91   });
  92
  93   return $self->_datetime_sql($part, $sql), @bind;
  94 }
  95
  96 for my $part (qw(month day year hour minute second)) {
  97    no strict 'refs';
  98    my $name = '_where_op_GET_DATETIME_' . uc($part);
  99    *{$name} = subname "DBIx::Class::SQLMaker::DateOps::$name", sub {
 100      my $self = shift;
 101      my ($op, $rhs) = splice @_, -2;
 102
 103      my $lhs = shift;
 104
 105      return $self->_where_op_GET_DATETIME($op, $lhs, [$part, $rhs])
 106    }
 107 }
 108
 109 sub _where_op_DATETIME_NOW {
 110   my ($self) = @_;
 111
 112   my ($k, $op, $vals);
 113
 114   if (@_ == 3) {
 115      $op = $_[1];
 116      $vals = $_[2];
 117      $k = '';
 118   } elsif (@_ == 4) {
 119      $k = $_[1];
 120      $op = $_[2];
 121      $vals = $_[3];
 122   }
 123
 124   croak "args to -$op must be an arrayref" unless ref $vals eq 'ARRAY';
 125   if (!exists $vals->[0]) {
 126      return $self->_datetime_now_sql()
 127   } elsif ($vals->[0] eq 'system') {
 128      require DateTime;
 129      return $self->_where_op_CONVERT_DATETIME('dt', DateTime->now);
 130   } else {
 131      croak "first arg to -$op must be a 'system' or non-existant"
 132   }
 133 }
 134
 135 sub _reorder_add_datetime_vars {
 136    my ($self, $amount, $date) = @_;
 137
 138    return ($amount, $date);
 139 }
 140
 141 sub _where_op_ADD_DATETIME {
 142   my ($self) = @_;
 143
 144   my ($k, $op, $vals);
 145
 146   if (@_ == 3) {
 147      $op = $_[1];
 148      $vals = $_[2];
 149      $k = '';
 150   } elsif (@_ == 4) {
 151      $k = $_[1];
 152      $op = $_[2];
 153      $vals = $_[3];
 154   }
 155
 156   croak "args to -$op must be an arrayref" unless ref $vals eq 'ARRAY';
 157   croak "first arg to -$op must be a scalar" unless !ref $vals->[0];
 158   croak "-$op must have two more arguments" unless scalar @$vals == 3;
 159
 160   my ($part, @rest) = @$vals;
 161
 162   my $placeholder = $self->_convert('?');
 163
 164   my (@all_sql, @all_bind);
 165   foreach my $val ($self->_reorder_add_datetime_vars(@rest)) {
 166     my ($sql, @bind) = $self->_SWITCH_refkind($val, {
 167        SCALAR => sub {
 168          return ($placeholder, $self->_bindtype($k, $val) );
 169        },
 170        SCALARREF => sub {
 171          return $$val;
 172        },
 173        ARRAYREFREF => sub {
 174          my ($sql, @bind) = @$$val;
 175          $self->_assert_bindval_matches_bindtype(@bind);
 176          return ($sql, @bind);
 177        },
 178        HASHREF => sub {
 179          my $method = $self->_METHOD_FOR_refkind("_where_hashpair", $val);
 180          $self->$method('', $val);
 181        }
 182     });
 183     push @all_sql, $sql;
 184     push @all_bind, @bind;
 185   }
 186
 187   return $self->_datetime_add_sql($part, $all_sql[0], $all_sql[1]), @all_bind
 188 }
 189
 190 sub _where_op_DIFF_DATETIME {
 191   my ($self) = @_;
 192
 193   my ($k, $op, $vals);
 194
 195   if (@_ == 3) {
 196      $op = $_[1];
 197      $vals = $_[2];
 198      $k = '';
 199   } elsif (@_ == 4) {
 200      $k = $_[1];
 201      $op = $_[2];
 202      $vals = $_[3];
 203   }
 204
 205   croak 'args to -dt_diff must be an arrayref' unless ref $vals eq 'ARRAY';
 206   croak 'first arg to -dt_diff must be a scalar' unless !ref $vals->[0];
 207   croak '-dt_diff must have two more arguments' unless scalar @$vals == 3;
 208
 209   my ($part, @val) = @$vals;
 210   my $placeholder = $self->_convert('?');
 211
 212   my (@all_sql, @all_bind);
 213   foreach my $val (@val) {
 214     my ($sql, @bind) = $self->_SWITCH_refkind($val, {
 215        SCALAR => sub {
 216          return ($placeholder, $self->_bindtype($k, $val) );
 217        },
 218        SCALARREF => sub {
 219          return $$val;
 220        },
 221        ARRAYREFREF => sub {
 222          my ($sql, @bind) = @$$val;
 223          $self->_assert_bindval_matches_bindtype(@bind);
 224          return ($sql, @bind);
 225        },
 226        HASHREF => sub {
 227          my $method = $self->_METHOD_FOR_refkind("_where_hashpair", $val);
 228          $self->$method('', $val);
 229        }
 230     });
 231     push @all_sql, $sql;
 232     push @all_bind, @bind;
 233   }
 234
 235   return $self->_datetime_diff_sql($part, $all_sql[0], $all_sql[1]), @all_bind
 236 }
 237
 238 1;