gitlab-org--gitlab-foss/db/post_migrate/20200930144340_set_job_waiter_ttl.rb

31 lines
616 B
Ruby

# frozen_string_literal: true
class SetJobWaiterTtl < ActiveRecord::Migration[6.0]
DOWNTIME = false
SCRIPT = <<~LUA.freeze
if redis.call("ttl", KEYS[1]) < 0 then
redis.call("expire", KEYS[1], 21600)
end
LUA
def up
Gitlab::Redis::SharedState.with do |redis|
cursor_init = '0'
cursor = cursor_init
loop do
cursor, keys = redis.scan(cursor, match: 'gitlab:job_waiter:*')
redis.pipelined do |redis|
keys.each { |k| redis.eval(SCRIPT, keys: [k]) }
end
break if cursor == cursor_init
end
end
end
def down
end
end