Commit 17306e8
Changed files (5)
app
controllers
agents
models
workers
app/controllers/agents/files_controller.rb
@@ -13,7 +13,7 @@ module Agents
@file = Disposition.find_by(fingerprint: params[:id])
message = {
agent_id: params[:id],
- type: 'lookup',
+ type: :lookup,
data: params[:data]
}
Publisher.publish("events.scanned.#{@agent.id}", message)
app/models/lookup.rb
@@ -0,0 +1,2 @@
+class Lookup < Event
+end
app/models/modified.rb
@@ -1,2 +1,4 @@
class Modified < Event
end
+class Lookup < Event
+end
app/workers/cloud_queries.rb
@@ -11,7 +11,7 @@ class CloudQueries
fingerprint = attributes["data"]["fingerprint"]
disposition = Disposition.find_by(fingerprint: fingerprint)
- Disposition.create!(fingerprint: fingerprint, state: :unknown) if disposition.nil?
+ disposition = Disposition.create!(fingerprint: fingerprint, state: :unknown) if disposition.nil?
FingerprintLookupJob.perform_later(fingerprint) if disposition.state == :unknown
ack!
app/workers/event_intake.rb
@@ -6,7 +6,9 @@ class EventIntake
def work(event_json)
logger.info event_json
- event = Event.create!(JSON.parse(event_json))
+ json = JSON.parse(event_json)
+ json['type'] = json['type'].capitalize
+ event = Event.create!(json)
logger.info("Create Event: #{event.id}")
ack!
end