kbaum-mongo 0.18.3p

data/bin/bson_benchmark.rb

@@ -0,0 +1,59 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+
+include Mongo
+
+TRIALS = 100000
+
+def encode(doc)
+  t0 = Time.new
+  b = BSON.new
+  TRIALS.times { |i|
+    b = BSON.new
+    b.serialize doc
+  }
+  print "took: #{Time.now.to_f - t0.to_f}\n"
+  return b
+end
+
+def decode(bson)
+  t0 = Time.new
+  doc = nil
+  TRIALS.times { |i|
+    doc = bson.deserialize
+  }
+  print "took: #{Time.now.to_f - t0.to_f}\n"
+  return doc
+end
+
+TEST_CASES = [{},
+              {
+                "hello" => "world"
+              },
+              {
+                "hello" => "world",
+                "mike" => "something",
+                "here's" => "another"
+              },
+              {
+                "int" => 200,
+                "bool" => true,
+                "an int" => 20,
+                "a bool" => false
+              },
+              {
+                "this" => 5,
+                "is" => {"a" => true},
+                "big" => [true, 5.5],
+                "object" => nil
+              }]
+
+TEST_CASES.each { |doc|
+  print "case #{doc.inspect}\n"
+  print "enc bson\n"
+  enc_bson = encode(doc)
+  print "dec bson\n"
+  raise "FAIL" unless doc == decode(enc_bson)
+}

data/bin/fail_if_no_c.rb

@@ -0,0 +1,11 @@
+# Fail if the C extension module isn't installed.
+#
+# Only really intended to be used by internal build scripts.
+
+require 'rubygems'
+require 'mongo'
+begin
+  require 'mongo_ext/cbson'
+rescue LoadError
+  Process.exit 1
+end

data/examples/admin.rb

@@ -0,0 +1,42 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+require 'pp'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Mongo::Connection.new(host, port).db('ruby-mongo-examples')
+coll = db.create_collection('test')
+
+# Erase all records from collection, if any
+coll.remove
+
+admin = db.admin
+
+# Profiling level set/get
+puts "Profiling level: #{admin.profiling_level}"
+
+# Start profiling everything
+admin.profiling_level = :all
+
+# Read records, creating a profiling event
+coll.find().to_a
+
+# Stop profiling
+admin.profiling_level = :off
+
+# Print all profiling info
+pp admin.profiling_info
+
+# Validate returns a hash if all is well and
+# raises an exception if there is a problem.
+info = admin.validate_collection(coll.name)
+puts "valid = #{info['ok']}"
+puts info['result']
+
+# Destroy the collection
+coll.drop

data/examples/capped.rb

@@ -0,0 +1,22 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'mongo'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+db.drop_collection('test')
+
+# A capped collection has a max size and, optionally, a max number of records.
+# Old records get pushed out by new ones once the size or max num records is reached.
+coll = db.create_collection('test', :capped => true, :size => 1024, :max => 12)
+
+100.times { |i| coll.insert('a' => i+1) }
+
+# We will only see the last 12 records
+coll.find().each { |row| p row }
+
+coll.drop

data/examples/cursor.rb

@@ -0,0 +1,48 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+require 'pp'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+coll = db.collection('test')
+
+# Erase all records from collection, if any
+coll.remove
+
+# Insert 3 records
+3.times { |i| coll.insert({'a' => i+1}) }
+
+# Cursors don't run their queries until you actually attempt to retrieve data
+# from them.
+
+# Find returns a Cursor, which is Enumerable. You can iterate:
+coll.find().each { |row| pp row }
+
+# You can turn it into an array:
+array = coll.find().to_a
+
+# You can iterate after turning it into an array (the cursor will iterate over
+# the copy of the array that it saves internally.)
+cursor = coll.find()
+array = cursor.to_a
+cursor.each { |row| pp row }
+
+# You can get the next object
+first_object = coll.find().next_document
+
+# next_document returns nil if there are no more objects that match
+cursor = coll.find()
+obj = cursor.next_document
+while obj
+  pp obj
+  obj = cursor.next_document
+end
+
+# Destroy the collection
+coll.drop

