animehunter-mongo 0.9

data/lib/mongo/message.rb

@@ -0,0 +1,20 @@
+# --
+# 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

@@ -0,0 +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
+
+      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
+    end
+  end
+end
+

data/lib/mongo/message/insert_message.rb

@@ -0,0 +1,35 @@
+# --
+# 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
+
+      class InsertMessage < Message
+
+        def initialize(db_name, collection_name, check_keys=true, *objs)
+          super(OP_INSERT)
+          write_int(0)
+          write_string("#{db_name}.#{collection_name}")
+          objs.each { |o| write_doc(o, check_keys) }
+        end
+      end
+    end
+  end
+end

data/lib/mongo/message/kill_cursors_message.rb

@@ -0,0 +1,36 @@
+# --
+# 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
+
+      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
+  end
+end
+

data/lib/mongo/message/message.rb

@@ -0,0 +1,84 @@
+# --
+# 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, 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
+  end
+end

data/lib/mongo/message/message_header.rb

@@ -0,0 +1,50 @@
+# --
+# 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
+
+      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
+  end
+end
+

data/lib/mongo/message/msg_message.rb

@@ -0,0 +1,33 @@
+# --
+# 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
+
+      class MsgMessage < Message
+
+        def initialize(msg)
+          super(OP_MSG)
+          write_string(msg)
+        end
+      end
+    end
+  end
+end

data/lib/mongo/message/opcodes.rb

@@ -0,0 +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.
+# ++
+
+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
+

data/lib/mongo/message/query_message.rb

@@ -0,0 +1,77 @@
+# --
+# 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/ordered_hash'
+
+module XGen
+  module Mongo
+    module Driver
+
+      class QueryMessage < Message
+
+        attr_reader :query
+
+        def initialize(db_name, collection_name, query)
+          super(OP_QUERY)
+          @query = query
+          write_int(0)
+          write_string("#{db_name}.#{collection_name}")
+          write_int(query.number_to_skip)
+          write_int(-query.number_to_return) # Negative means hard limit
+          sel = query.selector
+          if query.contains_special_fields
+            sel = OrderedHash.new
+            sel['query'] = query.selector
+            if query.order_by && query.order_by.length > 0
+              sel['orderby'] = case query.order_by
+                               when String
+                                 {query.order_by => 1}
+                               when Array
+                                 h = OrderedHash.new
+                                 query.order_by.each { |ob|
+                                   case ob
+                                   when String
+                                     h[ob] = 1
+                                   when Hash # should have one entry; will handle all
+                                     ob.each { |k,v| h[k] = v }
+                                   else
+                                     raise "illegal query order_by value #{query.order_by.inspect}"
+                                   end
+                                 }
+                                 h
+                               when Hash # Should be an ordered hash, but this message doesn't care
+                                 query.order_by
+                               else
+                                 raise "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
+                               
+          end
+          write_doc(sel)
+          write_doc(query.fields) if query.fields
+        end
+
+        def first_key(key)
+          @first_key = key
+        end
+      end
+    end
+  end
+end

data/lib/mongo/message/remove_message.rb

@@ -0,0 +1,36 @@
+# --
+# 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
+
+      class RemoveMessage < Message
+
+        def initialize(db_name, collection_name, sel)
+          super(OP_DELETE)
+          write_int(0)
+          write_string("#{db_name}.#{collection_name}")
+          write_int(0)              # flags?
+          write_doc(sel)
+        end
+      end
+    end
+  end
+end