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

@@ -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, *objs)
-          super(OP_INSERT)
-          write_int(0)
-          write_string("#{db_name}.#{collection_name}")
-          objs.each { |o| write_doc(o) }
-        end
-      end
+    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 to_s
+      "db.#{@collection_name}.insert(#{@objs.inspect})"
     end
   end
 end