mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-27 17:02:55 +08:00
53d7a756d6
There are concrete implementations for a simple set, a key-value store, and nested sets with 2 or 3 keys. The API stays the same for all implementations and the performances is more or less the same as without the wrapper (at least with YJIT enabled).
58 lines
1.2 KiB
Ruby
Vendored
58 lines
1.2 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module Migrations::SetStore
|
|
class TwoKeySet
|
|
include Interface
|
|
|
|
def initialize
|
|
@store = {}
|
|
end
|
|
|
|
def add(key1, key2, value)
|
|
h1 = @store[key1] ||= {}
|
|
set = h1[key2] ||= Set.new
|
|
set.add(value)
|
|
self
|
|
end
|
|
|
|
def add?(key1, key2, value)
|
|
h1 = @store[key1] ||= {}
|
|
set = h1[key2] ||= Set.new
|
|
!!set.add?(value)
|
|
end
|
|
|
|
def include?(key1, key2, value)
|
|
h1 = @store[key1] or return false
|
|
set = h1[key2] or return false
|
|
set.include?(value)
|
|
end
|
|
|
|
def bulk_add(records)
|
|
current_key1 = nil
|
|
current_key2 = nil
|
|
current_h1 = nil
|
|
current_set = nil
|
|
|
|
records.each do |record|
|
|
key1, key2, value = record
|
|
|
|
if key1 != current_key1
|
|
current_key1 = key1
|
|
current_h1 = @store[key1] ||= {}
|
|
current_key2 = key2
|
|
current_set = current_h1[key2] ||= Set.new
|
|
elsif key2 != current_key2
|
|
current_key2 = key2
|
|
current_set = current_h1[key2] ||= Set.new
|
|
end
|
|
|
|
current_set.add(value)
|
|
end
|
|
nil
|
|
end
|
|
|
|
def empty?
|
|
@store.empty?
|
|
end
|
|
end
|
|
end
|
