pg_audit_log 0.4.5 → 0.5.0

data/lib/pg_audit_log/entry.rb

@@ -41,12 +41,45 @@ class PgAuditLog::Entry < ActiveRecord::Base
         );
 
         ALTER SEQUENCE #{self.table_name}_id_seq OWNED BY #{self.table_name}.id;
+
+        CREATE OR REPLACE FUNCTION audit_log_insert_trigger()
+        RETURNS TRIGGER AS $$
+        DECLARE
+          tablename TEXT;
+          insert_sql TEXT;
+          create_table_sql TEXT;
+          month_start DATE;
+          month_end DATE;
+        BEGIN
+          tablename := '#{self.table_name}_' || to_char(NEW.occurred_at, 'YYYYMM');
+          insert_sql := 'INSERT INTO ' || tablename || ' VALUES($1.*)';
+          EXECUTE insert_sql USING NEW;
+          RETURN NULL;
+        EXCEPTION
+          WHEN null_value_not_allowed THEN
+            RETURN NULL;
+          WHEN undefined_table THEN
+            EXECUTE 'SELECT to_char($1, ''YYYY-MM-01'')::DATE' INTO month_start USING NEW.occurred_at;
+            EXECUTE 'SELECT ($1 +  INTERVAL ''1 MONTH'')' INTO month_end USING month_start;
+            create_table_sql :=  'CREATE TABLE ' || tablename || ' ( CHECK ( date(occurred_at) >= DATE ''' || month_start || ''' AND date(occurred_at) < DATE ''' ||
+              month_end || ''' ) ) INHERITS (#{self.table_name})';
+            EXECUTE create_table_sql;
+            EXECUTE 'CREATE INDEX ' || tablename || '_occurred_at ON ' || tablename || ' (date(occurred_at))';
+            EXECUTE insert_sql USING NEW;
+            RETURN NULL;
+        END;
+        $$
+        LANGUAGE plpgsql;
+
+        CREATE TRIGGER insert_audit_log_trigger
+          BEFORE INSERT ON audit_log
+          FOR EACH ROW EXECUTE PROCEDURE audit_log_insert_trigger();
       SQL
       connection.execute_without_auditing(sql)
     end
 
     def uninstall
-      connection.execute("DROP TABLE IF EXISTS #{self.table_name}")
+      connection.execute("DROP TABLE IF EXISTS #{self.table_name} CASCADE")
     end
 
     def delete(id)

data/lib/pg_audit_log/triggers.rb

@@ -11,7 +11,7 @@ module PgAuditLog
 
     class << self
       def tables
-        skip_tables = (PgAuditLog::IGNORED_TABLES + [PgAuditLog::Entry.table_name])
+        skip_tables = (PgAuditLog::IGNORED_TABLES + [PgAuditLog::Entry.table_name, /#{PgAuditLog::Entry.table_name}_[0-9]{6}/])
         connection.tables.reject do |table|
           skip_tables.include?(table) ||
             skip_tables.any? { |skip_table| skip_table =~ table if skip_table.is_a? Regexp }

data/lib/pg_audit_log/version.rb

@@ -1,3 +1,3 @@
 module PgAuditLog
-  VERSION = "0.4.5"
+  VERSION = "0.5.0"
 end

data/spec/spec_helper.rb

@@ -18,12 +18,12 @@ rescue PGError => e
   raise e
 end
 
-
 RSpec.configure do |config|
   config.mock_with :rspec
   config.extend WithModel
 
   config.before(:each) do
+    PgAuditLog::Entry.uninstall rescue nil
     connection.tables.each do |table|
       connection.drop_table_without_auditing(table)
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_audit_log
 version: !ruby/object:Gem::Version
-  version: 0.4.5
+  version: 0.5.0
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-02-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
-  requirement: &70252020501060 !ruby/object:Gem::Requirement
+  requirement: &70242168359480 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: 3.0.0
   type: :runtime
   prerelease: false
-  version_requirements: *70252020501060
+  version_requirements: *70242168359480
 - !ruby/object:Gem::Dependency
   name: pg
-  requirement: &70252020500560 !ruby/object:Gem::Requirement
+  requirement: &70242168358980 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: 0.9.0
   type: :runtime
   prerelease: false
-  version_requirements: *70252020500560
+  version_requirements: *70242168358980
 - !ruby/object:Gem::Dependency
   name: rspec-rails
-  requirement: &70252020500100 !ruby/object:Gem::Requirement
+  requirement: &70242168358520 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - =
@@ -43,10 +43,10 @@ dependencies:
         version: '2.7'
   type: :development
   prerelease: false
-  version_requirements: *70252020500100
+  version_requirements: *70242168358520
 - !ruby/object:Gem::Dependency
   name: with_model
-  requirement: &70252020499640 !ruby/object:Gem::Requirement
+  requirement: &70242168358060 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: 0.1.3
   type: :development
   prerelease: false
-  version_requirements: *70252020499640
+  version_requirements: *70242168358060
 - !ruby/object:Gem::Dependency
   name: autotest
-  requirement: &70252020499260 !ruby/object:Gem::Requirement
+  requirement: &70242168357680 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,7 +65,7 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70252020499260
+  version_requirements: *70242168357680
 description: A completely transparent audit logging component for your application
   using a stored procedure and triggers. Comes with specs for your project and a rake
   task to generate the reverse SQL to undo changes logged
@@ -120,7 +120,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3581204670683200493
+      hash: 1658347057273507033
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -129,7 +129,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3581204670683200493
+      hash: 1658347057273507033
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.15