mongodb-mongo 0.1.3

data/README.rdoc

@@ -0,0 +1,216 @@
+= Introduction
+
+This is a Ruby driver for the 10gen Mongo DB. For more information about
+Mongo, see http://www.mongodb.org.
+
+Note: this driver is still alpha quality. The API will change, as *may* the
+data saved to the database (especially primary key values). Do *_not_* use
+this for any production data yet.
+
+Start by reading the XGen::Mongo::Driver::Mongo and XGen::Mongo::Driver::DB
+documentation, then move on to XGen::Mongo::Driver::Collection and
+XGen::Mongo::Driver::Cursor.
+
+A quick code sample:
+
+  require 'mongo'
+
+  include XGen::Mongo::Driver
+
+  db = Mongo.new('localhost').db('sample-db')
+  coll = db.collection('test')
+
+  coll.clear
+  3.times { |i| coll.insert({'a' => i+1}) }
+  puts "There are #{coll.count()} records. Here they are:"
+  coll.find().each { |doc| puts doc.inspect }
+
+= Installation
+
+Install the "mongo" gem by typing
+
+  $ gem sources -a http://gems.github.com
+  $ sudo gem install mongodb-mongo-ruby-driver
+
+The source code is available at http://github.com/mongodb/mongo-ruby-driver.
+You can either clone the git repository or download a tarball or zip file.
+Once you have the source, you can use it from wherever you downloaded it or
+you can install it as a gem from the source by typing
+
+  $ rake gem:install
+
+
+= Demo
+
+You can see and run the examples if you've downloaded the source. Mongo must
+be running, of course.
+
+  $ ruby examples/simple.rb
+
+See also the test code, especially tests/test_db_api.rb.
+
+
+= Notes
+
+== String Encoding
+
+The BSON ("Binary JSON") format used to communicate with Mongo requires that
+strings be UTF-8 (http://en.wikipedia.org/wiki/UTF-8).
+
+Ruby 1.9 has built-in character encoding support. All strings sent to Mongo
+and received from Mongo are converted to UTF-8 when necessary, and strings
+read from Mongo will have their character encodings set to UTF-8.
+
+When used with Ruby 1.8, the bytes in each string are written to and read from
+Mongo as-is. If the string is ASCII all is well, because ASCII is a subset of
+UTF-8. If the string is not ASCII then it may not be a well-formed UTF-8
+string.
+
+== The DB class
+
+=== Primary Key Factories
+
+A basic Mongo driver is not responsible for creating primary keys or knowing
+how to interpret them. You can tell the Ruby Mongo driver how to create
+primary keys by passing in the :pk option to the Mongo#db method.
+
+  include XGen::Mongo::Driver
+  db = Mongo.new.db('dbname', :pk => MyPKFactory)
+
+A primary key factory object must respond to :create_pk, which should take a
+hash and return a hash which merges the original hash with any primary key
+fields the factory wishes to inject. NOTE: if the object already has a primary
+key, the factory should not inject a new key; this means that the object is
+being used in a repsert but it already exists. The idea here is that when ever
+a record is inserted, the :pk object's +create_pk+ method will be called and
+the new hash returned will be inserted.
+
+Here is a sample primary key factory, taken from the tests:
+
+  class TestPKFactory
+    def create_pk(row)
+      row['_id'] ||= XGen::Mongo::Driver::ObjectID.new
+      row
+    end
+  end
+
+A database's PK factory object may be changed, but this is not recommended.
+The only reason it is changeable is so that libraries such as MongoRecord that
+use this driver can set the PK factory after obtaining the database but before
+using it for the first time.
+
+=== Strict mode
+
+Each database has an optional strict mode. If strict mode is on, then asking
+for a collection that does not exist will raise an error, as will asking to
+create a collection that already exists. Note that both these operations are
+completely harmless; strict mode is a programmer convenience only.
+
+To turn on strict mode, either pass in :strict => true when obtaining a DB
+object or call the :strict= method:
+
+  db = XGen::Mongo::Driver::Mongo.new.db('dbname', :strict => true)
+  # I'm feeling lax
+  db.strict = false
+  # No, I'm not!
+  db.strict = true
+
+The method DB#strict? returns the current value of that flag.
+
+
+= Testing
+
+If you have the source code, you can run the tests.
+
+  $ rake test
+
+The tests assume that the Mongo database is running on the default port.
+
+The project mongo-qa (http://github.com/mongodb/mongo-qa) contains many more
+Mongo driver tests that are language independent. To run thoses tests as part
+of the "rake test" task, run
+
+  $ rake mongo_qa
+  $ rake test
+
+The mongo_qa task uses the "git clone" command to make a copy of that project
+in a directory named mongo-qa. If the directory already exists, then the
+mongo_qa task uses "git pull" to updated the code that's there. The Ruby
+driver tests will then use some of the data files from that project when it
+runs BSON tests. You can delete this directory at any time if you don't want
+to run those tests any more.
+
+Additionally, the script bin/validate is used by the mongo-qa project's
+validator script.
+
+
+= Documentation
+
+This documentation is available online at http://mongo.rubyforge.org. You can
+generate the documentation if you have the source by typing
+
+  $ rake rdoc
+
+Then open the file html/index.html.
+
+
+= Release Notes
+
+See the git log comments.
+
+
+= To Do
+
+* Add group_by. Need to figure out how we are going to send functions. The
+  current thinking is that Mongo will allow a subset of JavaScript (which we
+  would have to send as a string), but this is still under discussion.
+
+* Tests for update and repsert.
+
+* Add a way to specify a collection of databases on startup (a simple array of
+  IP address/port numbers, perhaps, or a hash or something). The driver would
+  then find the master and, on each subsequent command, ask that machine if it
+  is the master before proceeding.
+
+* Introduce optional per-database and per-collection PKInjector.
+
+* More tests.
+
+== Optimizations
+
+* Only update message sizes once, not after every write of a value. This will
+  require an explicit call to update_message_length in each message subclass.
+
+* ensure_index commands should be cached to prevent excessive communication
+  with the database. (Or, the driver user should be informed that ensure_index
+  is not a lightweight operation for the particular driver.)
+
+
+= Credits
+
+Adrian Madrid, aemadrid@gmail.com
+* bin/mongo_console
+* examples/benchmarks.rb
+* examples/irb.rb
+* Modifications to examples/simple.rb
+* Found plenty of bugs and missing features.
+* Ruby 1.9 support.
+* Gem support.
+* Many other code suggestions and improvements.
+
+
+= License
+
+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.
+
+See http://www.gnu.org/licenses for a copy of the GNU Affero General Public
+License.

data/Rakefile

@@ -0,0 +1,54 @@
+require 'rubygems'
+require 'rubygems/specification'
+require 'fileutils'
+require 'rake'
+require 'rake/testtask'
+require 'rake/gempackagetask'
+require 'rake/contrib/rubyforgepublisher'
+
+# NOTE: some of the tests assume Mongo is running
+Rake::TestTask.new do |t|
+  t.test_files = FileList['tests/test*.rb']
+end
+
+desc "Clone or pull (update) the mongo-qa project used for testing"
+task :mongo_qa do
+  if File.exist?('mongo-qa')
+    Dir.chdir('mongo-qa') do
+      system('git pull')
+    end
+  else
+    system('git clone git://github.com/mongodb/mongo-qa.git')
+  end
+end
+
+desc "Generate documentation"
+task :rdoc do
+  FileUtils.rm_rf('html')
+  system "rdoc --main README.rdoc --op html --inline-source --quiet README.rdoc `find lib -name '*.rb'`"
+end
+
+desc "Publish documentation to mongo.rubyforge.org"
+task :publish => [:rdoc] do
+  # Assumes docs are in ./html
+  Rake::RubyForgePublisher.new(GEM, RUBYFORGE_USER).upload
+end
+
+namespace :gem do 
+
+  desc "Install the gem locally"
+  task :install do
+    sh <<EOS
+gem build mongo-ruby-driver.gemspec &&
+    sudo gem install mongo-*.gem &&
+    rm mongo-*.gem
+EOS
+  end
+
+end
+
+task :default => :list
+
+task :list do
+  system 'rake -T'
+end

data/bin/mongo_console

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+org_argv = ARGV.dup
+ARGV.clear
+
+require 'irb'
+
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+
+include XGen::Mongo::Driver
+
+host = org_argv[0] || ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = org_argv[1] || ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
+dbnm = org_argv[2] || ENV['MONGO_RUBY_DRIVER_DB']   || 'ruby-mongo-console'
+
+puts "Connecting to #{host}:#{port} (CONN) on with database #{dbnm} (DB)"
+CONN = Mongo.new(host, port)
+DB = CONN.db(dbnm)
+
+puts "Starting IRB session..."
+IRB.start(__FILE__)

data/bin/validate

@@ -0,0 +1,51 @@
+#!/usr/bin/env ruby
+#
+# usage: validate somefile.xson somefile.bson
+#
+# Reads somefile.xson file (XML that describes a Mongo-type document),
+# converts it into a Ruby OrderedHash, runs that through the BSON
+# serialization code, and writes the BSON bytes to somefile.bson.
+#
+# In addition, this script takes the generated BSON, reads it in then writes
+# it back out to a temp BSON file. If they are different, we report that error
+# to STDOUT.
+#
+# This script is used by the mongo-qa project
+# (http://github.com/mongodb/mongo-qa).
+
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+require 'mongo/util/xml_to_ruby'
+
+if ARGV.length < 2
+  $stderr.puts "usage: validate somefile.xson somefile.bson"
+  exit 1
+end
+
+# Translate the .xson XML into a Ruby object, turn that object into BSON, and
+# write the BSON to the file as requested.
+obj = File.open(ARGV[0], 'rb') { |f| XMLToRuby.new.xml_to_ruby(f) }
+bson = BSON.new.serialize(obj).to_a
+File.open(ARGV[1], 'wb') { |f| bson.each { |b| f.putc(b) } }
+
+# Now the additional testing. Read the generated BSON back in, deserialize it,
+# and re-serialize the results. Compare that BSON with the BSON from the file
+# we output.
+bson = File.open(ARGV[1], 'rb') { |f| f.read }
+bson = if RUBY_VERSION >= '1.9'
+         bson.bytes.to_a
+       else
+         bson.split(//).collect { |c| c[0] }
+       end
+
+# Turn the Ruby object into BSON bytes and compare with the BSON bytes from
+# the file.
+bson_from_ruby = BSON.new.serialize(obj).to_a
+
+if bson.length != bson_from_ruby.length
+  $stderr.puts "error: round-trip BSON lengths differ when testing #{ARGV[0]}"
+  exit 1
+elsif bson != bson_from_ruby
+  $stderr.puts "error: round-trip BSON contents differ when testing #{ARGV[0]}"
+  exit 1
+end

data/examples/benchmarks.rb

@@ -0,0 +1,38 @@
+require "rubygems"
+require "benchmark"
+  
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = XGen::Mongo::Driver::Mongo.new(host, port).db('ruby-mongo-examples-complex')
+coll = db.collection('test')
+coll.clear
+
+OBJS_COUNT = 100
+TEST_COUNT = 100
+
+puts "Generating benchmark data"
+msgs = %w{hola hello aloha ciao}
+arr = OBJS_COUNT.times.map {|x| { :number => x, :rndm => (rand(5)+1), :msg => msgs[rand(4)] }}
+
+puts "Running benchmark"
+Benchmark.bmbm do |results|
+  results.report("single object inserts:  ") { 
+    TEST_COUNT.times {
+      coll.clear
+      arr.each {|x| coll.insert(x)} 
+    }
+  }
+  results.report("multiple object insert: ") { 
+    TEST_COUNT.times {
+      coll.clear
+      coll.insert(arr)
+    }
+  }
+end
+
+coll.clear

data/examples/blog.rb

@@ -0,0 +1,76 @@
+require "rubygems"
+
+class Exception
+  def errmsg
+    "%s: %s\n%s" % [self.class, message, (backtrace || []).join("\n") << "\n"]
+  end
+end
+  
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+
+include XGen::Mongo::Driver
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
+
+puts ">> Connecting to #{host}:#{port}"
+DB = Mongo.new(host, port).db('ruby-mongo-blog')
+
+LINE_SIZE = 120
+puts "=" * LINE_SIZE
+puts "Adding authors"
+authors = DB.collection "authors"
+authors.clear
+authors.create_index "meta", '_id' => 1, 'name' => 1, 'age' => 1
+puts "-" * LINE_SIZE
+shaksp = authors << { :name => "William Shakespeare", :email => "william@shakespeare.com", :age => 587 }
+puts "shaksp : #{shaksp.inspect}"
+borges = authors << { :name => "Jorge Luis Borges", :email => "jorge@borges.com", :age => 123 }
+puts "borges : #{borges.inspect}"
+puts "-" * LINE_SIZE
+puts "authors ordered by age ascending"
+puts "-" * LINE_SIZE
+authors.find({}, :sort => [{'age' => 1}]).each {|x| puts "%-25.25s : %-25.25s : %3i" % [x['name'], x['email'], x['age']]}
+
+puts "=" * LINE_SIZE
+puts "Adding users"
+users = DB.collection "users"
+users.clear
+# users.create_index "meta", :_id => 1, :login => 1, :name => 1
+puts "-" * LINE_SIZE
+jdoe = users << { :login => "jdoe", :name => "John Doe", :email => "john@doe.com" }
+puts "jdoe : #{jdoe.inspect}"
+lsmt = users << { :login => "lsmith", :name => "Lucy Smith", :email => "lucy@smith.com" }
+puts "lsmt : #{lsmt.inspect}"
+puts "-" * LINE_SIZE
+puts "users ordered by login ascending"
+puts "-" * LINE_SIZE
+users.find({}, :sort => [{'login' => 1}]).each {|x| puts "%-10.10s : %-25.25s : %-25.25s" % [x['login'], x['name'], x['email']]}
+
+puts "=" * LINE_SIZE
+puts "Adding articles"
+articles = DB.collection "articles"
+articles.clear
+# articles.create_index "meta", :_id => 1, :author_id => 1, :title => 1
+puts "-" * LINE_SIZE
+begin
+  art1 = articles << { :title => "Caminando por Buenos Aires", :body => "Las callecitas de Buenos Aires tienen ese no se que...", :author_id => borges["_id"].to_s }
+  puts "art1 : #{art1.inspect}"
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+begin
+  art2 = articles << { :title => "I must have seen thy face before", :body => "Thine eyes call me in a new way", :author_id => shaksp["_id"].to_s, :comments => [ { :user_id => jdoe["_id"].to_s, :body => "great article!" } ] }
+  puts "art2 : #{art2.inspect}"
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+puts "-" * LINE_SIZE
+puts "articles ordered by title ascending"
+puts "-" * LINE_SIZE
+articles.find({}, :sort => [{'title' => 1}]).each {|x| puts "%-25.25s : %-25.25s" % [x['title'], x['author_id']]}
+
+puts ">> Closing connection"
+DB.close
+puts "closed"

data/examples/index_test.rb

@@ -0,0 +1,128 @@
+require "rubygems"
+require "benchwarmer"
+
+class Exception
+  def errmsg
+    "%s: %s\n%s" % [self.class, message, (backtrace || []).join("\n") << "\n"]
+  end
+end
+  
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+
+include XGen::Mongo::Driver
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
+
+puts ">> Connecting to #{host}:#{port}"
+db = Mongo.new(host, port).db('ruby-mongo-index_test')
+
+puts ">> Dropping collection test"
+begin
+  res = db.drop_collection('test')
+  puts "dropped : #{res.inspect}" 
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+
+puts ">> Creating collection test"
+begin
+  coll = db.collection('test')
+  puts "created : #{coll.inspect}" 
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+
+OBJS_COUNT = 100
+
+puts ">> Generating test data"
+msgs = %w{hola hello aloha ciao}
+arr = OBJS_COUNT.times.map {|x| { :number => x, :rndm => (rand(5)+1), :msg => msgs[rand(4)] }}
+puts "generated"
+
+puts ">> Inserting data (#{arr.size})"
+coll.insert(arr)
+puts "inserted"
+
+puts ">> Creating index"
+res = coll.create_index "all", :_id => 1, :number => 1, :rndm => 1, :msg => 1
+# res = coll.create_index "all", '_id' => 1, 'number' => 1, 'rndm' => 1, 'msg' => 1
+puts "created index: #{res.inspect}"
+# ============================ Mongo Log ============================
+# Fri Dec  5 14:45:02 Adding all existing records for ruby-mongo-console.test to new index
+# ***
+# Bad data or size in BSONElement::size()
+# bad type:30
+# totalsize:11 fieldnamesize:4
+# lastrec:
+# Fri Dec  5 14:45:02 ruby-mongo-console.system.indexes Assertion failure false jsobj.cpp a0
+# Fri Dec  5 14:45:02  database: ruby-mongo-console op:7d2 0
+# Fri Dec  5 14:45:02  ns: ruby-mongo-console.system.indexes
+
+puts ">> Gathering index information"
+begin
+  res = coll.index_information     
+  puts "index_information : #{res.inspect}" 
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+# ============================ Console Output ============================
+# RuntimeError: Keys for index on return from db was nil. Coll = ruby-mongo-console.test
+#         from ./bin/../lib/mongo/db.rb:135:in `index_information'
+#         from (irb):11:in `collect'
+#         from ./bin/../lib/mongo/cursor.rb:47:in `each'
+#         from ./bin/../lib/mongo/db.rb:130:in `collect'
+#         from ./bin/../lib/mongo/db.rb:130:in `index_information'
+#         from ./bin/../lib/mongo/collection.rb:74:in `index_information'
+#         from (irb):11
+
+puts ">> Dropping index"
+begin
+  res = coll.drop_index "all"
+  puts "dropped : #{res.inspect}"
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+
+# ============================ Console Output ============================
+# => {"nIndexesWas"=>2.0, "ok"=>1.0}
+# ============================ Mongo Log ============================
+# 0x41802a 0x411549 0x42bac6 0x42c1f6 0x42c55b 0x42e6f7 0x41631e 0x41a89d 0x41ade2 0x41b448 0x4650d2 0x4695ad
+#  db/db(_Z12sayDbContextPKc+0x17a) [0x41802a]
+#  db/db(_Z8assertedPKcS0_j+0x9) [0x411549]
+#  db/db(_ZNK11BSONElement4sizeEv+0x1f6) [0x42bac6]
+#  db/db(_ZN7BSONObj8getFieldEPKc+0xa6) [0x42c1f6]
+#  db/db(_ZN7BSONObj14getFieldDottedEPKc+0x11b) [0x42c55b]
+#  db/db(_ZN7BSONObj19extractFieldsDottedES_R14BSONObjBuilder+0x87) [0x42e6f7]
+#  db/db(_ZN12IndexDetails17getKeysFromObjectER7BSONObjRSt3setIS0_St4lessIS0_ESaIS0_EE+0x24e) [0x41631e]
+#  db/db(_Z12_indexRecordR12IndexDetailsR7BSONObj7DiskLoc+0x5d) [0x41a89d]
+#  db/db(_Z18addExistingToIndexPKcR12IndexDetails+0xb2) [0x41ade2]
+#  db/db(_ZN11DataFileMgr6insertEPKcPKvib+0x508) [0x41b448]
+#  db/db(_Z14receivedInsertR7MessageRSt18basic_stringstreamIcSt11char_traitsIcESaIcEE+0x112) [0x4650d2]
+#  db/db(_Z10connThreadv+0xb4d) [0x4695ad]
+# Fri Dec  5 14:45:02 ruby-mongo-console.system.indexes  Caught Assertion insert, continuing
+# Fri Dec  5 14:47:59 CMD: deleteIndexes ruby-mongo-console.test
+#   d->nIndexes was 2
+#   alpha implementation, space not reclaimed
+
+puts ">> Gathering index information"
+begin
+  res = coll.index_information     
+  puts "index_information : #{res.inspect}" 
+rescue => e
+  puts "Error: #{e.errmsg}"
+end
+# ============================ Console Output ============================
+# RuntimeError: Keys for index on return from db was nil. Coll = ruby-mongo-console.test
+#         from ./bin/../lib/mongo/db.rb:135:in `index_information'
+#         from (irb):15:in `collect'
+#         from ./bin/../lib/mongo/cursor.rb:47:in `each'
+#         from ./bin/../lib/mongo/db.rb:130:in `collect'
+#         from ./bin/../lib/mongo/db.rb:130:in `index_information'
+#         from ./bin/../lib/mongo/collection.rb:74:in `index_information'
+#         from (irb):15
+
+puts ">> Closing connection"
+db.close
+puts "closed"

data/examples/simple.rb

@@ -0,0 +1,17 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+
+include XGen::Mongo::Driver
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || XGen::Mongo::Driver::Mongo::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Mongo.new(host, port).db('ruby-mongo-examples-simple')
+coll = db.collection('test')
+coll.clear
+
+3.times { |i| coll.insert({'a' => i+1}) }
+
+puts "There are #{coll.count()} records in the test collection. Here they are:"
+coll.find().each { |doc| puts doc.inspect }

data/lib/mongo/admin.rb

@@ -0,0 +1,86 @@
+# --
+# 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/util/ordered_hash'
+
+module XGen
+  module Mongo
+    module Driver
+
+      # Provide administrative database methods: those having to do with
+      # profiling and validation.
+      class Admin
+
+        def initialize(db)
+          @db = db
+        end
+
+        # Return the current database profiling level.
+        def profiling_level
+          oh = OrderedHash.new
+          oh[:profile] = -1
+          doc = @db.db_command(oh)
+          raise "Error with profile command: #{doc.inspect}" unless @db.ok?(doc) && doc['was'].kind_of?(Numeric)
+          case doc['was'].to_i
+          when 0
+            :off
+          when 1
+            :slow_only
+          when 2
+            :all
+          else
+            raise "Error: illegal profiling level value #{doc['was']}"
+          end
+        end
+
+        # Set database profiling level to :off, :slow_only, or :all.
+        def profiling_level=(level)
+          oh = OrderedHash.new
+          oh[:profile] = case level
+                         when :off
+                           0
+                         when :slow_only
+                           1
+                         when :all
+                           2
+                         else
+                           raise "Error: illegal profiling level value #{level}"
+                         end
+          doc = @db.db_command(oh)
+          raise "Error with profile command: #{doc.inspect}" unless @db.ok?(doc)
+        end
+
+        # Return an array contining current profiling information from the
+        # database.
+        def profiling_info
+          @db.query(Collection.new(@db, DB::SYSTEM_PROFILE_COLLECTION), Query.new({})).to_a
+        end
+
+        # Validate a named collection by raising an exception if there is a
+        # problem or returning +true+ if all is well.
+        def validate_collection(name)
+          doc = @db.db_command(:validate => name)
+          raise "Error with validate command: #{doc.inspect}" unless @db.ok?(doc)
+          result = doc['result']
+          raise "Error with validation data: #{doc.inspect}" unless result.kind_of?(String)
+          raise "Error: invalid collection #{name}: #{doc.inspect}" if result =~ /\b(exception|corrupt)\b/i
+          true
+        end
+
+      end
+    end
+  end
+end