Upstream/SaneEconomy
1
0
Fork 0
mirror of https://github.com/AppleDash/SaneEconomy.git synced 2024-11-26 03:55:46 +01:00
Code Issues Projects Releases Wiki Activity

Retry MySQL queries if they fail.

Browse Source
This commit is contained in:
AppleDash 2017-01-20 00:48:05 -05:00
parent feb88b4519
commit 9d71bb5a21
1 changed files with 21 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
SaneEconomyCore/src/main/java/org/appledash/saneeconomy/utils/database/MySQLConnection.java
View File

@ -9,12 +9,14 @@ import java.sql.PreparedStatement;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.logging.Logger;
/** /**
* Created by appledash on 9/19/16. * Created by appledash on 9/19/16.
* Blackjack is best pony. * Blackjack is best pony.
*/ */
public class MySQLConnection { public class MySQLConnection {
private static final Logger LOGGER = Logger.getLogger("MySQLConnection");
private static final int MAX_OPEN_TRANSACTIONS = 5; private static final int MAX_OPEN_TRANSACTIONS = 5;
private final DatabaseCredentials dbCredentials; private final DatabaseCredentials dbCredentials;
private final AtomicInteger openTransactions = new AtomicInteger(0); private final AtomicInteger openTransactions = new AtomicInteger(0);
@ -60,18 +62,28 @@ public class MySQLConnection {
public void executeAsyncOperation(Consumer<Connection> callback) { public void executeAsyncOperation(Consumer<Connection> callback) {
Bukkit.getServer().getScheduler().scheduleAsyncDelayedTask(SaneEconomy.getInstance(), () -> { Bukkit.getServer().getScheduler().scheduleAsyncDelayedTask(SaneEconomy.getInstance(), () -> {
waitForSlot(); doExecuteAsyncOperation(1, callback);
openTransactions.incrementAndGet();
try (Connection conn = openConnection()) {
callback.accept(conn);
} catch (Exception e) {
throw new RuntimeException("This shouldn't happen (database error)", e);
} finally {
openTransactions.decrementAndGet();
}
}); });
} }
private void doExecuteAsyncOperation(int levels, Consumer<Connection> callback) {
waitForSlot();
openTransactions.incrementAndGet();
try (Connection conn = openConnection()) {
callback.accept(conn);
} catch (Exception e) {
if (levels < 5) {
LOGGER.severe("An internal SQL error has occured, trying up to " + (5 - levels) + " more times...");
e.printStackTrace();
levels++;
doExecuteAsyncOperation(levels, callback);
}
throw new RuntimeException("This shouldn't happen (database error)", e);
} finally {
openTransactions.decrementAndGet();
}
}
public DatabaseCredentials getCredentials() { public DatabaseCredentials getCredentials() {
return dbCredentials; return dbCredentials;
} }