AGRFを1回のみ使う

2024-07-19 14:57:01 +09:00 · 2024-07-19 14:57:01 +09:00 · 3e8b029d8e
commit 3e8b029d8e
parent d38889ea48
1 changed files with 5 additions and 6 deletions
--- a/wc.rb
+++ b/wc.rb
@ -24,12 +24,11 @@ def count_file_stats(input)
 end

 def process_input(source)
-  input = if source == '-'
-            $stdin.read
+  if source == '-'
+    count_file_stats(ARGF)
  else
-            File.read(source)
+    count_file_stats(File.read(source))
  end
-  count_file_stats(input)
 end

 def update_totals(totals, stats)