cassandra_db 0.1.1 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 361482a345f38fb4bf81842cf9042d8b701e9be9fafc49882a56d04d2e004437
-  data.tar.gz: 94c351a9a73e97a358ff0bc04133602d2239fd6fef965f37e59aa4876334ab43
+  metadata.gz: ed2d4c361a64638cb1efd26d311f63a7867bcafcb5e7efdea0c72e72bddce90c
+  data.tar.gz: 5376a2d931db76bd7a1e7d71217c3351c8298cc185dc181901c20ccf433c2ac3
 SHA512:
-  metadata.gz: 9aec4396f0c99714d314d9c31df7849cddbb07c9e9dfc4338d2a729ee8b685b4342477b64a64ddf1e0aae3253f397b0755fdfc6aafef8d9b22314e914b092e17
-  data.tar.gz: 7524878056971f8f493d9dd85ca5b149915b42a2764ddfc75d78f9e7b44f35a3c7d59b966d4613352a2c0bc39059a7d3cd5f94613cd84aea979f8658ffa6c313
+  metadata.gz: 8bdc20dc8bcfa76967847f32327a5f42f334063ec4a1a424ecce8c44e69a559fb0b77940a4e4e5d81e1cf55ed5ef77ce7fa62e4cdec277cb56877df9fe1734a4
+  data.tar.gz: 3c0624835af96bf2b1c92a54052612303e280a320c3a9cf21afe54d4c5af1853073a9aa30969b48a0d5fa25db29183f27313cdead32bbbf6f8f913c006bc010e

data/README.md

