Initial commit.
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
package org.mobarena.stats;
|
||||
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
import org.mobarena.stats.store.StatsStoreRegistry;
|
||||
|
||||
import java.io.File;
|
||||
import java.util.concurrent.Executor;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
public interface MobArenaStats {
|
||||
|
||||
Logger getLogger();
|
||||
|
||||
Executor getSyncExecutor();
|
||||
|
||||
Executor getAsyncExecutor();
|
||||
|
||||
StatsStore getStatsStore();
|
||||
|
||||
StatsStoreRegistry getStatsStoreRegistry();
|
||||
|
||||
File getDataFolder();
|
||||
|
||||
}
|
||||
@@ -0,0 +1,193 @@
|
||||
package org.mobarena.stats;
|
||||
|
||||
import com.garbagemule.MobArena.MobArena;
|
||||
import com.garbagemule.MobArena.commands.CommandHandler;
|
||||
import org.bukkit.command.PluginCommand;
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.bukkit.event.HandlerList;
|
||||
import org.bukkit.plugin.PluginManager;
|
||||
import org.mobarena.stats.command.ArenaStatsCommand;
|
||||
import org.mobarena.stats.command.DeleteSessionStatsCommand;
|
||||
import org.mobarena.stats.command.ExportCommand;
|
||||
import org.mobarena.stats.command.GlobalStatsCommand;
|
||||
import org.mobarena.stats.command.ImportCommand;
|
||||
import org.mobarena.stats.command.PlayerStatsCommand;
|
||||
import org.mobarena.stats.platform.AsyncBukkitExecutor;
|
||||
import org.mobarena.stats.platform.SyncBukkitExecutor;
|
||||
import org.mobarena.stats.session.SessionListener;
|
||||
import org.mobarena.stats.session.SessionStore;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
import org.bukkit.plugin.java.JavaPlugin;
|
||||
import org.bukkit.scheduler.BukkitScheduler;
|
||||
import org.mobarena.stats.store.StatsStoreRegistry;
|
||||
import org.mobarena.stats.store.csv.CsvStatsStore;
|
||||
import org.mobarena.stats.store.jdbc.JdbcStatsStore;
|
||||
import org.mobarena.stats.store.mariadb.MariadbStatsStore;
|
||||
import org.mobarena.stats.store.mysql.MysqlStatsStore;
|
||||
import org.mobarena.stats.store.sqlite.SqliteStatsStore;
|
||||
|
||||
import java.io.File;
|
||||
import java.util.concurrent.Executor;
|
||||
import java.util.logging.Level;
|
||||
|
||||
public class MobArenaStatsPlugin extends JavaPlugin implements MobArenaStats {
|
||||
|
||||
// The sad state of affairs is that MobArena's command framework has no
|
||||
// support for registering commands by instance, but only by class, which
|
||||
// means that we can't properly inject dependencies and have to resort to
|
||||
// the Singleton Pattern.
|
||||
private static MobArenaStats instance;
|
||||
|
||||
private StatsStoreRegistry statsStoreRegistry;
|
||||
|
||||
private Executor syncExecutor;
|
||||
private Executor asyncExecutor;
|
||||
|
||||
private SessionStore sessionStore;
|
||||
private StatsStore statsStore;
|
||||
private SessionListener sessionListener;
|
||||
|
||||
@Override
|
||||
public void onLoad() {
|
||||
createStatsStoreRegistry();
|
||||
registerStatsStoreFactories();
|
||||
}
|
||||
|
||||
private void createStatsStoreRegistry() {
|
||||
statsStoreRegistry = StatsStoreRegistry.create(this);
|
||||
}
|
||||
|
||||
private void registerStatsStoreFactories() {
|
||||
statsStoreRegistry.register("csv", CsvStatsStore::create);
|
||||
statsStoreRegistry.register("jdbc", JdbcStatsStore::create);
|
||||
statsStoreRegistry.register("sqlite", SqliteStatsStore::create);
|
||||
statsStoreRegistry.register("mysql", MysqlStatsStore::create);
|
||||
statsStoreRegistry.register("mariadb", MariadbStatsStore::create);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onEnable() {
|
||||
setup();
|
||||
reload();
|
||||
}
|
||||
|
||||
private void setup() {
|
||||
try {
|
||||
instance = this;
|
||||
|
||||
createDataFolder();
|
||||
createConfigFile();
|
||||
setupExecutors();
|
||||
setupCommands();
|
||||
} catch (Exception up) {
|
||||
// If setup fails, we can't recover, so throw up
|
||||
throw new RuntimeException(up);
|
||||
}
|
||||
}
|
||||
|
||||
private void createDataFolder() {
|
||||
File dir = getDataFolder();
|
||||
if (!dir.exists()) {
|
||||
if (!dir.mkdir()) {
|
||||
throw new IllegalStateException("Failed to create plugin data folder!");
|
||||
}
|
||||
getLogger().info("Data folder created.");
|
||||
}
|
||||
}
|
||||
|
||||
private void createConfigFile() {
|
||||
File file = new File(getDataFolder(), "config.yml");
|
||||
if (!file.exists()) {
|
||||
saveResource("config.yml", false);
|
||||
getLogger().info("config.yml created.");
|
||||
}
|
||||
}
|
||||
|
||||
private void setupExecutors() {
|
||||
BukkitScheduler scheduler = getServer().getScheduler();
|
||||
if (syncExecutor == null) {
|
||||
syncExecutor = new SyncBukkitExecutor(this, scheduler);
|
||||
}
|
||||
if (asyncExecutor == null) {
|
||||
asyncExecutor = new AsyncBukkitExecutor(this, scheduler);
|
||||
}
|
||||
}
|
||||
|
||||
private void setupCommands() {
|
||||
PluginManager manager = getServer().getPluginManager();
|
||||
MobArena mobarena = (MobArena) manager.getPlugin("MobArena");
|
||||
|
||||
PluginCommand command = mobarena.getCommand("ma");
|
||||
CommandHandler handler = (CommandHandler) command.getExecutor();
|
||||
|
||||
// User commands
|
||||
handler.register(ArenaStatsCommand.class);
|
||||
handler.register(GlobalStatsCommand.class);
|
||||
handler.register(PlayerStatsCommand.class);
|
||||
|
||||
// Admin commands
|
||||
handler.register(DeleteSessionStatsCommand.class);
|
||||
handler.register(ExportCommand.class);
|
||||
handler.register(ImportCommand.class);
|
||||
}
|
||||
|
||||
private void reload() {
|
||||
try {
|
||||
createSessionStore();
|
||||
loadStatsStore();
|
||||
registerSessionListener();
|
||||
} catch (Exception e) {
|
||||
getLogger().log(Level.SEVERE, "Load failure", e);
|
||||
}
|
||||
}
|
||||
|
||||
private void createSessionStore() {
|
||||
if (sessionStore != null) {
|
||||
sessionStore.clear();
|
||||
}
|
||||
|
||||
sessionStore = SessionStore.createNew();
|
||||
getLogger().info("Session store created.");
|
||||
}
|
||||
|
||||
private void loadStatsStore() throws Exception {
|
||||
ConfigurationSection config = getConfig();
|
||||
ConfigurationSection section = config.getConfigurationSection("store");
|
||||
if (section == null) {
|
||||
throw new IllegalArgumentException("No store section in config-file.");
|
||||
}
|
||||
|
||||
statsStore = statsStoreRegistry.create(section);
|
||||
}
|
||||
|
||||
private void registerSessionListener() {
|
||||
if (sessionListener != null) {
|
||||
HandlerList.unregisterAll(sessionListener);
|
||||
}
|
||||
|
||||
sessionListener = new SessionListener(sessionStore, statsStore, asyncExecutor, getLogger());
|
||||
getServer().getPluginManager().registerEvents(sessionListener, this);
|
||||
getLogger().info("Session listener registered.");
|
||||
}
|
||||
|
||||
public StatsStoreRegistry getStatsStoreRegistry() {
|
||||
return statsStoreRegistry;
|
||||
}
|
||||
|
||||
public Executor getSyncExecutor() {
|
||||
return syncExecutor;
|
||||
}
|
||||
|
||||
public Executor getAsyncExecutor() {
|
||||
return asyncExecutor;
|
||||
}
|
||||
|
||||
public StatsStore getStatsStore() {
|
||||
return statsStore;
|
||||
}
|
||||
|
||||
public static MobArenaStats getInstance() {
|
||||
return instance;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,57 @@
|
||||
package org.mobarena.stats.command;
|
||||
|
||||
import com.garbagemule.MobArena.commands.Command;
|
||||
import com.garbagemule.MobArena.commands.CommandInfo;
|
||||
import com.garbagemule.MobArena.framework.ArenaMaster;
|
||||
import com.garbagemule.MobArena.util.Slugs;
|
||||
import org.bukkit.command.CommandSender;
|
||||
import org.bukkit.entity.Player;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
import org.mobarena.stats.store.ArenaStats;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@CommandInfo(
|
||||
name = "arena-stats",
|
||||
pattern = "arena-stats",
|
||||
usage = "/ma arena-stats <arena>",
|
||||
desc = "show overall stats for the given arena",
|
||||
permission = "mobarenastats.command.arena-stats"
|
||||
)
|
||||
public class ArenaStatsCommand implements Command {
|
||||
|
||||
@Override
|
||||
public boolean execute(ArenaMaster am, CommandSender sender, String... args) {
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
if (args.length < 1) {
|
||||
return false;
|
||||
}
|
||||
String slug = Slugs.create(args[0]);
|
||||
|
||||
plugin.getAsyncExecutor().execute(() -> {
|
||||
StatsStore store = plugin.getStatsStore();
|
||||
ArenaStats stats = store.getArenaStats(slug);
|
||||
sender.sendMessage("Stats for arena " + slug + ":");
|
||||
sender.sendMessage("- Highest wave: " + stats.highestWave);
|
||||
sender.sendMessage("- Longest duration: " + stats.highestSeconds + " secs");
|
||||
sender.sendMessage("- Most kills: " + stats.highestKills);
|
||||
sender.sendMessage("- Total sessions: " + stats.totalSessions);
|
||||
sender.sendMessage("- Total duration: " + stats.totalSeconds + " secs");
|
||||
sender.sendMessage("- Total kills: " + stats.totalKills);
|
||||
sender.sendMessage("- Total waves: " + stats.totalWaves);
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> tab(ArenaMaster am, Player player, String... args) {
|
||||
// TODO: tab complete arena slugs?
|
||||
return Command.super.tab(am, player, args);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,56 @@
|
||||
package org.mobarena.stats.command;
|
||||
|
||||
import com.garbagemule.MobArena.Messenger;
|
||||
import com.garbagemule.MobArena.commands.Command;
|
||||
import com.garbagemule.MobArena.commands.CommandInfo;
|
||||
import com.garbagemule.MobArena.framework.ArenaMaster;
|
||||
import org.bukkit.ChatColor;
|
||||
import org.bukkit.command.CommandSender;
|
||||
import org.bukkit.entity.Player;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
|
||||
@CommandInfo(
|
||||
name = "delete-session-stats",
|
||||
pattern = "delete-session-stats",
|
||||
usage = "/ma delete-session-stats <session-id>",
|
||||
desc = "delete all stats collected for the given session",
|
||||
permission = "mobarenastats.command.delete-session-stats"
|
||||
)
|
||||
public class DeleteSessionStatsCommand implements Command {
|
||||
|
||||
@Override
|
||||
public boolean execute(ArenaMaster am, CommandSender sender, String... args) {
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
// TODO: check args, handle non-UUID input error
|
||||
UUID sessionId = UUID.fromString(args[0]);
|
||||
|
||||
StatsStore store = plugin.getStatsStore();
|
||||
Messenger messenger = am.getGlobalMessenger();
|
||||
|
||||
plugin.getAsyncExecutor().execute(() -> {
|
||||
store.delete(sessionId);
|
||||
messenger.tell(sender, String.format(
|
||||
"Stats for session %s%s%s deleted.",
|
||||
ChatColor.YELLOW,
|
||||
sessionId,
|
||||
ChatColor.RESET
|
||||
));
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> tab(ArenaMaster am, Player player, String... args) {
|
||||
// TODO: tab complete session IDs?
|
||||
return Command.super.tab(am, player, args);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,57 @@
|
||||
package org.mobarena.stats.command;
|
||||
|
||||
import com.garbagemule.MobArena.Messenger;
|
||||
import com.garbagemule.MobArena.commands.Command;
|
||||
import com.garbagemule.MobArena.commands.CommandInfo;
|
||||
import com.garbagemule.MobArena.framework.ArenaMaster;
|
||||
import org.bukkit.ChatColor;
|
||||
import org.bukkit.command.CommandSender;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
import org.mobarena.stats.store.StatsExport;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
import org.mobarena.stats.store.StatsStoreRegistry;
|
||||
|
||||
@CommandInfo(
|
||||
name = "export-stats",
|
||||
pattern = "export-stats",
|
||||
usage = "/ma export-stats",
|
||||
desc = "export the current stats store to a file in the given format",
|
||||
permission = "mobarenastats.command.export-stats"
|
||||
)
|
||||
public class ExportCommand implements Command {
|
||||
|
||||
@Override
|
||||
public boolean execute(ArenaMaster am, CommandSender sender, String... args) {
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
StatsStore store = plugin.getStatsStore();
|
||||
StatsStoreRegistry registry = plugin.getStatsStoreRegistry();
|
||||
|
||||
Messenger messenger = am.getGlobalMessenger();
|
||||
messenger.tell(sender, "Exporting stats. This may take a while...");
|
||||
plugin.getAsyncExecutor().execute(() -> {
|
||||
try {
|
||||
String filename = StatsExport.run(store, registry);
|
||||
|
||||
messenger.tell(sender, String.format(
|
||||
"Export to %s%s%s complete.",
|
||||
ChatColor.YELLOW,
|
||||
filename,
|
||||
ChatColor.RESET
|
||||
));
|
||||
} catch (Exception e) {
|
||||
messenger.tell(sender, String.format(
|
||||
"Export %sfailed%s because:\n%s",
|
||||
ChatColor.RED,
|
||||
ChatColor.RESET,
|
||||
e.getMessage()
|
||||
));
|
||||
}
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,39 @@
|
||||
package org.mobarena.stats.command;
|
||||
|
||||
import com.garbagemule.MobArena.commands.Command;
|
||||
import com.garbagemule.MobArena.commands.CommandInfo;
|
||||
import com.garbagemule.MobArena.framework.ArenaMaster;
|
||||
import org.bukkit.command.CommandSender;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
import org.mobarena.stats.store.GlobalStats;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
|
||||
@CommandInfo(
|
||||
name = "global-stats",
|
||||
pattern = "global-stats",
|
||||
usage = "/ma global-stats",
|
||||
desc = "show stats across all sessions",
|
||||
permission = "mobarenastats.command.global-stats"
|
||||
)
|
||||
public class GlobalStatsCommand implements Command {
|
||||
|
||||
@Override
|
||||
public boolean execute(ArenaMaster am, CommandSender sender, String... args) {
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
plugin.getAsyncExecutor().execute(() -> {
|
||||
StatsStore store = plugin.getStatsStore();
|
||||
GlobalStats stats = store.getGlobalStats();
|
||||
sender.sendMessage("Global stats:");
|
||||
sender.sendMessage("- Total sessions: " + stats.totalSessions);
|
||||
sender.sendMessage("- Total duration: " + stats.totalSeconds + " secs");
|
||||
sender.sendMessage("- Total kills: " + stats.totalKills);
|
||||
sender.sendMessage("- Total waves: " + stats.totalWaves);
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,130 @@
|
||||
package org.mobarena.stats.command;
|
||||
|
||||
import com.garbagemule.MobArena.Messenger;
|
||||
import com.garbagemule.MobArena.commands.Command;
|
||||
import com.garbagemule.MobArena.commands.CommandInfo;
|
||||
import com.garbagemule.MobArena.framework.ArenaMaster;
|
||||
import org.bukkit.ChatColor;
|
||||
import org.bukkit.command.CommandSender;
|
||||
import org.bukkit.entity.Player;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
import org.mobarena.stats.store.StatsExport;
|
||||
import org.mobarena.stats.store.StatsImport;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
import org.mobarena.stats.store.StatsStoreRegistry;
|
||||
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
@CommandInfo(
|
||||
name = "import-stats",
|
||||
pattern = "import-stats",
|
||||
usage = "/ma import-stats <filename>",
|
||||
desc = "import stats from an database export file into the current stats store",
|
||||
permission = "mobarenastats.command.import-stats"
|
||||
)
|
||||
public class ImportCommand implements Command {
|
||||
|
||||
@Override
|
||||
public boolean execute(ArenaMaster am, CommandSender sender, String... args) {
|
||||
if (args.length < 1) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
Path data = plugin.getDataFolder().toPath();
|
||||
Path file = data.resolve(args[0]);
|
||||
if (!Files.exists(file)) {
|
||||
sender.sendMessage(String.format(
|
||||
"File %s%s%s not found.",
|
||||
ChatColor.YELLOW,
|
||||
args[0],
|
||||
ChatColor.RESET
|
||||
));
|
||||
return false;
|
||||
}
|
||||
|
||||
String filename = file.getFileName().toString();
|
||||
if (!filename.startsWith(StatsExport.FILENAME_PREFIX)) {
|
||||
sender.sendMessage(String.format(
|
||||
"Not a valid database export; filename must start with %s%s%s.",
|
||||
ChatColor.YELLOW,
|
||||
StatsExport.FILENAME_PREFIX,
|
||||
ChatColor.RESET
|
||||
));
|
||||
return true;
|
||||
}
|
||||
if (!filename.endsWith(StatsExport.FILENAME_SUFFIX)) {
|
||||
sender.sendMessage(String.format(
|
||||
"Not a valid database export; filename must end with %s%s%s.",
|
||||
ChatColor.YELLOW,
|
||||
StatsExport.FILENAME_SUFFIX,
|
||||
ChatColor.RESET
|
||||
));
|
||||
return true;
|
||||
}
|
||||
|
||||
StatsStore store = plugin.getStatsStore();
|
||||
StatsStoreRegistry registry = plugin.getStatsStoreRegistry();
|
||||
|
||||
Messenger messenger = am.getGlobalMessenger();
|
||||
messenger.tell(sender, String.format(
|
||||
"Importing stats from %s%s%s. This may take a while...",
|
||||
ChatColor.YELLOW,
|
||||
filename,
|
||||
ChatColor.RESET
|
||||
));
|
||||
plugin.getAsyncExecutor().execute(() -> {
|
||||
try {
|
||||
StatsImport.run(registry, filename, store);
|
||||
|
||||
messenger.tell(sender, String.format(
|
||||
"Import from %s%s%s complete.",
|
||||
ChatColor.YELLOW,
|
||||
filename,
|
||||
ChatColor.RESET
|
||||
));
|
||||
} catch (Exception e) {
|
||||
messenger.tell(sender, String.format(
|
||||
"Import %sfailed%s because:\n%s",
|
||||
ChatColor.RED,
|
||||
ChatColor.RESET,
|
||||
e.getMessage()
|
||||
));
|
||||
}
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> tab(ArenaMaster am, Player player, String... args) {
|
||||
if (args.length > 1) {
|
||||
return Collections.emptyList();
|
||||
}
|
||||
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
String[] files = plugin.getDataFolder().list();
|
||||
if (files == null || files.length == 0) {
|
||||
return Collections.emptyList();
|
||||
}
|
||||
|
||||
String prefix = (args.length == 1) ? args[0] : "";
|
||||
|
||||
return Arrays.stream(files)
|
||||
.filter(filename -> filename.startsWith(prefix))
|
||||
.filter(filename -> filename.startsWith(StatsExport.FILENAME_PREFIX))
|
||||
.filter(filename -> filename.endsWith(StatsExport.FILENAME_SUFFIX))
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,58 @@
|
||||
package org.mobarena.stats.command;
|
||||
|
||||
import com.garbagemule.MobArena.commands.Command;
|
||||
import com.garbagemule.MobArena.commands.CommandInfo;
|
||||
import com.garbagemule.MobArena.framework.ArenaMaster;
|
||||
import org.bukkit.command.CommandSender;
|
||||
import org.bukkit.entity.Player;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
import org.mobarena.stats.store.PlayerStats;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@CommandInfo(
|
||||
name = "player-stats",
|
||||
pattern = "player-stats",
|
||||
usage = "/ma player-stats <player>",
|
||||
desc = "show overall stats for the given player",
|
||||
permission = "mobarenastats.command.player-stats"
|
||||
)
|
||||
public class PlayerStatsCommand implements Command {
|
||||
|
||||
@Override
|
||||
public boolean execute(ArenaMaster am, CommandSender sender, String... args) {
|
||||
// :(
|
||||
MobArenaStats plugin = MobArenaStatsPlugin.getInstance();
|
||||
|
||||
String name;
|
||||
if (args.length == 0) {
|
||||
if (!(sender instanceof Player)) {
|
||||
return false;
|
||||
}
|
||||
name = sender.getName();
|
||||
} else {
|
||||
name = args[0];
|
||||
}
|
||||
|
||||
plugin.getAsyncExecutor().execute(() -> {
|
||||
StatsStore store = plugin.getStatsStore();
|
||||
PlayerStats stats = store.getPlayerStats(name);
|
||||
sender.sendMessage("Stats for player " + name + ":");
|
||||
sender.sendMessage("- Total sessions: " + stats.totalSessions);
|
||||
sender.sendMessage("- Total duration: " + stats.totalSeconds + " secs");
|
||||
sender.sendMessage("- Total kills: " + stats.totalKills);
|
||||
sender.sendMessage("- Total waves: " + stats.totalWaves);
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> tab(ArenaMaster am, Player player, String... args) {
|
||||
// TODO: tab complete player names?
|
||||
return Command.super.tab(am, player, args);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
package org.mobarena.stats.platform;
|
||||
|
||||
import org.bukkit.plugin.Plugin;
|
||||
import org.bukkit.scheduler.BukkitScheduler;
|
||||
|
||||
import java.util.concurrent.Executor;
|
||||
|
||||
public class AsyncBukkitExecutor implements Executor {
|
||||
|
||||
private final Plugin plugin;
|
||||
private final BukkitScheduler scheduler;
|
||||
|
||||
public AsyncBukkitExecutor(Plugin plugin, BukkitScheduler scheduler) {
|
||||
this.plugin = plugin;
|
||||
this.scheduler = scheduler;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void execute(Runnable command) {
|
||||
scheduler.runTaskAsynchronously(plugin, command);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
package org.mobarena.stats.platform;
|
||||
|
||||
import org.bukkit.plugin.Plugin;
|
||||
import org.bukkit.scheduler.BukkitScheduler;
|
||||
|
||||
import java.util.concurrent.Executor;
|
||||
|
||||
public class SyncBukkitExecutor implements Executor {
|
||||
|
||||
private final Plugin plugin;
|
||||
private final BukkitScheduler scheduler;
|
||||
|
||||
public SyncBukkitExecutor(Plugin plugin, BukkitScheduler scheduler) {
|
||||
this.plugin = plugin;
|
||||
this.scheduler = scheduler;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void execute(Runnable command) {
|
||||
scheduler.runTask(plugin, command);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
public enum PlayerConclusion {
|
||||
|
||||
/**
|
||||
* When an arena has a final wave and the given player reaches and
|
||||
* completes that wave, the player session concludes in a victory.
|
||||
*/
|
||||
VICTORY,
|
||||
|
||||
/**
|
||||
* When the given player dies in an arena, the player's session will
|
||||
* concludes in a defeat, even if other players are still alive.
|
||||
*/
|
||||
DEFEAT,
|
||||
|
||||
/**
|
||||
* When the given player leaves an ongoing arena, the player's session
|
||||
* will conclude in a retreat, even if other players are still alive.
|
||||
*/
|
||||
RETREAT,
|
||||
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
import java.time.Instant;
|
||||
import java.util.UUID;
|
||||
|
||||
public class PlayerSessionStats {
|
||||
|
||||
public final UUID sessionId;
|
||||
public final UUID playerId;
|
||||
public final String playerName;
|
||||
|
||||
public String className;
|
||||
|
||||
public Instant joinTime;
|
||||
public Instant readyTime;
|
||||
public Instant leaveTime;
|
||||
public Instant deathTime;
|
||||
|
||||
public int kills;
|
||||
public int dmgDone;
|
||||
public int dmgTaken;
|
||||
public int swings;
|
||||
public int hits;
|
||||
public int lastWave;
|
||||
|
||||
public PlayerConclusion conclusion;
|
||||
|
||||
public PlayerSessionStats(
|
||||
UUID sessionId,
|
||||
UUID playerId,
|
||||
String playerName
|
||||
) {
|
||||
this.sessionId = sessionId;
|
||||
this.playerId = playerId;
|
||||
this.playerName = playerName;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,130 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
import com.garbagemule.MobArena.framework.Arena;
|
||||
import org.bukkit.entity.Player;
|
||||
|
||||
import java.time.Instant;
|
||||
import java.util.Collection;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.UUID;
|
||||
|
||||
public class Session {
|
||||
|
||||
final UUID sessionId;
|
||||
final String arenaSlug;
|
||||
|
||||
final SessionStats sessionStats;
|
||||
final Map<UUID, PlayerSessionStats> playerStats;
|
||||
|
||||
public Session(UUID sessionId, String arenaSlug) {
|
||||
this.sessionId = sessionId;
|
||||
this.arenaSlug = arenaSlug;
|
||||
|
||||
this.sessionStats = new SessionStats(sessionId, arenaSlug);
|
||||
this.playerStats = new HashMap<>();
|
||||
}
|
||||
|
||||
public void playerJoin(Player player) {
|
||||
UUID playerId = player.getUniqueId();
|
||||
String playerName = player.getName();
|
||||
PlayerSessionStats stats = new PlayerSessionStats(sessionId, playerId, playerName);
|
||||
playerStats.put(playerId, stats);
|
||||
|
||||
stats.joinTime = Instant.now();
|
||||
}
|
||||
|
||||
public void playerReady(Player player, String className) {
|
||||
UUID playerId = player.getUniqueId();
|
||||
PlayerSessionStats stats = playerStats.get(playerId);
|
||||
if (stats == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
stats.readyTime = Instant.now();
|
||||
stats.className = className;
|
||||
}
|
||||
|
||||
public void playerLeave(Arena arena, Player player) {
|
||||
UUID playerId = player.getUniqueId();
|
||||
PlayerSessionStats stats = playerStats.get(playerId);
|
||||
if (stats == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
stats.leaveTime = Instant.now();
|
||||
|
||||
if (stats.conclusion == null) {
|
||||
stats.conclusion = PlayerConclusion.RETREAT;
|
||||
}
|
||||
|
||||
StatsUtil.copy(arena, player, stats);
|
||||
}
|
||||
|
||||
public void playerDeath(Arena arena, Player player) {
|
||||
UUID playerId = player.getUniqueId();
|
||||
PlayerSessionStats stats = playerStats.get(playerId);
|
||||
if (stats == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
stats.deathTime = Instant.now();
|
||||
|
||||
if (stats.conclusion == null) {
|
||||
stats.conclusion = PlayerConclusion.DEFEAT;
|
||||
}
|
||||
|
||||
StatsUtil.copy(arena, player, stats);
|
||||
}
|
||||
|
||||
public void start() {
|
||||
sessionStats.startTime = Instant.now();
|
||||
}
|
||||
|
||||
public void wave(int wave) {
|
||||
sessionStats.lastWave = wave;
|
||||
}
|
||||
|
||||
public void complete() {
|
||||
sessionStats.conclusion = SessionConclusion.VICTORY;
|
||||
|
||||
for (PlayerSessionStats playerStats : playerStats.values()) {
|
||||
if (playerStats.conclusion == null) {
|
||||
playerStats.conclusion = PlayerConclusion.VICTORY;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void end() {
|
||||
sessionStats.endTime = Instant.now();
|
||||
|
||||
if (sessionStats.conclusion == null) {
|
||||
sessionStats.conclusion = SessionConclusion.DEFEAT;
|
||||
}
|
||||
}
|
||||
|
||||
public UUID getSessionId() {
|
||||
return sessionId;
|
||||
}
|
||||
|
||||
public String getArenaSlug() {
|
||||
return arenaSlug;
|
||||
}
|
||||
|
||||
public SessionStats getSessionStats() {
|
||||
return sessionStats;
|
||||
}
|
||||
|
||||
public Collection<PlayerSessionStats> getPlayerStats() {
|
||||
return playerStats.values();
|
||||
}
|
||||
|
||||
public PlayerSessionStats getPlayerStats(UUID playerId) {
|
||||
return playerStats.get(playerId);
|
||||
}
|
||||
|
||||
public void setPlayerStats(UUID playerId, PlayerSessionStats stats) {
|
||||
playerStats.put(playerId, stats);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,17 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
public enum SessionConclusion {
|
||||
|
||||
/**
|
||||
* When an arena has a final wave and one or more players reach and
|
||||
* complete that wave, the session concludes in a victory.
|
||||
*/
|
||||
VICTORY,
|
||||
|
||||
/**
|
||||
* When the last player alive in an arena dies, the session concludes
|
||||
* in a defeat, meaning the players "lost" the session.
|
||||
*/
|
||||
DEFEAT,
|
||||
|
||||
}
|
||||
@@ -0,0 +1,190 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
import com.garbagemule.MobArena.ArenaClass;
|
||||
import com.garbagemule.MobArena.ArenaPlayer;
|
||||
import com.garbagemule.MobArena.events.ArenaCompleteEvent;
|
||||
import com.garbagemule.MobArena.events.ArenaEndEvent;
|
||||
import com.garbagemule.MobArena.events.ArenaPlayerDeathEvent;
|
||||
import com.garbagemule.MobArena.events.ArenaPlayerJoinEvent;
|
||||
import com.garbagemule.MobArena.events.ArenaPlayerLeaveEvent;
|
||||
import com.garbagemule.MobArena.events.ArenaPlayerReadyEvent;
|
||||
import com.garbagemule.MobArena.events.ArenaStartEvent;
|
||||
import com.garbagemule.MobArena.events.NewWaveEvent;
|
||||
import com.garbagemule.MobArena.framework.Arena;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
import org.bukkit.entity.Player;
|
||||
import org.bukkit.event.EventHandler;
|
||||
import org.bukkit.event.EventPriority;
|
||||
import org.bukkit.event.Listener;
|
||||
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
import java.util.concurrent.Executor;
|
||||
import java.util.logging.Level;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
public class SessionListener implements Listener {
|
||||
|
||||
private final SessionStore sessionStore;
|
||||
private final StatsStore statsStore;
|
||||
private final Executor asyncExecutor;
|
||||
private final Logger log;
|
||||
|
||||
public SessionListener(
|
||||
SessionStore sessionStore,
|
||||
StatsStore statsStore,
|
||||
Executor asyncExecutor,
|
||||
Logger log
|
||||
) {
|
||||
this.sessionStore = sessionStore;
|
||||
this.statsStore = statsStore;
|
||||
this.asyncExecutor = asyncExecutor;
|
||||
this.log = log;
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaPlayerJoinEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
Player player = event.getPlayer();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
session = sessionStore.create(arena);
|
||||
}
|
||||
|
||||
session.playerJoin(player);
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaPlayerReadyEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
Player player = event.getPlayer();
|
||||
String className = getClassName(arena, player);
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected ready event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.playerReady(player, className);
|
||||
}
|
||||
|
||||
private String getClassName(Arena arena, Player player) {
|
||||
ArenaPlayer ap = arena.getArenaPlayer(player);
|
||||
if (ap == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
ArenaClass ac = ap.getArenaClass();
|
||||
if (ac == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return ac.getSlug();
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaPlayerLeaveEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
Player player = event.getPlayer();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected leave event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.playerLeave(arena, player);
|
||||
|
||||
if (!arena.isRunning()) {
|
||||
if (arena.getPlayersInLobby().size() <= 1) {
|
||||
sessionStore.delete(session);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaPlayerDeathEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
Player player = event.getPlayer();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected death event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.playerDeath(arena, player);
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaStartEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected start event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.start();
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(NewWaveEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
int wave = event.getWaveNumber();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected wave event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.wave(wave);
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaCompleteEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected complete event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.complete();
|
||||
}
|
||||
|
||||
@EventHandler(priority = EventPriority.MONITOR, ignoreCancelled = true)
|
||||
public void on(ArenaEndEvent event) {
|
||||
Arena arena = event.getArena();
|
||||
|
||||
Session session = sessionStore.getByArena(arena);
|
||||
if (session == null) {
|
||||
log.warning("Unexpected end event for non-existent session of arena " + arena.getSlug());
|
||||
return;
|
||||
}
|
||||
|
||||
session.end();
|
||||
|
||||
if (!arena.isRunning()) {
|
||||
// Session never started, so just clean up and bail
|
||||
sessionStore.delete(session);
|
||||
return;
|
||||
}
|
||||
|
||||
CompletableFuture.runAsync(() -> {
|
||||
try {
|
||||
statsStore.save(session);
|
||||
sessionStore.delete(session);
|
||||
log.info("Session (" + session.sessionId + ") for arena " + session.arenaSlug + " saved.");
|
||||
} catch (Exception e) {
|
||||
log.log(Level.SEVERE, "Failed to save session (" + session.sessionId + ") for arena " + session.arenaSlug, e);
|
||||
sessionStore.delete(session);
|
||||
}
|
||||
}, asyncExecutor);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
import java.time.Instant;
|
||||
import java.util.UUID;
|
||||
|
||||
public class SessionStats {
|
||||
|
||||
public final UUID sessionId;
|
||||
public final String arenaSlug;
|
||||
|
||||
public Instant startTime;
|
||||
public Instant endTime;
|
||||
|
||||
public int lastWave;
|
||||
|
||||
public SessionConclusion conclusion;
|
||||
|
||||
public SessionStats(UUID sessionId, String arenaSlug) {
|
||||
this.sessionId = sessionId;
|
||||
this.arenaSlug = arenaSlug;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,86 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
import com.garbagemule.MobArena.framework.Arena;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.UUID;
|
||||
|
||||
/**
|
||||
* In-memory store for on-going session data.
|
||||
* <p>
|
||||
* The session store is the top-level bookkeeping entity for current sessions
|
||||
* in that all {@link Session} objects are created and kept track of by this
|
||||
* store. Unlike {@link org.mobarena.stats.store.StatsStore}, which persists
|
||||
* its data, the session store is just an in-memory collection.
|
||||
*/
|
||||
public class SessionStore {
|
||||
|
||||
private final Map<String, Session> slugToSession;
|
||||
|
||||
private SessionStore() {
|
||||
this.slugToSession = new HashMap<>();
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new {@link Session} instance for the given {@link Arena}.
|
||||
* <p>
|
||||
* Note that only one session can be active per arena. The method throws
|
||||
* if it is called with an arena instance that already has an associated
|
||||
* on-going session.
|
||||
*
|
||||
* @param arena the arena to create a new session for
|
||||
* @return a new session for the given arena
|
||||
* @throws IllegalStateException if a session exists for the given arena
|
||||
*/
|
||||
public Session create(Arena arena) {
|
||||
String arenaSlug = arena.getSlug();
|
||||
|
||||
if (slugToSession.containsKey(arenaSlug)) {
|
||||
throw new IllegalStateException("A session for arena " + arenaSlug + " already exists");
|
||||
}
|
||||
|
||||
UUID sessionId = UUID.randomUUID();
|
||||
Session session = new Session(sessionId, arenaSlug);
|
||||
slugToSession.put(arenaSlug, session);
|
||||
|
||||
return session;
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete the given {@link Session} from the store.
|
||||
* <p>
|
||||
* When a session is deleted, it opens up the possibility of starting a
|
||||
* new one for the associated arena.
|
||||
*
|
||||
* @param session a session to delete
|
||||
*/
|
||||
public void delete(Session session) {
|
||||
slugToSession.remove(session.getArenaSlug());
|
||||
}
|
||||
|
||||
/**
|
||||
* Look up a {@link Session} by its associated {@link Arena}.
|
||||
*
|
||||
* @param arena the arena whose session to look up
|
||||
* @return the associated session instance, or null
|
||||
*/
|
||||
public Session getByArena(Arena arena) {
|
||||
return slugToSession.get(arena.getSlug());
|
||||
}
|
||||
|
||||
/**
|
||||
* Clear the internal session map.
|
||||
* <p>
|
||||
* This method is called by MobArenaStats on reloads to try to clear any
|
||||
* residue from old sessions.
|
||||
*/
|
||||
public void clear() {
|
||||
slugToSession.clear();
|
||||
}
|
||||
|
||||
public static SessionStore createNew() {
|
||||
return new SessionStore();
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,33 @@
|
||||
package org.mobarena.stats.session;
|
||||
|
||||
import com.garbagemule.MobArena.ArenaPlayer;
|
||||
import com.garbagemule.MobArena.ArenaPlayerStatistics;
|
||||
import com.garbagemule.MobArena.framework.Arena;
|
||||
import org.bukkit.entity.Player;
|
||||
|
||||
final class StatsUtil {
|
||||
|
||||
private StatsUtil() {
|
||||
// OK BOSS
|
||||
}
|
||||
|
||||
static void copy(Arena arena, Player player, PlayerSessionStats target) {
|
||||
ArenaPlayer ap = arena.getArenaPlayer(player);
|
||||
if (ap == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
ArenaPlayerStatistics aps = ap.getStats();
|
||||
if (aps == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
target.kills = aps.getInt("kills");
|
||||
target.dmgDone = aps.getInt("dmgDone");
|
||||
target.dmgTaken = aps.getInt("dmgTaken");
|
||||
target.swings = aps.getInt("swings");
|
||||
target.hits = aps.getInt("hits");
|
||||
target.lastWave = aps.getInt("lastWave");
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
public class ArenaStats {
|
||||
|
||||
public final int highestWave;
|
||||
public final int highestSeconds;
|
||||
public final int highestKills;
|
||||
public final int totalSessions;
|
||||
public final long totalSeconds;
|
||||
public final long totalKills;
|
||||
public final long totalWaves;
|
||||
|
||||
public ArenaStats(
|
||||
int highestWave,
|
||||
int highestSeconds,
|
||||
int highestKills,
|
||||
int totalSessions,
|
||||
long totalSeconds,
|
||||
long totalKills,
|
||||
long totalWaves
|
||||
) {
|
||||
this.highestWave = highestWave;
|
||||
this.highestSeconds = highestSeconds;
|
||||
this.highestKills = highestKills;
|
||||
this.totalSessions = totalSessions;
|
||||
this.totalSeconds = totalSeconds;
|
||||
this.totalKills = totalKills;
|
||||
this.totalWaves = totalWaves;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,67 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
import org.mobarena.stats.session.Session;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.UUID;
|
||||
|
||||
public class CachingStatsStore implements StatsStore {
|
||||
|
||||
private final StatsStore delegate;
|
||||
|
||||
private GlobalStats globalStats;
|
||||
private final Map<String, ArenaStats> arenaStats;
|
||||
private final Map<String, PlayerStats> playerStats;
|
||||
|
||||
public CachingStatsStore(StatsStore delegate) {
|
||||
this.delegate = delegate;
|
||||
|
||||
this.globalStats = null;
|
||||
this.arenaStats = new HashMap<>();
|
||||
this.playerStats = new HashMap<>();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void save(Session session) throws IOException {
|
||||
delegate.save(session);
|
||||
|
||||
globalStats = null;
|
||||
arenaStats.remove(session.getArenaSlug());
|
||||
session.getPlayerStats().forEach(stats -> playerStats.remove(stats.playerName));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void delete(UUID sessionId) {
|
||||
delegate.delete(sessionId);
|
||||
|
||||
globalStats = null;
|
||||
arenaStats.clear();
|
||||
playerStats.clear();
|
||||
}
|
||||
|
||||
@Override
|
||||
public GlobalStats getGlobalStats() {
|
||||
if (globalStats == null) {
|
||||
globalStats = delegate.getGlobalStats();
|
||||
}
|
||||
return globalStats;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ArenaStats getArenaStats(String slug) {
|
||||
return arenaStats.computeIfAbsent(slug, delegate::getArenaStats);
|
||||
}
|
||||
|
||||
@Override
|
||||
public PlayerStats getPlayerStats(String name) {
|
||||
return playerStats.computeIfAbsent(name, delegate::getPlayerStats);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void export(StatsStore target) throws IOException {
|
||||
delegate.export(target);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,22 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
public class GlobalStats {
|
||||
|
||||
public final int totalSessions;
|
||||
public final long totalSeconds;
|
||||
public final long totalKills;
|
||||
public final long totalWaves;
|
||||
|
||||
public GlobalStats(
|
||||
int totalSessions,
|
||||
long totalSeconds,
|
||||
long totalKills,
|
||||
long totalWaves
|
||||
) {
|
||||
this.totalSessions = totalSessions;
|
||||
this.totalSeconds = totalSeconds;
|
||||
this.totalKills = totalKills;
|
||||
this.totalWaves = totalWaves;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,22 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
public class PlayerStats {
|
||||
|
||||
public final int totalSessions;
|
||||
public final long totalSeconds;
|
||||
public final long totalKills;
|
||||
public final long totalWaves;
|
||||
|
||||
public PlayerStats(
|
||||
int totalSessions,
|
||||
long totalSeconds,
|
||||
long totalKills,
|
||||
long totalWaves
|
||||
) {
|
||||
this.totalSessions = totalSessions;
|
||||
this.totalSeconds = totalSeconds;
|
||||
this.totalKills = totalKills;
|
||||
this.totalWaves = totalWaves;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,27 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.bukkit.configuration.file.YamlConfiguration;
|
||||
|
||||
public class StatsExport {
|
||||
|
||||
public static final String FILENAME_PREFIX = "stats.export-";
|
||||
public static final String FILENAME_SUFFIX = ".db";
|
||||
|
||||
public static String run(
|
||||
StatsStore store,
|
||||
StatsStoreRegistry registry
|
||||
) throws Exception {
|
||||
String filename = FILENAME_PREFIX + System.currentTimeMillis() + FILENAME_SUFFIX;
|
||||
|
||||
ConfigurationSection config = new YamlConfiguration();
|
||||
config.set("type", "sqlite");
|
||||
config.set("filename", filename);
|
||||
StatsStore output = registry.create(config);
|
||||
|
||||
store.export(output);
|
||||
|
||||
return filename;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,21 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.bukkit.configuration.file.YamlConfiguration;
|
||||
|
||||
public class StatsImport {
|
||||
|
||||
public static void run(
|
||||
StatsStoreRegistry registry,
|
||||
String filename,
|
||||
StatsStore store
|
||||
) throws Exception {
|
||||
ConfigurationSection config = new YamlConfiguration();
|
||||
config.set("type", "sqlite");
|
||||
config.set("filename", filename);
|
||||
StatsStore source = registry.create(config);
|
||||
|
||||
source.export(store);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
import org.mobarena.stats.session.Session;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.UUID;
|
||||
|
||||
/**
|
||||
* Persistent data store for session and player stats.
|
||||
* <p>
|
||||
* All store operations are <i>blocking</i>, meaning any calls to the store
|
||||
* should be handled off the main thread to prevent performance impacts.
|
||||
*/
|
||||
public interface StatsStore {
|
||||
|
||||
/**
|
||||
* Store the given {@link Session}'s data in the store.
|
||||
* <p>
|
||||
* This method should only be called with a "finished" session, i.e. a
|
||||
* session that has concluded and won't be altered after saving.
|
||||
*
|
||||
* @param session a session to store
|
||||
* @throws IOException if the operation fails due to I/O
|
||||
*/
|
||||
void save(Session session) throws IOException;
|
||||
|
||||
/**
|
||||
* Delete all data about the session with the given ID.
|
||||
* <p>
|
||||
* Removes all session and player data associated with the session of
|
||||
* the given ID, meaning these stats will be lost forever.
|
||||
*
|
||||
* @param sessionId the ID of the session whose data to delete
|
||||
*/
|
||||
void delete(UUID sessionId);
|
||||
|
||||
// TODO: docs
|
||||
GlobalStats getGlobalStats();
|
||||
|
||||
// TODO: docs
|
||||
ArenaStats getArenaStats(String slug);
|
||||
|
||||
// TODO: docs
|
||||
PlayerStats getPlayerStats(String name);
|
||||
|
||||
// TODO: docs
|
||||
void export(StatsStore target) throws IOException;
|
||||
|
||||
}
|
||||
@@ -0,0 +1,22 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
|
||||
@FunctionalInterface
|
||||
public interface StatsStoreFactory {
|
||||
|
||||
/**
|
||||
* Create a new stats store from the given configuration.
|
||||
*
|
||||
* @param config a configuration to set up the stats store
|
||||
* @param plugin a plugin instance for dependencies
|
||||
* @return a new stats store instance
|
||||
* @throws Exception if store creation fails
|
||||
*/
|
||||
StatsStore create(
|
||||
ConfigurationSection config,
|
||||
MobArenaStats plugin
|
||||
) throws Exception;
|
||||
|
||||
}
|
||||
@@ -0,0 +1,41 @@
|
||||
package org.mobarena.stats.store;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.mobarena.stats.MobArenaStatsPlugin;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
public class StatsStoreRegistry {
|
||||
|
||||
private final Map<String, StatsStoreFactory> typeToFactory;
|
||||
private final MobArenaStatsPlugin plugin;
|
||||
|
||||
StatsStoreRegistry(MobArenaStatsPlugin plugin) {
|
||||
this.typeToFactory = new HashMap<>();
|
||||
this.plugin = plugin;
|
||||
}
|
||||
|
||||
public void register(String type, StatsStoreFactory factory) {
|
||||
typeToFactory.put(type.toLowerCase(), factory);
|
||||
}
|
||||
|
||||
public StatsStore create(ConfigurationSection config) throws Exception {
|
||||
String type = config.getString("type");
|
||||
if (type == null || type.isEmpty()) {
|
||||
throw new IllegalArgumentException("Missing 'type' in store configuration");
|
||||
}
|
||||
|
||||
StatsStoreFactory factory = typeToFactory.get(type.toLowerCase());
|
||||
if (factory == null) {
|
||||
throw new IllegalArgumentException("Unknown store type: " + type);
|
||||
}
|
||||
|
||||
return factory.create(config, plugin);
|
||||
}
|
||||
|
||||
public static StatsStoreRegistry create(MobArenaStatsPlugin plugin) {
|
||||
return new StatsStoreRegistry(plugin);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,253 @@
|
||||
package org.mobarena.stats.store.csv;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.session.PlayerConclusion;
|
||||
import org.mobarena.stats.session.PlayerSessionStats;
|
||||
import org.mobarena.stats.session.Session;
|
||||
import org.mobarena.stats.session.SessionConclusion;
|
||||
import org.mobarena.stats.session.SessionStats;
|
||||
import org.mobarena.stats.store.ArenaStats;
|
||||
import org.mobarena.stats.store.GlobalStats;
|
||||
import org.mobarena.stats.store.PlayerStats;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
|
||||
import java.io.BufferedWriter;
|
||||
import java.io.File;
|
||||
import java.io.FileWriter;
|
||||
import java.io.IOException;
|
||||
import java.io.PrintWriter;
|
||||
import java.nio.file.Files;
|
||||
import java.time.Instant;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
import java.util.function.Function;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
public class CsvStatsStore implements StatsStore {
|
||||
|
||||
private static final String[] SESSION_FIELDS = {
|
||||
"session_id",
|
||||
"arena_slug",
|
||||
"start_time",
|
||||
"end_time",
|
||||
"last_wave",
|
||||
"conclusion"
|
||||
};
|
||||
|
||||
private static final String[] PLAYER_SESSION_FIELDS = {
|
||||
"session_id",
|
||||
"player_id",
|
||||
"player_name",
|
||||
"class",
|
||||
"join_time",
|
||||
"ready_time",
|
||||
"leave_time",
|
||||
"death_time",
|
||||
"kills",
|
||||
"dmg_done",
|
||||
"dmg_taken",
|
||||
"swings",
|
||||
"hits",
|
||||
"last_wave",
|
||||
"conclusion"
|
||||
};
|
||||
|
||||
private final File folder;
|
||||
private final File sessionsFile;
|
||||
private final File playersFile;
|
||||
private final String separator;
|
||||
private final DateTimeFormatter formatter;
|
||||
private final Logger log;
|
||||
|
||||
private CsvStatsStore(
|
||||
File folder,
|
||||
String separator,
|
||||
Logger log
|
||||
) {
|
||||
this.folder = folder;
|
||||
this.sessionsFile = new File(folder, "sessions.csv");
|
||||
this.playersFile = new File(folder, "players.csv");
|
||||
this.separator = separator;
|
||||
this.formatter = DateTimeFormatter.ISO_INSTANT;
|
||||
this.log = log;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void save(Session session) throws IOException {
|
||||
try {
|
||||
createDataFolder();
|
||||
saveArenaSession(session);
|
||||
savePlayerSessions(session);
|
||||
} catch (Exception e) {
|
||||
throw new IOException(e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void delete(UUID sessionId) {
|
||||
throw new UnsupportedOperationException("Session deletion is not supported by the CSV data store");
|
||||
}
|
||||
|
||||
private void createDataFolder() {
|
||||
if (!folder.exists()) {
|
||||
if (!folder.mkdirs()) {
|
||||
throw new IllegalStateException("Failed to create stats data folder");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void saveArenaSession(Session session) throws Exception {
|
||||
boolean writeHeader = !sessionsFile.exists();
|
||||
|
||||
try (PrintWriter writer = new PrintWriter(new BufferedWriter(new FileWriter(sessionsFile, true)))) {
|
||||
if (writeHeader) {
|
||||
String line = String.join(separator, SESSION_FIELDS);
|
||||
writer.println(line);
|
||||
}
|
||||
|
||||
SessionStats stats = session.getSessionStats();
|
||||
String line = String.join(
|
||||
separator,
|
||||
stats.sessionId.toString(),
|
||||
stats.arenaSlug,
|
||||
formatter.format(stats.startTime),
|
||||
formatter.format(stats.endTime),
|
||||
String.valueOf(stats.lastWave),
|
||||
String.valueOf(stats.conclusion)
|
||||
);
|
||||
writer.println(line);
|
||||
log.info("Session stats written to disk (" + stats.sessionId + ").");
|
||||
}
|
||||
}
|
||||
|
||||
private void savePlayerSessions(Session session) throws Exception {
|
||||
boolean writeHeader = !playersFile.exists();
|
||||
|
||||
try (PrintWriter writer = new PrintWriter(new BufferedWriter(new FileWriter(playersFile, true)))) {
|
||||
if (writeHeader) {
|
||||
String line = String.join(separator, PLAYER_SESSION_FIELDS);
|
||||
writer.println(line);
|
||||
}
|
||||
|
||||
for (PlayerSessionStats stats : session.getPlayerStats()) {
|
||||
String line = String.join(
|
||||
separator,
|
||||
stats.sessionId.toString(),
|
||||
stats.playerId.toString(),
|
||||
stats.playerName,
|
||||
stats.className,
|
||||
formatter.format(stats.joinTime),
|
||||
formatter.format(stats.readyTime),
|
||||
stats.leaveTime != null ? formatter.format(stats.leaveTime) : "",
|
||||
stats.deathTime != null ? formatter.format(stats.deathTime) : "",
|
||||
String.valueOf(stats.kills),
|
||||
String.valueOf(stats.dmgDone),
|
||||
String.valueOf(stats.dmgTaken),
|
||||
String.valueOf(stats.swings),
|
||||
String.valueOf(stats.hits),
|
||||
String.valueOf(stats.lastWave),
|
||||
String.valueOf(stats.conclusion)
|
||||
);
|
||||
writer.println(line);
|
||||
}
|
||||
log.info("Player stats written to disk (" + session.getSessionStats().sessionId + ").");
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public GlobalStats getGlobalStats() {
|
||||
throw new UnsupportedOperationException("Queries are not supported by the CSV data store");
|
||||
}
|
||||
|
||||
@Override
|
||||
public ArenaStats getArenaStats(String slug) {
|
||||
throw new UnsupportedOperationException("Queries are not supported by the CSV data store");
|
||||
}
|
||||
|
||||
@Override
|
||||
public PlayerStats getPlayerStats(String name) {
|
||||
throw new UnsupportedOperationException("Queries are not supported by the CSV data store");
|
||||
}
|
||||
|
||||
@Override
|
||||
public void export(StatsStore target) throws IOException {
|
||||
List<String> sessionLines = Files.readAllLines(sessionsFile.toPath());
|
||||
List<String> playerLines = Files.readAllLines(playersFile.toPath());
|
||||
|
||||
for (int i = 1; i < sessionLines.size(); i++) {
|
||||
String sessionLine = sessionLines.get(i);
|
||||
String[] sessionParts = sessionLine.split(separator);
|
||||
|
||||
UUID sessionId = UUID.fromString(sessionParts[0]);
|
||||
String arenaSlug = sessionParts[1];
|
||||
Session session = new Session(sessionId, arenaSlug);
|
||||
{
|
||||
SessionStats stats = session.getSessionStats();
|
||||
stats.startTime = Instant.parse(sessionParts[2]);
|
||||
stats.endTime = Instant.parse(sessionParts[3]);
|
||||
stats.lastWave = Integer.parseInt(sessionParts[4]);
|
||||
stats.conclusion = SessionConclusion.valueOf(sessionParts[5]);
|
||||
}
|
||||
|
||||
String prefix = sessionId + separator;
|
||||
for (int j = 1; j < playerLines.size(); j++) {
|
||||
String playerLine = playerLines.get(j);
|
||||
if (!playerLine.startsWith(prefix)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
String[] playerParts = playerLine.split(separator);
|
||||
{
|
||||
UUID playerId = UUID.fromString(playerParts[1]);
|
||||
String playerName = playerParts[2];
|
||||
|
||||
PlayerSessionStats stats = new PlayerSessionStats(sessionId, playerId, playerName);
|
||||
stats.className = playerParts[3];
|
||||
stats.joinTime = safe(playerParts[4], Instant::parse);
|
||||
stats.readyTime = safe(playerParts[5], Instant::parse);
|
||||
stats.leaveTime = safe(playerParts[6], Instant::parse);
|
||||
stats.deathTime = safe(playerParts[7], Instant::parse);
|
||||
stats.kills = Integer.parseInt(playerParts[8]);
|
||||
stats.dmgDone = Integer.parseInt(playerParts[9]);
|
||||
stats.dmgTaken = Integer.parseInt(playerParts[10]);
|
||||
stats.swings = Integer.parseInt(playerParts[11]);
|
||||
stats.hits = Integer.parseInt(playerParts[12]);
|
||||
stats.lastWave = Integer.parseInt(playerParts[13]);
|
||||
stats.conclusion = safe(playerParts[14], PlayerConclusion::valueOf, PlayerConclusion.DEFEAT);
|
||||
|
||||
session.setPlayerStats(playerId, stats);
|
||||
}
|
||||
}
|
||||
|
||||
target.save(session);
|
||||
}
|
||||
}
|
||||
|
||||
private static <T, R> R safe(T value, Function<T, R> parser) {
|
||||
return safe(value, parser, null);
|
||||
}
|
||||
|
||||
private static <T, R> R safe(T value, Function<T, R> parser, R def) {
|
||||
try {
|
||||
return parser.apply(value);
|
||||
} catch (Exception e) {
|
||||
return def;
|
||||
}
|
||||
}
|
||||
|
||||
public static CsvStatsStore create(
|
||||
ConfigurationSection config,
|
||||
MobArenaStats plugin
|
||||
) {
|
||||
String folder = config.getString("folder", "data");
|
||||
String separator = config.getString("separator", ";");
|
||||
|
||||
File root = new File(plugin.getDataFolder(), folder);
|
||||
Logger log = plugin.getLogger();
|
||||
|
||||
return new CsvStatsStore(root, separator, log);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,245 @@
|
||||
package org.mobarena.stats.store.jdbc;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.jdbi.v3.core.Jdbi;
|
||||
import org.jdbi.v3.core.mapper.RowMapper;
|
||||
import org.jdbi.v3.core.statement.PreparedBatch;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.session.PlayerConclusion;
|
||||
import org.mobarena.stats.session.PlayerSessionStats;
|
||||
import org.mobarena.stats.session.Session;
|
||||
import org.mobarena.stats.session.SessionConclusion;
|
||||
import org.mobarena.stats.session.SessionStats;
|
||||
import org.mobarena.stats.store.ArenaStats;
|
||||
import org.mobarena.stats.store.GlobalStats;
|
||||
import org.mobarena.stats.store.PlayerStats;
|
||||
import org.mobarena.stats.store.StatsStore;
|
||||
import org.mobarena.stats.util.ResourceLoader;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.sql.Timestamp;
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
import java.util.function.Function;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
import static org.mobarena.stats.store.jdbc.Statement.DELETE_SESSION_DATA;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.FIND_ARENA_STATS;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.FIND_GLOBAL_STATS;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.FIND_PLAYER_SESSIONS_BY_ID;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.FIND_PLAYER_STATS;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.FIND_SESSIONS;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.INSERT_PLAYER_DATA;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.INSERT_SESSION_DATA;
|
||||
|
||||
public class JdbcStatsStore implements StatsStore {
|
||||
|
||||
private final Jdbi jdbi;
|
||||
private final Statements statements;
|
||||
|
||||
private JdbcStatsStore(Jdbi jdbi, Statements statements) {
|
||||
this.jdbi = jdbi;
|
||||
this.statements = statements;
|
||||
}
|
||||
|
||||
@Override
|
||||
public synchronized void save(Session session) {
|
||||
jdbi.useTransaction(handle -> {
|
||||
// First the session data
|
||||
handle.createUpdate(statements.get(INSERT_SESSION_DATA))
|
||||
.bind("session_id", session.getSessionId().toString())
|
||||
.bind("arena_slug", session.getArenaSlug())
|
||||
.bind("start_time", session.getSessionStats().startTime)
|
||||
.bind("end_time", session.getSessionStats().endTime)
|
||||
.bind("last_wave", session.getSessionStats().lastWave)
|
||||
.bind("conclusion", session.getSessionStats().conclusion)
|
||||
.execute();
|
||||
|
||||
// Then all of the player data
|
||||
PreparedBatch batch = handle.prepareBatch(statements.get(INSERT_PLAYER_DATA));
|
||||
for (PlayerSessionStats player : session.getPlayerStats()) {
|
||||
batch.bind("session_id", session.getSessionId().toString());
|
||||
batch.bind("player_id", player.playerId.toString());
|
||||
batch.bind("player_name", player.playerName);
|
||||
batch.bind("class", player.className);
|
||||
batch.bind("join_time", player.joinTime);
|
||||
batch.bind("ready_time", player.readyTime);
|
||||
batch.bind("leave_time", player.leaveTime);
|
||||
batch.bind("death_time", player.deathTime);
|
||||
batch.bind("kills", player.kills);
|
||||
batch.bind("dmg_done", player.dmgDone);
|
||||
batch.bind("dmg_taken", player.dmgTaken);
|
||||
batch.bind("swings", player.swings);
|
||||
batch.bind("hits", player.hits);
|
||||
batch.bind("last_wave", player.lastWave);
|
||||
batch.bind("conclusion", player.conclusion);
|
||||
batch.add();
|
||||
}
|
||||
batch.execute();
|
||||
});
|
||||
}
|
||||
|
||||
@Override
|
||||
public void delete(UUID sessionId) {
|
||||
jdbi.useTransaction(handle -> handle
|
||||
.createUpdate(statements.get(DELETE_SESSION_DATA))
|
||||
.bind("session_id", sessionId.toString())
|
||||
.execute()
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public GlobalStats getGlobalStats() {
|
||||
return jdbi.withHandle(handle -> handle
|
||||
.createQuery(statements.get(FIND_GLOBAL_STATS))
|
||||
.map((rs, ctx) -> new GlobalStats(
|
||||
rs.getInt("total_sessions"),
|
||||
rs.getLong("total_seconds"),
|
||||
rs.getLong("total_kills"),
|
||||
rs.getLong("total_waves")
|
||||
))
|
||||
.first()
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public ArenaStats getArenaStats(String slug) {
|
||||
return jdbi.withHandle(handle -> handle
|
||||
.createQuery(statements.get(FIND_ARENA_STATS))
|
||||
.bind("arena_slug", slug)
|
||||
.map((rs, ctx) -> new ArenaStats(
|
||||
rs.getInt("highest_wave"),
|
||||
rs.getInt("highest_seconds"),
|
||||
rs.getInt("highest_kills"),
|
||||
rs.getInt("total_sessions"),
|
||||
rs.getLong("total_seconds"),
|
||||
rs.getLong("total_kills"),
|
||||
rs.getLong("total_waves")
|
||||
))
|
||||
.first()
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public PlayerStats getPlayerStats(String name) {
|
||||
return jdbi.withHandle(handle -> handle
|
||||
.createQuery(statements.get(FIND_PLAYER_STATS))
|
||||
.bind("player_name", name)
|
||||
.map((rs, ctx) -> new PlayerStats(
|
||||
rs.getInt("total_sessions"),
|
||||
rs.getLong("total_seconds"),
|
||||
rs.getLong("total_kills"),
|
||||
rs.getLong("total_waves")
|
||||
))
|
||||
.first()
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public synchronized void export(StatsStore target) throws IOException {
|
||||
jdbi.useHandle(handle -> {
|
||||
int limit = 100;
|
||||
int offset = 0;
|
||||
|
||||
while (true) {
|
||||
List<Session> sessions = handle.createQuery(statements.get(FIND_SESSIONS))
|
||||
.bind("limit", limit)
|
||||
.bind("offset", offset)
|
||||
.map(toSession())
|
||||
.list();
|
||||
|
||||
for (Session session : sessions) {
|
||||
UUID sessionId = session.getSessionId();
|
||||
|
||||
handle.createQuery(statements.get(FIND_PLAYER_SESSIONS_BY_ID))
|
||||
.bind("session_id", session.getSessionId().toString())
|
||||
.map(toPlayerStats(sessionId))
|
||||
.forEach(stats -> session.setPlayerStats(stats.playerId, stats));
|
||||
|
||||
target.save(session);
|
||||
}
|
||||
|
||||
if (sessions.size() < limit) {
|
||||
break;
|
||||
}
|
||||
|
||||
offset += limit;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private static RowMapper<Session> toSession() {
|
||||
return (r, ctx) -> {
|
||||
UUID sessionId = UUID.fromString(r.getString("session_id"));
|
||||
String arenaSlug = r.getString("arena_slug");
|
||||
Session session = new Session(sessionId, arenaSlug);
|
||||
|
||||
SessionStats stats = session.getSessionStats();
|
||||
stats.startTime = r.getTimestamp("start_time").toInstant();
|
||||
stats.endTime = r.getTimestamp("end_time").toInstant();
|
||||
stats.lastWave = r.getInt("last_wave");
|
||||
stats.conclusion = SessionConclusion.valueOf(r.getString("conclusion"));
|
||||
|
||||
return session;
|
||||
};
|
||||
}
|
||||
|
||||
private static RowMapper<PlayerSessionStats> toPlayerStats(UUID sessionId) {
|
||||
return (r, ctx) -> {
|
||||
UUID playerId = UUID.fromString(r.getString("player_id"));
|
||||
String playerName = r.getString("player_name");
|
||||
|
||||
PlayerSessionStats stats = new PlayerSessionStats(sessionId, playerId, playerName);
|
||||
stats.className = r.getString("class");
|
||||
stats.joinTime = safe(r.getTimestamp("join_time"), Timestamp::toInstant);
|
||||
stats.readyTime = safe(r.getTimestamp("ready_time"), Timestamp::toInstant);
|
||||
stats.leaveTime = safe(r.getTimestamp("leave_time"), Timestamp::toInstant);
|
||||
stats.deathTime = safe(r.getTimestamp("death_time"), Timestamp::toInstant);
|
||||
stats.kills = r.getInt("kills");
|
||||
stats.dmgDone = r.getInt("dmg_done");
|
||||
stats.dmgTaken = r.getInt("dmg_taken");
|
||||
stats.swings = r.getInt("swings");
|
||||
stats.hits = r.getInt("hits");
|
||||
stats.lastWave = r.getInt("last_wave");
|
||||
stats.conclusion = safe(r.getString("conclusion"), PlayerConclusion::valueOf, PlayerConclusion.DEFEAT);
|
||||
|
||||
return stats;
|
||||
};
|
||||
}
|
||||
|
||||
private static <T, R> R safe(T value, Function<T, R> parser) {
|
||||
return safe(value, parser, null);
|
||||
}
|
||||
|
||||
private static <T, R> R safe(T value, Function<T, R> parser, R def) {
|
||||
try {
|
||||
return parser.apply(value);
|
||||
} catch (Exception e) {
|
||||
return def;
|
||||
}
|
||||
}
|
||||
|
||||
public static JdbcStatsStore create(
|
||||
ConfigurationSection config,
|
||||
MobArenaStats plugin
|
||||
) throws Exception {
|
||||
String type = config.getString("type");
|
||||
String url = config.getString("url");
|
||||
String username = config.getString("username");
|
||||
String password = config.getString("password");
|
||||
Jdbi jdbi = Jdbi.create(url, username, password);
|
||||
|
||||
// Load up migrations and statements for the given type
|
||||
ResourceLoader loader = ResourceLoader.create(plugin.getClass().getClassLoader());
|
||||
Migrations migrations = Migrations.create(loader, type);
|
||||
Statements statements = Statements.create(loader, type);
|
||||
Logger log = plugin.getLogger();
|
||||
|
||||
// Bring database schema up to speed
|
||||
SchemaMigrator migrator = new SchemaMigrator(jdbi, migrations, statements, log);
|
||||
migrator.migrate();
|
||||
|
||||
return new JdbcStatsStore(jdbi, statements);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
package org.mobarena.stats.store.jdbc;
|
||||
|
||||
import org.mobarena.stats.util.ResourceLoader;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.URISyntaxException;
|
||||
import java.util.List;
|
||||
|
||||
class Migrations {
|
||||
|
||||
private final ResourceLoader loader;
|
||||
private final String type;
|
||||
|
||||
private Migrations(ResourceLoader loader, String type) {
|
||||
this.loader = loader;
|
||||
this.type = type;
|
||||
}
|
||||
|
||||
List<String> list() throws URISyntaxException, IOException {
|
||||
return loader.list(type + "/migration");
|
||||
}
|
||||
|
||||
String get(String filename) throws IOException {
|
||||
return loader.loadString(type + "/migration/" + filename);
|
||||
}
|
||||
|
||||
static Migrations create(ResourceLoader loader, String type) {
|
||||
return new Migrations(loader, type);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,139 @@
|
||||
package org.mobarena.stats.store.jdbc;
|
||||
|
||||
import org.jdbi.v3.core.Jdbi;
|
||||
import org.jdbi.v3.core.statement.Batch;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.URISyntaxException;
|
||||
import java.sql.Connection;
|
||||
import java.sql.DatabaseMetaData;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.time.Instant;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.logging.Logger;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import static org.mobarena.stats.store.jdbc.Statement.FIND_ALL_MIGRATIONS;
|
||||
import static org.mobarena.stats.store.jdbc.Statement.INSERT_MIGRATION;
|
||||
|
||||
class SchemaMigrator {
|
||||
|
||||
private final Jdbi jdbi;
|
||||
private final Migrations migrations;
|
||||
private final Statements statements;
|
||||
private final Logger log;
|
||||
|
||||
SchemaMigrator(
|
||||
Jdbi jdbi,
|
||||
Migrations migrations,
|
||||
Statements statements,
|
||||
Logger log
|
||||
) {
|
||||
this.jdbi = jdbi;
|
||||
this.migrations = migrations;
|
||||
this.statements = statements;
|
||||
this.log = log;
|
||||
}
|
||||
|
||||
void migrate() throws IOException, SQLException, URISyntaxException {
|
||||
List<String> filenames = migrations.list();
|
||||
List<String> completed = getCompletedMigrations();
|
||||
|
||||
filenames.removeAll(completed);
|
||||
|
||||
if (filenames.isEmpty()) {
|
||||
log.info("Schema is up to date.");
|
||||
return;
|
||||
}
|
||||
|
||||
if (completed.isEmpty()) {
|
||||
log.info("Schema is has not yet been initialized, migrating...");
|
||||
} else {
|
||||
log.info("Schema is " + filenames.size() + " version(s) behind, migrating...");
|
||||
}
|
||||
|
||||
for (String filename : filenames) {
|
||||
execute(filename);
|
||||
}
|
||||
|
||||
log.info("Schema migration complete.");
|
||||
}
|
||||
|
||||
private List<String> getCompletedMigrations() throws SQLException {
|
||||
return jdbi.withHandle(handle -> {
|
||||
// We don't really have a good way to check if the database has
|
||||
// migration info without making some actual queries, which will
|
||||
// fail if it doesn't. Instead, we can use the database metadata
|
||||
// (available via the underlying JDBC connection object) to find
|
||||
// out if the schema migrations table exists.
|
||||
Connection connection = handle.getConnection();
|
||||
DatabaseMetaData meta = connection.getMetaData();
|
||||
try (ResultSet tables = meta.getTables(null, null, "schema_migrations", null)) {
|
||||
while (tables.next()) {
|
||||
String name = tables.getString("TABLE_NAME");
|
||||
if (name.equals("schema_migrations")) {
|
||||
// Jackpot! We found the table, now query it.
|
||||
String sql = statements.get(FIND_ALL_MIGRATIONS);
|
||||
return handle.createQuery(sql)
|
||||
.map((r, ctx) -> r.getString("filename"))
|
||||
.list();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// No migrations table means fresh database.
|
||||
return Collections.emptyList();
|
||||
});
|
||||
}
|
||||
|
||||
private void execute(String filename) throws IOException {
|
||||
String content = migrations.get(filename);
|
||||
|
||||
// Migration files may contain several different statements,
|
||||
// but not all databases support multiple statements in a
|
||||
// single update, so we split the file contents by semicolon
|
||||
// and hold our breath while we invoke each part...
|
||||
List<String> parts = Arrays.stream(content.split(";"))
|
||||
.map(String::trim)
|
||||
.filter(s -> !s.isEmpty())
|
||||
.collect(Collectors.toList());
|
||||
|
||||
jdbi.useTransaction(handle -> {
|
||||
Instant executed = Instant.now();
|
||||
try {
|
||||
if (parts.size() == 1) {
|
||||
handle.execute(parts.get(0));
|
||||
} else {
|
||||
Batch batch = handle.createBatch();
|
||||
parts.forEach(batch::add);
|
||||
batch.execute();
|
||||
}
|
||||
|
||||
String sql = statements.get(INSERT_MIGRATION);
|
||||
handle.createUpdate(sql)
|
||||
.bind("filename", filename)
|
||||
.bind("executed", executed)
|
||||
.bind("success", true)
|
||||
.bind("error", (String) null)
|
||||
.execute();
|
||||
|
||||
log.info("\u2713 " + filename);
|
||||
} catch (Exception e) {
|
||||
String sql = statements.get(INSERT_MIGRATION);
|
||||
handle.createUpdate(sql)
|
||||
.bind("filename", filename)
|
||||
.bind("executed", executed)
|
||||
.bind("success", false)
|
||||
.bind("error", e.getMessage())
|
||||
.execute();
|
||||
|
||||
log.severe("\u2717 " + filename);
|
||||
throw new IllegalStateException("Migration failed: " + filename, e);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,19 @@
|
||||
package org.mobarena.stats.store.jdbc;
|
||||
|
||||
public enum Statement {
|
||||
|
||||
FIND_ALL_MIGRATIONS,
|
||||
INSERT_MIGRATION,
|
||||
|
||||
INSERT_SESSION_DATA,
|
||||
INSERT_PLAYER_DATA,
|
||||
DELETE_SESSION_DATA,
|
||||
|
||||
FIND_ARENA_STATS,
|
||||
FIND_GLOBAL_STATS,
|
||||
FIND_PLAYER_STATS,
|
||||
|
||||
FIND_SESSIONS,
|
||||
FIND_PLAYER_SESSIONS_BY_ID,
|
||||
|
||||
}
|
||||
@@ -0,0 +1,32 @@
|
||||
package org.mobarena.stats.store.jdbc;
|
||||
|
||||
import org.mobarena.stats.util.ResourceLoader;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.EnumMap;
|
||||
import java.util.Map;
|
||||
|
||||
class Statements {
|
||||
|
||||
private final Map<Statement, String> sql;
|
||||
|
||||
private Statements(Map<Statement, String> sql) {
|
||||
this.sql = sql;
|
||||
}
|
||||
|
||||
String get(Statement statement) {
|
||||
return sql.get(statement);
|
||||
}
|
||||
|
||||
static Statements create(ResourceLoader loader, String type) throws IOException {
|
||||
EnumMap<Statement, String> result = new EnumMap<>(Statement.class);
|
||||
for (Statement statement : Statement.values()) {
|
||||
// SCREAMING_SNAKE_CASE -> kebab-case, .sql file extension
|
||||
String filename = statement.toString().toLowerCase().replace('_', '-') + ".sql";
|
||||
String sql = loader.loadString(type + "/" + filename);
|
||||
result.put(statement, sql);
|
||||
}
|
||||
return new Statements(result);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,39 @@
|
||||
package org.mobarena.stats.store.mariadb;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.bukkit.configuration.MemoryConfiguration;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.store.jdbc.JdbcStatsStore;
|
||||
|
||||
public class MariadbStatsStore {
|
||||
|
||||
public static JdbcStatsStore create(
|
||||
ConfigurationSection config,
|
||||
MobArenaStats plugin
|
||||
) throws Exception {
|
||||
ConfigurationSection copy = new MemoryConfiguration();
|
||||
for (String key : config.getKeys(false)) {
|
||||
copy.set(key, config.get(key));
|
||||
}
|
||||
|
||||
// Note the type override of "mysql" here. This ensures that we
|
||||
// reuse the MySQL SQL files from the resources folder.
|
||||
String url = getUrl(copy);
|
||||
copy.set("type", "mysql");
|
||||
copy.set("url", url);
|
||||
|
||||
return JdbcStatsStore.create(copy, plugin);
|
||||
}
|
||||
|
||||
static String getUrl(ConfigurationSection config) {
|
||||
String host = config.getString("host", "localhost");
|
||||
int port = config.getInt("port", 3306);
|
||||
String database = config.getString("database", "mobarena_stats");
|
||||
boolean ssl = config.getBoolean("ssl", false);
|
||||
|
||||
String params = "useSSL=" + ssl;
|
||||
|
||||
return "jdbc:mariadb://" + host + ":" + port + "/" + database + "?" + params;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
package org.mobarena.stats.store.mysql;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.bukkit.configuration.MemoryConfiguration;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.store.jdbc.JdbcStatsStore;
|
||||
|
||||
public class MysqlStatsStore {
|
||||
|
||||
public static JdbcStatsStore create(
|
||||
ConfigurationSection config,
|
||||
MobArenaStats plugin
|
||||
) throws Exception {
|
||||
ConfigurationSection copy = new MemoryConfiguration();
|
||||
for (String key : config.getKeys(false)) {
|
||||
copy.set(key, config.get(key));
|
||||
}
|
||||
|
||||
String url = getUrl(copy);
|
||||
copy.set("type", "mysql");
|
||||
copy.set("url", url);
|
||||
|
||||
return JdbcStatsStore.create(copy, plugin);
|
||||
}
|
||||
|
||||
static String getUrl(ConfigurationSection config) {
|
||||
String host = config.getString("host", "localhost");
|
||||
int port = config.getInt("port", 3306);
|
||||
String database = config.getString("database", "mobarena_stats");
|
||||
boolean ssl = config.getBoolean("ssl", false);
|
||||
|
||||
String params = "useSSL=" + ssl;
|
||||
|
||||
return "jdbc:mysql://" + host + ":" + port + "/" + database + "?" + params;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
package org.mobarena.stats.store.sqlite;
|
||||
|
||||
import org.bukkit.configuration.ConfigurationSection;
|
||||
import org.bukkit.configuration.MemoryConfiguration;
|
||||
import org.mobarena.stats.MobArenaStats;
|
||||
import org.mobarena.stats.store.jdbc.JdbcStatsStore;
|
||||
|
||||
import java.io.File;
|
||||
|
||||
public class SqliteStatsStore {
|
||||
|
||||
public static JdbcStatsStore create(
|
||||
ConfigurationSection config,
|
||||
MobArenaStats plugin
|
||||
) throws Exception {
|
||||
ConfigurationSection copy = new MemoryConfiguration();
|
||||
for (String key : config.getKeys(false)) {
|
||||
copy.set(key, config.get(key));
|
||||
}
|
||||
|
||||
File data = plugin.getDataFolder();
|
||||
String url = getUrl(copy, data);
|
||||
copy.set("type", "sqlite");
|
||||
copy.set("url", url);
|
||||
copy.addDefault("username", "sa");
|
||||
copy.addDefault("password", "");
|
||||
|
||||
return JdbcStatsStore.create(copy, plugin);
|
||||
}
|
||||
|
||||
static String getUrl(ConfigurationSection config, File data) {
|
||||
String folder = data.getPath();
|
||||
String filename = config.getString("filename", "stats.db");
|
||||
|
||||
return "jdbc:sqlite:" + folder + "/" + filename;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,118 @@
|
||||
package org.mobarena.stats.util;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.FileNotFoundException;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.net.URI;
|
||||
import java.net.URISyntaxException;
|
||||
import java.net.URL;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.nio.file.FileSystem;
|
||||
import java.nio.file.FileSystems;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.Paths;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.NoSuchElementException;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* Lists and loads resources via a given {@link ClassLoader}.
|
||||
* <p>
|
||||
* The primary goal of this class is to provide a developer-friendly
|
||||
* abstraction over Java's complex concept of "resources", allowing
|
||||
* client components to focus their efforts on their own context.
|
||||
* <p>
|
||||
* In general, we know what resources we're looking for, and we just
|
||||
* want to load them into memory and apply them where needed, but we
|
||||
* also want to "scan" resource "folders". The latter is fairly easy
|
||||
* in a file system context, but in a jar-file, while still somewhat
|
||||
* doable, becomes a nightmare to have to do again and again. That's
|
||||
* where this class comes in, as a mild wrapper around something that
|
||||
* can best be described as infuriating.
|
||||
*/
|
||||
public class ResourceLoader {
|
||||
|
||||
private final ClassLoader loader;
|
||||
|
||||
ResourceLoader(ClassLoader loader) {
|
||||
this.loader = loader;
|
||||
}
|
||||
|
||||
/**
|
||||
* Find all resources that match the given path.
|
||||
*
|
||||
* @param prefix a resource "prefix" to filter resources by
|
||||
* @return a list of all resources that match the given prefix
|
||||
* @throws URISyntaxException if the given path isn't a valid URI
|
||||
* in the context of the class loader
|
||||
* @throws IOException if an I/O error occurs during traversal
|
||||
*/
|
||||
public List<String> list(String prefix) throws URISyntaxException, IOException {
|
||||
URL url = loader.getResource(prefix);
|
||||
if (url == null) {
|
||||
throw new NoSuchElementException("No resources found at " + prefix);
|
||||
}
|
||||
|
||||
URI uri = url.toURI();
|
||||
if (uri.getScheme().equals("jar")) {
|
||||
try (FileSystem fs = FileSystems.newFileSystem(uri, Collections.emptyMap())) {
|
||||
Path path = fs.getPath(prefix);
|
||||
return walk(path);
|
||||
}
|
||||
} else {
|
||||
Path path = Paths.get(uri);
|
||||
return walk(path);
|
||||
}
|
||||
}
|
||||
|
||||
private List<String> walk(Path path) throws IOException {
|
||||
// When we traverse the path, we want to skip the folder
|
||||
// denoted by the path itself, and this is always first
|
||||
// in the stream.
|
||||
return Files.walk(path, 1)
|
||||
.skip(1)
|
||||
.map(Path::getFileName)
|
||||
.map(Path::toString)
|
||||
.sorted()
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
|
||||
/**
|
||||
* Load the resource at the given path, interpreting its contents
|
||||
* as UTF-8 encoded text.
|
||||
*
|
||||
* @param path the path to the resource to load
|
||||
* @return the contents of the resource at the given path
|
||||
* @throws IOException if an I/O error occurs during loading
|
||||
*/
|
||||
public String loadString(String path) throws IOException {
|
||||
InputStream is = loader.getResourceAsStream(path);
|
||||
if (is == null) {
|
||||
throw new FileNotFoundException("Resource not found: " + path);
|
||||
}
|
||||
|
||||
ByteArrayOutputStream output = new ByteArrayOutputStream();
|
||||
byte[] buffer = new byte[1024];
|
||||
int length;
|
||||
while ((length = is.read(buffer)) != -1) {
|
||||
output.write(buffer, 0, length);
|
||||
}
|
||||
|
||||
return output.toString(StandardCharsets.UTF_8.name());
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new resource loader with the given {@link ClassLoader}
|
||||
* as its source.
|
||||
*
|
||||
* @param loader a class loader to use as a source of resources
|
||||
* @return a new resource loader
|
||||
*/
|
||||
public static ResourceLoader create(ClassLoader loader) {
|
||||
return new ResourceLoader(loader);
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user