66 lines
1.8 KiB
Ruby
66 lines
1.8 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
module Gitlab
|
||
|
|
module PhabricatorImport
|
||
|
|
module Cache
|
||
|
|
class Map
|
||
|
|
def initialize(project)
|
||
|
|
@project = project
|
||
|
|
end
|
||
|
|
|
||
|
|
def get_gitlab_model(phabricator_id)
|
||
|
|
cached_info = get(phabricator_id)
|
||
|
|
return unless cached_info[:classname] && cached_info[:database_id]
|
||
|
|
|
||
|
|
cached_info[:classname].constantize.find_by_id(cached_info[:database_id])
|
||
|
|
end
|
||
|
|
|
||
|
|
def set_gitlab_model(object, phabricator_id)
|
||
|
|
set(object.class, object.id, phabricator_id)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
attr_reader :project
|
||
|
|
|
||
|
|
def set(klass_name, object_id, phabricator_id)
|
||
|
|
key = cache_key_for_phabricator_id(phabricator_id)
|
||
|
|
|
||
|
|
redis.with do |r|
|
||
|
|
r.multi do |multi|
|
||
|
|
multi.mapped_hmset(key,
|
||
|
|
{ classname: klass_name, database_id: object_id })
|
||
|
|
multi.expire(key, timeout)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def get(phabricator_id)
|
||
|
|
key = cache_key_for_phabricator_id(phabricator_id)
|
||
|
|
|
||
|
|
redis.with do |r|
|
||
|
|
r.pipelined do |pipe|
|
||
|
|
# Extend the TTL when a key was
|
||
|
|
pipe.expire(key, timeout)
|
||
|
|
pipe.mapped_hmget(key, :classname, :database_id)
|
||
|
|
end.last
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def cache_key_for_phabricator_id(phabricator_id)
|
||
|
|
"#{Redis::Cache::CACHE_NAMESPACE}/phabricator-import/#{project.id}/#{phabricator_id}"
|
||
|
|
end
|
||
|
|
|
||
|
|
def redis
|
||
|
|
Gitlab::Redis::Cache
|
||
|
|
end
|
||
|
|
|
||
|
|
def timeout
|
||
|
|
# Setting the timeout to the same one as we do for clearing stuck jobs
|
||
|
|
# this makes sure all cache is available while the import is running.
|
||
|
|
StuckImportJobsWorker::IMPORT_JOBS_EXPIRATION
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|