better_record 0.17.9 → 0.18.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88614c5435066753d21aab6f1cdbf445362ce968c9832758c8286c308caa5660
-  data.tar.gz: ef4e6cbee750ffbfb48baf1f33e97dd1fc9377d5d35585bc80f081dcc47a5ea1
+  metadata.gz: c07da71971172205aecfda94df6fb1ca1213d32569053fcd8892dda8a91076fa
+  data.tar.gz: a64508bffff376045804d34726d45bfa6a0ab80948f45198dac6a6c1a8a57d5f
 SHA512:
-  metadata.gz: 7f1c870552dbc7a6fbd7b009a79c601b5241c5f1b7541ce0caf45298c9bd45ef5564b43c931548ea313ca52ffe49bc9428643fd32eae3775d49bb9e3c5c12fbd
-  data.tar.gz: fbb33a161af131964a041994c02b127432aea7fc9b1e0df0c09399a2f27ad2abe6f867f837bf8fd3911de387f6cf0c28bca9b1b59a63a2d7b2b73b99e2edee56
+  metadata.gz: b47ff7753e4775cd1d95f4741833e4719e51d0c4624d5139b68828c84c51e40eaeece53dcd60ac5052b288f83d3f6ca8689c3a7b0ca7fcdbcf0cd8ee8259d5b9
+  data.tar.gz: 0ccc0c6aab836bba068ba45d014555b4bb08d8cf88b27a9d673489d0c36fcd6d3f5fa016525797e6ee57c3da2b150c36663f5a2b5b450410ecb06749b11a02fb

data/app/models/better_record/base.rb

@@ -14,16 +14,25 @@ module BetterRecord
 
     # == Relationships ========================================================
     if (ha = BetterRecord.has_auditing_relation_by_default)
-      has_many self.audit_relation_name,
-        class_name: 'BetterRecord::LoggedAction',
-        primary_type: :table_name,
-        foreign_key: :row_id,
-        foreign_type: :table_name,
-        as: self.audit_relation_name
-
       class << self
         define_method BetterRecord.audit_relation_name do |*args, &block|
-          base_q = BetterRecord::LoggedAction.where(table_name: self.table_name)
+          @logger_model ||=
+            begin
+              connection.execute(%Q(SELECT 1 FROM #{BetterRecord::LoggedAction.table_name}_#{self.table_name} LIMIT 1))
+
+              class self.to_s.constantize::LoggedAction < BetterRecord::LoggedAction
+                self.table_name = "#{BetterRecord::LoggedAction.table_name}_#{self.to_s.deconstantize.constantize.table_name}"
+                self
+              end
+            rescue ActiveRecord::StatementInvalid
+              class self.to_s.constantize::LoggedAction < BetterRecord::LoggedAction
+                self
+              end
+            end
+
+          return @logger_model if args.present? && args.first == 'SETTING_INHERITANCE'
+
+          base_q = @logger_model.where(table_name: self.table_name)
           base_q = base_q.where(*args) if args.present?
 
           if block
@@ -37,6 +46,16 @@ module BetterRecord
           end
         end
       end
+
+      def self.inherited(child)
+        super
+        TracePoint.trace(:end) do |t|
+          if child == t.self
+            child.set_audits_methods!
+            t.disable
+          end
+        end
+      end
     end
     # == Validations ==========================================================
 
@@ -47,6 +66,16 @@ module BetterRecord
     # == Boolean Class Methods ================================================
 
     # == Class Methods ========================================================
+    def self.set_audits_methods!
+      m = __send__ BetterRecord.audit_relation_name, 'SETTING_INHERITANCE'
+      self.has_many self.audit_relation_name,
+        class_name: m.to_s,
+        primary_type: :table_name,
+        foreign_key: :row_id,
+        foreign_type: :table_name,
+        as: self.audit_relation_name
+    end
+
     def self.gender_enum(col)
       enum col, BetterRecord::Gender::ENUM
     end

data/app/models/better_record/logged_action.rb

@@ -8,6 +8,7 @@ module BetterRecord
       I: 'INSERT',
       U: 'UPDATE',
       T: 'TRUNCATE',
+      A: 'ARCHIVE',
     }.with_indifferent_access
 
     # == Attributes ===========================================================
@@ -43,6 +44,15 @@ module BetterRecord
       ]
     end
 
