cat.nyaa.nyaacore.orm.DatabaseUtils Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of rpgitems Show documentation

The legacy RPGItems continues. (fork)

There is a newer version: 3.12.2

package cat.nyaa.nyaacore.orm;

import cat.nyaa.nyaacore.orm.backends.BackendConfig;
import cat.nyaa.nyaacore.orm.backends.IConnectedDatabase;
import cat.nyaa.nyaacore.orm.backends.SQLiteDatabase;
import org.bukkit.plugin.Plugin;

import java.io.File;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;

public class DatabaseUtils {
    /**
     * @param cfg
     * @return
     */
    public static IConnectedDatabase connect(Plugin plugin, BackendConfig cfg) throws ClassNotFoundException, SQLException {
        if ("sqlite".equalsIgnoreCase(cfg.provider)) {
            return new SQLiteDatabase(newJdbcConnection(plugin, cfg));
        } else if ("mysql".equalsIgnoreCase(cfg.provider)) {
            throw new RuntimeException("NyaaCore ORM MySQL backend is not implemented");
        } else {
            throw new IllegalArgumentException("Invalid provider: " + cfg.provider);
        }
    }


    /**
     * @param plugin
     * @param cfg
     * @return
     * @throws ClassNotFoundException
     * @throws SQLException
     * @throws IllegalArgumentException
     */
    public static Connection newJdbcConnection(Plugin plugin, BackendConfig cfg) throws ClassNotFoundException, SQLException, IllegalArgumentException {
        String provider = cfg.provider;
        if ("sqlite".equalsIgnoreCase(provider)) {

            Class.forName("org.sqlite.JDBC");
            File f = new File(plugin.getDataFolder(), cfg.sqlite_file);
            return DriverManager.getConnection("jdbc:sqlite:" + f.getAbsolutePath());

        } else if ("mysql".equalsIgnoreCase(provider)) {

            Class.forName(cfg.mysql_jdbc_driver);
            return DriverManager.getConnection(cfg.mysql_url, cfg.mysql_username, cfg.mysql_password);

        } else {
            throw new IllegalArgumentException("Invalid provider: " + provider);
        }
    }
}