Class: DatabaseRecorder::Recording

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Defined in:
lib/database_recorder/recording.rb

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options: {}) ⇒ Recording

Returns a new instance of Recording.



10
11
12
13
14
15
16
17
18
 
# File 'lib/database_recorder/recording.rb', line 10

def initialize(options: {})
  (@@instances ||= {})[Process.pid] = self
  @cache = nil
  @entities = []
  @options = options
  @queries = []
  @search_index = 0
  @@prepared_queries ||= {}
end

Instance Attribute Details

#cache ⇒ Object

Returns the value of attribute cache.



7
8
9
 
# File 'lib/database_recorder/recording.rb', line 7

def cache
  @cache
end

#entities ⇒ Object

Returns the value of attribute entities.



7
8
9
 
# File 'lib/database_recorder/recording.rb', line 7

def entities
  @entities
end

#from_cache ⇒ Object (readonly)

Returns the value of attribute from_cache.



8
9
10
 
# File 'lib/database_recorder/recording.rb', line 8

def from_cache
  @from_cache
end

#options ⇒ Object (readonly)

Returns the value of attribute options.



8
9
10
 
# File 'lib/database_recorder/recording.rb', line 8

def options
  @options
end

#prepared_queries ⇒ Object (readonly)

Returns the value of attribute prepared_queries.



8
9
10
 
# File 'lib/database_recorder/recording.rb', line 8

def prepared_queries
  @prepared_queries
end

#queries ⇒ Object (readonly)

Returns the value of attribute queries.



8
9
10
 
# File 'lib/database_recorder/recording.rb', line 8

def queries
  @queries
end

#started ⇒ Object (readonly)

Returns the value of attribute started.



8
9
10
 
# File 'lib/database_recorder/recording.rb', line 8

def started
  @started
end

Class Method Details

.current_instance ⇒ Object 



79
80
81
 
# File 'lib/database_recorder/recording.rb', line 79

def current_instance
  (@@instances ||= {})[Process.pid]
end

.started? ⇒ Boolean

Returns:

  • (Boolean) 


83
84
85
 
# File 'lib/database_recorder/recording.rb', line 83

def started?
  current_instance&.started
end

Instance Method Details

#cached_query_for(sql) ⇒ Object 



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

def cached_query_for(sql)
  current = @search_index
  match = cache[@search_index..].find do |item|
    current += 1
    item['sql'] == sql
  end
  return unless match

  @search_index = current
  match

  # cache.shift # TMP
end

#new_entity(model:, id:) ⇒ Object 



34
35
36
 
# File 'lib/database_recorder/recording.rb', line 34

def new_entity(model:, id:)
  @entities.push('model' => model, 'id' => id)
end

#pull_entity ⇒ Object 



38
39
40
 
# File 'lib/database_recorder/recording.rb', line 38

def pull_entity
  @entities.shift
end

#push(sql:, name: nil, binds: nil, result: nil, source: nil) ⇒ Object 



42
43
44
45
 
# File 'lib/database_recorder/recording.rb', line 42

def push(sql:, name: nil, binds: nil, result: nil, source: nil)
  query = { 'name' => name, 'sql' => sql, 'binds' => binds, 'result' => result }.compact
  @queries.push(query)
end

#push_prepared(name: nil, sql: nil, binds: nil, result: nil, source: nil) ⇒ Object 



47
48
49
50
 
# File 'lib/database_recorder/recording.rb', line 47

def push_prepared(name: nil, sql: nil, binds: nil, result: nil, source: nil)
  query = { 'name' => name, 'sql' => sql, 'binds' => binds, 'result' => result }.compact
  @@prepared_queries[name || sql] = query
end

#start ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/database_recorder/recording.rb', line 52

def start
  @started = true
  storage = Config.storage&.new(self, name: options[:name])
  @from_cache = storage&.load
  yield
  storage&.save unless from_cache
  @started = false
  result = { current_queries: queries.map { _1['sql'] } }
  result[:stored_queries] = cache.map { _1['sql'] } if from_cache
  result
end

#update_prepared(name: nil, sql: nil, binds: nil, result: nil, source: nil) ⇒ Object 



64
65
66
67
68
69
70
71
 
# File 'lib/database_recorder/recording.rb', line 64

def update_prepared(name: nil, sql: nil, binds: nil, result: nil, source: nil)
  query = @@prepared_queries[name || sql]
  query['sql'] = sql if sql
  query['binds'] = binds if binds
  query['result'] = result if result
  @queries.push(query)
  query
end