multi column PK support

[dbsrgits/DBIx-Class-Journal.git] / lib / DBIx / Class / Schema / Journal.pm

diff --git a/lib/DBIx/Class/Schema/Journal.pm b/lib/DBIx/Class/Schema/Journal.pm
index f688d39..eaa4808 100644 (file)
--- a/lib/DBIx/Class/Schema/Journal.pm
+++ b/lib/DBIx/Class/Schema/Journal.pm
@@ -131,11 +131,13 @@ sub create_journal_for
     my ($self, $s_name) = @_;
 
     my $source = $self->source($s_name);
-    my $newclass = $self->get_audit_log_class_name($s_name);
-    DBIx::Class::Componentised->inject_base($newclass, 'DBIx::Class::Schema::Journal::DB::AuditLog');
-    $newclass->table(lc($s_name) . "_audit_log");
+    my $logclass = $self->get_audit_log_class_name($s_name);
+
+    DBIx::Class::Componentised->inject_base($logclass, 'DBIx::Class::Schema::Journal::DB::AuditLog');
+    $logclass->journal_define_table($source);
+
     my $log_source = "${s_name}AuditLog";
-    $self->_journal_schema->register_class($log_source, $newclass);
+    $self->_journal_schema->register_class($log_source, $logclass);
                            
 
     my $histclass = $self->get_audit_history_class_name($s_name);