lib/mongorilla/collection.rb in mongorilla-0.5.0 vs lib/mongorilla/collection.rb in mongorilla-0.6.0

- old
+ new

@@ -4,11 +4,11 @@ class Collection @@master = nil @@slaves = nil @@slave_index = 0 @@config = nil - @@loger = nil + @@logger = nil def self.master @@master end @@ -22,15 +22,19 @@ def self.load_config(path) @@config = YAML.load(File.read(path)) end + def self.output_log(method,contents) + @@logger.send(method,contents) if @@logger + end + def self.build(path=File.expand_path("../config.yml",__FILE__),logger=nil) load_config(path) @@config["max_retries"] ||= 10 @@config["meantime"] ||= 0.5 - @@loger = logger + @@logger = logger if @@config["hosts"] @@master = Mongo::ReplSetConnection.new(*@@config["hosts"]).db(@@config["database"]) elsif @@config["slaves"] @@master = Mongo::Connection.new(@@config["host"],@@config["port"]).db(@@config["database"]) @@slaves = @@config["slaves"].map{|s| Mongo::Connection.new(s["host"],s["port"]).db(@@config["database"])} @@ -67,11 +71,11 @@ ret = find(cond,opt) ret.first end def count(cond={},opt={}) - @@loger.info("count #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("count #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@logger find(cond,opt).count end def find(cond={},opt={}) if opt[:master] || opt["master"] @@ -79,39 +83,39 @@ opt.delete("master") if @@config["hosts"] opt[:read] = :primary end rescue_connection_failure do - @@loger.info("find(master) #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("find(master) #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@logger w_col.find(cond,opt) end else if @@config["hosts"] && @@config["read_secondary"] opt[:read] = :secondary end begin rescue_connection_failure do - @@loger.info("find(secondary) #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("find(secondary) #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@logger r_col.find(cond,opt) end rescue - @@loger.info("find(master) #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("find(master) #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@logger w_col.find(cond,opt) end end end def insert(data,opt={}) rescue_connection_failure do - @@loger.info("insert #{@name} data:#{data.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("insert #{@name} data:#{data.inspect} opt:#{opt.inspect}") if @@logger w_col.insert(data,opt) end end def update(cond,data,opt) rescue_connection_failure do - @@loger.info("update #{@name} cond:#{cond.inspect} data:#{data.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("update #{@name} cond:#{cond.inspect} data:#{data.inspect} opt:#{opt.inspect}") if @@logger w_col.update(cond,data,opt) end end def remove(cond={},opt={}) @@ -119,10 +123,10 @@ cond = {:_id => BSON::ObjectId(cond)} elsif cond.is_a? BSON::ObjectId cond = {:_id => cond} end rescue_connection_failure do - @@loger.info("remove #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@loger + @@logger.info("remove #{@name} cond:#{cond.inspect} opt:#{opt.inspect}") if @@logger w_col.remove(cond,opt) end end def rescue_connection_failure(max_retries=@@config["max_retries"])