diff --git a/src/main/java/tech/ailef/dbadmin/external/dbmapping/fields/DbField.java b/src/main/java/tech/ailef/dbadmin/external/dbmapping/fields/DbField.java index 6189304..d503155 100644 --- a/src/main/java/tech/ailef/dbadmin/external/dbmapping/fields/DbField.java +++ b/src/main/java/tech/ailef/dbadmin/external/dbmapping/fields/DbField.java @@ -245,10 +245,17 @@ public class DbField { } public Set getAllValues() { - List findAll = schema.getJpaRepository().findAll(); - return findAll.stream() - .map(o -> new DbObject(o, schema).get(this)) - .collect(Collectors.toSet()); + if (type instanceof EnumFieldType) { + List values = type.getValues(); + return values.stream().map(v -> { + return new DbFieldValue(v, this); + }).collect(Collectors.toSet()); + } else { + List findAll = schema.getJpaRepository().findAll(); + return findAll.stream() + .map(o -> new DbObject(o, schema).get(this)) + .collect(Collectors.toSet()); + } } @Override