Class: IP2ProxyOrderedHash

Inherits:
Object
  • Object
show all
Defined in:
lib/logstash/filters/ip2proxy.rb

Overview

class LogStash::Filters::IP2Proxy

Constant Summary collapse

ONE = 
1

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ IP2ProxyOrderedHash

Returns a new instance of IP2ProxyOrderedHash.



81
82
83
84
 
# File 'lib/logstash/filters/ip2proxy.rb', line 81

def initialize
  @times_queried = Hash.new(0) # ip -> times queried
  @hash = {} # number of hits -> array of ips
end

Instance Attribute Details

#hash ⇒ Object (readonly)

Returns the value of attribute hash.



79
80
81
 
# File 'lib/logstash/filters/ip2proxy.rb', line 79

def hash
  @hash
end

#times_queried ⇒ Object (readonly)

ip -> times queried



78
79
80
 
# File 'lib/logstash/filters/ip2proxy.rb', line 78

def times_queried
  @times_queried
end

Instance Method Details

#add(key) ⇒ Object 



86
87
88
89
90
 
# File 'lib/logstash/filters/ip2proxy.rb', line 86

def add(key)
  hash[ONE] ||= []
  hash[ONE] << key
  times_queried[key] = ONE
end

#delete_least_used ⇒ Object 



108
109
110
 
# File 'lib/logstash/filters/ip2proxy.rb', line 108

def delete_least_used
  first_pile_with_something.shift.tap { |key| times_queried.delete(key) }
end

#first_pile_with_something ⇒ Object 



112
113
114
 
# File 'lib/logstash/filters/ip2proxy.rb', line 112

def first_pile_with_something
  hash[hash.keys.min]
end

#increment(key) ⇒ Object 



102
103
104
105
106
 
# File 'lib/logstash/filters/ip2proxy.rb', line 102

def increment(key)
  add(key) unless times_queried.has_key?(key)
  reorder(key)
  times_queried[key] += 1
end

#reorder(key) ⇒ Object 



92
93
94
95
96
97
98
99
100
 
# File 'lib/logstash/filters/ip2proxy.rb', line 92

def reorder(key)
  number_of_queries = times_queried[key]

  hash[number_of_queries].delete(key)
  hash.delete(number_of_queries) if hash[number_of_queries].empty?

  hash[number_of_queries + 1] ||= []
  hash[number_of_queries + 1] << key
end