From 40338c4269fc3535e13440041f97af57d08265b2 Mon Sep 17 00:00:00 2001
From: "Erik C. Thauvin" <erik@thauvin.net>
Date: Mon, 6 May 2019 15:07:26 -0700
Subject: [PATCH] Improved main method.

---
 .../net/thauvin/erik/httpstatus/Reasons.java  | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/main/java/net/thauvin/erik/httpstatus/Reasons.java b/src/main/java/net/thauvin/erik/httpstatus/Reasons.java
index 7adb2c1..77c6149 100644
--- a/src/main/java/net/thauvin/erik/httpstatus/Reasons.java
+++ b/src/main/java/net/thauvin/erik/httpstatus/Reasons.java
@@ -32,8 +32,12 @@
 
 package net.thauvin.erik.httpstatus;
 
+import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
+
 import java.util.Map;
 import java.util.ResourceBundle;
+import java.util.SortedSet;
+import java.util.TreeSet;
 import java.util.concurrent.ConcurrentHashMap;
 
 /**
@@ -96,11 +100,21 @@ public final class Reasons {
      * @param args The command line arguments.
      */
     @SuppressWarnings("PMD.SystemPrintln")
+    @SuppressFBWarnings("MUI_CONTAINSKEY_BEFORE_GET")
     public static void main(final String... args) {
-        for (final Map.Entry<String, String> entry : REASON_PHRASES.entrySet()) {
-            System.out.println(entry.getKey() + ": " + entry.getValue());
-        }
+        if (args.length >= 1) {
+            for (final String key : args) {
+                if (REASON_PHRASES.containsKey(key)) {
+                    System.out.println(key + ": " + REASON_PHRASES.get(key));
+                }
+            }
+        } else {
+            final SortedSet<String> keys = new TreeSet<>(REASON_PHRASES.keySet());
+            for (final String key : keys) {
+                System.out.println(key + ": " + REASON_PHRASES.get(key));
+            }
 
-        System.out.println("Total: " + REASON_PHRASES.size());
+            System.out.println("Total: " + REASON_PHRASES.size());
+        }
     }
 }
\ No newline at end of file