blob: 9a52a4f6e23a1233ce1736457ff2cff8c0b27e0e (
plain)
1
2
3
4
5
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
# frozen_string_literal: true
module Gitlab
module Database
module QueryAnalyzers
class Base
# `Exception` to ensure that is not easily rescued when running in test env
QueryAnalyzerError = Class.new(Exception) # rubocop:disable Lint/InheritException
def self.suppressed?
Thread.current[self.suppress_key] || @suppress_in_rspec
end
def self.requires_tracking?(parsed)
false
end
def self.suppress=(value)
Thread.current[self.suppress_key] = value
end
# The other suppress= method stores the
# value in Thread.current because it is
# meant to work in a multi-threaded puma
# environment but this does not work
# correctly in capybara tests where we
# suppress in the rspec runner context but
# this does not take effect in the puma
# thread. As such we just suppress
# globally in RSpec since we don't run
# different tests concurrently.
class << self
attr_writer :suppress_in_rspec
end
def self.with_suppressed(value = true, &blk)
previous = self.suppressed?
self.suppress = value
yield
ensure
self.suppress = previous
end
def self.begin!
Thread.current[self.context_key] = {}
end
def self.end!
Thread.current[self.context_key] = nil
end
def self.context
Thread.current[self.context_key]
end
def self.enabled?
raise NotImplementedError
end
def self.analyze(parsed)
raise NotImplementedError
end
def self.context_key
@context_key ||= "analyzer_#{self.analyzer_key}_context".to_sym
end
def self.suppress_key
@suppress_key ||= "analyzer_#{self.analyzer_key}_suppressed".to_sym
end
def self.analyzer_key
@analyzer_key ||= self.name.demodulize.underscore.to_sym
end
end
end
end
end
|