826fb0d4db
Makes driver ~2x faster for simple benchmarks. DEPRECATED pool_size, auto_start_request and timeout parameters to Connection DEPRECATED Connection.start_request Each thread now gets it's own socket reserved on it's first operation. Those sockets are held until Connection.end_request is called by that thread, or Connection.disconnect is called by any thread, or the thread dies. Calling Connection.end_request allows the socket to be returned to the pool, and to be used by other threads instead of creating a new socket. Judicious use of this method is important for applications with many threads or with long running threads that make few calls to PyMongo operations.
115 lines
2.7 KiB
Python
115 lines
2.7 KiB
Python
# Copyright 2009 10gen, 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.
|
|
|
|
"""Test built in connection-pooling."""
|
|
|
|
import unittest
|
|
import threading
|
|
import os
|
|
import random
|
|
import sys
|
|
sys.path[0:0] = [""]
|
|
|
|
from test_connection import get_connection
|
|
|
|
N = 50
|
|
DB = "pymongo-pooling-tests"
|
|
|
|
class MongoThread(threading.Thread):
|
|
|
|
def __init__(self, test_case):
|
|
threading.Thread.__init__(self)
|
|
self.connection = test_case.c
|
|
self.db = self.connection[DB]
|
|
self.ut = test_case
|
|
|
|
|
|
class SaveAndFind(MongoThread):
|
|
|
|
def run(self):
|
|
for _ in xrange(N):
|
|
rand = random.randint(0, N)
|
|
id = self.db.sf.save({"x": rand})
|
|
self.ut.assertEqual(rand, self.db.sf.find_one(id)["x"])
|
|
self.connection.end_request()
|
|
|
|
|
|
class Unique(MongoThread):
|
|
|
|
def run(self):
|
|
for _ in xrange(N):
|
|
self.db.unique.insert({})
|
|
self.ut.assertEqual(None, self.db.error())
|
|
self.connection.end_request()
|
|
|
|
|
|
class NonUnique(MongoThread):
|
|
|
|
def run(self):
|
|
for _ in xrange(N):
|
|
self.db.unique.insert({"_id": "mike"})
|
|
self.ut.assertNotEqual(None, self.db.error())
|
|
self.connection.end_request()
|
|
|
|
|
|
class Disconnect(MongoThread):
|
|
|
|
def run(self):
|
|
for _ in xrange(N):
|
|
self.connection.disconnect()
|
|
|
|
|
|
class NoRequest(MongoThread):
|
|
|
|
def run(self):
|
|
errors = 0
|
|
for _ in xrange(N):
|
|
self.db.unique.insert({"_id": "mike"})
|
|
if self.db.error() is None:
|
|
errors += 1
|
|
|
|
self.ut.assertEqual(0, errors)
|
|
|
|
|
|
def run_cases(ut, cases):
|
|
threads = []
|
|
for case in cases:
|
|
for i in range(10):
|
|
thread = case(ut)
|
|
thread.start()
|
|
threads.append(thread)
|
|
|
|
for t in threads:
|
|
t.join()
|
|
|
|
class TestPooling(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.c = get_connection()
|
|
|
|
# reset the db
|
|
self.c.drop_database(DB)
|
|
self.c[DB].unique.insert({"_id": "mike"})
|
|
self.c[DB].unique.find_one()
|
|
|
|
def test_no_disconnect(self):
|
|
run_cases(self, [NoRequest, NonUnique, Unique, SaveAndFind])
|
|
|
|
def test_disconnect(self):
|
|
run_cases(self, [SaveAndFind, Disconnect, Unique])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|