mongo 0.0.1

data/lib/mongo/message.rb

@@ -0,0 +1,4 @@
+%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,21 @@
+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,19 @@
+require 'mongo/message/message'
+require 'mongo/message/opcodes'
+
+module XGen
+  module Mongo
+    module Driver
+
+      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
+    end
+  end
+end

data/lib/mongo/message/kill_cursors_message.rb

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

data/lib/mongo/message/message_header.rb

@@ -0,0 +1,34 @@
+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(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 dump
+          @buf.dump
+        end
+      end
+    end
+  end
+end
+

data/lib/mongo/message/msg_message.rb

@@ -0,0 +1,17 @@
+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,16 @@
+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,45 @@
+require 'mongo/message/message'
+require 'mongo/message/opcodes'
+require 'mongo/util/ordered_hash'
+
+module XGen
+  module Mongo
+    module Driver
+
+      class QueryMessage < Message
+
+        def initialize(db_name, collection_name, query)
+          super(OP_QUERY)
+          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.order_by && query.order_by.length > 0
+            sel = OrderedHash.new
+            sel['query'] = query.selector
+            sel['orderby'] = case query.order_by
+                             when String
+                               {query.order_by => 1}
+                             when Array
+                               h = OrderedHash.new
+                               query.order_by.each { |ob| h[ob] = 1 }
+                               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
+          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,20 @@
+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

data/lib/mongo/message/update_message.rb

@@ -0,0 +1,21 @@
+require 'mongo/message/message'
+require 'mongo/message/opcodes'
+
+module XGen
+  module Mongo
+    module Driver
+
+      class UpdateMessage < Message
+
+        def initialize(db_name, collection_name, sel, obj, repsert)
+          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
+      end
+    end
+  end
+end

data/lib/mongo/mongo.rb

@@ -0,0 +1,52 @@
+# --
+# Copyright (C) 2008-2009 10gen Inc.
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU Affero General Public License, version 3, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License
+# for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+# ++
+
+require 'mongo/db'
+
+module XGen
+  module Mongo
+    module Driver
+
+      # Represents a Mongo database server.
+      class Mongo
+
+        DEFAULT_PORT = 27017
+
+        # Host default is 'localhost', port default is DEFAULT_PORT.
+        def initialize(host='localhost', port=DEFAULT_PORT)
+          @host, @port = host, port
+        end
+
+        # Return the XGen::Mongo::Driver::DB named +db_name+.
+        def db(db_name)
+          XGen::Mongo::Driver::DB.new(db_name, @host, @port)
+        end
+
+        # Not implemented.
+        def clone_database(from)
+          raise "not implemented"
+        end
+
+        # Not implemented.
+        def copy_database(from_host, from_db, to_db)
+          raise "not implemented"
+        end
+
+      end
+    end
+  end
+end
+

data/lib/mongo/objectid.rb

@@ -0,0 +1,107 @@
+# --
+# Copyright (C) 2008-2009 10gen Inc.
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU Affero General Public License, version 3, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License
+# for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+# ++
+
+require 'mutex_m'
+require 'mongo/util/byte_buffer'
+
+module XGen
+  module Mongo
+    module Driver
+
+      # Implementation of the Babble OID. Object ids are not required by
+      # Mongo, but they make certain operations more efficient.
+      #
+      # The driver does not automatically assign ids to records that are
+      # inserted. (An upcoming feature will allow you to give an id "factory"
+      # to a database and/or a collection.)
+      # 
+      #   12 bytes
+      #   ---
+      #    0 time
+      #    1
+      #    2
+      #    3
+      #    4 machine
+      #    5
+      #    6
+      #    7 pid
+      #    8
+      #    9 inc
+      #   10
+      #   11
+      class ObjectID
+
+        MACHINE = ( val = rand(0x1000000); [val & 0xff, (val >> 8) & 0xff, (val >> 16) & 0xff] )
+        PID = ( val = rand(0x10000); [val & 0xff, (val >> 8) & 0xff]; )
+
+        LOCK = Object.new
+        LOCK.extend Mutex_m
+
+        @@index_time = Time.new.to_i
+        @@index = 0
+
+        # +data+ is an array of bytes. If nil, a new id will be generated.
+        # The time +t+ is only used for testing; leave it nil.
+        def initialize(data=nil, t=nil)
+          @data = data || generate_id(t)
+        end
+
+        def eql?(other)
+          @data == other.to_a
+        end
+        alias_method :==, :eql?
+
+        def to_a
+          @data.dup
+        end
+
+        def to_s
+          @data.collect { |b| '%02x' % b }.join
+        end
+
+        # (Would normally be private, but isn't so we can test it.)
+        def generate_id(t=nil)
+          t ||= Time.new.to_i
+          buf = ByteBuffer.new
+          buf.put_int(t & 0xffffffff)
+          buf.put_array(MACHINE)
+          buf.put_array(PID)
+          i = index_for_time(t)
+          buf.put(i & 0xff)
+          buf.put((i >> 8) & 0xff)
+          buf.put((i >> 16) & 0xff)
+
+          buf.rewind
+          buf.to_a.dup
+        end
+
+        # (Would normally be private, but isn't so we can test it.)
+        def index_for_time(t)
+          LOCK.mu_synchronize {
+            if t != @@index_time
+              @@index = 0
+              @@index_time = t
+            end
+            retval = @@index
+            @@index += 1
+            retval
+          }
+        end
+
+      end
+    end
+  end
+end