diff --git a/src/main/java/ru/touchin/templates/logansquare/LoganSquareEnumConverter.java b/src/main/java/ru/touchin/templates/logansquare/LoganSquareEnumConverter.java index b10f32a..f94921a 100644 --- a/src/main/java/ru/touchin/templates/logansquare/LoganSquareEnumConverter.java +++ b/src/main/java/ru/touchin/templates/logansquare/LoganSquareEnumConverter.java @@ -35,10 +35,17 @@ public class LoganSquareEnumConverter extends @NonNull private final T[] enumValues; + @Nullable + private final T defaultValue; public LoganSquareEnumConverter(@NonNull final T[] enumValues) { + this(enumValues, null); + } + + public LoganSquareEnumConverter(@NonNull final T[] enumValues, @Nullable final T defaultValue) { super(); this.enumValues = enumValues; + this.defaultValue = defaultValue; } @Nullable @@ -52,6 +59,9 @@ public class LoganSquareEnumConverter extends return value; } } + if (defaultValue != null) { + return defaultValue; + } throw new ShouldNotHappenException(); }