|
require 'location'
# Events are how Huginn Agents communicate and log information about the world. Events can be emitted and received by
# Agents. They contain a serialized `payload` of arbitrary JSON data, as well as optional `lat`, `lng`, and `expires_at`
# fields.
class Event < ActiveRecord::Base
include JSONSerializedField
include LiquidDroppable
acts_as_mappable
json_serialize :payload
belongs_to :user, optional: true
belongs_to :agent, :counter_cache => true
has_many :agent_logs_as_inbound_event, :class_name => "AgentLog", :foreign_key => :inbound_event_id, :dependent => :nullify
has_many :agent_logs_as_outbound_event, :class_name => "AgentLog", :foreign_key => :outbound_event_id, :dependent => :nullify
scope :recent, lambda { |timespan = 12.hours.ago|
where("events.created_at > ?", timespan)
}
after_create :update_agent_last_event_at
after_create :possibly_propagate
scope :expired, lambda {
where("expires_at IS NOT NULL AND expires_at < ?", Time.now)
}
scope :with_location, -> {
where.not(lat: nil).where.not(lng: nil)
}
def location
@location ||= Location.new(
# lat and lng are BigDecimal, but converted to Float by the Location class
lat: lat,
lng: lng,
radius:
begin
h = payload[:horizontal_accuracy].presence
v = payload[:vertical_accuracy].presence
if h && v
(h.to_f + v.to_f) / 2
else
(h || v || payload[:accuracy]).to_f
end
end,
course: payload[:course],
speed: payload[:speed].presence)
end
def location=(location)
case location
when nil
self.lat = self.lng = nil
return
when Location
else
location = Location.new(location)
end
self.lat, self.lng = location.lat, location.lng
location
end
# Emit this event again, as a new Event.
def reemit!
agent.create_event :payload => payload, :lat => lat, :lng => lng
end
# Look for Events whose `expires_at` is present and in the past. Remove those events and then update affected Agents'
# `events_counts` cache columns. This method is called by bin/schedule.rb periodically.
def self.cleanup_expired!
affected_agents = Event.expired.group("agent_id").pluck(:agent_id)
Event.expired.delete_all
Agent.where(:id => affected_agents).update_all "events_count = (select count(*) from events where agent_id = agents.id)"
end
protected
def update_agent_last_event_at
agent.touch :last_event_at
end
def possibly_propagate
#immediately schedule agents that want immediate updates
propagate_ids = agent.receivers.where(:propagate_immediately => true).pluck(:id)
Agent.receive!(:only_receivers => propagate_ids) unless propagate_ids.empty?
end
end
class EventDrop
def initialize(object)
@payload = object.payload
super
end
def before_method(key)
@payload[key]
end
def each(&block)
@payload.each(&block)
end
def agent
@payload.fetch(__method__) {
@object.agent
}
end
def created_at
@payload.fetch(__method__) {
@object.created_at
}
end
def _location_
@object.location
end
def as_json
{location: _location_.as_json, agent: @object.agent.to_liquid.as_json, payload: @payload.as_json, created_at: created_at.as_json}
end
end
|