cassandra-cql 1.0.1

data/lib/cassandra-cql.rb

@@ -0,0 +1,39 @@
+=begin
+Copyright 2011 Inside Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+=end
+
+here = File.dirname(__FILE__)
+require "#{here}/../vendor/gen-rb/cassandra"
+
+require 'bigdecimal'
+require 'thrift_client'
+require 'cassandra-cql/types/abstract_type'
+require 'cassandra-cql/types/ascii_type'
+require 'cassandra-cql/types/boolean_type'
+require 'cassandra-cql/types/bytes_type'
+require 'cassandra-cql/types/decimal_type'
+require 'cassandra-cql/types/double_type'
+require 'cassandra-cql/types/float_type'
+require 'cassandra-cql/types/integer_type'
+require 'cassandra-cql/types/long_type'
+require 'cassandra-cql/types/utf8_type'
+require 'cassandra-cql/types/uuid_type'
+require 'cassandra-cql/utility'
+require 'cassandra-cql/uuid'
+require 'cassandra-cql/database'
+require 'cassandra-cql/schema'
+require 'cassandra-cql/statement'
+require 'cassandra-cql/result'
+require 'cassandra-cql/row'

data/lib/cassandra-cql/database.rb

@@ -0,0 +1,107 @@
+=begin
+Copyright 2011 Inside Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+=end
+
+module CassandraCQL
+  module Error
+    class InvalidRequestException < Exception; end
+  end
+
+  class Database
+    attr_reader :connection, :schema, :keyspace
+  
+    def initialize(servers, options={}, thrift_client_options={})
+      @options = {
+        :keyspace => 'system'
+      }.merge(options)
+
+      @thrift_client_options = {
+        :exception_class_overrides => CassandraCQL::Thrift::InvalidRequestException
+      }.merge(thrift_client_options)
+
+      @keyspace = @options[:keyspace]
+      @servers = servers
+      connect!
+      execute("USE #{@keyspace}")
+    end
+
+    def connect!
+      @connection = ThriftClient.new(CassandraCQL::Thrift::Client, @servers, @thrift_client_options)
+      obj = self
+      @connection.add_callback(:post_connect) do
+        execute("USE #{@keyspace}")
+      end
+    end
+  
+    def disconnect!
+      @connection.disconnect! if active?
+    end
+
+    def active?
+      # TODO: This should be replaced with a CQL call that doesn't exist yet
+      @connection.describe_version 
+      true
+    rescue Exception
+      false
+    end
+    alias_method :ping, :active?
+
+    def reset!
+      disconnect!
+      reconnect!
+    end
+    alias_method :reconnect!, :reset!
+
+    def prepare(statement, options={}, &block)
+      stmt = Statement.new(self, statement)
+      if block_given?
+        yield stmt
+      else
+        stmt
+      end
+    end
+
+    def execute(statement, *bind_vars)
+      result = Statement.new(self, statement).execute(bind_vars)
+      if block_given?
+        yield result
+      else
+        result
+      end
+    rescue CassandraCQL::Thrift::InvalidRequestException
+      raise Error::InvalidRequestException.new($!.why)
+    end
+
+    def execute_cql_query(cql, compression=CassandraCQL::Thrift::Compression::NONE)
+      @connection.execute_cql_query(cql, compression)
+    rescue CassandraCQL::Thrift::InvalidRequestException
+      raise Error::InvalidRequestException.new($!.why)
+    end
+    
+    def keyspace=(ks)
+      @keyspace = (ks.nil? ? nil : ks.to_s)
+    end
+  
+    def keyspaces
+      # TODO: This should be replaced with a CQL call that doesn't exist yet
+      @connection.describe_keyspaces.map { |keyspace| Schema.new(keyspace) }
+    end
+    
+    def schema
+      # TODO: This should be replaced with a CQL call that doesn't exist yet
+      Schema.new(@connection.describe_keyspace(@keyspace))
+    end
+  end
+end

data/lib/cassandra-cql/result.rb

