Class: OrderedHash

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 ⇒ OrderedHash

Returns a new instance of OrderedHash.



64
65
66
67
 
# File 'lib/logstash/filters/ip2proxy.rb', line 64

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.



62
63
64
 
# File 'lib/logstash/filters/ip2proxy.rb', line 62

def hash
  @hash
end

#times_queried ⇒ Object (readonly)

ip -> times queried



61
62
63
 
# File 'lib/logstash/filters/ip2proxy.rb', line 61

def times_queried
  @times_queried
end

Instance Method Details

#add(key) ⇒ Object 



69
70
71
72
73
 
# File 'lib/logstash/filters/ip2proxy.rb', line 69

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

#delete_least_used ⇒ Object 



91
92
93
 
# File 'lib/logstash/filters/ip2proxy.rb', line 91

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

#first_pile_with_something ⇒ Object 



95
96
97
 
# File 'lib/logstash/filters/ip2proxy.rb', line 95

def first_pile_with_something
  hash[hash.keys.min]
end

#increment(key) ⇒ Object 



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

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

#reorder(key) ⇒ Object 



75
76
77
78
79
80
81
82
83
 
# File 'lib/logstash/filters/ip2proxy.rb', line 75

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