From 6da9686a50c52ed579e7a1243f7ef0aa1b1ae125 Mon Sep 17 00:00:00 2001 From: ruslanpa Date: Mon, 9 Feb 2015 22:19:38 +0200 Subject: [PATCH] [refactor] Update StarType enum in memento pattern. --- .../src/main/java/com/iluwatar/StarType.java | 33 +++++-------------- 1 file changed, 9 insertions(+), 24 deletions(-) diff --git a/memento/src/main/java/com/iluwatar/StarType.java b/memento/src/main/java/com/iluwatar/StarType.java index 85f599b1c..5914e8528 100644 --- a/memento/src/main/java/com/iluwatar/StarType.java +++ b/memento/src/main/java/com/iluwatar/StarType.java @@ -2,31 +2,16 @@ package com.iluwatar; public enum StarType { - SUN, RED_GIANT, WHITE_DWARF, SUPERNOVA, DEAD; + SUN("sun"), RED_GIANT("red giant"), WHITE_DWARF("white dwarf"), SUPERNOVA("supernova"), DEAD("dead star"), UNDEFINED(""); - @Override + private String title; + + StarType(String title) { + this.title = title; + } + + @Override public String toString() { - String s = ""; - switch (this) { - case RED_GIANT: - s = "red giant"; - break; - case SUN: - s = "sun"; - break; - case SUPERNOVA: - s = "supernova"; - break; - case WHITE_DWARF: - s = "white dwarf"; - break; - case DEAD: - s = "dead star"; - break; - default: - break; - } - return s; + return title; } - }