diff --git a/factory-method/src/main/java/com/iluwatar/WeaponType.java b/factory-method/src/main/java/com/iluwatar/WeaponType.java index 4097fd3c7..8d5ab54cc 100644 --- a/factory-method/src/main/java/com/iluwatar/WeaponType.java +++ b/factory-method/src/main/java/com/iluwatar/WeaponType.java @@ -2,23 +2,16 @@ package com.iluwatar; 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; } - }