Fashionscape/discord-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed accessors and added getAll method

Browse Source
This commit is contained in:
Salmonllama 2020-08-19 15:14:25 -04:00
parent 32e74b77c1
commit 43b1570b9d
1 changed files with 29 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/main/java/dev/salmonllama/fsbot/database/controllers/ColorRoleController.java
View File

@ -5,13 +5,15 @@ import dev.salmonllama.fsbot.database.models.ColorRole;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Optional; import java.util.Optional;
import java.util.concurrent.CompletableFuture; import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException; import java.util.concurrent.CompletionException;
public class ColorRoleController { public class ColorRoleController {
// Need insert, get by color, exists by color, exists by role, get by server, count, and delete // Need insert, get by color, exists by color, exists by role, get by server, count, and delete
public CompletableFuture<Void> insert(ColorRole cr) { public static CompletableFuture<Void> insert(ColorRole cr) {
return CompletableFuture.runAsync(() -> { return CompletableFuture.runAsync(() -> {
try { try {
insertExec(cr); insertExec(cr);
@ -21,7 +23,7 @@ public class ColorRoleController {
}); });
} }
public CompletableFuture<Boolean> exists(String color) { public static CompletableFuture<Boolean> exists(String color) {
return CompletableFuture.supplyAsync(() -> { return CompletableFuture.supplyAsync(() -> {
try { try {
return existsExec(color); return existsExec(color);
@ -31,7 +33,7 @@ public class ColorRoleController {
}); });
} }
public CompletableFuture<Boolean> exists(long roleId) { public static CompletableFuture<Boolean> exists(long roleId) {
return CompletableFuture.supplyAsync(() -> { return CompletableFuture.supplyAsync(() -> {
try { try {
return existsExec(roleId); return existsExec(roleId);
@ -41,7 +43,7 @@ public class ColorRoleController {
}); });
} }
public CompletableFuture<Optional<ColorRole>> get(String color) { public static CompletableFuture<Optional<ColorRole>> get(String color) {
return CompletableFuture.supplyAsync(() -> { return CompletableFuture.supplyAsync(() -> {
try { try {
return getExec(color); return getExec(color);
@ -51,7 +53,7 @@ public class ColorRoleController {
}); });
} }
public CompletableFuture<Optional<ColorRole>> get(long roleId) { public static CompletableFuture<Optional<ColorRole>> get(long roleId) {
return CompletableFuture.supplyAsync(() -> { return CompletableFuture.supplyAsync(() -> {
try { try {
return getExec(roleId); return getExec(roleId);
@ -61,17 +63,17 @@ public class ColorRoleController {
}); });
} }
public CompletableFuture<Integer> count(long serverId) { public static CompletableFuture<Optional<Collection<ColorRole>>> getAll() {
return CompletableFuture.supplyAsync(() -> { return CompletableFuture.supplyAsync(() -> {
try { try {
return countExec(serverId); return getAllExec();
} catch (SQLException e) { } catch (SQLException e) {
throw new CompletionException(e); throw new CompletionException(e);
} }
}); });
} }
public CompletableFuture<Void> delete(long roleId) { public static CompletableFuture<Void> delete(long roleId) {
return CompletableFuture.runAsync(() -> { return CompletableFuture.runAsync(() -> {
try { try {
deleteExec(roleId); deleteExec(roleId);
@ -81,7 +83,7 @@ public class ColorRoleController {
}); });
} }
public CompletableFuture<Void> delete(String color) { public static CompletableFuture<Void> delete(String color) {
return CompletableFuture.runAsync(() -> { return CompletableFuture.runAsync(() -> {
try { try {
deleteExec(color); deleteExec(color);
@ -91,7 +93,7 @@ public class ColorRoleController {
}); });
} }
private void insertExec(ColorRole cr) throws SQLException { private static void insertExec(ColorRole cr) throws SQLException {
FSDB.get().insert("INSERT INTO color_roles (role_id, server_id, color) VALUES (?, ?, ?)", FSDB.get().insert("INSERT INTO color_roles (role_id, server_id, color) VALUES (?, ?, ?)",
cr.getRoleId(), cr.getRoleId(),
cr.getServerId(), cr.getServerId(),
@ -99,7 +101,7 @@ public class ColorRoleController {
); );
} }
private boolean existsExec(String color) throws SQLException { private static boolean existsExec(String color) throws SQLException {
ResultSet rs = FSDB.get().select("SELECT EXISTS(SELECT 1 FROM color_roles WHERE color = ?) AS hmm", color); ResultSet rs = FSDB.get().select("SELECT EXISTS(SELECT 1 FROM color_roles WHERE color = ?) AS hmm", color);
boolean exists = rs.getBoolean("hmm"); boolean exists = rs.getBoolean("hmm");
@ -107,7 +109,7 @@ public class ColorRoleController {
return exists; return exists;
} }
private boolean existsExec(long roleId) throws SQLException { private static boolean existsExec(long roleId) throws SQLException {
ResultSet rs = FSDB.get().select("SELECT EXISTS(SELECT 1 FROM color_roles WHERE role_id = ?) AS hmm", roleId); ResultSet rs = FSDB.get().select("SELECT EXISTS(SELECT 1 FROM color_roles WHERE role_id = ?) AS hmm", roleId);
boolean exists = rs.getBoolean("hmm"); boolean exists = rs.getBoolean("hmm");
@ -115,7 +117,7 @@ public class ColorRoleController {
return exists; return exists;
} }
private Optional<ColorRole> getExec(String color) throws SQLException { private static Optional<ColorRole> getExec(String color) throws SQLException {
ResultSet rs = FSDB.get().select("SELECT * FROM color_roles WHERE color = ?", color); ResultSet rs = FSDB.get().select("SELECT * FROM color_roles WHERE color = ?", color);
if (rs.next()) { if (rs.next()) {
@ -128,7 +130,7 @@ public class ColorRoleController {
return Optional.empty(); return Optional.empty();
} }
private Optional<ColorRole> getExec(Long roleId) throws SQLException { private static Optional<ColorRole> getExec(Long roleId) throws SQLException {
ResultSet rs = FSDB.get().select("SELECT * FROM color_roles WHERE role_id = ?", roleId); ResultSet rs = FSDB.get().select("SELECT * FROM color_roles WHERE role_id = ?", roleId);
if (rs.next()) { if (rs.next()) {
@ -141,28 +143,31 @@ public class ColorRoleController {
return Optional.empty(); return Optional.empty();
} }
private int countExec(long serverId) throws SQLException { private static Optional<Collection<ColorRole>> getAllExec() throws SQLException {
ResultSet rs = FSDB.get().select("SELECT COUNT(*) AS count FROM color_roles WHERE server_id = ?", serverId); ResultSet rs = FSDB.get().select("SELECT * FROM color_roles");
if (rs.next()) { Collection<ColorRole> roles = new ArrayList<>();
int count = rs.getInt("count"); while (rs.next()) {
roles.add(mapObject(rs));
FSDB.get().close(rs); FSDB.get().close(rs);
return count;
} }
FSDB.get().close(rs); if (roles.isEmpty()) {
return 0; return Optional.empty();
} }
private void deleteExec(long roleId) throws SQLException { return Optional.of(roles);
}
private static void deleteExec(long roleId) throws SQLException {
FSDB.get().query("DELETE FROM color_roles WHERE role_id = ?", roleId); FSDB.get().query("DELETE FROM color_roles WHERE role_id = ?", roleId);
} }
private void deleteExec(String color) throws SQLException { private static void deleteExec(String color) throws SQLException {
FSDB.get().query("DELETE FROM color_roles WHERE color = ?", color); FSDB.get().query("DELETE FROM color_roles WHERE color = ?", color);
} }
private ColorRole mapObject(ResultSet rs) throws SQLException { private static ColorRole mapObject(ResultSet rs) throws SQLException {
return new ColorRole.ColorRoleBuilder(rs.getLong("role_id")) return new ColorRole.ColorRoleBuilder(rs.getLong("role_id"))
.setServerId(rs.getLong("server_id")) .setServerId(rs.getLong("server_id"))
.setColor(rs.getString("color")) .setColor(rs.getString("color"))