Class: MysqlRewinder

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Defined in:
lib/mysql_rewinder.rb,
lib/mysql_rewinder/cleaner.rb,
lib/mysql_rewinder/version.rb,
lib/mysql_rewinder/ext/trilogy.rb,
lib/mysql_rewinder/cleaner/adapter.rb,
lib/mysql_rewinder/ext/mysql2_client.rb,
lib/mysql_rewinder/cleaner/mysql2_adapter.rb,
lib/mysql_rewinder/cleaner/trilogy_adapter.rb

Defined Under Namespace

Modules: Ext Classes: Cleaner

Constant Summary collapse

VERSION = 
"0.1.0"

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(db_configs:, except_tables:, adapter:) ⇒ MysqlRewinder 



20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/mysql_rewinder.rb', line 20

def initialize(db_configs:, except_tables:, adapter:)
  @initialized_pid = Process.pid
  @inserted_table_record_dir = Pathname(Dir.tmpdir)
  @cleaners = db_configs.map do |db_config|
    Cleaner.new(
      db_config.transform_keys(&:to_sym),
      except_tables: except_tables,
      adapter: adapter
    )
  end
  reset_inserted_tables
end

Class Method Details

.setup(db_configs, except_tables: [], adapter: :trilogy) ⇒ Object 



15
16
17
 
# File 'lib/mysql_rewinder.rb', line 15

def setup(db_configs, except_tables: [], adapter: :trilogy)
  @instance = new(db_configs: db_configs, except_tables: except_tables, adapter: adapter)
end

Instance Method Details

#calculate_inserted_tables ⇒ Object 



59
60
61
62
63
64
65
66
67
 
# File 'lib/mysql_rewinder.rb', line 59

def calculate_inserted_tables
  unless @initialized_pid == Process.pid
    raise "MysqlRewinder is initialize in process #{@initialized_pid}, but calculate_inserted_tables is called in process #{Process.pid}"
  end

  Dir.glob(@inserted_table_record_dir.join("#{@initialized_pid}.*.inserted_tables").to_s).flat_map do |fname|
    File.read(fname).strip.split(',')
  end.uniq
end

#clean ⇒ Object 



74
75
76
77
78
 
# File 'lib/mysql_rewinder.rb', line 74

def clean
  aggregated_inserted_tables = calculate_inserted_tables
  @cleaners.each { |c| c.clean(tables: aggregated_inserted_tables) }
  reset_inserted_tables
end

#clean_all ⇒ Object 



69
70
71
72
 
# File 'lib/mysql_rewinder.rb', line 69

def clean_all
  @cleaners.each(&:clean_all)
  reset_inserted_tables
end

#record_inserted_table(sql) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/mysql_rewinder.rb', line 33

def record_inserted_table(sql)
  return unless @initialized_pid

  @inserted_tables ||= Set.new
  sql.split(';').each do |statement|
    match = statement.match(/\A\s*INSERT(?:\s+IGNORE)?(?:\s+INTO)?\s+(?:\.*[`"]?([^.\s`"(]+)[`"]?)*/i)
    next unless match

    table = match[1]
    @inserted_tables << table if table
  end
  File.write(
    @inserted_table_record_dir.join("#{@initialized_pid}.#{Process.pid}.inserted_tables").to_s,
    @inserted_tables.to_a.join(',')
  )
end

#reset_inserted_tables ⇒ Object 



50
51
52
53
54
55
56
57
 
# File 'lib/mysql_rewinder.rb', line 50

def reset_inserted_tables
  unless @initialized_pid == Process.pid
    raise "MysqlRewinder is initialize in process #{@initialized_pid}, but reset_inserted_tables is called in process #{Process.pid}"
  end

  @inserted_tables = Set.new
  FileUtils.rm(Dir.glob(@inserted_table_record_dir.join("#{@initialized_pid}.*.inserted_tables").to_s))
end