diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/controllers/agents/files_controller.rb | 2 | ||||
| -rw-r--r-- | app/models/lookup.rb | 2 | ||||
| -rw-r--r-- | app/models/modified.rb | 2 | ||||
| -rw-r--r-- | app/workers/cloud_queries.rb | 2 | ||||
| -rw-r--r-- | app/workers/event_intake.rb | 4 |
5 files changed, 9 insertions, 3 deletions
diff --git a/app/controllers/agents/files_controller.rb b/app/controllers/agents/files_controller.rb index 95421e0..0f493fb 100644 --- a/app/controllers/agents/files_controller.rb +++ b/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) diff --git a/app/models/lookup.rb b/app/models/lookup.rb new file mode 100644 index 0000000..5ccf622 --- /dev/null +++ b/app/models/lookup.rb @@ -0,0 +1,2 @@ +class Lookup < Event +end diff --git a/app/models/modified.rb b/app/models/modified.rb index 46e0f14..d95dd42 100644 --- a/app/models/modified.rb +++ b/app/models/modified.rb @@ -1,2 +1,4 @@ class Modified < Event end +class Lookup < Event +end diff --git a/app/workers/cloud_queries.rb b/app/workers/cloud_queries.rb index fbbe5c4..a1fa68d 100644 --- a/app/workers/cloud_queries.rb +++ b/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! diff --git a/app/workers/event_intake.rb b/app/workers/event_intake.rb index 7e70497..79be810 100644 --- a/app/workers/event_intake.rb +++ b/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 |