Sql.java

package fr.revivemc.data;

import fr.revivemc.Main;
import fr.revivemc.data.wrapper.Players;
import org.bukkit.entity.Player;

import java.io.File;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Map;
import java.util.UUID;

public class Sql {

    // Information of connection
    private Connection connection;

    /**
     * This function allows you to create the different tables
     * @param stmt A parameter to run SQL commands
     * @throws SQLException Returns an exception if a sql error occurred
     */
    private void createTable(Statement stmt) throws SQLException {
        stmt.execute("CREATE TABLE IF NOT EXISTS players (uuid TEXT PRIMARY KEY, kill INTEGER, death INTEGER)");
    }

    /**
     * Allows you to close the db
     */
    public void close() {
        try {
            if (connection != null && !connection.isClosed())
                connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    /**
     * Allows you to initialize the db when the server is launched
     */
    public void init() {
        try {
            Class.forName("org.sqlite.JDBC");

            File data_folder = Main.getPlugin(Main.class).getDataFolder();

            connection = DriverManager.getConnection("jdbc:sqlite:" + data_folder + "/data.db");
            Statement statement = connection.createStatement();
            createTable(statement);
            statement.close();
            System.out.println("Database connection established");
        } catch (SQLException | ClassNotFoundException e) {
            e.printStackTrace();
        }
    }


    /**
     * Save data in the db
     */
    public void saveData() {
        for (Map.Entry<UUID, Cache> cache : Main.cache.entrySet()) {
            Players.addPlayer(cache.getKey());
            Players.setKillByUuid(cache.getKey(), cache.getValue().getKills());
            Players.setDeathByUuid(cache.getKey(), cache.getValue().getDeaths());
        }
    }

    /**
     * @return Returns the connection
     */
    public Connection getConnection() {
        return (connection);
    }

}