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

Completed insert statement

Browse Source
This commit is contained in:
Aleksei 2020-02-19 22:25:06 -05:00
parent ec5960321e
commit 504630e191
2 changed files with 29 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/dev/salmonllama/fsbot/database/FSDB.java
View File

@ -7,7 +7,7 @@
package dev.salmonllama.fsbot.database;
import dev.salmonllama.fsbot.config.BotConfig;
import dev.salmonllama.fsbot.database.models.OutfitModel;
import dev.salmonllama.fsbot.database.models.Outfit;
import java.sql.SQLException;
import java.util.HashMap;
@ -38,7 +38,7 @@ public class FSDB {
private static void prepareTables() {
try {
get().query(OutfitModel.schema());
get().query(Outfit.schema());
} catch (SQLException e) {
e.printStackTrace();
}

40
src/main/java/dev/salmonllama/fsbot/database/controllers/OutfitController.java
View File

@ -6,22 +6,36 @@
package dev.salmonllama.fsbot.database.controllers;
import dev.salmonllama.fsbot.database.FSDB;
import dev.salmonllama.fsbot.database.models.OutfitModel;
import dev.salmonllama.fsbot.database.models.Outfit;
import java.sql.ResultSet;
import java.sql.SQLException;
public class OutfitController {
public static void insert(OutfitModel outfit) {
public static void insert(Outfit outfit) {
try {
FSDB.get().insert("INSERT INTO outfits('id', 'link') VALUES (?, ?)", outfit.id, outfit.link);
FSDB.get().insert(
"INSERT INTO " +
"outfits('id', 'link', 'submitter', 'tag', 'created', 'updated', 'deleted', 'featured', 'display_count', 'deletion_hash') " +
"VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
outfit.id,
outfit.link,
outfit.submitter,
outfit.tag,
outfit.created,
outfit.updated,
outfit.deleted,
outfit.featured,
outfit.displayCount,
outfit.deletionHash
);
} catch (SQLException e) {
e.printStackTrace();
}
}
public static OutfitModel findById(String id) throws SQLException {
OutfitModel outfit = new OutfitModel();
public static Outfit findById(String id) throws SQLException {
Outfit outfit = new Outfit();
try (ResultSet rs = FSDB.get().select("SELECT * FROM outfits WHERE id = ?", id)) {
if (rs.next()) {
outfit = mapObject(rs);
@ -33,8 +47,8 @@ public class OutfitController {
return outfit;
}
public static OutfitModel findRandom() throws SQLException {
OutfitModel outfit = new OutfitModel();
public static Outfit findRandom() throws SQLException {
Outfit outfit = new Outfit();
try (ResultSet rs = FSDB.get().select("SELECT * FROM outfits WHERE deleted = 0 ORDERBY random() LIMIT 1")) {
if (rs.next()) {
outfit = mapObject(rs);
@ -46,8 +60,8 @@ public class OutfitController {
return outfit;
}
public static OutfitModel findRandomByTag(String tag) {
OutfitModel outfit = new OutfitModel();
public static Outfit findRandomByTag(String tag) {
Outfit outfit = new Outfit();
try (ResultSet rs = FSDB.get().select("SELECT * FROM outfits WHERE tag = ? AND deleted = 0 ORDERBY random() LIMIT 1", tag)) {
if (rs.next()) {
outfit = mapObject(rs);
@ -59,8 +73,8 @@ public class OutfitController {
return outfit;
}
public static OutfitModel findRandomBySubmitter(String submitterId) {
OutfitModel outfit = new OutfitModel();
public static Outfit findRandomBySubmitter(String submitterId) {
Outfit outfit = new Outfit();
try (ResultSet rs = FSDB.get().select("SELECT * FROM outfits WHERE submitter = ? AND deleted = 0 ORDERBY random() LIMIT 1", submitterId)) {
if (rs.next()) {
outfit = mapObject(rs);
@ -98,8 +112,8 @@ public class OutfitController {
return count;
}
private static OutfitModel mapObject(ResultSet rs) throws SQLException {
OutfitModel outfit = new OutfitModel();
private static Outfit mapObject(ResultSet rs) throws SQLException {
Outfit outfit = new Outfit();
outfit.id = rs.getString("id");
outfit.link = rs.getString("link");
outfit.tag = rs.getString("tag");