cassandra-cql 1.0.1

data/lib/cassandra-cql/statement.rb

@@ -0,0 +1,111 @@
+=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={})
+      if options[:compression]
+        res = Result.new(@handle.execute_cql_query(Utility.compress(self.class.sanitize(@statement, bind_vars)), CassandraCQL::Thrift::Compression::GZIP))
+      else
+        res = Result.new(@handle.execute_cql_query(self.class.sanitize(@statement, bind_vars), CassandraCQL::Thrift::Compression::NONE))
+      end
+
+      # 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?(Time)
+        UUID.new(obj).to_guid
+      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/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

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

@@ -0,0 +1,27 @@
+=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 IntegerType < AbstractType
+      def self.cast(value)
+        bytes_to_int(value)
+      end
+    end
+    
+    class Int32Type < IntegerType; end
+  end
+end

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

@@ -0,0 +1,27 @@
+=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 LongType < AbstractType
+      def self.cast(value)
+        bytes_to_long(value)
+      end
+    end
+    
+    class CounterColumnType < LongType; end
+  end
+end

data/lib/cassandra-cql/types/utf8_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 UTF8Type < AbstractType
+      def self.cast(value)
+        RUBY_VERSION >= "1.9" ? value.to_s.dup.force_encoding('UTF-8') : value.to_s.dup
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+=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 UUIDType < AbstractType
+      def self.cast(value)
+        UUID.new(value)
+      end
+    end
+    
+    class TimeUUIDType < UUIDType; end
+  end
+end

data/lib/cassandra-cql/utility.rb

@@ -0,0 +1,37 @@
+=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
+
+require 'zlib'
+
+module CassandraCQL
+  class Utility
+    def self.compress(source, level=2)
+      Zlib::Deflate.deflate(source, level)
+    end
+    
+    def self.decompress(source)
+      Zlib::Inflate.inflate(source)
+    end
+
+    def self.binary_data?(string)
+      if RUBY_VERSION >= "1.9"
+        string.encoding.name == "ASCII-8BIT"
+      else
+        string.count("\x00-\x7F", "^ -~\t\r\n").fdiv(string.size) > 0.3 || string.index("\x00") unless string.empty?
+      end
+    end
+  end
+end