lib/DBIx/Class/Schema/Journal.pm

   1 package DBIx::Class::Schema::Journal;
   2
   3 use base qw/DBIx::Class/;
   4
   5 use Scalar::Util 'blessed';
   6 use DBIx::Class::Schema::Journal::DB;
   7
   8 __PACKAGE__->mk_classdata('journal_storage_type');
   9 __PACKAGE__->mk_classdata('journal_connection');
  10 __PACKAGE__->mk_classdata('journal_sources'); ## [ source names ]
  11 __PACKAGE__->mk_classdata('journal_user'); ## [ class, field for user id ]
  12 __PACKAGE__->mk_classdata('_journal_schema'); ## schema object for journal
  13
  14 our $VERSION = '0.01';
  15
  16 use strict;
  17 use warnings;
  18
  19 # sub throw_exception
  20 # {
  21 # }
  22
  23 # sub exception_action
  24 # {
  25 #     my $self = shift;
  26 # #    print STDERR Carp::longmess;
  27
  28 #     $self->next::method(@_);
  29 # }
  30
  31 # sub load_classes
  32 # {
  33 #     my $class = shift;
  34
  35
  36 #     $class->next::method(@_);
  37
  38 # }
  39
  40 sub connection
  41 {
  42     my $self = shift;
  43     my $schema = $self->next::method(@_);
  44
  45 #   print STDERR join(":", $self->sources), "\n";
  46
  47     my $journal_schema;
  48     if(!defined($self->journal_connection))
  49     {
  50         ## If no connection, use the same schema/storage etc as the user
  51         DBIx::Class::Componentised->inject_base(ref $self, 'DBIx::Class::Schema::Journal::DB');
  52           $journal_schema = $self;
  53     }
  54     else
  55     {
  56         $journal_schema = DBIx::Class::Schema::Journal::DB->connect(@{ $self->journal_connection });
  57         if($self->journal_storage_type)
  58         {
  59             $journal_schema->storage_type($self->journal_storage_type);
  60         }
  61     }
  62
  63     ## get our own private version of the journaling sources
  64    $self->_journal_schema($journal_schema->compose_namespace(blessed($self) . '::Journal'));
  65
  66     ## Create auditlog+history per table
  67     my %j_sources = map { $_ => 1 } $self->journal_sources
  68                                       ? @{$self->journal_sources}
  69                                       : $self->sources;
  70     foreach my $s_name ($self->sources)
  71     {
  72         next unless($j_sources{$s_name});
  73         $self->create_journal_for($s_name);
  74         $self->class($s_name)->load_components('Journal');
  75 #        print STDERR "$s_name :", $self->class($s_name), "\n";
  76     }
  77
  78     ## Set up relationship between changeset->user_id and this schema's user
  79     if(!@{$self->journal_user})
  80     {
  81         warn "No Journal User set!";
  82         return;
  83     }
  84
  85     $self->_journal_schema->deploy();
  86     $self->_journal_schema->class('ChangeSet')->belongs_to('user', @{$self->journal_user});
  87     $self->_journal_schema->storage->disconnect();
  88
  89     return $schema;
  90 }
  91
  92 sub get_audit_log_class_name
  93 {
  94     my ($self, $sourcename) = @_;
  95
  96     return blessed($self->_journal_schema) . "::${sourcename}AuditLog";
  97 }
  98
  99 sub get_audit_history_class_name
 100 {
 101     my ($self, $sourcename) = @_;
 102
 103     return blessed($self->_journal_schema) . "::${sourcename}AuditHistory";
 104 }
 105
 106 sub create_journal_for
 107 {
 108     my ($self, $s_name) = @_;
 109
 110     my $source = $self->source($s_name);
 111     my $newclass = $self->get_audit_log_class_name($s_name);
 112     DBIx::Class::Componentised->inject_base($newclass, 'DBIx::Class::Schema::Journal::DB::AuditLog');
 113     $newclass->table(lc($s_name) . "_audit_log");
 114     $self->_journal_schema->register_class("${s_name}AuditLog", $newclass);
 115
 116
 117     my $histclass = $self->get_audit_history_class_name($s_name);
 118     DBIx::Class::Componentised->inject_base($histclass, 'DBIx::Class::Schema::Journal::DB::AuditHistory');
 119     $histclass->table(lc($s_name) . "_audit_history");
 120 #    $histclass->result_source_instance->name(lc($s_name) . "_audit_hisory");
 121     $histclass->add_columns(
 122                             map { $_ => $source->column_info($_) } $source->columns
 123                            );
 124
 125     $self->_journal_schema->register_class("${s_name}AuditHistory", $histclass);
 126 }
 127
 128 sub txn_do
 129 {
 130     my ($self, $code) = @_;
 131
 132     ## Create a new changeset, then run $code as a transaction
 133     my $cs = $self->_journal_schema->resultset('ChangeSet');
 134
 135     $self->txn_begin;
 136     my %changesetdata;
 137     if( defined $self->_journal_schema->current_user() )
 138     {
 139         $changesetdata{user_id} = $self->_journal_schema->current_user();
 140     }
 141     if( defined $self->_journal_schema->current_session() )
 142     {
 143         $changesetdata{session_id} = $self->_journal_schema->current_session();
 144     }
 145
 146 #         (
 147 #           $self->_journal_schema->current_user()
 148 #           ? ( user_id => $self->_journal_schema->current_user())
 149 #           : (),
 150 #           $self->_journal_schema->current_session()
 151 #           ? ( session_id => $self->_journal_schema->current_session() )
 152 #           : ()
 153 #         );
 154     if(!%changesetdata)
 155     {
 156         %changesetdata = ( ID => undef );
 157     }
 158     my $changeset = $cs->create({ %changesetdata });
 159     $self->_journal_schema->current_changeset($changeset->ID);
 160
 161     $self->next::method($code);
 162 }
 163
 164 sub changeset_user
 165 {
 166     my ($self, $userid) = @_;
 167
 168     return $self->_journal_schema->current_user() if(@_ == 1);
 169
 170     $self->_journal_schema->current_user($userid);
 171 }
 172
 173 sub changeset_session
 174 {
 175     my ($self, $sessionid) = @_;
 176
 177     return $self->_journal_schema->current_session() if(@_ == 1);
 178
 179     $self->_journal_schema->current_session($sessionid);
 180 }
 181
 182
 183 1;