Gracefully handle a PyMongo AutoReconnect

graceful_auto_reconnect.py

import functools
import pymongo
import logging
import time

MAX_AUTO_RECONNECT_ATTEMPTS = 5

def graceful_auto_reconnect(mongo_op_func):
  """Gracefully handle a reconnection event."""
  @functools.wraps(mongo_op_func)
  def wrapper(*args, **kwargs):
    for attempt in xrange(MAX_AUTO_RECONNECT_ATTEMPTS):
      try:
        return mongo_op_func(*args, **kwargs)
      except pymongo.errors.AutoReconnect as e:
        wait_t = 0.5 * pow(2, attempt) # exponential back off
        logging.warning("PyMongo auto-reconnecting... %s. Waiting %.1f seconds.", str(e), wait_t)
        time.sleep(wait_t)

  return wrapper

Interesting. I'm building something similar by subclassing pymongo.connection.Connection. However, I'm having trouble with it looping forever between 0 and 1 tries. I tried this https://gist.github.com/1057236 with the same result.

What exactly is mongo_op_func here, if you don't mind?

Might want to look at this https://github.com/arngarden/MongoDBProxy. Its pretty cool and gets incorporated into existing code with minimal changes.