AllMusic/core/sql/DataSql.java

package coloryr.allmusic.core.sql;

import coloryr.allmusic.core.AllMusic;
import coloryr.allmusic.core.music.play.PlayMusic;
import coloryr.allmusic.core.objs.enums.HudDirType;
import coloryr.allmusic.core.utils.HudUtils;
import coloryr.allmusic.core.objs.config.SaveObj;
import coloryr.allmusic.core.objs.hud.PosObj;

import java.io.File;
import java.sql.*;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.Semaphore;

public class DataSql {
    private static final Queue<Runnable> tasks = new ConcurrentLinkedQueue<>();
    private static final Semaphore semaphore = new Semaphore(0);
    /**
     * 创建表用
     */
    private static final String table = "CREATE TABLE IF NOT EXISTS \"allmusic\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"name\" TEXT(20),\n" +
            "  \"info_x\" integer(6),\n" +
            "  \"info_y\" integer(6),\n" +
            "  \"info_color\" integer(20),\n" +
            "  \"info_dir\" integer(6),\n" +
            "  \"info_shadow\" integer(1),\n" +
            "  \"info_enable\" integer(1),\n" +
            "  \"lyric_x\" integer(6),\n" +
            "  \"lyric_y\" integer(6),\n" +
            "  \"lyric_color\" integer(20),\n" +
            "  \"lyric_dir\" integer(6),\n" +
            "  \"lyric_shadow\" integer(1),\n" +
            "  \"lyric_enable\" integer(1),\n" +
            "  \"list_x\" integer(6),\n" +
            "  \"list_y\" integer(6),\n" +
            "  \"list_color\" integer(20),\n" +
            "  \"list_dir\" integer(6),\n" +
            "  \"list_shadow\" integer(1),\n" +
            "  \"list_enable\" integer(1),\n" +
            "  \"pic_x\" integer(6),\n" +
            "  \"pic_y\" integer(6),\n" +
            "  \"pic_dir\" integer(6),\n" +
            "  \"pic_size\" integer(6),\n" +
            "  \"pic_enable\" integer(1),\n" +
            "  \"pic_rotate\" integer(1),\n" +
            "  \"pic_rotate_speed\" integer(6)\n" +
            ");";

    private static final String table1 = "CREATE TABLE IF NOT EXISTS \"allmusic_list\" (\n" +
            "  \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n" +
            "  \"sid\" TEXT(40)\n" +
            ");";

    /**
     * 数据库文件
     */
    public static File sqlFile;
    private static boolean isRun;
    /**
     * 数据库链接对象
     */
    private static Connection connection;

