mongo 1.3.0 → 1.12.5

data/lib/mongo/connection/sharding_pool_manager.rb

@@ -0,0 +1,67 @@
+# 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.
+
+module Mongo
+  class ShardingPoolManager < PoolManager
+    def inspect
+      "<Mongo::ShardingPoolManager:0x#{self.object_id.to_s(16)} @seeds=#{@seeds}>"
+    end
+
+    # "Best" should be the member with the fastest ping time
+    # but connect/connect_to_members reinitializes @members
+    def best(members)
+      Array(members.first)
+    end
+
+    def connect
+      @connect_mutex.synchronize do
+        begin
+          thread_local[:locks][:connecting_manager] = true
+          @refresh_required = false
+          disconnect_old_members
+          connect_to_members
+          initialize_pools best(@members)
+          update_max_sizes
+          @seeds = discovered_seeds
+        ensure
+          thread_local[:locks][:connecting_manager] = false
+        end
+      end
+    end
+
+    # Checks that each node is healthy (via check_is_master) and that each
+    # node is in fact a mongos. If either criteria are not true, a refresh is
+    # set to be triggered and close() is called on the node.
+    #
+    # @return [Boolean] indicating if a refresh is required.
+    def check_connection_health
+      @refresh_required = false
+      @members.each do |member|
+        begin
+          config = @client.check_is_master([member.host, member.port])
+          unless config && config.has_key?('msg')
+            @refresh_required = true
+            member.close
+          end
+        rescue OperationTimeout
+          @refresh_required = true
+          member.close
+        end
+        break if @refresh_required
+      end
+      @refresh_required
+    end
+
+  end
+end

data/lib/mongo/connection/socket/socket_util.rb

@@ -0,0 +1,37 @@
+# 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 'socket'
+require 'timeout'
+
+module SocketUtil
+
+  attr_accessor :pool, :pid, :auths
+
+  def checkout
+    @pool.checkout if @pool
+  end
+
+  def checkin
+    @pool.checkin(self) if @pool
+  end
+
+  def close
+    @socket.close unless closed?
+  end
+
+  def closed?
+    @socket.closed?
+  end
+end

data/lib/mongo/connection/socket/ssl_socket.rb

@@ -0,0 +1,95 @@
+# 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 'openssl'
+
+module Mongo
+
+  # A basic wrapper over Ruby's SSLSocket that initiates
+  # a TCP connection over SSL and then provides an basic interface
+  # mirroring Ruby's TCPSocket, vis., TCPSocket#send and TCPSocket#read.
+  class SSLSocket
+    include SocketUtil
+
+    def initialize(host, port, op_timeout=nil, connect_timeout=nil, opts={})
+      @op_timeout      = op_timeout
+      @connect_timeout = connect_timeout
+      @pid             = Process.pid
+      @auths           = Set.new
+
+      @tcp_socket = ::TCPSocket.new(host, port)
+      @tcp_socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+
+      @context = OpenSSL::SSL::SSLContext.new
+
+      if opts[:cert]
+        @context.cert = OpenSSL::X509::Certificate.new(File.open(opts[:cert]))
+      end
+
+      if opts[:key]
+        if opts[:key_pass_phrase]
+          @context.key = OpenSSL::PKey::RSA.new(File.open(opts[:key]), opts[:key_pass_phrase])
+        else
+          @context.key = OpenSSL::PKey::RSA.new(File.open(opts[:key]))
+        end
+      end
+
+      if opts[:verify]
+        @context.ca_file = opts[:ca_cert]
+        @context.verify_mode = OpenSSL::SSL::VERIFY_PEER
+      end
+
+      begin
+        @socket = OpenSSL::SSL::SSLSocket.new(@tcp_socket, @context)
+        @socket.sync_close = true
+        connect
+      rescue OpenSSL::SSL::SSLError
+        raise ConnectionFailure, "SSL handshake failed. MongoDB may " +
+                                 "not be configured with SSL support."
+      end
+
+      if opts[:verify]
+        unless OpenSSL::SSL.verify_certificate_identity(@socket.peer_cert, host)
+          raise ConnectionFailure, "SSL handshake failed. Hostname mismatch."
+        end
+      end
+
+      self
+    end
+
+    def connect
+      if @connect_timeout
+        Timeout::timeout(@connect_timeout, ConnectionTimeoutError) do
+          @socket.connect
+        end
+      else
+        @socket.connect
+      end
+    end
+
+    def send(data)
+      @socket.syswrite(data)
+    end
+
+    def read(length, buffer)
+      if @op_timeout
+        Timeout::timeout(@op_timeout, OperationTimeout) do
+          @socket.sysread(length, buffer)
+        end
+      else
+        @socket.sysread(length, buffer)
+      end
+    end
+  end
+end

