244 lines
10 KiB
Java
244 lines
10 KiB
Java
package com.craftaro.ultimatestacker.database;
|
|
|
|
import com.craftaro.ultimatestacker.api.stack.block.BlockStack;
|
|
import com.craftaro.ultimatestacker.api.stack.spawner.SpawnerStack;
|
|
import com.craftaro.ultimatestacker.stackable.block.BlockStackImpl;
|
|
import com.craftaro.ultimatestacker.stackable.spawner.SpawnerStackImpl;
|
|
import com.songoda.core.compatibility.CompatibleMaterial;
|
|
import com.songoda.core.database.DataManagerAbstract;
|
|
import com.songoda.core.database.DatabaseConnector;
|
|
import com.songoda.core.database.DatabaseType;
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.Location;
|
|
import org.bukkit.World;
|
|
import org.bukkit.plugin.Plugin;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.PreparedStatement;
|
|
import java.sql.ResultSet;
|
|
import java.sql.SQLException;
|
|
import java.sql.Statement;
|
|
import java.util.Collection;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.function.Consumer;
|
|
|
|
public class DataManager extends DataManagerAbstract {
|
|
|
|
public DataManager(DatabaseConnector databaseConnector, Plugin plugin) {
|
|
super(databaseConnector, plugin);
|
|
}
|
|
|
|
public void bulkUpdateSpawners(Collection<SpawnerStack> spawnerStacks) {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String updateSpawner = "UPDATE " + this.getTablePrefix() + "spawners SET amount = ? WHERE id = ?";
|
|
PreparedStatement statement = connection.prepareStatement(updateSpawner);
|
|
for (SpawnerStack spawnerStack : spawnerStacks) {
|
|
statement.setInt(1, spawnerStack.getAmount());
|
|
statement.setInt(2, spawnerStack.getId());
|
|
statement.addBatch();
|
|
}
|
|
statement.executeBatch();
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public void updateSpawner(SpawnerStack spawnerStack) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
|
|
String updateSpawner = "UPDATE " + this.getTablePrefix() + "spawners SET amount = ? WHERE id = ?";
|
|
|
|
PreparedStatement statement = connection.prepareStatement(updateSpawner);
|
|
statement.setInt(1, spawnerStack.getAmount());
|
|
statement.setInt(2, spawnerStack.getId());
|
|
statement.executeUpdate();
|
|
} catch (SQLException exception) {
|
|
exception.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void createSpawner(SpawnerStack spawnerStack) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String createSpawner = "INSERT " + getSyntax("INTO ", DatabaseType.MYSQL) + getSyntax("OR REPLACE INTO ", DatabaseType.SQLITE) + this.getTablePrefix() + "spawners (amount, world, x, y, z) VALUES (?, ?, ?, ?, ?)";
|
|
PreparedStatement statement = connection.prepareStatement(createSpawner);
|
|
statement.setInt(1, spawnerStack.getAmount());
|
|
|
|
statement.setString(2, spawnerStack.getWorld().getName());
|
|
statement.setInt(3, spawnerStack.getX());
|
|
statement.setInt(4, spawnerStack.getY());
|
|
statement.setInt(5, spawnerStack.getZ());
|
|
statement.executeUpdate();
|
|
int spawnerId = this.lastInsertedId(connection, "spawners");
|
|
this.sync(() -> spawnerStack.setId(spawnerId));
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void updateBlock(BlockStack blockStack) {
|
|
if (blockStack.getAmount() == 0) return;
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String updateBlock = "UPDATE " + this.getTablePrefix() + "blocks SET amount = ? WHERE id = ?";
|
|
PreparedStatement statement = connection.prepareStatement(updateBlock);
|
|
if (blockStack.getAmount() == 0) return;
|
|
statement.setInt(1, blockStack.getAmount());
|
|
statement.setInt(2, blockStack.getId());
|
|
statement.executeUpdate();
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void createBlock(BlockStack blockStack) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String createSpawner = "INSERT " + getSyntax("INTO ", DatabaseType.MYSQL) + getSyntax("OR REPLACE INTO ", DatabaseType.SQLITE) + this.getTablePrefix() + "blocks (amount, material, world, x, y, z) VALUES (?, ?, ?, ?, ?, ?)";
|
|
PreparedStatement statement = connection.prepareStatement(createSpawner);
|
|
statement.setInt(1, blockStack.getAmount());
|
|
statement.setString(2, blockStack.getMaterial().name());
|
|
|
|
statement.setString(3, blockStack.getLocation().getWorld().getName());
|
|
statement.setInt(4, (int) blockStack.getLocation().getX());
|
|
statement.setInt(5, (int) blockStack.getLocation().getY());
|
|
statement.setInt(6, (int) blockStack.getLocation().getZ());
|
|
statement.executeUpdate();
|
|
int blockId = this.lastInsertedId(connection, "blocks");
|
|
this.sync(() -> ((BlockStackImpl)blockStack).setId(blockId));
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void deleteSpawner(SpawnerStack spawnerStack) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String deleteSpawner = "DELETE FROM " + this.getTablePrefix() + "spawners WHERE id = ?";
|
|
PreparedStatement statement = connection.prepareStatement(deleteSpawner);
|
|
statement.setInt(1, spawnerStack.getId());
|
|
statement.executeUpdate();
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void deleteBlock(BlockStackImpl blockStack) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String deleteBlock = "DELETE FROM " + this.getTablePrefix() + "blocks WHERE id = ?";
|
|
PreparedStatement statement = connection.prepareStatement(deleteBlock);
|
|
statement.setInt(1, blockStack.getId());
|
|
statement.executeUpdate();
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void getSpawners(Consumer<Map<Location, SpawnerStack>> callback) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()){
|
|
String selectSpawners = "SELECT * FROM " + this.getTablePrefix() + "spawners";
|
|
|
|
Map<Location, SpawnerStack> spawners = new HashMap<>();
|
|
|
|
try (Statement statement = connection.createStatement()) {
|
|
ResultSet result = statement.executeQuery(selectSpawners);
|
|
while (result.next()) {
|
|
World world = Bukkit.getWorld(result.getString("world"));
|
|
|
|
if (world == null)
|
|
continue;
|
|
|
|
int spawnerId = result.getInt("id");
|
|
|
|
int amount = result.getInt("amount");
|
|
|
|
int x = result.getInt("x");
|
|
int y = result.getInt("y");
|
|
int z = result.getInt("z");
|
|
Location location = new Location(world, x, y, z);
|
|
|
|
SpawnerStack spawnerStack = new SpawnerStackImpl(location, amount);
|
|
spawnerStack.setId(spawnerId);
|
|
spawners.put(location, spawnerStack);
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
this.sync(() -> callback.accept(spawners));
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void getBlocks(Consumer<Map<Location, BlockStack>> callback) {
|
|
this.runAsync(() -> {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String selectBlocks = "SELECT * FROM " + this.getTablePrefix() + "blocks";
|
|
|
|
Map<Location, BlockStack> blocks = new HashMap<>();
|
|
|
|
Statement statement = connection.createStatement();
|
|
|
|
ResultSet result = statement.executeQuery(selectBlocks);
|
|
while (result.next()) {
|
|
World world = Bukkit.getWorld(result.getString("world"));
|
|
|
|
if (world == null)
|
|
continue;
|
|
|
|
int blockId = result.getInt("id");
|
|
|
|
CompatibleMaterial material = CompatibleMaterial.getMaterial(result.getString("material"));
|
|
|
|
int amount = result.getInt("amount");
|
|
|
|
int x = result.getInt("x");
|
|
int y = result.getInt("y");
|
|
int z = result.getInt("z");
|
|
Location location = new Location(world, x, y, z);
|
|
|
|
BlockStackImpl blockStack = new BlockStackImpl(material, location, amount);
|
|
blockStack.setId(blockId);
|
|
if (amount == 0) {
|
|
//remove from database
|
|
this.deleteBlock(blockStack);
|
|
continue;
|
|
}
|
|
blocks.put(location, blockStack);
|
|
|
|
this.sync(() -> callback.accept(blocks));
|
|
}
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
});
|
|
}
|
|
|
|
public void bulkUpdateBlocks(Collection<BlockStack> stacks) {
|
|
try (Connection connection = this.databaseConnector.getConnection()) {
|
|
String updateSpawner = "UPDATE " + this.getTablePrefix() + "blocks SET amount = ? WHERE id = ?";
|
|
PreparedStatement statement = connection.prepareStatement(updateSpawner);
|
|
for (BlockStack spawnerStack : stacks) {
|
|
statement.setInt(1, spawnerStack.getAmount());
|
|
statement.setInt(2, spawnerStack.getId());
|
|
statement.addBatch();
|
|
}
|
|
statement.executeBatch();
|
|
} catch (Exception ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
}
|
|
}
|