mirrors/untisbot-discord
1
0
Fork 0
mirror of https://github.com/bytedream/untisbot-discord.git synced 2025-06-06 07:44:07 +02:00
Code Issues Packages Projects Releases Wiki Activity

Created anti sql timeouter

Browse Source
This commit is contained in:
ByteDream 2020-12-05 00:19:51 +01:00
parent 5d4e5e634c
commit 74274fa2d7
1 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/org/bytedream/untisbot/discord/DiscordCommandListener.java
View File

@ -39,6 +39,7 @@ import java.time.LocalDateTime;
import java.time.LocalTime; import java.time.LocalTime;
import java.time.format.DateTimeFormatter; import java.time.format.DateTimeFormatter;
import java.util.*; import java.util.*;
import java.util.concurrent.TimeUnit;
/** /**
* Adapter to handle all events * Adapter to handle all events
@ -73,8 +74,16 @@ public class DiscordCommandListener extends ListenerAdapter {
statsDataConnector = dataConnector.statsConnector(); statsDataConnector = dataConnector.statsConnector();
this.languages = languages; this.languages = languages;
EmbedBuilder embedBuilder = new EmbedBuilder(); new Timer().scheduleAtFixedRate(new TimerTask() { // just execute this so that the connect won't have a timeout
embedBuilder.setColor(Color.GREEN); @Override
public void run() {
Thread.currentThread().setName("Anti sql timeout");
try {
Main.getConnection().createStatement().execute("SELECT * FROM Guilds WHERE GUILDID = 0");
} catch (SQLException ignore) {
}
}
}, 0, TimeUnit.HOURS.toMillis(1));
} }
/** /**
@ -161,9 +170,12 @@ public class DiscordCommandListener extends ListenerAdapter {
language = languages.getJSONObject(data.getLanguage()); language = languages.getJSONObject(data.getLanguage());
} }
String className = "-"; String className = "-";
try { try {
className = session.getKlassen().findById(data.getKlasseId()).getName(); session.reconnect();
className = session.getKlassen().findById(classId).getName();
} catch (IOException ignore) {} } catch (IOException ignore) {}
String finalClassName = className; // yea java... String finalClassName = className; // yea java...
LocalDate finalDate = date; // yea java part two... LocalDate finalDate = date; // yea java part two...
embedBuilder.setTitle(Utils.advancedFormat(language.getString("timetable-title"), new HashMap<String, Object>() {{ embedBuilder.setTitle(Utils.advancedFormat(language.getString("timetable-title"), new HashMap<String, Object>() {{
@ -279,7 +291,7 @@ public class DiscordCommandListener extends ListenerAdapter {
if (data.getUsername() != null && data.getServer() != null && data.getSchool() != null) { if (data.getUsername() != null && data.getServer() != null && data.getSchool() != null) {
dataSet = "✅ Set"; dataSet = "✅ Set";
if (!data.isCheckActive()) { if (!data.isCheckActive()) {
embedBuilder.setFooter("The timetable checker is deactivated. Type `" + data.getPrefix() + "start` - use `" + data.getPrefix() + "help start` for more details"); embedBuilder.setFooter("The timetable checker is deactivated. Type `" + data.getPrefix() + "start` to re-enable it - use `" + data.getPrefix() + "help start` for more details");
} }
} }
else { else {
@ -430,7 +442,7 @@ public class DiscordCommandListener extends ListenerAdapter {
channel.sendMessage("Updated prefix to `" + prefix + "`" + note).queue(); channel.sendMessage("Updated prefix to `" + prefix + "`" + note).queue();
} }
} else { } else {
channel.sendMessage("Wrong number of arguments were given (expected 3, got " + args.length + "), type `" + data.getPrefix() + "help prefix` for help").queue(); channel.sendMessage("Wrong number of arguments were given (expected 1, got " + args.length + "), type `" + data.getPrefix() + "help prefix` for help").queue();
} }
break; break;
case "start": // `start` command case "start": // `start` command
@ -440,7 +452,6 @@ public class DiscordCommandListener extends ListenerAdapter {
} else { } else {
guildDataConnector.update(guildId, null, null, null, null, null, null, null, null, null, true, null); guildDataConnector.update(guildId, null, null, null, null, null, null, null, null, null, true, null);
runTimetableChecker(guild); runTimetableChecker(guild);
logger.info(guildName + " started timetable listening");
channel.sendMessage("✅ Timetable listening has been started").queue(); channel.sendMessage("✅ Timetable listening has been started").queue();
} }
} else { } else {
@ -550,6 +561,7 @@ public class DiscordCommandListener extends ListenerAdapter {
for (; i <= daysToCheck; i++) { for (; i <= daysToCheck; i++) {
LocalDate localDate = now.plusDays(i); LocalDate localDate = now.plusDays(i);
try { try {
LocalDate lastChecked = guildDataConnector.get(guildId).getLastChecked();
CheckCallback checkCallback = timetableChecker.check(localDate); CheckCallback checkCallback = timetableChecker.check(localDate);
ArrayList<Timetable.Lesson> cancelledLessons = checkCallback.getCancelled(); ArrayList<Timetable.Lesson> cancelledLessons = checkCallback.getCancelled();
@ -636,14 +648,11 @@ public class DiscordCommandListener extends ListenerAdapter {
textChannel.sendMessage(embedBuilder.build()).queue(); textChannel.sendMessage(embedBuilder.build()).queue();
} }
LocalDate lastChecked = guildDataConnector.get(guildId).getLastChecked();
Short totalDays = stats.getTotalDays(); Short totalDays = stats.getTotalDays();
int totalLessons = stats.getTotalLessons(); int totalLessons = stats.getTotalLessons();
if (lastChecked == null || lastChecked.isBefore(now.plusDays(i))) { if (lastChecked == null || lastChecked.isBefore(now.plusDays(i))) {
totalDays++; totalDays++;
totalLessons += checkCallback.getAllLessons().size(); totalLessons += checkCallback.getAllLessons().size();
guildDataConnector.update(guildId, null, null, null, null, null, null, null, null, null, null, now.plusDays(i));
} }
short totalCancelledLessons = (short) (stats.getTotalCancelledLessons() + cancelledLessons.size() - notCancelledLessons.size()); short totalCancelledLessons = (short) (stats.getTotalCancelledLessons() + cancelledLessons.size() - notCancelledLessons.size());
short totalMovedLessons = (short) (stats.getTotalMovedLessons() + movedLessons.size() - notMovedLessons.size()); short totalMovedLessons = (short) (stats.getTotalMovedLessons() + movedLessons.size() - notMovedLessons.size());
@ -671,6 +680,8 @@ public class DiscordCommandListener extends ListenerAdapter {
if (error) { if (error) {
error = false; error = false;
} }
} else if (lastChecked == null || lastChecked.isBefore(now.plusDays(i))) {
guildDataConnector.update(guildId, null, null, null, null, null, null, null, null, null, null, now.plusDays(i));
} }
} catch (Exception e) { } catch (Exception e) {
logger.warn(guildName + " ran into an exception while trying to check the timetable for the " + localDate.format(DateTimeFormatter.ofPattern("dd.MM.yyyy")), e); logger.warn(guildName + " ran into an exception while trying to check the timetable for the " + localDate.format(DateTimeFormatter.ofPattern("dd.MM.yyyy")), e);
@ -704,12 +715,6 @@ public class DiscordCommandListener extends ListenerAdapter {
if (latestImportTime < sessionLatestImportTime) { if (latestImportTime < sessionLatestImportTime) {
latestImportTime = sessionLatestImportTime; latestImportTime = sessionLatestImportTime;
main(); main();
} else {
try {
Main.getConnection().createStatement().execute("SELECT * FROM Guilds WHERE GUILDID = 0");
// just execute this so that the connect won't have a timeout
} catch (SQLException ignore) {
}
} }
} catch (IOException e) { } catch (IOException e) {
logger.info("Running main through IOException", e); logger.info("Running main through IOException", e);