lib/yogi_berra/catcher.rb in yogi_berra-0.0.6 vs lib/yogi_berra/catcher.rb in yogi_berra-0.0.8

@@ -28,14 +28,18 @@
           end
           @@settings
         end
       end
 
-      def db_client(host, port)
+      def db_client(host, port, replica_set = nil)
         # :w => 0 set the default write concern to 0, this allows writes to be non-blocking
         # by not waiting for a response from mongodb
-        @@mongo_client = Mongo::MongoClient.new(host, port, :w => 0)
+        if replica_set
+          @@mongo_client = Mongo::MongoReplicaSetClient.new(replica_set, :w => 0)
+        else
+          @@mongo_client = Mongo::MongoClient.new(host, port, :w => 0)
+        end
       rescue
         YogiBerra::Logger.log("Couldn't connect to the mongo database on host: #{host} port: #{port}.", :error)
         nil
       end
 
@@ -43,10 +47,11 @@
         load_db_settings if load_settings
 
         if @@settings
           host = @@settings["host"]
           port = @@settings["port"]
-          client = db_client(host, port)
+          replica_set = @@settings["replica_set"]
+          client = db_client(host, port, replica_set)
           if client
             @@connection = client[@@settings["database"]]
           else
             YogiBerra::Logger.log("Couldn't connect to the mongo database on host: #{host} port: #{port}.", :error)
           end
\ No newline at end of file