Server/src/engine/db/handlers/dbRuneBaseHandler.java

//  • ▌ ▄ ·.  ▄▄▄·  ▄▄ • ▪   ▄▄· ▄▄▄▄·  ▄▄▄·  ▐▄▄▄  ▄▄▄ .
//  ·██ ▐███▪▐█ ▀█ ▐█ ▀ ▪██ ▐█ ▌▪▐█ ▀█▪▐█ ▀█ •█▌ ▐█▐▌·
//  ▐█ ▌▐▌▐█·▄█▀▀█ ▄█ ▀█▄▐█·██ ▄▄▐█▀▀█▄▄█▀▀█ ▐█▐ ▐▌▐▀▀▀
//  ██ ██▌▐█▌▐█ ▪▐▌▐█▄▪▐█▐█▌▐███▌██▄▪▐█▐█ ▪▐▌██▐ █▌▐█▄▄▌
//  ▀▀  █▪▀▀▀ ▀  ▀ ·▀▀▀▀ ▀▀▀·▀▀▀ ·▀▀▀▀  ▀  ▀ ▀▀  █▪ ▀▀▀
//      Magicbane Emulator Project © 2013 - 2026
//                https://www.magicbane.com


package engine.db.handlers;

import engine.gameManager.DbManager;
import engine.mbEnums;
import engine.objects.AbstractGameObject;
import engine.objects.BaseClass;
import engine.objects.PromotionClass;
import engine.objects.Race;
import engine.objects.RuneBase;
import engine.powers.RunePowerEntry;
import org.pmw.tinylog.Logger;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;

public class dbRuneBaseHandler extends dbHandlerBase {

    public dbRuneBaseHandler() {
        this.localClass = RuneBase.class;
        this.localObjectType = mbEnums.GameObjectType.valueOf(this.localClass.getSimpleName());
    }

    public static HashMap<Integer, ArrayList<RunePowerEntry>> LOAD_RUNE_POWERS() {

        HashMap<Integer, ArrayList<RunePowerEntry>> mobPowers = new HashMap<>();
        RunePowerEntry runePowerEntry;

        int rune_id;
        int recordsRead = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM static_rune_powers")) {

            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {

                recordsRead++;

                rune_id = rs.getInt("rune_id");
                runePowerEntry = new RunePowerEntry(rs);

                if (mobPowers.get(rune_id) == null) {
                    ArrayList<RunePowerEntry> runePowerList = new ArrayList<>();
                    runePowerList.add(runePowerEntry);
                    mobPowers.put(rune_id, runePowerList);
                } else {
                    ArrayList<RunePowerEntry> powerList = mobPowers.get(rune_id);
                    powerList.add(runePowerEntry);
                    mobPowers.put(rune_id, powerList);
                }
            }
        } catch (SQLException e) {
            Logger.error(e);
            return mobPowers;
        }

        Logger.info("read: " + recordsRead + " cached: " + mobPowers.size());
        return mobPowers;
    }

    // static_rune_lookup is generated for the DB views; a failed write must never stop the boot
    public static void WRITE_RUNE_LOOKUP() {

        int rowsWritten = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement deleteStatement = connection.prepareStatement("DELETE FROM `static_rune_lookup`");
             PreparedStatement insertStatement = connection.prepareStatement("INSERT INTO `static_rune_lookup` (`ID`, `type`, `name`) VALUES (?, ?, ?)")) {

            connection.setAutoCommit(false);

            try {
                deleteStatement.executeUpdate();

                for (Race race : Race.getAllRaces()) {
                    addRuneLookupRow(insertStatement, race.getRaceRuneID(), "RACE", race.getName());
                    rowsWritten++;
                }

                for (AbstractGameObject baseClass : DbManager.getMap(mbEnums.GameObjectType.BaseClass).values()) {
                    addRuneLookupRow(insertStatement, baseClass.getObjectUUID(), "BASECLASS", ((BaseClass) baseClass).getName());
                    rowsWritten++;
                }

                for (AbstractGameObject promotionClass : DbManager.getMap(mbEnums.GameObjectType.PromotionClass).values()) {
                    addRuneLookupRow(insertStatement, promotionClass.getObjectUUID(), "PROMOTION", ((PromotionClass) promotionClass).getName());
                    rowsWritten++;
                }

                insertStatement.executeBatch();
                connection.commit();
            } catch (SQLException | RuntimeException e) {
                // Roll back before setAutoCommit(true), which would otherwise commit the delete
                connection.rollback();
                throw e;
            } finally {
                connection.setAutoCommit(true);
            }

        } catch (Exception e) {
            Logger.warn("static_rune_lookup not written; DB views keep their old names: " + e.getMessage());
            return;
        }

        Logger.info("static_rune_lookup: wrote " + rowsWritten + " rows");
    }

    private static void addRuneLookupRow(PreparedStatement insertStatement, int runeID, String type, String name) throws SQLException {
        insertStatement.setInt(1, runeID);
        insertStatement.setString(2, type);
        insertStatement.setString(3, name);
        insertStatement.addBatch();
    }
}