Commit 17306e8

mo khan <mo@mokhan.ca>
2015-02-08 23:15:56
capitalize event type for polymorphic lookup.
1 parent ef0868d
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