projekt kivitendo

use SL::DB::Object::Hooks;

# These three functions cannot sit in SL::DB::Object::Hooks because

# mixins don't deal well with super classes (SUPER is the current

# package's super class, not $self's).

sub load {

  my ($self, @args) = @_;

  SL::DB::Object::Hooks::run_hooks($self, 'before_load');

  my $result = $self->SUPER::load(@args);

  SL::DB::Object::Hooks::run_hooks($self, 'after_load', $result);

  return $result;

}

sub save {

  my ($self, @args) = @_;

  my $worker        = sub {

    SL::DB::Object::Hooks::run_hooks($self, 'before_save');

    my $result = $self->SUPER::save(@args);

    SL::DB::Object::Hooks::run_hooks($self, 'after_save', $result);

  };

  return $self->db->in_transaction ? $worker->() : $self->db->do_transaction($worker);

}

sub delete {

  my ($self, @args) = @_;

  my $worker        = sub {

    SL::DB::Object::Hooks::run_hooks($self, 'before_delete');

    my $result = $self->SUPER::delete(@args);

    SL::DB::Object::Hooks::run_hooks($self, 'after_delete', $result);

  };

  return $self->db->in_transaction ? $worker->() : $self->db->do_transaction($worker);

}