clear specific logs when using rake log:clear

1 files changed, 18 insertions, 4 deletions

diff --git a/railties/lib/rails/tasks/log.rake b/railties/lib/rails/tasks/log.rake
index 6e1334692e..6c3f02eb0c 100644
--- a/railties/lib/rails/tasks/log.rake
+++ b/railties/lib/rails/tasks/log.rake
@@ -1,9 +1,23 @@
 namespace :log do
-  desc "Truncates all *.log files in log/ to zero bytes"
+  desc "Truncates all *.log files in log/ to zero bytes (specify which logs with LOGS=test,development)"
   task :clear do
-    FileList["log/*.log"].each do |log_file|
-      f = File.open(log_file, "w")
-      f.close
+    log_files.each do |file|
+      clear_log_file(file)
     end
   end
+
+  def log_files
+    if ENV['LOGS']
+      ENV['LOGS'].split(',')
+                  .map    { |file| "log/#{file.strip}.log" }
+                  .select { |file| File.exists?(file) }
+    else
+      FileList["log/*.log"]
+    end
+  end
+
+  def clear_log_file(file)
+    f = File.open(file, "w")
+    f.close
+  end
 end