@@ -70,6 +70,15 @@ dataset.map
 dataset.all
 ```
 
+### Insert, Update, Delete
+```ruby
+dataset.insert id: 1, name: 'Name 1'
+
+dataset.where(id: 1).update name: 'Name 2'
+
+dataset.where(id: 1).delete
+```
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/gabynaiman/cassandra_db.

data/cassandra_db.gemspec

@@ -28,6 +28,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'minitest', '~> 5.0', '< 5.11'
   spec.add_development_dependency 'minitest-colorin', '~> 0.1'
   spec.add_development_dependency 'minitest-line', '~> 0.6'
+  spec.add_development_dependency 'minitest-great_expectations', '~> 0.0', '>= 0.0.5'
   spec.add_development_dependency 'simplecov', '~> 0.14'
   spec.add_development_dependency 'coveralls', '~> 0.8'
   spec.add_development_dependency 'pry-nav', '~> 0.2'

data/lib/cassandra_db/database.rb

@@ -5,6 +5,8 @@ module CassandraDB
       keyspace: DEFAULT_KEYSPACE
     }.freeze
 
+    DEFAULT_LOGGER = Logger.new '/dev/null'
+
     def initialize(options={})
       @options = DEFAULTS.merge(options)
       keyspace = @options.delete :keyspace
@@ -31,7 +33,7 @@ module CassandraDB
     end
 
     def from(table)
-      Dataset.new self, from: table
+      Dataset.new self, table: table
     end
     alias_method :[], :from
 
@@ -48,8 +50,9 @@ module CassandraDB
       execute "DROP KEYSPACE \"#{name}\";"
     end
 
-    def execute(*args)
-      session.execute(*args)
+    def execute(cql, options={})
+      logger.debug(self.class) { "#{cql} #{options.any? ? options : ''}".strip }
+      session.execute cql, options
     end
 
     def inspect
@@ -60,5 +63,9 @@ module CassandraDB
 
     attr_reader :cluster, :session, :options
 
+    def logger
+      @logger ||= options.fetch(:logger, DEFAULT_LOGGER)
+    end
+
   end
 end

data/lib/cassandra_db/dataset.rb

@@ -3,22 +3,18 @@ module CassandraDB
 
     include Enumerable
 
-    def initialize(db, from:, where:[])
+    def initialize(db, table:, conditions:[])
       @db = db
-      @table = from
-      @conditions = where
+      @table = table
+      @conditions = conditions
     end
 
     def where(filters)
       new_conditions = filters.map do |field, value|
-        {
-          field: field,
-          operator: value.is_a?(Array) ? 'IN' : '=',
-          value: value
-        }
+        QueryCondition.new field, value
       end
 
-      Dataset.new db, from: table, where: conditions + new_conditions
+      Dataset.new db, table: table, conditions: conditions + new_conditions
     end
 
     def each(&block)
@@ -26,27 +22,34 @@ module CassandraDB
     end
 
     def all
-      db.execute(*cql).to_a
+      statement = Statements::Select.new table: table, conditions: conditions
+      result = db.execute statement.cql, statement.options
+      result.to_a
     end
 
-    def cql
-      select = 'SELECT *'
-      from = "FROM #{table}"
-      where = conditions.any? ? "WHERE #{conditions.map { |c| "#{c[:field]} #{c[:operator]} :#{c[:field]}" }.join(' AND ')}" : ''
-      arguments = conditions.each_with_object({}) { |c,h| h[c[:field]] = c[:value] }
+    def insert(attributes)
+      statement = Statements::Insert.new table: table, attributes: attributes
+      db.execute statement.cql, statement.options
+    end
+
+    def update(attributes)
+      statement = Statements::Update.new table: table, attributes: attributes, conditions: conditions
+      db.execute statement.cql, statement.options
+    end
 
-      [
-        "#{select} #{from} #{where}".strip,
-        {arguments: arguments}
-      ]
+    def delete
+      statement = Statements::Delete.new table: table, conditions: conditions
+      db.execute statement.cql, statement.options
     end
 
     def inspect
-      "#<#{self.class.name}: #{cql.join(', ')}>"
+      statement = Statements::Select.new table: table, conditions: conditions
+      "#<#{self.class.name}: #{statement.to_s}>"
     end
 
     private
 
     attr_reader :db, :table, :conditions
+
   end
 end

data/lib/cassandra_db/query_condition.rb

@@ -0,0 +1,21 @@
+module CassandraDB
+  class QueryCondition
+
+    attr_reader :field, :value, :operator
+
+    def initialize(field, value)
+      @field = field
+      @value = value
+      @operator = value.is_a?(Array) ? 'IN' : '='
+    end
+
+    def to_cql
+      "#{field} #{operator} :#{field}"
+    end
+
+    def to_argument
+      {field => value}
+    end
+
+  end
+end

data/lib/cassandra_db/statements.rb

@@ -0,0 +1,121 @@
+module CassandraDB
+  module Statements
+    class Base
+
+      def options
+        {arguments: arguments}
+      end
+
+      def to_s
+        "#{cql} #{options}"
+      end
+
+      private
+
+      def where_cql
+        conditions.any? ? "WHERE #{conditions.map(&:to_cql).join(' AND ')}" : ''
+      end
+
+      def conditions_agruments
+        conditions.each_with_object({}) do |condition, agruments|
+          agruments.merge! condition.to_argument
+        end
+      end
+
+    end
+
+    class Select < Base
+
+      attr_reader :table, :conditions
+
+      def initialize(table:, conditions:[])
+        @table = table
+        @conditions = conditions
+      end
+
+      def cql
+        @cql ||= "SELECT * FROM #{table} #{where_cql}".strip
+      end
+
+      def arguments
+        @arguments ||= conditions_agruments
+      end
+
+    end
+
+    class Insert < Base
+
+      attr_reader :table, :attributes
+
+      def initialize(table:, attributes:)
+        @table = table
+        @attributes = attributes
+      end
+
+      def cql
+        @cql ||= "INSERT INTO #{table} (#{field_names}) VALUES (#{field_references})"
+      end
+
+      def arguments
+        attributes
+      end
+
+      private
+
+      def field_names
+        attributes.keys.join(', ')
+      end
+
+      def field_references
+        attributes.keys.map { |f| ":#{f}" }.join(', ')
+      end
+
+    end
+
+    class Update < Base
+
+      attr_reader :table, :attributes, :conditions
+
+      def initialize(table:, attributes:, conditions:)
+        @table = table
+        @attributes = attributes
+        @conditions = conditions
+      end
+
+      def cql
+        @cql ||= "UPDATE #{table} SET #{filed_mappings} #{where_cql}"
+      end
+
+      def arguments
+        @arguments ||= attributes.merge(conditions_agruments)
+      end
+
+      private
+
+      def filed_mappings
+        attributes.keys.map { |f| "#{f} = :#{f}" }.join(', ')
+      end
+
+    end
+
+    class Delete < Base
+
+      attr_reader :table, :conditions
+
+      def initialize(table:, conditions:)
+        @table = table
+        @conditions = conditions
+      end
+
+      def cql
+        @cql ||= "DELETE FROM #{table} #{where_cql}"
+      end
+
+      def arguments
+        @arguments ||= conditions_agruments
+      end
+
+    end
+
+  end
+end

data/lib/cassandra_db/version.rb

@@ -1,3 +1,3 @@
 module CassandraDB
-  VERSION = '0.1.1'
+  VERSION = '1.0.0'
 end

data/lib/cassandra_db.rb

@@ -1,5 +1,6 @@
 require 'cassandra'
 require 'json'
+require 'logger'
 
 module CassandraDB
 
@@ -18,4 +19,6 @@ end
 
 require_relative 'cassandra_db/version'
 require_relative 'cassandra_db/database'
-require_relative 'cassandra_db/dataset'
+require_relative 'cassandra_db/dataset'
+require_relative 'cassandra_db/statements'
+require_relative 'cassandra_db/query_condition'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cassandra_db
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 1.0.0
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-09-25 00:00:00.000000000 Z
+date: 2024-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: cassandra-driver
@@ -86,6 +86,26 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: minitest-great_expectations
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.0.5
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.0.5
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
@@ -148,6 +168,8 @@ files:
 - lib/cassandra_db.rb
 - lib/cassandra_db/database.rb
 - lib/cassandra_db/dataset.rb
+- lib/cassandra_db/query_condition.rb
+- lib/cassandra_db/statements.rb
 - lib/cassandra_db/version.rb
 homepage: https://github.com/gabynaiman/cassandra_db
 licenses: