fluent-plugin-cassandra-driver 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7ac22674557433f0ea85c00863cee8ac7c59be11
-  data.tar.gz: 7e2f89d4571c7cc7801e6386458212ea3a2f6d33
+  metadata.gz: fca7029206d19969af19c9cf4efd8ea790039d83
+  data.tar.gz: 2aac1bc2b292515c8257b9d3a2d9ef0718238398
 SHA512:
-  metadata.gz: 5c9d9fac0898971fac9967c015af167e0155dd0fd8cffe4fa12a1c1005c94c5e44c6000609a428fe45e1c9d5ebe9d8d64b666577830d8734a776b44040e1c0d6
-  data.tar.gz: 90128ed2a057a4618d1ce3b9104a8f635daaf7a4aeeb69e9b46d9e6dc6dcdeb763cb86238803b11ec557f0e8588270a088431ae6d008d52d8dc5e2ab533c3c56
+  metadata.gz: 37410f2a120900a86717907432110bfd39c1f56396dc99f531a023464325764448b817a564812c312c23937001226cad6be269b6cfd3309ea6bb1c7a01ca34b7
+  data.tar.gz: 100aadb2db5c245b5b387f57f001e7124fd15835de9be4862dfea64da1faaa2c98591c14c8920c474ba79ae22339c20ef899009455a0c3dfb19d988901e0595e

data/VERSION

@@ -1 +1 @@
-0.0.2
+0.0.3

data/fluent-plugin-cassandra-driver.gemspec

@@ -2,11 +2,11 @@
 # DO NOT EDIT THIS FILE DIRECTLY
 # Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
 # -*- encoding: utf-8 -*-
-# stub: fluent-plugin-cassandra-driver 0.0.2 ruby lib
+# stub: fluent-plugin-cassandra-driver 0.0.3 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "fluent-plugin-cassandra-driver".freeze
-  s.version = "0.0.2"
+  s.version = "0.0.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -27,6 +27,8 @@ Gem::Specification.new do |s|
     "Rakefile",
     "VERSION",
     "fluent-plugin-cassandra-driver.gemspec",
+    "lib/fluent/plugin/mutators/string_mutator.rb",
+    "lib/fluent/plugin/mutators/timeuuid_mutator.rb",
     "lib/fluent/plugin/out_cassandra_driver.rb",
     "spec/spec.opts",
     "spec/spec_helper.rb"

data/lib/fluent/plugin/mutators/string_mutator.rb

@@ -0,0 +1,7 @@
+module CassandraDriver
+  class StringMutator
+    def mutate(value)
+      "'#{value}'"
+    end
+  end
+end

data/lib/fluent/plugin/mutators/timeuuid_mutator.rb

@@ -0,0 +1,11 @@
+require 'cassandra'
+
+module CassandraDriver
+  class TimeuuidMutator
+    @generator = Cassandra::Uuid::Generator.new
+
+    def mutate(value)
+      @generator.at(value).to_s
+    end
+  end
+end

data/lib/fluent/plugin/out_cassandra_driver.rb

@@ -1,6 +1,8 @@
 require 'cassandra'
 require 'msgpack'
 require 'json'
+require 'mutators/string_mutator'
+require 'mutators/timeuuid_mutator'
 
 module Fluent
   class CassandraCqlOutput < BufferedOutput
@@ -60,7 +62,7 @@ module Fluent
 
     def write(chunk)
       chunk.msgpack_each { |record|
-        $log.info "Sending a new record to Cassandra: #{record.to_json}"
+        $log.debug "Sending a new record to Cassandra: #{record.to_json}"
 
         values = build_insert_values_string(self.schema.keys, self.data_keys, record, self.pop_data_keys)
 
@@ -76,6 +78,8 @@ module Fluent
 
     private
 
+    @mutators_map = {:string => ::CassandraDriver::StringMutator.new, :timeuuid => ::CassandraDriver::TimeuuidMutator.new}
+
     def get_session(hosts, keyspace)
       cluster = ::Cassandra.cluster(hosts: hosts)
 
@@ -84,11 +88,11 @@ module Fluent
 
     def build_insert_values_string(schema_keys, data_keys, record, pop_data_keys)
       values = data_keys.map.with_index do |key, index|
-        if pop_data_keys
-          self.schema[schema_keys[index]] == :string ? "'#{record.delete(key)}'" : record.delete(key)
-        else
-          self.schema[schema_keys[index]] == :string ? "'#{record[key]}'" : record[key]
-        end
+        value = pop_data_keys ? record.delete(key) : record[key]
+        type = self.schema[schema_keys[index]]
+        mutator = @mutators_map.key?(type) ? @mutators_map[type] : nil
+
+        mutator ? mutator.mutate(value) : value
       end
 
       # if we have one more schema key than data keys,

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-cassandra-driver
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Yaroslav Lukyanov
@@ -125,6 +125,8 @@ files:
 - Rakefile
 - VERSION
 - fluent-plugin-cassandra-driver.gemspec
+- lib/fluent/plugin/mutators/string_mutator.rb
+- lib/fluent/plugin/mutators/timeuuid_mutator.rb
 - lib/fluent/plugin/out_cassandra_driver.rb
 - spec/spec.opts
 - spec/spec_helper.rb