@@ -0,0 +1,133 @@
+=begin
+Copyright 2011 Inside Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+=end
+
+module CassandraCQL
+  module Error
+    class InvalidResultType < Exception; end
+    class InvalidCursor < Exception; end
+  end
+
+  class ResultSchema
+    attr_reader :names, :values
+
+    def initialize(schema)
+      # When https://issues.apache.org/jira/browse/CASSANDRA-3436 is resolve, no more need to split/last
+      @names = Hash.new(schema.default_name_type.split(".").last)
+      schema.name_types.each_pair { |key, type|
+        @names[key] = type.split(".").last
+      }
+      @values = Hash.new(schema.default_value_type.split(".").last)
+      schema.value_types.each_pair { |key, type|
+        @values[key] = type.split(".").last
+      }
+    end
+  end
+  
+  class Result
+    attr_reader :result, :schema, :cursor
+
+    def initialize(result)
+      @result = result
+      @schema = ResultSchema.new(result.schema) if rows?
+      @cursor = 0
+    end
+  
+    def void?
+      @result.type == CassandraCQL::Thrift::CqlResultType::VOID
+    end
+    
+    def int?
+      @result.type == CassandraCQL::Thrift::CqlResultType::INT
+    end
+
+    def rows?
+      @result.type == CassandraCQL::Thrift::CqlResultType::ROWS
+    end
+
+    def rows
+      @result.rows.size
+    end
+
+    def cursor=(cursor)
+      @cursor = cursor.to_i
+    rescue Exception => e
+      raise Error::InvalidCursor, e.to_s
+    end
+    
+    def fetch_row
+      case @result.type
+      when CassandraCQL::Thrift::CqlResultType::ROWS
+        return nil if @cursor >= rows
+
+        row = Row.new(@result.rows[@cursor], @schema)
+        @cursor += 1
+        return row
+      when CassandraCQL::Thrift::CqlResultType::VOID
+        return nil
+      when CassandraCQL::Thrift::CqlResultType::INT
+        return @result.num
+      else
+        raise Error::InvalidResultType, "Expects one of 0, 1, 2; was #{@result.type} "
+      end
+    end
+
+    def fetch
+      if block_given?
+        while row = fetch_row
+          yield row
+        end
+      else
+        fetch_row
+      end
+    end
+
+    def fetch_hash
+      if block_given?
+        while row = fetch_row
+          if row.kind_of?(Fixnum)
+            yield({row => row})
+          else
+            yield row.to_hash
+          end
+        end
+      else
+        if (row = fetch_row).kind_of?(Fixnum)
+          {row => row}
+        else
+          row.to_hash
+        end
+      end
+    end
+  
+    def fetch_array
+      if block_given?
+        while row = fetch_row
+          if row.kind_of?(Fixnum)
+            yield [row]
+          else
+            yield row.to_a
+          end
+        end
+      else
+        if (row = fetch_row).kind_of?(Fixnum)
+          [row]
+        else
+          row.to_a
+        end
+      end
+    end
+  end
+end

data/lib/cassandra-cql/row.rb

@@ -0,0 +1,59 @@
+=begin
+Copyright 2011 Inside Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+=end
+
+module CassandraCQL
+  class Row
+    attr_reader :row
+    
+    def initialize(row, schema)
+      @row, @schema = row, schema
+    end
+  
+    def [](obj)
+      # Rows include the row key so we skip the first one
+      column_index = obj.kind_of?(Fixnum) ? obj : column_names.index(obj)
+      return nil if column_index.nil?
+      column_values[column_index]
+    end
+
+    def column_names
+      @names ||= @row.columns.map do |column|
+        ColumnFamily.cast(column.name, @schema.names[column.name])
+      end
+    end
+  
+    def column_values
+      @values ||= @row.columns.map { |column| ColumnFamily.cast(column.value, @schema.values[column.name]) }
+    end
+  
+    def columns
+      @row.columns.size
+    end
+  
+    def to_a
+      column_values
+    end
+  
+    # TODO: This should be an ordered hash
+    def to_hash
+      Hash[([column_names, column_values]).transpose]
+    end
+  
+    def key
+      ColumnFamily.cast(@row.key, @schema.values[@row.key])
+    end
+  end
+end

data/lib/cassandra-cql/schema.rb

@@ -0,0 +1,108 @@
+=begin
+Copyright 2011 Inside Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+=end
+
+module CassandraCQL
+  class Schema
+    attr_reader :schema, :column_families
+
+    def initialize(schema)
+      @schema = schema
+      @column_families = {}
+      @schema.cf_defs.each { |cf|
+        @column_families[cf.name] = ColumnFamily.new(cf)
+      }
+    end
+
+    def method_missing(method, *args, &block)
+      if @schema.respond_to?(method)
+        @schema.send(method)
+      else
+        super(method, *args, &block)
+      end
+    end
+
+    def to_s
+      keyspace
+    end
+
+    def keyspace
+      name
+    end
+
+    def column_family_names
+      @column_families.keys
+    end
+    alias_method :tables, :column_family_names
+  end
+
+  class ColumnFamily
+    attr_reader :cf_def
+
+    def initialize(cf_def)
+      @cf_def = cf_def
+    end
+
+    def method_missing(method, *args, &block)
+      if @cf_def.respond_to?(method)
+        @cf_def.send(method)
+      else
+        super(method, *args, &block)
+      end
+    end
+
+    def columns
+      return @columns if @columns
+
+      @columns = Hash.new(default_validation_class)
+      @cf_def.column_metadata.each do |col|
+        @columns[col.name] = col.validation_class
+      end
+      @columns[key_alias] = key_validation_class
+
+      @columns
+    end
+
+    def self.cast(value, type)
+      return nil if value.nil?
+
+      if CassandraCQL::Types.const_defined?(type)
+        CassandraCQL::Types.const_get(type).cast(value)
+      else
+        CassandraCQL::Types::AbstractType.cast(value)
+      end
+    end
+
+    def name
+      @cf_def.name
+    end
+
+    def type
+      @cf_def.column_type
+    end
+
+    def id
+      @cf_def.id
+    end
+
+    def standard?
+      type == 'Standard'
+    end
+
+    def super?
+      type == 'Super'
+    end
+  end
+end