lib/fluent/plugin/out_cassandra_driver.rb in fluent-plugin-cassandra-driver-0.0.23 vs lib/fluent/plugin/out_cassandra_driver.rb in fluent-plugin-cassandra-driver-1.0.0

@@ -92,11 +92,11 @@
           record_key, type = mapping.first
         else
           record_key, type = column_family_key, mapping
         end
 
-        value = record[record_key]
+        value = record[record_key.to_s]
 
         case type
           when :integer
             value = value.to_i
           when :timeuuid
@@ -110,10 +110,12 @@
 
         [column_family_key.to_s, value]
       }.to_h
 
       self.schema.each { |column_family_key, mapping|
-        record.delete(mapping.class == Hash ? mapping.first.first : column_family_key)
+        record_key = mapping.class == Hash ? mapping.first.first : column_family_key
+
+        record.delete(record_key.to_s)
       } if self.pop_data_keys
 
       # if we have one more data in record and json column
       # then store all remaining data into that column
       values[self.json_column] = record.to_json if self.json_column and record.length > 0