Upstream/LuckPerms
1
0
Fork 0
mirror of https://github.com/LuckPerms/LuckPerms.git synced 2024-12-29 12:37:40 +01:00
Code Issues Projects Releases Wiki Activity

Don't attempt further poll operations after SQL messenger has closed (#2516)

Browse Source
This commit is contained in:
Luck 2020-08-13 10:49:18 +01:00
parent ee3bb22c2c
commit 6f90e545fc
No known key found for this signature in database
GPG Key ID: EFA9B3EC5FD90F8B
1 changed files with 38 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
common/src/main/java/me/lucko/luckperms/common/messaging/sql/AbstractSqlMessenger.java
View File

@ -36,6 +36,8 @@ import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/**
* An implementation of {@link Messenger} using SQL.
@ -45,6 +47,9 @@ public abstract class AbstractSqlMessenger implements Messenger {
private final IncomingMessageConsumer consumer;
private long lastId = -1;
private final ReadWriteLock lock = new ReentrantReadWriteLock();
private boolean closed = false;
protected AbstractSqlMessenger(IncomingMessageConsumer consumer) {
this.consumer = consumer;
}
@ -82,6 +87,12 @@ public abstract class AbstractSqlMessenger implements Messenger {
@Override
public void sendOutgoingMessage(@NonNull OutgoingMessage outgoingMessage) {
this.lock.readLock().lock();
if (this.closed) {
this.lock.readLock().unlock();
return;
}
try (Connection c = getConnection()) {
try (PreparedStatement ps = c.prepareStatement("INSERT INTO `" + getTableName() + "` (`time`, `msg`) VALUES(NOW(), ?)")) {
ps.setString(1, outgoingMessage.asEncodedString());
@ -89,10 +100,18 @@ public abstract class AbstractSqlMessenger implements Messenger {
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
this.lock.readLock().unlock();
}
}
public void pollMessages() {
this.lock.readLock().lock();
if (this.closed) {
this.lock.readLock().unlock();
return;
}
try (Connection c = getConnection()) {
try (PreparedStatement ps = c.prepareStatement("SELECT `id`, `msg` FROM `" + getTableName() + "` WHERE `id` > ? AND (NOW() - `time` < 30)")) {
ps.setLong(1, this.lastId);
@ -108,18 +127,36 @@ public abstract class AbstractSqlMessenger implements Messenger {
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
this.lock.readLock().unlock();
}
}
public void runHousekeeping() {
this.lock.readLock().lock();
if (this.closed) {
this.lock.readLock().unlock();
return;
}
try (Connection c = getConnection()) {
try (PreparedStatement ps = c.prepareStatement("DELETE FROM `" + getTableName() + "` WHERE (NOW() - `time` > 60)")) {
ps.execute();
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
this.lock.readLock().unlock();
}
}
@Override
public void close() {
this.lock.writeLock().lock();
try {
this.closed = true;
} finally {
this.lock.writeLock().unlock();
}
}
}