mongo 0.15.1 → 0.16

data/lib/mongo/message/kill_cursors_message.rb

@@ -1,31 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/message/message'
-require 'mongo/message/opcodes'
-
-module Mongo
-
-  class KillCursorsMessage < Message
-
-    def initialize(*cursors)
-      super(OP_KILL_CURSORS)
-      write_int(0)
-      write_int(cursors.length)
-      cursors.each { |c| write_long c }
-    end
-  end
-end

data/lib/mongo/message/message.rb

@@ -1,80 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/util/bson'
-require 'mongo/util/byte_buffer'
-
-module Mongo
-
-  class Message
-
-    HEADER_SIZE = 16        # size, id, response_to, opcode
-
-    @@class_req_id = 0
-
-    attr_reader :buf        # for testing
-
-    def initialize(op)
-      @op = op
-      @message_length = HEADER_SIZE
-      @data_length = 0
-      @request_id = (@@class_req_id += 1)
-      @response_id = 0
-      @buf = ByteBuffer.new
-
-      @buf.put_int(16)      # holder for length
-      @buf.put_int(@request_id)
-      @buf.put_int(0)       # response_to
-      @buf.put_int(op)
-    end
-
-    def write_int(i)
-      @buf.put_int(i)
-      update_message_length
-    end
-
-    def write_long(i)
-      @buf.put_long(i)
-      update_message_length
-    end
-
-    def write_string(s)
-      BSON.serialize_cstr(@buf, s)
-      update_message_length
-    end
-
-    def write_doc(hash, check_keys=false)
-      @buf.put_array(BSON.new.serialize(hash, check_keys).to_a)
-      update_message_length
-    end
-
-    def to_a
-      @buf.to_a
-    end
-
-    def dump
-      @buf.dump
-    end
-
-    # Do not call. Private, but kept public for testing.
-    def update_message_length
-      pos = @buf.position
-      @buf.put_int(@buf.size, 0)
-      @buf.position = pos
-    end
-
-  end
-end

data/lib/mongo/message/message_header.rb

@@ -1,45 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/util/byte_buffer'
-
-module Mongo
-
-  class MessageHeader
-
-    HEADER_SIZE = 16
-
-    def initialize()
-      @buf = ByteBuffer.new
-    end
-
-    def read_header(db)
-      @buf.rewind
-      @buf.put_array(db.receive_full(HEADER_SIZE).unpack("C*"))
-      raise "Short read for DB response header: expected #{HEADER_SIZE} bytes, saw #{@buf.size}" unless @buf.size == HEADER_SIZE
-      @buf.rewind
-      @size = @buf.get_int
-      @request_id = @buf.get_int
-      @response_to = @buf.get_int
-      @op = @buf.get_int
-      self
-    end
-
-    def dump
-      @buf.dump
-    end
-  end
-end

data/lib/mongo/message/msg_message.rb

@@ -1,29 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/message/message'
-require 'mongo/message/opcodes'
-
-module Mongo
-
-  class MsgMessage < Message
-
-    def initialize(msg)
-      super(OP_MSG)
-      write_string(msg)
-    end
-  end
-end

data/lib/mongo/message/opcodes.rb

@@ -1,27 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-module Mongo
-  OP_REPLY = 1              # reply. responseTo is set.
-  OP_MSG = 1000             # generic msg command followed by a string
-  OP_UPDATE = 2001          # update object
-  OP_INSERT = 2002
-  # GET_BY_OID = 2003
-  OP_QUERY = 2004
-  OP_GET_MORE = 2005
-  OP_DELETE = 2006
-  OP_KILL_CURSORS = 2007
-end

data/lib/mongo/message/query_message.rb

@@ -1,69 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/message/message'
-require 'mongo/message/opcodes'
-require 'mongo/util/conversions'
-require 'mongo/util/ordered_hash'
-
-module Mongo
-  class QueryMessage < Message
-    include Mongo::Conversions
-
-    attr_reader :query
-
-    def initialize(db_name, collection_name, query)
-      super(OP_QUERY)
-      @query = query
-      @collection_name = collection_name
-      write_int(0)
-      write_string("#{db_name}.#{collection_name}")
-      write_int(query.number_to_skip)
-      write_int(query.number_to_return)
-      sel = query.selector
-      if query.contains_special_fields
-        sel = OrderedHash.new
-        sel['query'] = query.selector
-        if query.order_by
-          order_by = query.order_by
-          sel['orderby'] = case order_by
-                           when String then string_as_sort_parameters(order_by)
-                           when Symbol then symbol_as_sort_parameters(order_by)
-                           when Array then array_as_sort_parameters(order_by)
-                           when Hash # Should be an ordered hash, but this message doesn't care
-                             warn_if_deprecated(order_by)
-                             order_by
-                           else
-                             raise InvalidSortValueError.new("illegal order_by: is a #{query.order_by.class.name}, must be String, Array, Hash, or OrderedHash")
-                           end
-        end
-        sel['$hint'] = query.hint if query.hint && query.hint.length > 0
-        sel['$explain'] = true if query.explain
-        sel['$snapshot'] = true if query.snapshot
-      end
-      write_doc(sel)
-      write_doc(query.fields) if query.fields
-    end
-
-    def first_key(key)
-      @first_key = key
-    end
-
-    def to_s
-      "db.#{@collection_name}.#{@query}"
-    end
-  end
-end