data/examples/gridfs.rb

@@ -0,0 +1,88 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+require 'mongo/gridfs'
+
+include Mongo
+include GridFS
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+
+def dump(db, fname)
+  GridStore.open(db, fname, 'r') { |f| puts f.read }
+end
+
+# Write a new file
+GridStore.open(db, 'foobar', 'w') { |f| f.write("hello, world!") }
+
+# Read it and print out the contents
+dump(db, 'foobar')
+
+# Append more data
+GridStore.open(db, 'foobar', 'w+') { |f| f.write("\n"); f.puts "line two" }
+dump(db, 'foobar')
+
+# Overwrite
+GridStore.open(db, 'foobar', 'w') { |f| f.puts "hello, sailor!" }
+dump(db, 'foobar')
+
+# File existence tests
+puts "File 'foobar' exists: #{GridStore.exist?(db, 'foobar')}"
+puts "File 'does-not-exist' exists: #{GridStore.exist?(db, 'does-not-exist')}"
+
+# Read with offset (uses seek)
+puts GridStore.read(db, 'foobar', 6, 7)
+
+# Rewind/seek/tell
+GridStore.open(db, 'foobar', 'w') { |f|
+  f.write "hello, world!"
+  f.rewind
+  f.write "xyzzz"
+  puts f.tell                   # => 5
+  f.seek(4)
+  f.write('y')
+}
+dump(db, 'foobar')              # => 'xyzzy'
+
+# Unlink (delete)
+GridStore.unlink(db, 'foobar')
+puts "File 'foobar' exists after delete: #{GridStore.exist?(db, 'foobar')}"
+
+# Metadata
+GridStore.open(db, 'foobar', 'w') { |f| f.write("hello, world!") }
+GridStore.open(db, 'foobar', 'r') { |f|
+  puts f.content_type
+  puts f.upload_date
+  puts f.chunk_size
+  puts f.metadata.inspect
+}
+
+# Add some metadata; change content type
+GridStore.open(db, 'foobar', 'w+') { |f|
+  f.content_type = 'text/xml'
+  f.metadata = {'a' => 1}
+}
+# Print it
+GridStore.open(db, 'foobar', 'r') { |f|
+  puts f.content_type
+  puts f.upload_date
+  puts f.chunk_size
+  puts f.metadata.inspect
+}
+
+# You can also set metadata when initially writing the file. Setting :root
+# means that the file and its chunks are stored in a different root
+# collection: instead of gridfs.files and gridfs.chunks, here we use
+# my_files.files and my_files.chunks.
+GridStore.open(db, 'foobar', 'w',
+               :content_type => 'text/plain',
+               :metadata => {'a' => 1},
+               :chunk_size => 1024 * 4,
+               :root => 'my_files') { |f|
+  f.puts 'hello, world'
+}
+

data/examples/index_test.rb

