103 lines
3.2 KiB
Java
103 lines
3.2 KiB
Java
/*
|
|
* License is provided in the jar as LICENSE also here:
|
|
* https://github.com/Rsl1122/Plan-PlayerAnalytics/blob/master/Plan/src/main/resources/LICENSE
|
|
*/
|
|
package com.djrapitops.plan;
|
|
|
|
import com.djrapitops.plan.api.exceptions.database.DBException;
|
|
import com.djrapitops.plan.api.exceptions.database.DBInitException;
|
|
import com.djrapitops.plan.api.exceptions.database.DBOpException;
|
|
import com.djrapitops.plan.data.container.Session;
|
|
import com.djrapitops.plan.data.store.keys.SessionKeys;
|
|
import com.djrapitops.plan.system.cache.SessionCache;
|
|
import com.djrapitops.plan.system.database.DBSystem;
|
|
import com.djrapitops.plan.system.database.databases.Database;
|
|
import com.djrapitops.plugin.logging.L;
|
|
import com.djrapitops.plugin.logging.error.ErrorHandler;
|
|
|
|
import javax.inject.Inject;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
import java.util.UUID;
|
|
|
|
/**
|
|
* Thread that is run when JVM shuts down.
|
|
* <p>
|
|
* Saves active sessions to the Database (PlayerQuitEvent is not called)
|
|
*
|
|
* @author Rsl1122
|
|
*/
|
|
public class ShutdownHook extends Thread {
|
|
|
|
private static ShutdownHook activated;
|
|
private final DBSystem dbSystem;
|
|
private final ErrorHandler errorHandler;
|
|
|
|
@Inject
|
|
public ShutdownHook(DBSystem dbSystem, ErrorHandler errorHandler) {
|
|
this.dbSystem = dbSystem;
|
|
this.errorHandler = errorHandler;
|
|
}
|
|
|
|
private static boolean isActivated() {
|
|
return activated != null;
|
|
}
|
|
|
|
private static void activate(ShutdownHook hook) {
|
|
activated = hook;
|
|
Runtime.getRuntime().addShutdownHook(hook);
|
|
}
|
|
|
|
private static void deactivate() {
|
|
Runtime.getRuntime().removeShutdownHook(activated);
|
|
activated = null;
|
|
}
|
|
|
|
public void register() {
|
|
if (isActivated()) {
|
|
deactivate();
|
|
}
|
|
activate(this);
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
Map<UUID, Session> activeSessions = SessionCache.getActiveSessions();
|
|
long now = System.currentTimeMillis();
|
|
saveActiveSessions(activeSessions, now);
|
|
} catch (IllegalStateException ignored) {
|
|
/* Database is not initialized */
|
|
} catch (DBInitException e) {
|
|
errorHandler.log(L.ERROR, this.getClass(), e);
|
|
} finally {
|
|
try {
|
|
dbSystem.getDatabase().close();
|
|
} catch (DBException e) {
|
|
errorHandler.log(L.ERROR, this.getClass(), e);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void saveActiveSessions(Map<UUID, Session> activeSessions, long now) throws DBInitException {
|
|
for (Map.Entry<UUID, Session> entry : activeSessions.entrySet()) {
|
|
UUID uuid = entry.getKey();
|
|
Session session = entry.getValue();
|
|
Optional<Long> end = session.getValue(SessionKeys.END);
|
|
if (!end.isPresent()) {
|
|
session.endSession(now);
|
|
}
|
|
Database database = dbSystem.getDatabase();
|
|
if (!database.isOpen()) {
|
|
database.init();
|
|
}
|
|
try {
|
|
database.save().session(uuid, session);
|
|
} catch (DBOpException e) {
|
|
errorHandler.log(L.ERROR, this.getClass(), e);
|
|
}
|
|
}
|
|
activeSessions.clear();
|
|
}
|
|
}
|