lib/mongo/message/message.rb in mongo-0.1.0 vs lib/mongo/message/message.rb in mongo-0.15

@@ -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
+module Mongo
 
-      class Message
+  class Message
 
-        HEADER_SIZE = 16        # size, id, response_to, opcode
+    HEADER_SIZE = 16        # size, id, response_to, opcode
 
-        @@class_req_id = 0
+    @@class_req_id = 0
 
-        attr_reader :buf        # for testing
+    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 initialize(op)
+      @op = op
+      @message_length = HEADER_SIZE
+      @data_length = 0
+      @request_id = (@@class_req_id += 1)
+      @response_id = 0
+      @buf = ByteBuffer.new
 
-        def write_int(i)
-          @buf.put_int(i)
-          update_message_length
-        end
+      @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_long(i)
-          @buf.put_long(i)
-          update_message_length
-        end
+    def write_int(i)
+      @buf.put_int(i)
+      update_message_length
+    end
 
-        def write_string(s)
-          BSON.serialize_cstr(@buf, s)
-          update_message_length
-        end
+    def write_long(i)
+      @buf.put_long(i)
+      update_message_length
+    end
 
-        def write_doc(hash)
-          @buf.put_array(BSON.new.serialize(hash).to_a)
-          update_message_length
-        end
+    def write_string(s)
+      BSON.serialize_cstr(@buf, s)
+      update_message_length
+    end
 
-        def to_a
-          @buf.to_a
-        end
+    def write_doc(hash, check_keys=false)
+      @buf.put_array(BSON.new.serialize(hash, check_keys).to_a)
+      update_message_length
+    end
 
-        def dump
-          @buf.dump
-        end
+    def to_a
+      @buf.to_a
+    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
+    def dump
+      @buf.dump
+    end
 
-      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