write queries to cassandra cluster.

3 files changed, 15 insertions, 0 deletions

diff --git a/app/models/query.rb b/app/models/query.rb
new file mode 100644
index 0000000..24836dc
--- /dev/null
+++ b/app/models/query.rb
@@ -0,0 +1,7 @@
+class Query
+  include Cequel::Record
+
+  key :agent_id, :uuid
+  column :path, :text
+  column :fingerprint, :text
+end
diff --git a/app/workers/cassandra_writer.rb b/app/workers/cassandra_writer.rb
index 0d4057a..c2f56a8 100644
--- a/app/workers/cassandra_writer.rb
+++ b/app/workers/cassandra_writer.rb
@@ -3,6 +3,13 @@ class CassandraWriter
   from_queue "worker.cassandra"
 
   def work(event_json)
+    attributes = JSON.parse(event_json)
+
+    Query.create!(
+      agent_id: attributes['agent_id'],
+      path: attributes['data']['path'],
+      fingerprint: attributes["data"]["fingerprint"],
+    )
     ack!
   end
 end
diff --git a/bin/setup b/bin/setup
index 00615ca..eaf43fb 100755
--- a/bin/setup
+++ b/bin/setup
@@ -37,4 +37,5 @@ Dir.chdir APP_ROOT do
 
   puts "\n== Preparing cassandra =="
   system "bin/rake cequel:keyspace:create"
+  system "bin/rake cequel:migrate"
 end