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:, occupied_segments: []) ⇒ 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
60
61
62
63
64
# 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:, occupied_segments: [])
  hot_bucket_set = hot_buckets.to_set
  segments = []
  middle_filler_index = 0
  index = 0

  while index < buckets.length
    bucket = buckets[index]
    occupant = OccupiedWindow.covering(occupied_segments, bucket)

    if occupant
      segments << occupant unless segments.any? { |segment| segment.name == occupant.name }
      index = OccupiedWindow.next_bucket_index(buckets, occupant.range_end, from: index)
      next
    end

    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]) && OccupiedWindow.covering(occupied_segments, buckets[index]).nil?
      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

  OccupiedWindow.finish_with_high_end(
    segments,
    occupied_segments: occupied_segments,
    table_name: table_name,
    active_end: active_end
  )
end