reconnect and datetime auto Model.pm

Model.pm

package Hamburger::Model;
use parent 'Teng';

use DateTime;

sub dbh {
    my $self = shift;
    $self->SUPER::dbh;
    $self->connect unless $self->{dbh};
    $self->{dbh};
}

sub insert {
    my ($self, $table_name, $args, $prefix) = @_;
    
    my $now = DateTime->now( time_zone => 'Asia/Tokyo' );
    my $table = $self->schema->get_table( $table_name );
    for my $column ( @{$table->columns} ) {
        $args->{$column} = $now if $column=~/^created_(on|at)$/;
    }

    $self->SUPER::insert($table_name, $args, $prefix);
}

sub update {
    my ($self, $table_name, $args, $prefix) = @_;
    
    my $now = DateTime->now( time_zone => 'Asia/Tokyo' );
    my $table = $self->schema->get_table( $table_name );
    for my $column ( @{$table->columns} ) {
        $args->{$column} = $now if $column=~/^modified_(on|at)$/;
    }

    $self->SUPER::update($table_name, $args, $prefix);
}

1;