    /**
     * 初始化数据库
     */
    public static void init() {
        try {
            AllMusic.log.info("正在初始化数据库");
            if (connection != null)
                connection.close();
            Class.forName("org.sqlite.JDBC");
            connection = DriverManager.getConnection("jdbc:sqlite:" + sqlFile.getPath());
            Statement stat = connection.createStatement();
            stat.execute(table);
            stat.execute(table1);
            stat.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 检查玩家是否在数据库
     *
     * @param name 用户名
     * @return 结果
     */
    public static boolean check(String name) {
        try {
            boolean have = false;
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT allmusic.id FROM allmusic WHERE allmusic.name ='" + name + "'");
            if (set.next()) {
                have = true;
            }
            set.close();
            stat.close();
            return have;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 更新玩家Hud数据
     *
     * @param name 用户名
     * @param hud  Hud数据
     */
    private static void update(String name, SaveObj hud) {
        String sql = "";
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            sql = MessageFormat.format("UPDATE allmusic SET info_x={0},info_y={1},info_color={15},info_dir={16},info_shadow={17},info_enable={2},lyric_x={3},lyric_y={4},lyric_color={18},lyric_dir={19},lyric_shadow={20},lyric_enable={5},list_x={6},list_y={7}, list_color={21},list_dir={22},list_shadow={23},list_enable={8},pic_x={9},pic_y={10},pic_dir={24},pic_enable={11},pic_size={12},pic_rotate={13},pic_rotate_speed={14} WHERE name=@name",
                    hud.info.x, hud.info.y, hud.info.enable ? 1 : 0,
                    hud.lyric.x, hud.lyric.y, hud.lyric.enable ? 1 : 0,
                    hud.list.x, hud.list.y, hud.list.enable ? 1 : 0,
                    hud.pic.x, hud.pic.y, hud.pic.enable ? 1 : 0,
                    hud.pic.color, hud.pic.shadow ? 1 : 0,
                    hud.picRotateSpeed, hud.info.color, hud.info.dir.ordinal(),
                    hud.info.shadow ? 1 : 0, hud.lyric.color, hud.lyric.dir.ordinal(),
                    hud.lyric.shadow ? 1 : 0, hud.list.color, hud.list.dir.ordinal(),
                    hud.list.shadow ? 1 : 0, hud.pic.dir.ordinal());
            sql = sql.replace("@name", "'" + name + "'");
            stat.execute(sql);
            stat.close();
        } catch (Exception e) {
            AllMusic.log.info(sql);
            e.printStackTrace();
        }
    }

    /**
     * 添加用户Hud数据
     *
     * @param name 用户名
     * @param hud  数据
     */
    public static void addUser(String name, SaveObj hud) {
        String sql = "";
        try {
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            if (check(name)) {
                update(name, hud);
            } else {
                Statement stat = connection.createStatement();
                sql = MessageFormat.format("INSERT INTO allmusic (name,info_x,info_y," +
                                "info_enable,lyric_x,lyric_y,lyric_enable,list_x,list_y,list_enable," +
                                "pic_x,pic_y,pic_enable,pic_size,pic_rotate,pic_rotate_speed," +
                                "info_color,info_dir,info_shadow,lyric_color,lyric_dir," +
                                "lyric_shadow,list_color,list_dir,list_shadow,pic_dir)" +
                                "VALUES (@name,{0},{1},{2},{3},{4},{5},{6},{7},{8},{9},{10},{11},{12},{13},{14},{15},{16},{17},{18},{19},{20},{21},{22},{23},{24})",
                        hud.info.x, hud.info.y, hud.info.enable ? 1 : 0,
                        hud.lyric.x, hud.lyric.y, hud.lyric.enable ? 1 : 0,
                        hud.list.x, hud.list.y, hud.list.enable ? 1 : 0,
                        hud.pic.x, hud.pic.y, hud.pic.enable ? 1 : 0,
                        hud.pic.color, hud.pic.shadow ? 1 : 0,
                        hud.picRotateSpeed, hud.info.color, hud.info.dir.ordinal(),
                        hud.info.shadow ? 1 : 0, hud.lyric.color, hud.lyric.dir.ordinal(),
                        hud.lyric.shadow ? 1 : 0, hud.list.color, hud.list.dir.ordinal(),
                        hud.list.shadow ? 1 : 0, hud.pic.dir.ordinal());
                sql = sql.replace("@name", "'" + name + "'");
                stat.execute(sql);
                stat.close();
            }
        } catch (Exception e) {
            AllMusic.log.info(sql);
            e.printStackTrace();
        }
    }

    /**
     * 读取所有数据
     */
    private static void readAll() {
        try {
            AllMusic.log.info("正在读取玩家数据");
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT name,info_x,info_y,info_enable,lyric_x," +
                    "lyric_y,lyric_enable,list_x,list_y,list_enable,pic_x,pic_y,pic_enable,pic_size," +
                    "pic_rotate,pic_rotate_speed,info_color,info_dir,info_shadow,lyric_color," +
                    "lyric_dir,lyric_shadow,list_color,list_dir,list_shadow,pic_dir FROM allmusic");
            HudDirType[] vas = HudDirType.values();
            while (set.next()) {
                String name = set.getString(1);
                SaveObj obj = new SaveObj();
                PosObj pos1 = new PosObj();
                pos1.x = set.getInt(2);
                pos1.y = set.getInt(3);
                obj.info = pos1;
                obj.info.enable = set.getInt(4) == 1;
                PosObj pos2 = new PosObj();
                pos2.x = set.getInt(5);
                pos2.y = set.getInt(6);
                obj.lyric = pos2;
                obj.lyric.enable = set.getInt(7) == 1;
                PosObj pos3 = new PosObj();
                pos3.x = set.getInt(8);
                pos3.y = set.getInt(9);
                obj.list = pos3;
                obj.list.enable = set.getInt(10) == 1;
                PosObj pos4 = new PosObj();
                pos4.x = set.getInt(11);
                pos4.y = set.getInt(12);
                obj.pic = pos4;
                obj.pic.enable = set.getInt(13) == 1;
                obj.pic.color = set.getInt(14);
                obj.pic.shadow = set.getInt(15) == 1;
                obj.picRotateSpeed = set.getInt(16);
                obj.info.color = set.getInt(17);
                obj.info.dir = vas[set.getInt(18)];
                obj.info.shadow = set.getInt(19) == 1;
                obj.lyric.color = set.getInt(20);
                obj.lyric.dir = vas[set.getInt(21)];
                obj.lyric.shadow = set.getInt(22) == 1;
                obj.list.color = set.getInt(23);
                obj.list.dir = vas[set.getInt(24)];
                obj.list.shadow = set.getInt(25) == 1;
                obj.list.dir = vas[set.getInt(26)];
                HudUtils.add(name, obj);
            }
            stat.close();
        } catch (Exception e) {
            AllMusic.log.warning("数据库读取错误，请删除关闭服务器删除数据库，在启动服务器");
            e.printStackTrace();
        }
    }

    /**
     * 读取空闲歌单列表
     */
    private static void readAllList(){
        try {
            AllMusic.log.info("正在读取空闲歌单");
            if (connection.isReadOnly() || connection.isClosed()) {
                init();
            }
            Statement stat = connection.createStatement();
            ResultSet set = stat.executeQuery("SELECT sid FROM allmusic_list");
            List<String> list = new ArrayList<>();
            while (set.next()) {
                String name = set.getString(1);
                list.add(name);
            }
            stat.close();

            PlayMusic.addIdleList(list);
        } catch (Exception e) {
            AllMusic.log.warning("数据库读取错误，请删除关闭服务器删除数据库，在启动服务器");
            e.printStackTrace();
        }
    }

    public static void addIdleList(List<String> list) {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                String arrayAsString = String.join(",", list);
                PreparedStatement pstmt = connection.prepareStatement("INSERT INTO allmusic_list (sid) VALUES (?)");
                pstmt.setString(1, arrayAsString);
                pstmt.execute();
                pstmt.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void clearIdleList() {
        task(() -> {
            try {
                if (connection.isReadOnly() || connection.isClosed()) {
                    init();
                }
                Statement stat = connection.createStatement();
                stat.execute("DELETE FROM allmusic_list");
                stat.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        });
    }

    public static void task(Runnable runnable) {
        tasks.add(runnable);
        semaphore.release();
    }

    /**
     * 启用Hud数据库
     */
    public static void start() {
        init();
        readAll();
        readAllList();
        Thread thread = new Thread(DataSql::run);
        isRun = true;
        thread.start();
    }

    /**
     * 停止数据库
     */
    public static void stop() {
        isRun = false;
        semaphore.release();
        HudUtils.save();
    }

    private static void run() {
        Runnable runnable;
        while (isRun) {
            try {
                semaphore.acquire();
                if (!isRun)
                    return;
                do {
                    runnable = tasks.poll();
                    if (runnable != null) {
                        runnable.run();
                    }
                }
                while (runnable != null);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}