Module: MultiTenant::FastTruncate

Defined in:
lib/activerecord-multi-tenant/fast_truncate.rb

Class Method Summary collapse

Class Method Details

.run(exclude: ['schema_migrations']) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/activerecord-multi-tenant/fast_truncate.rb', line 6

def self.run(exclude: ['schema_migrations'])
  # This is a slightly faster version of DatabaseCleaner.clean_with(:truncation, pre_count: true)
  ActiveRecord::Base.connection.execute format(%(
  DO LANGUAGE plpgsql $$
  DECLARE
    t record;
    tables text[];
    seq_exists boolean;
    needs_truncate boolean;
  BEGIN
    FOR t IN SELECT schemaname, tablename FROM pg_tables WHERE schemaname = 'public' AND tablename NOT IN (%s) LOOP
      EXECUTE 'SELECT EXISTS (SELECT * from pg_class c WHERE c.relkind = ''S''
       AND c.relname=''' || t.tablename || '_id_seq'')' into seq_exists;
      IF seq_exists THEN
        EXECUTE 'SELECT is_called FROM ' || t.tablename || '_id_seq' INTO needs_truncate;
      ELSE
        needs_truncate := true;
      END IF;

      IF needs_truncate THEN
        tables := array_append(tables, quote_ident(t.schemaname) || '.' || quote_ident(t.tablename));
      END IF;
    END LOOP;

    IF array_length(tables, 1) > 0 THEN
      EXECUTE 'TRUNCATE TABLE ' || array_to_string(tables, ', ') || ' RESTART IDENTITY CASCADE';
    END IF;
  END$$;), exclude.map { |t| "'#{t}'" }.join('\n'))
end