hallelujah-cassandra-cql 1.0.4

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

data/lib/cassandra-cql/statement.rb

@@ -0,0 +1,116 @@
+=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 InvalidBindVariable < Exception; end
+    class UnescapableObject < Exception; end
+  end
+
+  class Statement
+
+    KS_CHANGE_RE = /^use (\w+)/i
+    KS_DROP_RE = /^drop keyspace (\w+)/i
+
+    attr_reader :statement
+
+    def initialize(handle, statement)
+      @handle = handle
+      prepare(statement)
+    end
+
+    def prepare(statement)
+      @statement = statement
+    end
+
+    def execute(bind_vars=[], options={})
+      sanitized_query = self.class.sanitize(@statement, bind_vars)
+      compression_type = CassandraCQL::Thrift::Compression::NONE
+      if options[:compression]
+        compression_type = CassandraCQL::Thrift::Compression::GZIP
+        sanitized_query = Utility.compress(sanitized_query)
+      end
+
+      res = Result.new(@handle.execute_cql_query(sanitized_query, compression_type))
+
+      # Change our keyspace if required
+      if @statement =~ KS_CHANGE_RE
+        @handle.keyspace = $1
+      elsif @statement =~ KS_DROP_RE
+        @handle.keyspace = nil
+      end
+
+      # We let ints be fetched for now because they'll probably be deprecated later
+      if res.void?
+        nil
+      else
+        res
+      end
+    end
+
+    def finish
+      true
+    end
+
+    def self.escape(obj)
+      obj.gsub("'", "''")
+    end
+
+    def self.quote(obj)
+      if obj.kind_of?(Array)
+        obj.map { |member| quote(member) }.join(",")
+      elsif obj.kind_of?(String)
+        "'" + obj + "'"
+      elsif obj.kind_of?(Fixnum) or obj.kind_of?(Float)
+        obj
+      else
+        raise Error::UnescapableObject, "Unable to escape object of class #{obj.class}"
+      end
+    end
+
+    def self.cast_to_cql(obj)
+      if obj.kind_of?(Array)
+        obj.map { |member| cast_to_cql(member) }
+      elsif obj.kind_of?(Fixnum) or obj.kind_of?(Float)
+        obj
+      elsif obj.kind_of?(Date)
+        obj.strftime('%Y-%m-%d')
+      elsif obj.kind_of?(Time)
+        (obj.to_f * 1000).to_i
+      elsif obj.kind_of?(UUID)
+        obj.to_guid
+      # There are corner cases where this is an invalid assumption but they are extremely rare.
+      # The alternative is to make the user pack the data on their own .. let's not do that until we have to
+      elsif obj.kind_of?(String) and Utility.binary_data?(obj)
+        escape(obj.unpack('H*')[0])
+      else
+        RUBY_VERSION >= "1.9" ? escape(obj.to_s.dup.force_encoding('ASCII-8BIT')) : escape(obj.to_s.dup)
+      end
+    end
+  
+    def self.sanitize(statement, bind_vars=[])
+      bind_vars = bind_vars.dup
+      expected_bind_vars = statement.count("?")
+
+      return statement if expected_bind_vars == 0 and bind_vars.empty?
+      raise Error::InvalidBindVariable, "Wrong number of bound variables (statement expected #{expected_bind_vars}, was #{bind_vars.size})" if expected_bind_vars != bind_vars.size
+    
+      statement.gsub(/\?/) {
+        quote(cast_to_cql(bind_vars.shift))
+      }
+    end
+  end
+end

data/lib/cassandra-cql/types/abstract_type.rb

@@ -0,0 +1,47 @@
+=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 Types
+    class AbstractType
+      def self.cast(value)
+        value
+      end
+
+      private
+      
+      def self.bytes_to_int(bytes)
+        int = 0
+        values = bytes.unpack('C*')
+        values.each {|v| int = int << 8; int += v; }
+        if bytes[0].ord & 128 != 0
+          int = int - (1 << bytes.length * 8)
+        end
+        int
+      end
+
+      def self.bytes_to_long(bytes)
+        ints = bytes.unpack("NN")
+        val = (ints[0] << 32) + ints[1]
+        if val & 2**63 == 2**63
+          val - 2**64
+        else
+          val
+        end
+      end
+    end
+  end
+end

data/lib/cassandra-cql/types/ascii_type.rb

@@ -0,0 +1,25 @@
+=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 Types
+    class AsciiType < AbstractType
+      def self.cast(value)
+        RUBY_VERSION >= "1.9" ? value.to_s.dup.force_encoding('ASCII-8BIT') : value.to_s.dup
+      end
+    end
+  end
+end

data/lib/cassandra-cql/types/boolean_type.rb

@@ -0,0 +1,25 @@
+=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 Types
+    class BooleanType < AbstractType
+      def self.cast(value)
+        value.unpack('C') == [1]
+      end
+    end
+  end
+end

data/lib/cassandra-cql/types/bytes_type.rb

@@ -0,0 +1,21 @@
+=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 Types
+    class BytesType < AbstractType; end
+  end
+end

data/lib/cassandra-cql/types/date_type.rb

@@ -0,0 +1,25 @@
+=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 Types
+    class DateType < AbstractType
+      def self.cast(value)
+        Time.at(bytes_to_long(value) / 1000.0)
+      end
+    end
+  end
+end

data/lib/cassandra-cql/types/decimal_type.rb

@@ -0,0 +1,25 @@
+=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 Types
+    class DecimalType < AbstractType
+      def self.cast(value)
+        BigDecimal.new(bytes_to_int(value[4..-1]).to_s) * BigDecimal.new('10')**(bytes_to_int(value[0..3])*-1)
+      end
+    end
+  end
+end

data/lib/cassandra-cql/types/double_type.rb

@@ -0,0 +1,25 @@
+=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 Types
+    class DoubleType < AbstractType
+      def self.cast(value)
+        value.unpack('G')[0]
+      end
+    end
+  end
+end

data/lib/cassandra-cql/types/float_type.rb

@@ -0,0 +1,25 @@
+=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 Types
+    class FloatType < AbstractType
+      def self.cast(value)
+        value.unpack('g')[0]
+      end
+    end
+  end
+end