32 lines
928 B
Ruby
32 lines
928 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module GithubImport
|
||
|
|
# PageCounter can be used to keep track of the last imported page of a
|
||
|
|
# collection, allowing workers to resume where they left off in the event of
|
||
|
|
# an error.
|
||
|
|
class PageCounter
|
||
|
|
attr_reader :cache_key
|
||
|
|
|
||
|
|
# The base cache key to use for storing the last page number.
|
||
|
|
CACHE_KEY = 'github-importer/page-counter/%{project}/%{collection}'.freeze
|
||
|
|
|
||
|
|
def initialize(project, collection)
|
||
|
|
@cache_key = CACHE_KEY % { project: project.id, collection: collection }
|
||
|
|
end
|
||
|
|
|
||
|
|
# Sets the page number to the given value.
|
||
|
|
#
|
||
|
|
# Returns true if the page number was overwritten, false otherwise.
|
||
|
|
def set(page)
|
||
|
|
Caching.write_if_greater(cache_key, page)
|
||
|
|
end
|
||
|
|
|
||
|
|
# Returns the current value from the cache.
|
||
|
|
def current
|
||
|
|
Caching.read_integer(cache_key) || 1
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|