debian-mirror-gitlab/lib/gitlab/database/query_analyzers/base.rb

65 lines
1.5 KiB
Ruby
Raw Normal View History

2021-12-11 22:18:48 +05:30
# frozen_string_literal: true
module Gitlab
module Database
module QueryAnalyzers
class Base
2022-01-26 12:08:38 +05:30
# `Exception` to ensure that is not easily rescued when running in test env
QueryAnalyzerError = Class.new(Exception) # rubocop:disable Lint/InheritException
2021-12-11 22:18:48 +05:30
def self.suppressed?
Thread.current[self.suppress_key]
end
2022-01-26 12:08:38 +05:30
def self.requires_tracking?(parsed)
false
end
2021-12-11 22:18:48 +05:30
def self.suppress=(value)
Thread.current[self.suppress_key] = value
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
2022-04-04 11:22:00 +05:30
@context_key ||= "analyzer_#{self.analyzer_key}_context".to_sym
2021-12-11 22:18:48 +05:30
end
def self.suppress_key
2022-04-04 11:22:00 +05:30
@suppress_key ||= "analyzer_#{self.analyzer_key}_suppressed".to_sym
end
def self.analyzer_key
@analyzer_key ||= self.name.demodulize.underscore.to_sym
2021-12-11 22:18:48 +05:30
end
end
end
end
end