ryanwood · November 28, 2018 15:31 · Nov 28, 2018 · Nov 28, 2018
diff --git a/multi_logger.rb → multi_io.rb b/multi_logger.rb → multi_io.rb
@@ -1,5 +1,5 @@
 # So we can log to STDOUT and a file at once. ;)
-class MultiLogger
+class MultiIO
   def initialize(*targets)
      @targets = targets
   end

diff --git a/script.rb b/script.rb
@@ -1,13 +1,11 @@
-require 'multi_logger'
+require 'multi_io'
 
 def logger
-  target = if DEBUG
-    STDOUT
-  else
+  @logger ||= begin
     log_file = File.open("my_log_file.log", "a")
-    MultiIO.new(STDOUT, log_file)
+    target = MultiIO.new(STDOUT, log_file)
+    Logger.new(target)
   end
-  @logger ||= Logger.new(target)
 end
 
 # Use

diff --git a/multi_logger.rb b/multi_logger.rb
@@ -0,0 +1,14 @@
+# So we can log to STDOUT and a file at once. ;)
+class MultiLogger
+  def initialize(*targets)
+     @targets = targets
+  end
+
+  def write(*args)
+    @targets.each {|t| t.write(*args)}
+  end
+
+  def close
+    @targets.each(&:close)
+  end
+end
diff --git a/script.rb b/script.rb
@@ -0,0 +1,16 @@
+require 'multi_logger'
+
+def logger
+  target = if DEBUG
+    STDOUT
+  else
+    log_file = File.open("my_log_file.log", "a")
+    MultiIO.new(STDOUT, log_file)
+  end
+  @logger ||= Logger.new(target)
+end
+
+# Use
+
+logger.debug("Would have run: #{sql}")
+logger.info('Checkpoint!')