|
|
- # frozen_string_literal: true
-
- class FastIpMap
- MAX_IPV4_PREFIX = 32
- MAX_IPV6_PREFIX = 128
-
- # @param [Enumerable<IPAddr>] addresses
- def initialize(addresses)
- @fast_lookup = {}
- @ranges = []
-
- # Hash look-up is faster but only works for exact matches, so we split
- # exact addresses from non-exact ones
- addresses.each do |address|
- if (address.ipv4? && address.prefix == MAX_IPV4_PREFIX) || (address.ipv6? && address.prefix == MAX_IPV6_PREFIX)
- @fast_lookup[address.to_s] = true
- else
- @ranges << address
- end
- end
-
- # We're more likely to hit wider-reaching ranges when checking for
- # inclusion, so make sure they're sorted first
- @ranges.sort_by!(&:prefix)
- end
-
- # @param [IPAddr] address
- # @return [Boolean]
- def include?(address)
- @fast_lookup[address.to_s] || @ranges.any? { |cidr| cidr.include?(address) }
- end
- end
|