@@ -0,0 +1,126 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts ">> Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-index_test')
+
+class Exception
+  def errmsg
+    "%s: %s\n%s" % [self.class, message, (backtrace || []).join("\n") << "\n"]
+  end
+end
+
+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 = (0...OBJS_COUNT).collect {|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 [[: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 "number_1_rndm_1_msg_1"
+  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/info.rb

@@ -0,0 +1,31 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+coll = db.collection('test')
+
+# Erase all records from collection, if any
+coll.remove
+
+# Insert 3 records
+3.times { |i| coll.insert({'a' => i+1}) }
+
+# Collection names in database
+p db.collection_names
+
+# More information about each collection
+p db.collections_info
+
+# Index information
+db.create_index('test', 'a')
+p db.index_information('test')
+
+# Destroy the collection
+coll.drop

data/examples/queries.rb

@@ -0,0 +1,70 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+require 'pp'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+coll = db.collection('test')
+
+# Remove all records, if any
+coll.remove
+
+# Insert three records
+coll.insert('a' => 1)
+coll.insert('a' => 2)
+coll.insert('b' => 3)
+
+# Count.
+puts "There are #{coll.count()} records."
+
+# Find all records. find() returns a Cursor.
+cursor = coll.find()
+
+# Print them. Note that all records have an _id automatically added by the
+# database. See pk.rb for an example of how to use a primary key factory to
+# generate your own values for _id.
+cursor.each { |row| pp row }
+
+# Cursor has a to_a method that slurps all records into memory.
+rows = coll.find().to_a
+rows.each { |row| pp row }
+
+# See Collection#find. From now on in this file, we won't be printing the
+# records we find.
+coll.find('a' => 1)
+
+# Find records sort by 'a', skip 1, limit 2 records.
+# Sort can be single name, array, or hash.
+coll.find({}, {:skip => 1, :limit => 2, :sort => 'a'})
+
+# Find all records with 'a' > 1. There is also $lt, $gte, and $lte.
+coll.find({'a' => {'$gt' => 1}})
+coll.find({'a' => {'$gt' => 1, '$lte' => 3}})
+
+# Find all records with 'a' in a set of values.
+coll.find('a' => {'$in' => [1,2]})
+
+# Find by regexp
+coll.find('a' => /[1|2]/)
+
+# Print query explanation
+pp coll.find('a' => /[1|2]/).explain()
+
+# Use a hint with a query. Need an index. Hints can be stored with the
+# collection, in which case they will be used with all queries, or they can be
+# specified per query, in which case that hint overrides the hint associated
+# with the collection if any.
+coll.create_index('a')
+coll.hint = 'a'
+
+# You will see a different explanation now that the hint is in place
+pp coll.find('a' => /[1|2]/).explain()
+
+# Override hint for single query
+coll.find({'a' => 1}, :hint => 'b')

data/examples/simple.rb

@@ -0,0 +1,24 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+coll = db.collection('test')
+
+# Erase all records from collection, if any
+coll.remove
+
+# Insert 3 records
+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 }
+
+# Destroy the collection
+coll.drop

data/examples/strict.rb

@@ -0,0 +1,35 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+
+db.drop_collection('does-not-exist')
+db.create_collection('test')
+
+db.strict = true
+
+begin
+  # Can't reference collection that does not exist
+  db.collection('does-not-exist')
+  puts "error: expected exception"
+rescue => ex
+  puts "expected exception: #{ex}"
+end
+
+begin
+  # Can't create collection that already exists
+  db.create_collection('test')
+  puts "error: expected exception"
+rescue => ex
+  puts "expected exception: #{ex}"
+end
+
+db.strict = false
+db.drop_collection('test')

data/examples/types.rb

@@ -0,0 +1,36 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'mongo'
+require 'pp'
+
+include Mongo
+
+host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+
+puts "Connecting to #{host}:#{port}"
+db = Connection.new(host, port).db('ruby-mongo-examples')
+coll = db.collection('test')
+
+# Remove all records, if any
+coll.remove
+
+# Insert record with all sorts of values
+coll.insert('array' => [1, 2, 3],
+            'string' => 'hello',
+            'hash' => {'a' => 1, 'b' => 2},
+            'date' => Time.now, # milliseconds only; microseconds are not stored
+            'oid' => ObjectID.new,
+            'binary' => Binary.new([1, 2, 3]),
+            'int' => 42,
+            'float' => 33.33333,
+            'regex' => /foobar/i,
+            'boolean' => true,
+            'where' => Code.new('this.x == 3'),
+            'dbref' => DBRef.new(coll.name, ObjectID.new),
+            'null' => nil,
+            'symbol' => :zildjian)
+
+pp coll.find().next_document
+
+coll.remove