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
|
# frozen_string_literal: true
module ClickHouse
class RecordSyncContext
attr_reader :last_record_id, :last_processed_id, :total_record_count, :record_count_in_current_batch
def initialize(
last_record_id:, max_records_per_batch:,
runtime_limiter: Gitlab::Metrics::RuntimeLimiter.new)
@last_record_id = last_record_id
@runtime_limiter = runtime_limiter
@max_records_per_batch = max_records_per_batch
@last_processed_id = nil
@record_count_in_current_batch = 0
@total_record_count = 0
end
delegate :over_time?, to: :@runtime_limiter
def new_batch!
@record_count_in_current_batch = 0
end
def no_more_records!
@no_more_records = true
end
def no_more_records?
!!@no_more_records
end
def last_processed_id=(value)
@record_count_in_current_batch += 1
@total_record_count += 1
@last_processed_id = value
@last_record_id = value
end
def record_limit_reached?
@record_count_in_current_batch == @max_records_per_batch
end
end
end
|
