mongo 1.3.0 → 1.12.5

data/lib/mongo.rb

@@ -1,33 +1,38 @@
-# encoding: UTF-8
-#
-# --
-# Copyright (C) 2008-2011 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, 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
+#   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.
-# ++
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 
 module Mongo
-  VERSION = "1.3.0"
-end
+  ASCENDING   =  1
+  DESCENDING  = -1
+  GEO2D       = '2d'
+  GEO2DSPHERE = '2dsphere'
+  GEOHAYSTACK = 'geoHaystack'
+  TEXT        = 'text'
+  HASHED      = 'hashed'
 
-module Mongo
-  ASCENDING  =  1
-  DESCENDING = -1
-  GEO2D      = '2d'
+  INDEX_TYPES = {
+    'ASCENDING'   => ASCENDING,
+    'DESCENDING'  => DESCENDING,
+    'GEO2D'       => GEO2D,
+    'GEO2DSPHERE' => GEO2DSPHERE,
+    'GEOHAYSTACK' => GEOHAYSTACK,
+    'TEXT'        => TEXT,
+    'HASHED'      => HASHED
+  }
 
   DEFAULT_MAX_BSON_SIZE = 4 * 1024 * 1024
+  MESSAGE_SIZE_FACTOR = 2
 
   module Constants
     OP_REPLY        = 1
@@ -45,53 +50,49 @@ module Mongo
     OP_QUERY_NO_CURSOR_TIMEOUT = 2 ** 4
     OP_QUERY_AWAIT_DATA        = 2 ** 5
     OP_QUERY_EXHAUST           = 2 ** 6
+    OP_QUERY_PARTIAL           = 2 ** 7
 
     REPLY_CURSOR_NOT_FOUND     = 2 ** 0
     REPLY_QUERY_FAILURE        = 2 ** 1
     REPLY_SHARD_CONFIG_STALE   = 2 ** 2
     REPLY_AWAIT_CAPABLE        = 2 ** 3
   end
+
+  module ErrorCode # MongoDB Core Server src/mongo/base/error_codes.err
+    BAD_VALUE                = 2
+    UNKNOWN_ERROR            = 8
+    INVALID_BSON             = 22
+    WRITE_CONCERN_FAILED     = 64
+    MULTIPLE_ERRORS_OCCURRED = 65
+    UNAUTHORIZED             = 13
+
+    # mongod/s 2.6 and above return code 59 when a command doesn't exist.
+    # mongod versions previous to 2.6 and mongos 2.4.x return no error code
+    # when a command does exist.
+    # mongos versions previous to 2.4.0 return code 13390 when a command
+    # does not exist.
+    COMMAND_NOT_FOUND_CODES  = [nil, 59, 13390]
+  end
 end
 
 require 'bson'
 
-require 'mongo/util/conversions'
-require 'mongo/util/support'
-require 'mongo/util/core_ext'
-require 'mongo/util/pool'
-require 'mongo/util/server_version'
-require 'mongo/util/uri_parser'
+require 'set'
+require 'thread'
+require 'monitor'
 
-require 'mongo/collection'
+require 'mongo/utils'
+require 'mongo/exception'
+require 'mongo/functional'
 require 'mongo/connection'
-require 'mongo/repl_set_connection'
+require 'mongo/collection_writer'
+require 'mongo/collection'
+require 'mongo/bulk_write_collection_view'
 require 'mongo/cursor'
 require 'mongo/db'
-require 'mongo/exceptions'
-require 'mongo/gridfs/grid_ext'
-require 'mongo/gridfs/grid'
-require 'mongo/gridfs/grid_io'
-if RUBY_PLATFORM =~ /java/
-  require 'mongo/gridfs/grid_io_fix'
-end
-require 'mongo/gridfs/grid_file_system'
-
-# Use SystemTimer on Ruby 1.8
-if !defined?(RUBY_ENGINE) || (RUBY_ENGINE == 'ruby' && RUBY_VERSION < '1.9.0')
-  begin
-    require 'system_timer'
-    if SystemTimer.method(:timeout).arity.abs != 2
-      raise LoadError
-    end
-    Mongo::TimeoutHandler = SystemTimer
-  rescue LoadError
-    warn "Could not load SystemTimer >= v1.2.0. Falling back to timeout.rb. " +
-         "SystemTimer is STRONGLY recommended for timeouts in Ruby 1.8.7. " +
-         "See http://ph7spot.com/blog/system-timer-1-2-release for details."
-    require 'timeout'
-    Mongo::TimeoutHandler = Timeout
-  end
-else
-  require 'timeout'
-  Mongo::TimeoutHandler = Timeout
-end
+require 'mongo/gridfs'
+require 'mongo/networking'
+require 'mongo/mongo_client'
+require 'mongo/mongo_replica_set_client'
+require 'mongo/mongo_sharded_client'
+require 'mongo/legacy'

data/mongo.gemspec

@@ -1,34 +1,31 @@
-require "./lib/mongo"
-
 Gem::Specification.new do |s|
