[refactor] Update WeaponType enum in factory-method pattern.

RepmujNetsik · Feb 9, 2015 · eb9b5fd · eb9b5fd
1 parent bd4cecd
commit eb9b5fd
Showing 1 changed file with 9 additions and 16 deletions.
diff --git a/factory-method/src/main/java/com/iluwatar/WeaponType.java b/factory-method/src/main/java/com/iluwatar/WeaponType.java
@@ -2,23 +2,16 @@
 
 public enum WeaponType {
 
-	SHORT_SWORD, SPEAR, AXE;
+	SHORT_SWORD("short sword"), SPEAR("spear"), AXE("axe"), UNDEFINED("");
 
-	@Override
+    private String title;
+
+    WeaponType(String title) {
+        this.title = title;
+    }
+
+    @Override
 	public String toString() {
-		String s = "";
-		switch (this) {
-		case SHORT_SWORD:
-			s = "short sword";
-			break;
-		case SPEAR:
-			s = "spear";
-			break;
-		case AXE:
-			s = "axe";
-			break;
-		}
-		return s;
+		return title;
 	}
-
 }