Module: PartitionGardener::Layout::ZoneSegments

Defined in:
lib/partition_gardener/layout/zone_segments.rb

Class Method Summary collapse

Class Method Details

.build_filler_and_hot_segments(table_name:, buckets:, hot_buckets:, active_start:, active_end:, hot_bucket_name:, bucket_end:) ⇒ Object



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
# File 'lib/partition_gardener/layout/zone_segments.rb', line 6

def build_filler_and_hot_segments(table_name:, buckets:, hot_buckets:, active_start:, active_end:, hot_bucket_name:, bucket_end:)
  hot_bucket_set = hot_buckets.to_set
  segments = []
  middle_filler_index = 0
  index = 0

  while index < buckets.length
    bucket = buckets[index]

    if hot_bucket_set.include?(bucket)
      segments << Plan::Segment.new(
        name: hot_bucket_name.call(bucket),
        range_start: bucket,
        range_end: bucket_end.call(bucket),
        kind: :hot_bucket
      )
      index += 1
      next
    end

    run_start = bucket
    while index < buckets.length && !hot_bucket_set.include?(buckets[index])
      index += 1
    end
    run_end = (index < buckets.length) ? buckets[index] : active_end

    segment_name = if run_start == active_start
      Naming.current_partition_name(table_name)
    else
      middle_filler_index += 1
      if middle_filler_index == 1
        Naming.open_partition_name(table_name)
      else
        "#{table_name}_open_#{middle_filler_index}"
      end
    end

    segments << Plan::Segment.new(
      name: segment_name,
      range_start: run_start,
      range_end: run_end,
      kind: :filler
    )
  end

  segments << Plan::Segment.new(
    name: Naming.future_partition_name(table_name),
    range_start: active_end,
    range_end: :max,
    kind: :future
  )

  segments
end