Class: LogStash::Util::WrappedSynchronousQueue::ReadBatch

Inherits:
Object
  • Object
show all
Defined in:
lib/logstash/util/wrapped_synchronous_queue.rb

Instance Method Summary collapse

Constructor Details

#initialize(queue, size, wait) ⇒ ReadBatch

Returns a new instance of ReadBatch.



178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 178

def initialize(queue, size, wait)
  @queue = queue
  @size = size
  @wait = wait

  @originals = Hash.new

  # TODO: disabled for https://github.com/elastic/logstash/issues/6055 - will have to properly refactor
  # @cancelled = Hash.new

  @generated = Hash.new
  @iterating_temp = Hash.new
  @iterating = false # Atomic Boolean maybe? Although batches are not shared across threads
  @acked_batch = nil
end

Instance Method Details

#cancel(event) ⇒ Object 



215
216
217
218
219
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 215

def cancel(event)
  # TODO: disabled for https://github.com/elastic/logstash/issues/6055 - will have to properly refactor
  raise("cancel is unsupported")
  # @cancelled[event] = true
end

#cancelled_size ⇒ Object 



255
256
257
258
259
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 255

def cancelled_size
# TODO: disabled for https://github.com/elastic/logstash/issues/6055 = will have to properly refactor
raise("cancelled_size is unsupported ")
  # @cancelled.size
end

#each(&blk) ⇒ Object 



227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 227

def each(&blk)
  # take care not to cause @originals or @generated to change during iteration
  @iterating = true

  # below the checks for @cancelled.include?(e) have been replaced by e.cancelled?
  # TODO: for https://github.com/elastic/logstash/issues/6055 = will have to properly refactor
  @originals.each do |e, _|
    blk.call(e) unless e.cancelled?
  end
  @generated.each do |e, _|
    blk.call(e) unless e.cancelled?
  end
  @iterating = false
  update_generated
end

#filtered_size ⇒ Object 



251
252
253
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 251

def filtered_size
  @originals.size + @generated.size
end

#merge(event) ⇒ Object 



203
204
205
206
207
208
209
210
211
212
213
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 203

def merge(event)
  return if event.nil? || @originals.key?(event)
  # take care not to cause @generated to change during iteration
  # @iterating_temp is merged after the iteration
  if iterating?
    @iterating_temp[event] = true
  else
    # the periodic flush could generate events outside of an each iteration
    @generated[event] = true
  end
end

#read_next ⇒ Object 



194
195
196
197
198
199
200
201
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 194

def read_next
  @size.times do |t|
    event = @queue.poll(@wait)
    return if event.nil? # queue poll timed out

    @originals[event] = true
  end
end

#size ⇒ Object 



243
244
245
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 243

def size
  filtered_size
end

#starting_size ⇒ Object 



247
248
249
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 247

def starting_size
  @originals.size
end

#to_a ⇒ Object 



221
222
223
224
225
 
# File 'lib/logstash/util/wrapped_synchronous_queue.rb', line 221

def to_a
  events = []
  each {|e| events << e}
  events
end