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
|
# frozen_string_literal: true
module WebHooks
module HasWebHooks
WEB_HOOK_CACHE_EXPIRY = 1.hour
def any_hook_failed?
hooks.disabled.exists?
end
def web_hook_failure_redis_key
"any_web_hook_failed:#{id}"
end
def last_failure_redis_key
"web_hooks:last_failure:#{self.class.name.underscore}-#{id}"
end
def get_web_hook_failure
Gitlab::Redis::SharedState.with do |redis|
current = redis.get(web_hook_failure_redis_key)
Gitlab::Utils.to_boolean(current) if current
end
end
def fetch_web_hook_failure
Gitlab::Redis::SharedState.with do |_redis|
current = get_web_hook_failure
next current unless current.nil?
cache_web_hook_failure
end
end
def cache_web_hook_failure(state = any_hook_failed?)
Gitlab::Redis::SharedState.with do |redis|
redis.set(web_hook_failure_redis_key, state.to_s, ex: WEB_HOOK_CACHE_EXPIRY)
state
end
end
def last_webhook_failure
last_failure = Gitlab::Redis::SharedState.with do |redis|
redis.get(last_failure_redis_key)
end
DateTime.parse(last_failure) if last_failure
end
def update_last_webhook_failure(hook)
if hook.executable?
cache_web_hook_failure if get_web_hook_failure # may need update
else
cache_web_hook_failure(true) # definitely failing, no need to check
Gitlab::Redis::SharedState.with do |redis|
last_failure_key = last_failure_redis_key
time = Time.current.utc.iso8601
prev = redis.get(last_failure_key)
redis.set(last_failure_key, time) if !prev || prev < time
end
end
end
end
end
|
