mongo 0.1.0 → 0.15

data/lib/mongo/message.rb

@@ -1,3 +1,19 @@
+# --
+# 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.
+# ++
+
 %w(get_more_message insert_message kill_cursors_message message_header
    msg_message query_message remove_message update_message).each { |f|
   require "mongo/message/#{f}"

data/lib/mongo/message/get_more_message.rb

@@ -1,21 +1,32 @@
+# --
+# 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 XGen
-  module Mongo
-    module Driver
+module Mongo
 
-      class GetMoreMessage < Message
+  class GetMoreMessage < Message
 
-        def initialize(db_name, collection_name, cursor)
-          super(OP_GET_MORE)
-          write_int(0)
-          write_string("#{db_name}.#{collection_name}")
-          write_int(0)              # num to return; leave it up to the db for now
-          write_long(cursor)
-        end
-      end
+    def initialize(db_name, collection_name, cursor)
+      super(OP_GET_MORE)
+      write_int(0)
+      write_string("#{db_name}.#{collection_name}")
+      write_int(0)              # num to return; leave it up to the db for now
+      write_long(cursor)
     end
   end
 end
-

data/lib/mongo/message/insert_message.rb

@@ -1,19 +1,37 @@
+# --
+# 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 XGen
-  module Mongo
-    module Driver
+module Mongo
+
+  class InsertMessage < Message
 
-      class InsertMessage < Message
+    def initialize(db_name, collection_name, check_keys=true, *objs)
+      @collection_name = collection_name
+      @objs = objs
+      super(OP_INSERT)
+      write_int(0)
+      write_string("#{db_name}.#{collection_name}")
+      objs.each { |o| write_doc(o, check_keys) }
+    end
 
-        def initialize(db_name, collection_name, *objs)
-          super(OP_INSERT)
-          write_int(0)
-          write_string("#{db_name}.#{collection_name}")
-          objs.each { |o| write_doc(o) }
-        end
-      end
+    def to_s
+      "db.#{@collection_name}.insert(#{@objs.inspect})"
     end
   end
 end

data/lib/mongo/message/kill_cursors_message.rb

@@ -1,20 +1,31 @@
+# --
+# 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 XGen
-  module Mongo
-    module Driver
+module Mongo
 
-      class KillCursorsMessage < Message
+  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
+    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,68 +1,80 @@
+# --
+# 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 XGen
-  module Mongo
-    module Driver
-
-      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)
-          @buf.put_array(BSON.new.serialize(hash).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
+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,34 +1,45 @@
+# --
+# 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 XGen
-  module Mongo
-    module Driver
+module Mongo
 
-      class MessageHeader
+  class MessageHeader
 
-        HEADER_SIZE = 16
+    HEADER_SIZE = 16
 
-        def initialize()
-          @buf = ByteBuffer.new
-        end
+    def initialize()
+      @buf = ByteBuffer.new
+    end
 
-        def read_header(socket)
-          @buf.rewind
-          @buf.put_array(socket.recv(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 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
+    def dump
+      @buf.dump
     end
   end
 end
-

data/lib/mongo/message/msg_message.rb

@@ -1,17 +1,29 @@
+# --
+# 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 XGen
-  module Mongo
-    module Driver
+module Mongo
 
-      class MsgMessage < Message
+  class MsgMessage < Message
 
-        def initialize(msg)
-          super(OP_MSG)
-          write_string(msg)
-        end
-      end
+    def initialize(msg)
+      super(OP_MSG)
+      write_string(msg)
     end
   end
 end

data/lib/mongo/message/opcodes.rb

@@ -1,16 +1,27 @@
-module XGen
-  module Mongo
-    module Driver
-      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
-  end
-end
+# --
+# 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