data/lib/mongo/message/remove_message.rb

@@ -1,37 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/message/message'
-require 'mongo/message/opcodes'
-
-module Mongo
-
-  class RemoveMessage < Message
-
-    def initialize(db_name, collection_name, sel)
-      @collection_name = collection_name
-      super(OP_DELETE)
-      write_int(0)
-      write_string("#{db_name}.#{collection_name}")
-      write_int(0)              # flags?
-      write_doc(sel)
-    end
-
-    def to_s
-      "#{@collection_name}.clear()"
-    end
-  end
-end

data/lib/mongo/message/update_message.rb

@@ -1,38 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/message/message'
-require 'mongo/message/opcodes'
-
-module Mongo
-
-  class UpdateMessage < Message
-
-    def initialize(db_name, collection_name, sel, obj, repsert)
-      @collection_name = collection_name
-      super(OP_UPDATE)
-      write_int(0)
-      write_string("#{db_name}.#{collection_name}")
-      write_int(repsert ? 1 : 0) # 1 if a repsert operation (upsert)
-      write_doc(sel)
-      write_doc(obj)
-    end
-
-    def to_s
-      "db.#{@collection_name}.update(#{@sel.inspect}, #{@obj.inspect})"
-    end
-  end
-end

data/lib/mongo/query.rb

@@ -1,118 +0,0 @@
-# --
-# Copyright (C) 2008-2009 10gen 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.
-# ++
-
-require 'mongo/collection'
-require 'mongo/message'
-require 'mongo/types/code'
-
-module Mongo
-
-  # A query against a collection. A query's selector is a hash. See the
-  # Mongo documentation for query details.
-  class Query
-
-    attr_accessor :number_to_skip, :number_to_return, :order_by, :snapshot
-    # If true, $explain will be set in QueryMessage that uses this query.
-    attr_accessor :explain
-    # Either +nil+ or a hash (preferably an OrderedHash).
-    attr_accessor :hint
-    attr_reader :selector   # writer defined below
-
-    # sel :: A hash describing the query. See the Mongo docs for details.
-    #
-    # return_fields :: If not +nil+, a single field name or an array of
-    #                  field names. Only those fields will be returned.
-    #                  (Called :fields in calls to Collection#find.)
-    #
-    # number_to_skip :: Number of records to skip before returning
-    #                   records. Default is 0.
-    #
-    # number_to_return :: Max number of records to return. (Called :limit
-    #                     in calls to Collection#find.) Default is 0 (all
-    #                     records).
-    #
-    # order_by :: If not +nil+, specifies record sort order. May be a
-    #             String, Hash, OrderedHash, or Array. If a string, the
-    #             results will be ordered by that field in ascending
-    #             order. If an array, it should be an array of field names
-    #             which will all be sorted in ascending order. If a hash,
-    #             it may be either a regular Hash or an OrderedHash. The
-    #             keys should be field names, and the values should be 1
-    #             (ascending) or -1 (descending). Note that if it is a
-    #             regular Hash then sorting by more than one field
-    #             probably will not be what you intend because key order
-    #             is not preserved. (order_by is called :sort in calls to
-    #             Collection#find.)
-    #
-    # hint :: If not +nil+, specifies query hint fields. Must be either
-    #                +nil+ or a hash (preferably an OrderedHash). See
-    #                Collection#hint.
-    def initialize(sel={}, return_fields=nil, number_to_skip=0, number_to_return=0, order_by=nil, hint=nil, snapshot=nil)
-      @number_to_skip, @number_to_return, @order_by, @hint, @snapshot =
-        number_to_skip, number_to_return, order_by, hint, snapshot
-      @explain = nil
-      self.selector = sel
-      self.fields = return_fields
-    end
-
-    # Set query selector hash. If sel is Code/string, it will be used as a
-    # $where clause. (See Mongo docs for details.)
-    def selector=(sel)
-      @selector = case sel
-                  when nil
-                    {}
-                  when Code
-                    {"$where" => sel}
-                  when String
-                    {"$where" => Code.new(sel)}
-                  when Hash
-                    sel
-                  end
-    end
-
-    # Set fields to return. If +val+ is +nil+ or empty, all fields will be
-    # returned.
-    def fields=(val)
-      @fields = val
-      @fields = nil if @fields && @fields.empty?
-    end
-
-    def fields
-      case @fields
-      when String
-        {@fields => 1}
-      when Array
-        if @fields.length == 0
-          nil
-        else
-          h = {}
-          @fields.each { |field| h[field] = 1 }
-          h
-        end
-      else                  # nil, anything else
-        nil
-      end
-    end
-
-    def contains_special_fields
-      @order_by || @explain || @hint || @snapshot
-    end
-
-    def to_s
-      "find(#{@selector.inspect})" + (@order_by ? ".sort(#{@order_by.inspect})" : "")
-    end
-  end
-end