-  s.name = 'mongo'
-
-  s.version = Mongo::VERSION
-
-  s.platform = Gem::Platform::RUBY
-  s.summary = 'Ruby driver for the MongoDB'
-  s.description = 'A Ruby driver for MongoDB. For more information about Mongo, see http://www.mongodb.org.'
-
-  s.require_paths = ['lib']
-
-  s.files  = ['README.md', 'Rakefile', 'mongo.gemspec', 'LICENSE.txt']
-  s.files += ['lib/mongo.rb'] + Dir['lib/mongo/**/*.rb']
-  s.files += Dir['docs/**/*.md'] + Dir['examples/**/*.rb'] + Dir['bin/**/*.rb']
-  s.files += Dir['bin/mongo_console']
-  s.test_files = Dir['test/**/*.rb']
-
-  s.executables = ['mongo_console']
-
-  s.has_rdoc = true
-  s.test_files = Dir['test/**/*.rb']
-
-  s.has_rdoc = true
-  s.rdoc_options = ['--main', 'README.md', '--inline-source']
-  s.extra_rdoc_files = ['README.md']
-
-  s.authors = ['Jim Menard', 'Mike Dirolf', 'Kyle Banker']
-  s.email = 'mongodb-dev@googlegroups.com'
-  s.homepage = 'http://www.mongodb.org'
-
-  s.add_dependency(%q<bson>, [">= #{Mongo::VERSION}"])
+  s.name              = 'mongo'
+
+  s.version           = File.read(File.join(File.dirname(__FILE__), 'VERSION'))
+  s.platform          = Gem::Platform::RUBY
+  s.authors           = ['Emily Stolfo', 'Durran Jordan', 'Gary Murakami', 'Tyler Brock', 'Brandon Black']
+  s.email             = 'mongodb-dev@googlegroups.com'
+  s.homepage          = 'http://www.mongodb.org'
+  s.summary           = 'Ruby driver for MongoDB'
+  s.description       = 'A Ruby driver for MongoDB. For more information about Mongo, see http://www.mongodb.org.'
+  s.rubyforge_project = 'mongo'
+  s.license           = 'Apache License Version 2.0'
+
+  if File.exists?('gem-private_key.pem')
+    s.signing_key = 'gem-private_key.pem'
+    s.cert_chain  = ['gem-public_cert.pem']
+  else
+    warn 'Warning: No private key present, creating unsigned gem.'
+  end
+
+  s.files             = ['mongo.gemspec', 'LICENSE', 'VERSION']
+  s.files             += ['README.md', 'Rakefile', 'bin/mongo_console']
+  s.files             += ['lib/mongo.rb'] + Dir['lib/mongo/**/*.rb']
+
+  s.test_files        = Dir['test/**/*.rb'] - Dir['test/bson/*']
+  s.executables       = ['mongo_console']
+  s.require_paths     = ['lib']
+  s.has_rdoc          = 'yard'
+
+  s.add_dependency('bson', "#{s.version}")
 end

data/test/functional/authentication_test.rb

@@ -0,0 +1,39 @@
+# Copyright (C) 2009-2013 MongoDB, 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 'test_helper'
+require 'shared/authentication/basic_auth_shared'
+require 'shared/authentication/sasl_plain_shared'
+require 'shared/authentication/bulk_api_auth_shared'
+require 'shared/authentication/gssapi_shared'
+require 'shared/authentication/scram_shared'
+
+
+class AuthenticationTest < Test::Unit::TestCase
+  include Mongo
+  include BasicAuthTests
+  include SASLPlainTests
+  include BulkAPIAuthTests
+  include GSSAPITests
+  include SCRAMTests
+
+  def setup
+    @client    = standard_connection
+    omit("auth not enabled on mongod") unless auth_enabled?(@client)
+    @admin     = @client['admin']
+    @version   = @client.server_version
+    @db        = @client['ruby-test']
+    @host_info = host_port
+  end
+end

data/test/functional/bulk_api_stress_test.rb

@@ -0,0 +1,133 @@
+# Copyright (C) 2009-2013 MongoDB, 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 'test_helper'
+
+class BulkApiStressTest < Test::Unit::TestCase
+
+  # Generate a large string of 'size' MB (estimated
+  # by a string of 'size' * 1024 * 1024 characters).
+  def generate_large_string(size)
+    s = "a" * (size * 1024 * 1024)
+  end
+
+  def setup
+    @client = standard_connection
+    @db = @client[TEST_DB]
+    @coll = @db["bulk-api-stress-tests"]
+    @coll.remove
+  end
+
+  def test_ordered_batch_large_inserts
+    bulk = @coll.initialize_ordered_bulk_op
+    s = generate_large_string(4)
+
+    for i in 0..5
+      bulk.insert({:_id => i, :msg => s})
+    end
+    bulk.insert({:_id => 3}) # error
+    bulk.insert({:_id => 100})
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 6, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 6, error["index"]
+    assert_equal 6, @coll.count()
+  end
+
+  def test_unordered_batch_large_inserts
+    bulk = @coll.initialize_unordered_bulk_op
+    s = generate_large_string(4)
+
+    for i in 0..5
+      bulk.insert({:_id => i, :msg => s})
+    end
+    bulk.insert({:_id => 3}) # error
+    bulk.insert({:_id => 100})
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 7, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 6, error["index"]
+    assert_equal 7, @coll.count()
+  end
+
+  def test_large_single_insert
+    bulk = @coll.initialize_unordered_bulk_op
+    s = generate_large_string(17)
+    bulk.insert({:a => s})
+# RUBY-730:
+#    ex = assert_raise BulkWriteError do
+#      bulk.execute
+#    end
+  end
+
+  def test_ordered_batch_large_batch
+    bulk = @coll.initialize_ordered_bulk_op
+
+    bulk.insert({:_id => 1600})
+    for i in 0..2000
+      bulk.insert({:_id => i})
+    end
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 1601, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 1601, error["index"]
+    assert_equal 1601, @coll.count()
+  end
+
+  def test_unordered_batch_large_batch
+    bulk = @coll.initialize_unordered_bulk_op
+
+    bulk.insert({:_id => 1600})
+    for i in 0..2000
+      bulk.insert({:_id => i})
+    end
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 2001, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 1601, error["index"]
+    assert_equal 2001, @coll.count()
+  end
+end