Skip to content

Commit c5f4a3f

Browse files
replaced ordinal in UserPreference with actual Preference
1 parent 1573e1a commit c5f4a3f

File tree

3 files changed

+6
-10
lines changed

3 files changed

+6
-10
lines changed

src/main/java/net/javadiscord/javabot/systems/user_preferences/UserPreferenceManager.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ public boolean setOrCreate(long userId, Preference preference, boolean enabled)
3636
} else {
3737
UserPreference userPreference = new UserPreference();
3838
userPreference.setUserId(userId);
39-
userPreference.setOrdinal(preference.ordinal());
39+
userPreference.setPreference(preference);
4040
userPreference.setEnabled(enabled);
4141
repo.insert(userPreference, false);
4242
return true;
@@ -63,7 +63,7 @@ public UserPreference getOrCreate(long userId, Preference preference) {
6363
} else {
6464
UserPreference userPreference = new UserPreference();
6565
userPreference.setUserId(userId);
66-
userPreference.setOrdinal(preference.ordinal());
66+
userPreference.setPreference(preference);
6767
userPreference.setEnabled(true);
6868
repo.insert(userPreference, false);
6969
return userPreference;

src/main/java/net/javadiscord/javabot/systems/user_preferences/dao/UserPreferenceRepository.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@ public class UserPreferenceRepository extends DatabaseRepository<UserPreference>
2424
*/
2525
public UserPreferenceRepository(Connection con) {
2626
super(con, UserPreference.class, "USER_PREFERENCES", List.of(
27-
TableProperty.of("user_id", H2Type.BIGINT, (x, y) -> x.setUserId((Long) y), UserPreference::getUserId),
28-
TableProperty.of("ordinal", H2Type.INTEGER, (x, y) -> x.setOrdinal((Integer) y), UserPreference::getOrdinal),
29-
TableProperty.of("enabled", H2Type.BOOLEAN, (x, y) -> x.setEnabled((Boolean) y), UserPreference::isEnabled)
27+
TableProperty.of("user_id", H2Type.BIGINT, (x, y) -> x.setUserId((Long) y), UserPreference::getUserId),
28+
TableProperty.of("ordinal", H2Type.INTEGER, (x, y) -> x.setPreference(Preference.values()[(Integer) y]), p -> p.getPreference().ordinal()),
29+
TableProperty.of("enabled", H2Type.BOOLEAN, (x, y) -> x.setEnabled((Boolean) y), UserPreference::isEnabled)
3030
));
3131
}
3232

src/main/java/net/javadiscord/javabot/systems/user_preferences/model/UserPreference.java

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,6 @@
88
@Data
99
public class UserPreference {
1010
private long userId;
11-
private int ordinal;
11+
private Preference preference;
1212
private boolean enabled;
13-
14-
public Preference getPreference() {
15-
return Preference.values()[ordinal];
16-
}
1713
}

0 commit comments

Comments
 (0)