data/lib/mongo/connection/socket/tcp_socket.rb

@@ -0,0 +1,87 @@
+# 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.
+
+module Mongo
+  # Wrapper class for Socket
+  #
+  # Emulates TCPSocket with operation and connection timeout
+  # sans Timeout::timeout
+  #
+  class TCPSocket
+    include SocketUtil
+
+    def initialize(host, port, op_timeout=nil, connect_timeout=nil, opts={})
+      @op_timeout      = op_timeout
+      @connect_timeout = connect_timeout
+      @pid             = Process.pid
+      @auths           = Set.new
+
+      @socket = handle_connect(host, port)
+    end
+
+    def handle_connect(host, port)
+      error = nil
+      # Following python's lead (see PYTHON-356)
+      family = host == 'localhost' ? Socket::AF_INET : Socket::AF_UNSPEC
+      addr_info = Socket.getaddrinfo(host, nil, family, Socket::SOCK_STREAM)
+      error = nil
+      addr_info.each do |info|
+        begin
+          sock = Socket.new(info[4], Socket::SOCK_STREAM, 0)
+          sock.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+          socket_address = Socket.pack_sockaddr_in(port, info[3])
+          connect(sock, socket_address)
+          return sock
+        rescue IOError, SystemCallError => e
+          error = e
+          sock.close if sock
+        end
+      end
+      raise error
+    end
+
+    def connect(socket, socket_address)
+      if @connect_timeout
+        Timeout::timeout(@connect_timeout, ConnectionTimeoutError) do
+          socket.connect(socket_address)
+        end
+      else
+        socket.connect(socket_address)
+      end
+    end
+
+    def send(data)
+      @socket.write(data)
+    end
+
+    def read(maxlen, buffer)
+      # Block on data to read for @op_timeout seconds
+      begin
+        ready = IO.select([@socket], nil, [@socket], @op_timeout)
+        unless ready
+          raise OperationTimeout
+        end
+      rescue IOError
+        raise ConnectionFailure
+      end
+
+      # Read data from socket
+      begin
+        @socket.sysread(maxlen, buffer)
+      rescue SystemCallError, IOError => ex
+        raise ConnectionFailure, ex
+      end
+    end
+  end
+end

data/lib/mongo/connection/socket/unix_socket.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.
+
+module Mongo
+
+  attr_accessor :auths
+
+  # Wrapper class for Socket
+  #
+  # Emulates UNIXSocket with operation and connection timeout
+  # sans Timeout::timeout
+  #
+  class UNIXSocket < TCPSocket
+    def initialize(socket_path, port=:socket, op_timeout=nil, connect_timeout=nil, opts={})
+      @op_timeout      = op_timeout
+      @connect_timeout = connect_timeout
+      @pid             = Process.pid
+      @auths           = Set.new
+
+      @address         = socket_path
+      @port            = :socket # purposely override input
+
+      @socket_address  = Socket.pack_sockaddr_un(@address)
+      @socket          = Socket.new(Socket::AF_UNIX, Socket::SOCK_STREAM, 0)
+      connect(@socket, @socket_address)
+    end
+  end
+end

data/lib/mongo/connection/socket.rb

@@ -0,0 +1,18 @@
+# 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 'mongo/connection/socket/socket_util.rb'
+require 'mongo/connection/socket/ssl_socket.rb'
+require 'mongo/connection/socket/tcp_socket.rb'
+require 'mongo/connection/socket/unix_socket.rb'