+    # def self.set_audits_methods!
+    #   self.has_many self.audit_relation_name,
+    #     class_name: 'BetterRecord::LoggedAction',
+    #     primary_type: :table_name,
+    #     foreign_key: :row_id,
+    #     foreign_type: :table_name,
+    #     as: self.audit_relation_name
+    # end
+
     # == Boolean Methods ======================================================
 
     # == Instance Methods =====================================================

data/db/migrate/20190209033946_update_better_record_audit_functions.rb

@@ -0,0 +1,103 @@
+class UpdateBetterRecordAuditFunctions < ActiveRecord::Migration[5.2]
+  def up
+    last_event = BetterRecord::LoggedAction.count
+
+    execute <<-SQL
+      ALTER TABLE #{BetterRecord.db_audit_schema}.logged_actions
+      RENAME TO old_logged_actions
+    SQL
+
+    sql = ""
+    source = File.new(BetterRecord::Engine.root.join('db', 'postgres-audit-v2-table.psql'), "r")
+    while (line = source.gets)
+      sql << line.gsub(/SELECTED_SCHEMA_NAME/, BetterRecord.db_audit_schema)
+    end
+    source.close
+
+    execute sql
+
+    execute <<-SQL
+      ALTER SEQUENCE #{BetterRecord.db_audit_schema}.logged_actions_event_id_seq START WITH #{last_event}
+    SQL
+
+    sql = ""
+    source = File.new(BetterRecord::Engine.root.join('db', 'postgres-audit-v2-trigger.psql'), "r")
+    while (line = source.gets)
+      sql << line.gsub(/SELECTED_SCHEMA_NAME/, BetterRecord.db_audit_schema)
+    end
+    source.close
+
+    execute sql
+
+    rows = execute <<-SQL
+      select trg.tgname,
+          CASE trg.tgtype::integer & 66
+              WHEN 2 THEN 'BEFORE'
+              WHEN 64 THEN 'INSTEAD OF'
+              ELSE 'AFTER'
+          end as trigger_type,
+         case trg.tgtype::integer & cast(28 as int2)
+           when 16 then 'UPDATE'
+           when 8 then 'DELETE'
+           when 4 then 'INSERT'
+           when 20 then 'INSERT, UPDATE'
+           when 28 then 'INSERT, UPDATE, DELETE'
+           when 24 then 'UPDATE, DELETE'
+           when 12 then 'INSERT, DELETE'
+         end as trigger_event,
+         tbl.relname as table_name,
+         obj_description(trg.oid) as remarks,
+           case
+            when trg.tgenabled='O' then 'ENABLED'
+              else 'DISABLED'
+          end as status,
+          case trg.tgtype::integer & 1
+            when 1 then 'ROW'::text
+            else 'STATEMENT'::text
+          end as trigger_level
+      FROM pg_trigger trg
+        JOIN pg_class tbl on trg.tgrelid = tbl.oid
+        JOIN pg_namespace ns ON ns.oid = tbl.relnamespace
+      WHERE trg.tgname not like 'RI_ConstraintTrigger%'
+        AND trg.tgname not like 'pg_sync_pg%'
+    SQL
+
+    rows.each do |r|
+      if r['tgname'].to_s =~ /audit_trigger/
+        execute <<-SQL
+          CREATE TABLE IF NOT EXISTS #{BetterRecord.db_audit_schema}.logged_actions_#{r['table_name']} (
+            CHECK (table_name = '#{r['table_name']}')
+          ) INHERITS (#{BetterRecord.db_audit_schema}.logged_actions);
+        SQL
+
+        puts "\n\nPLEASE RECREATE ALL AUDIT TRIGGERS FOR #{r['table_name']}\n\n#{r}\n\n"
+      end
+    end
+
+    puts "\n\nTo insert old audits back into logged_actions run:\n\n"
+
+    puts <<-RUBY
+      class BetterRecord::OldLoggedAction < BetterRecord::LoggedAction
+        self.table_name = "#{BetterRecord.db_audit_schema}.old_logged_actions"
+      end
+
+      while BetterRecord::OldLoggedAction.count > 0
+        p BetterRecord::OldLoggedAction.count
+        BetterRecord::OldLoggedAction.order(:event_id).limit(100).each do |r|
+          klass = nil
+          begin
+            BetterRecord::LoggedAction.connection.execute(%Q(SELECT 1 FROM #{BetterRecord.db_audit_schema}.logged_actions_#\{r.table_name}))
+
+            klass = Class.new(BetterRecord::LoggedAction)
+            klass.table_name = "#{BetterRecord.db_audit_schema}.logged_actions_#\{r.table_name}"
+          rescue ActiveRecord::StatementInvalid
+            klass = BetterRecord::LoggedAction
+          end
+          klass.create!(r.attributes)
+          p klass.count
+          r.delete
+        end
+      end
+    RUBY
+  end
+end

data/db/postgres-audit-v2-table.psql

@@ -0,0 +1,77 @@
+CREATE EXTENSION IF NOT EXISTS hstore;
+
+CREATE SCHEMA IF NOT EXISTS SELECTED_SCHEMA_NAME;
+REVOKE ALL ON SCHEMA SELECTED_SCHEMA_NAME FROM public;
+
+COMMENT ON SCHEMA SELECTED_SCHEMA_NAME IS 'Out-of-table audit/history logging tables and trigger functions';
+
+--
+-- Audited data. Lots of information is available, it's just a matter of how much
+-- you really want to record. See:
+--
+--   http://www.postgresql.org/docs/9.1/static/functions-info.html
+--
+-- Remember, every column you add takes up more audit table space and slows audit
+-- inserts.
+--
+-- Every index you add has a big impact too, so avoid adding indexes to the
+-- audit table unless you REALLY need them. The hstore GIST indexes are
+-- particularly expensive.
+--
+-- It is sometimes worth copying the audit table, or a coarse subset of it that
+-- you're interested in, into a temporary table where you CREATE any useful
+-- indexes and do your analysis.
+--
+CREATE TABLE IF NOT EXISTS SELECTED_SCHEMA_NAME.logged_actions (
+  event_id bigserial primary key,
+  schema_name text not null,
+  table_name text not null,
+  relid oid not null,
+  session_user_name text,
+  app_user_id integer,
+  app_user_type text,
+  app_ip_address inet,
+  action_tstamp_tx TIMESTAMP WITH TIME ZONE NOT NULL,
+  action_tstamp_stm TIMESTAMP WITH TIME ZONE NOT NULL,
+  action_tstamp_clk TIMESTAMP WITH TIME ZONE NOT NULL,
+  transaction_id bigint,
+  application_name text,
+  client_addr inet,
+  client_port integer,
+  client_query text,
+  action TEXT NOT NULL CHECK (action IN ('I','D','U', 'T', 'A')),
+  row_id bigint,
+  row_data hstore,
+  changed_fields hstore,
+  statement_only boolean not null
+);
+
+REVOKE ALL ON SELECTED_SCHEMA_NAME.logged_actions FROM public;
+
+COMMENT ON TABLE SELECTED_SCHEMA_NAME.logged_actions IS 'History of auditable actions on audited tables, from SELECTED_SCHEMA_NAME.if_modified_func()';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.event_id IS 'Unique identifier for each auditable event';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.schema_name IS 'Database schema audited table for this event is in';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.table_name IS 'Non-schema-qualified table name of table event occured in';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.relid IS 'Table OID. Changes with drop/create. Get with ''tablename''::regclass';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.session_user_name IS 'Login / session user whose statement caused the audited event';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.app_user_id IS 'Application-provided polymorphic user id';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.app_user_type IS 'Application-provided polymorphic user type';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.app_ip_address IS 'Application-provided ip address of user whose statement caused the audited event';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.action_tstamp_tx IS 'Transaction start timestamp for tx in which audited event occurred';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.action_tstamp_stm IS 'Statement start timestamp for tx in which audited event occurred';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.action_tstamp_clk IS 'Wall clock time at which audited event''s trigger call occurred';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.transaction_id IS 'Identifier of transaction that made the change. May wrap, but unique paired with action_tstamp_tx.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.client_addr IS 'IP address of client that issued query. Null for unix domain socket.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.client_port IS 'Remote peer IP port address of client that issued query. Undefined for unix socket.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.client_query IS 'Top-level query that caused this auditable event. May be more than one statement.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.application_name IS 'Application name set when this audit event occurred. Can be changed in-session by client.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.action IS 'Action type; I = insert, D = delete, U = update, T = truncate, A = archive';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.row_id IS 'Record primary_key. Null for statement-level trigger. Prefers NEW.id if exists';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.row_data IS 'Record value. Null for statement-level trigger. For INSERT this is the new tuple. For DELETE and UPDATE it is the old tuple.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.changed_fields IS 'New values of fields changed by UPDATE. Null except for row-level UPDATE events.';
+COMMENT ON COLUMN SELECTED_SCHEMA_NAME.logged_actions.statement_only IS '''t'' if audit event is from an FOR EACH STATEMENT trigger, ''f'' for FOR EACH ROW';
+
+CREATE INDEX IF NOT EXISTS logged_actions_relid_idx ON SELECTED_SCHEMA_NAME.logged_actions(relid);
+CREATE INDEX IF NOT EXISTS logged_actions_action_tstamp_tx_stm_idx ON SELECTED_SCHEMA_NAME.logged_actions(action_tstamp_stm);
+CREATE INDEX IF NOT EXISTS logged_actions_action_idx ON SELECTED_SCHEMA_NAME.logged_actions(action);
+CREATE INDEX IF NOT EXISTS logged_actions_row_id_idx ON SELECTED_SCHEMA_NAME.logged_actions(row_id);

data/db/postgres-audit-v2-trigger.psql

@@ -0,0 +1,278 @@
+-- An audit history is important on most tables. Provide an audit trigger that logs to
+-- a dedicated audit table for the major relations.
+--
+-- This file should be generic and not depend on application roles or structures,
+-- as it's being listed here:
+--
+--  https://wiki.postgresql.org/wiki/Audit_trigger_91plus
+--
+-- This trigger was originally based on
+--   http://wiki.postgresql.org/wiki/Audit_trigger
+-- but has been completely rewritten.
+--
+-- Should really be converted into a relocatable EXTENSION, with control and upgrade files.
+
+CREATE OR REPLACE FUNCTION SELECTED_SCHEMA_NAME.if_modified_func()
+RETURNS TRIGGER AS
+$$
+  DECLARE
+    audit_row SELECTED_SCHEMA_NAME.logged_actions;
+    include_values boolean;
+    log_diffs boolean;
+    h_old hstore;
+    h_new hstore;
+    user_row record;
+    excluded_cols text[] = ARRAY[]::text[];
+    pk_val_query text;
+  BEGIN
+    IF TG_WHEN <> 'AFTER' THEN
+      RAISE EXCEPTION 'SELECTED_SCHEMA_NAME.if_modified_func() may only run as an AFTER trigger';
+    END IF;
+
+    audit_row = ROW(
+      nextval('SELECTED_SCHEMA_NAME.logged_actions_event_id_seq'), -- event_id
+      TG_TABLE_SCHEMA::text,                        -- schema_name
+      TG_TABLE_NAME::text,                          -- table_name
+      TG_RELID,                                     -- relation OID for much quicker searches
+      session_user::text,                           -- session_user_name
+      NULL, NULL, NULL,                             -- app_user_id, app_user_type, app_ip_address
+      current_timestamp,                            -- action_tstamp_tx
+      statement_timestamp(),                        -- action_tstamp_stm
+      clock_timestamp(),                            -- action_tstamp_clk
+      txid_current(),                               -- transaction ID
+      current_setting('application_name'),          -- client application
+      inet_client_addr(),                           -- client_addr
+      inet_client_port(),                           -- client_port
+      current_query(),                              -- top-level query or queries (if multistatement) from client
+      substring(TG_OP,1,1),                         -- action
+      NULL, NULL, NULL,                             -- row_id, row_data, changed_fields
+      'f'                                           -- statement_only
+    );
+
+    IF NOT TG_ARGV[0]::boolean IS DISTINCT FROM 'f'::boolean THEN
+      audit_row.client_query = NULL;
+    END IF;
+
+    IF ((TG_ARGV[1] IS NOT NULL) AND (TG_LEVEL = 'ROW')) THEN
+      pk_val_query = 'SELECT $1.' || quote_ident(TG_ARGV[1]::text);
+
+      IF (TG_OP IS DISTINCT FROM 'DELETE') THEN
+        EXECUTE pk_val_query INTO audit_row.row_id USING NEW;
+      END IF;
+
+      IF audit_row.row_id IS NULL THEN
+        EXECUTE pk_val_query INTO audit_row.row_id USING OLD;
+      END IF;
+    END IF;
+
+    IF TG_ARGV[2] IS NOT NULL THEN
+      excluded_cols = TG_ARGV[2]::text[];
+    END IF;
+
+    CREATE TEMP TABLE IF NOT EXISTS
+      "_app_user" (user_id integer, user_type text, ip_address inet);
+
+    IF (TG_OP = 'UPDATE' AND TG_LEVEL = 'ROW') THEN
+      audit_row.row_data = hstore(OLD.*) - excluded_cols;
+      audit_row.changed_fields =  (hstore(NEW.*) - audit_row.row_data) - excluded_cols;
+      IF audit_row.changed_fields = hstore('') THEN
+        -- All changed fields are ignored. Skip this update.
+        RETURN NULL;
+      END IF;
+    ELSIF (TG_OP = 'DELETE' AND TG_LEVEL = 'ROW') THEN
+      audit_row.row_data = hstore(OLD.*) - excluded_cols;
+    ELSIF (TG_OP = 'INSERT' AND TG_LEVEL = 'ROW') THEN
+      audit_row.row_data = hstore(NEW.*) - excluded_cols;
+    ELSIF (TG_LEVEL = 'STATEMENT' AND TG_OP IN ('INSERT','UPDATE','DELETE','TRUNCATE')) THEN
+      audit_row.statement_only = 't';
+    ELSE
+      RAISE EXCEPTION '[SELECTED_SCHEMA_NAME.if_modified_func] - Trigger func added as trigger for unhandled case: %, %',TG_OP, TG_LEVEL;
+      RETURN NULL;
+    END IF;
+
+    -- inject app_user data into audit
+    BEGIN
+      PERFORM
+      n.nspname, c.relname
+      FROM
+      pg_catalog.pg_class c
+      LEFT JOIN
+      pg_catalog.pg_namespace n
+      ON n.oid = c.relnamespace
+      WHERE
+      n.nspname like 'pg_temp_%'
+      AND
+      c.relname = '_app_user';
+
+      IF FOUND THEN
+      FOR user_row IN SELECT * FROM _app_user LIMIT 1 LOOP
+        audit_row.app_user_id = user_row.user_id;
+        audit_row.app_user_type = user_row.user_type;
+        audit_row.app_ip_address = user_row.ip_address;
+      END LOOP;
+      END IF;
+    END;
+    -- end app_user data
+
+    EXECUTE 'CREATE TABLE IF NOT EXISTS SELECTED_SCHEMA_NAME.logged_actions_' || quote_ident(TG_TABLE_NAME::TEXT) || '(
+      CHECK (table_name = ' || quote_literal(TG_TABLE_NAME::TEXT) || ')
+    ) INHERITS (SELECTED_SCHEMA_NAME.logged_actions)';
+
+    EXECUTE 'INSERT INTO SELECTED_SCHEMA_NAME.logged_actions_' || quote_ident(TG_TABLE_NAME::TEXT) || ' VALUES ($1.*)' USING audit_row;
+    RETURN NULL;
+  END;
+$$
+LANGUAGE plpgsql
+SECURITY DEFINER
+SET search_path = pg_catalog, public;
+
+
+COMMENT ON FUNCTION SELECTED_SCHEMA_NAME.if_modified_func() IS
+$$
+  Track changes to a table at the statement and/or row level.
+
+  Optional parameters to trigger in CREATE TRIGGER call:
+
+  param 0: boolean, whether to log the query text. Default 't'.
+
+  param 1: text, primary_key_column of audited table if bigint.
+
+  param 2: text[], columns to ignore in updates. Default [].
+
+       Updates to ignored cols are omitted from changed_fields.
+
+       Updates with only ignored cols changed are not inserted
+       into the audit log.
+
+       Almost all the processing work is still done for updates
+       that ignored. If you need to save the load, you need to use
+       WHEN clause on the trigger instead.
+
+       No warning or error is issued if ignored_cols contains columns
+       that do not exist in the target table. This lets you specify
+       a standard set of ignored columns.
+
+  There is no parameter to disable logging of values. Add this trigger as
+  a 'FOR EACH STATEMENT' rather than 'FOR EACH ROW' trigger if you do not
+  want to log row values.
+
+  Note that the user name logged is the login role for the session. The audit trigger
+  cannot obtain the active role because it is reset by the SECURITY DEFINER invocation
+  of the audit trigger its self.
+$$;
+
+
+CREATE OR REPLACE FUNCTION SELECTED_SCHEMA_NAME.get_primary_key_column(target_table text)
+RETURNS text AS
+$$
+  DECLARE
+    _pk_query_text text;
+    _pk_column_name text;
+  BEGIN
+    _pk_query_text =  'SELECT a.attname ' ||
+                      'FROM   pg_index i ' ||
+                      'JOIN   pg_attribute a ON a.attrelid = i.indrelid ' ||
+                      '                    AND a.attnum = ANY(i.indkey) ' ||
+                      'WHERE  i.indrelid = ' || quote_literal(target_table::TEXT) || '::regclass ' ||
+                      'AND    i.indisprimary ' ||
+                      'AND format_type(a.atttypid, a.atttypmod) = ' || quote_literal('bigint'::TEXT) ||
+                      'LIMIT 1';
+
+    EXECUTE _pk_query_text INTO _pk_column_name;
+    raise notice 'Value %', _pk_column_name;
+    return _pk_column_name;
+  END;
+$$
+LANGUAGE plpgsql;
+
+COMMENT ON FUNCTION SELECTED_SCHEMA_NAME.get_primary_key_column(text) IS
+$$
+  Get primary key column name if single PK and type bigint.
+
+  Arguments:
+      target_table:   Table name, schema qualified if not on search_path
+$$;
+
+CREATE OR REPLACE FUNCTION SELECTED_SCHEMA_NAME.audit_table(target_table regclass, audit_rows boolean, audit_query_text boolean, ignored_cols text[])
+RETURNS void AS
+$$
+  DECLARE
+    stm_targets text = 'INSERT OR UPDATE OR DELETE OR TRUNCATE';
+    _q_txt text;
+    _pk_column_name text;
+    _pk_column_snip text;
+    _ignored_cols_snip text = '';
+  BEGIN
+    EXECUTE 'DROP TRIGGER IF EXISTS audit_trigger_row ON ' || quote_ident(target_table::TEXT);
+    EXECUTE 'DROP TRIGGER IF EXISTS audit_trigger_stm ON ' || quote_ident(target_table::TEXT);
+
+    EXECUTE 'CREATE TABLE IF NOT EXISTS SELECTED_SCHEMA_NAME.logged_actions_' || quote_ident(target_table::TEXT) || '(
+      CHECK (table_name = ' || quote_literal(target_table::TEXT) || ')
+    ) INHERITS (SELECTED_SCHEMA_NAME.logged_actions)';
+
+    IF audit_rows THEN
+      _pk_column_name = SELECTED_SCHEMA_NAME.get_primary_key_column(target_table::TEXT);
+
+      IF _pk_column_name IS NOT NULL THEN
+        _pk_column_snip = ', ' || quote_literal(_pk_column_name);
+      ELSE
+        _pk_column_snip = ', NULL';
+      END IF;
+
+      IF array_length(ignored_cols,1) > 0 THEN
+        _ignored_cols_snip = ', ' || quote_literal(ignored_cols);
+      END IF;
+      _q_txt = 'CREATE TRIGGER audit_trigger_row AFTER INSERT OR UPDATE OR DELETE ON ' ||
+          quote_ident(target_table::TEXT) ||
+          ' FOR EACH ROW EXECUTE PROCEDURE SELECTED_SCHEMA_NAME.if_modified_func(' ||
+          quote_literal(audit_query_text) || _pk_column_snip || _ignored_cols_snip || ');';
+      RAISE NOTICE '%',_q_txt;
+      EXECUTE _q_txt;
+      stm_targets = 'TRUNCATE';
+    ELSE
+    END IF;
+
+    _q_txt = 'CREATE TRIGGER audit_trigger_stm AFTER ' || stm_targets || ' ON ' ||
+        target_table ||
+        ' FOR EACH STATEMENT EXECUTE PROCEDURE SELECTED_SCHEMA_NAME.if_modified_func('||
+        quote_literal(audit_query_text) || ');';
+    RAISE NOTICE '%',_q_txt;
+    EXECUTE _q_txt;
+
+  END;
+$$
+LANGUAGE plpgsql;
+
+COMMENT ON FUNCTION SELECTED_SCHEMA_NAME.audit_table(regclass, boolean, boolean, text[]) IS
+$$
+  Add auditing support to a table.
+
+  Arguments:
+      target_table:   Table name, schema qualified if not on search_path
+      audit_rows:     Record each row change, or only audit at a statement level
+      audit_query_text: Record the text of the client query that triggered the audit event?
+      ignored_cols:   Columns to exclude from update diffs, ignore updates that change only ignored cols.
+$$;
+
+-- Pg doesn't allow variadic calls with 0 params, so provide a wrapper
+CREATE OR REPLACE FUNCTION SELECTED_SCHEMA_NAME.audit_table(target_table regclass, audit_rows boolean, audit_query_text boolean)
+RETURNS void AS
+$$
+  SELECT SELECTED_SCHEMA_NAME.audit_table($1, $2, $3, ARRAY[]::text[]);
+$$
+LANGUAGE SQL;
+
+-- And provide a convenience call wrapper for the simplest case
+-- of row-level logging with no excluded cols and query logging enabled.
+--
+CREATE OR REPLACE FUNCTION SELECTED_SCHEMA_NAME.audit_table(target_table regclass)
+RETURNS void AS
+$$
+  SELECT SELECTED_SCHEMA_NAME.audit_table($1, BOOLEAN 't', BOOLEAN 't');
+$$
+LANGUAGE SQL;
+
+COMMENT ON FUNCTION SELECTED_SCHEMA_NAME.audit_table(regclass) IS
+$$
+  Add auditing support to the given table. Row-level changes will be logged with full client query text. No cols are ignored.
+$$;

data/lib/better_record/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module BetterRecord
-  VERSION = '0.17.9'
+  VERSION = '0.18.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: better_record
 version: !ruby/object:Gem::Version
-  version: 0.17.9
+  version: 0.18.0
 platform: ruby
 authors:
 - Sampson Crowley
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-01-31 00:00:00.000000000 Z
+date: 2019-02-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -270,7 +270,10 @@ files:
 - db/migrate/20181228204403_create_better_record_attachment_validations.rb
 - db/migrate/20190107202602_add_updated_at_to_better_record_table_sizes.rb
 - db/migrate/20190123225641_add_exchange_rate_integer_type.rb
+- db/migrate/20190209033946_update_better_record_audit_functions.rb
 - db/postgres-audit-trigger.psql
+- db/postgres-audit-v2-table.psql
+- db/postgres-audit-v2-trigger.psql
 - lib/better_record.rb
 - lib/better_record/batches.rb
 - lib/better_record/concerns/active_record_extensions/associations_extensions/association_scope_extensions.rb