Class: Vivarium::MapStore

Inherits:

Object

• Object
• Vivarium::MapStore

Defined in:

lib/vivarium.rb

Instance Method Summary

• #drain_events ⇒ Object
• #initialize(pin_dir: Vivarium.bpf_pin_dir) ⇒ MapStore constructor

  A new instance of MapStore.

• #register_pid(pid) ⇒ Object
• #unregister_pid(pid) ⇒ Object

Constructor Details

#initialize(pin_dir: Vivarium.bpf_pin_dir) ⇒ MapStore

Returns a new instance of MapStore.

# File 'lib/vivarium.rb', line 163

def initialize(pin_dir: Vivarium.bpf_pin_dir)
  @pin_dir = pin_dir
  @config_root_targets = RbBCC::HashTable.from_pin(
    File.join(@pin_dir, "config_root_targets"),
    "unsigned int",
    "unsigned char",
    keysize: 4,
    leafsize: 1
  )
  @config_spawned_targets = RbBCC::HashTable.from_pin(
    File.join(@pin_dir, "config_spawned_targets"),
    "unsigned int",
    "unsigned char",
    keysize: 4,
    leafsize: 1
  )
  @event_invoked = RbBCC::ArrayTable.from_pin(
    File.join(@pin_dir, "event_invoked"),
    "unsigned int",
    "char[#{EVENT_STRUCT_SIZE}]",
    keysize: 4,
    leafsize: EVENT_STRUCT_SIZE
  )
  @event_write_pos = RbBCC::ArrayTable.from_pin(
    File.join(@pin_dir, "event_write_pos"),
    "unsigned int",
    "unsigned int",
    keysize: 4,
    leafsize: 4
  )
rescue StandardError => e
  raise Error, "failed to open pinned maps under #{@pin_dir}: #{e.class}: #{e.message}"
end

Instance Method Details

#drain_events ⇒ Object

# File 'lib/vivarium.rb', line 208

def drain_events
  events = []
  EVENT_CAPACITY.times do |idx|
    ptr = @event_invoked[idx]
    next unless ptr

    event = Event.from_binary(ptr[0, EVENT_STRUCT_SIZE])
    next if event.empty?

    events << event
    @event_invoked[idx] = zeroed_event_ptr
  end

  @event_write_pos[0] = 0
  events
end

#register_pid(pid) ⇒ Object

# File 'lib/vivarium.rb', line 197

def register_pid(pid)
  @config_root_targets[pid] = 1
end

#unregister_pid(pid) ⇒ Object

# File 'lib/vivarium.rb', line 201

def unregister_pid(pid)
  @config_root_targets.delete(pid)
  @config_spawned_targets.clear
rescue KeyError
  nil
end