package scripts.qa;

import java.io.File;
import java.nio.file.Files;
import java.sql.*;

public class CheckRound3aDatabase {
    public static void main(String[] args) throws Exception {
        Class.forName("com.kingbase8.Driver");
        File yml = new File("RuoYi-Vue/ruoyi-admin/src/main/resources/application-druid.yml");
        String pass = "";
        for (String line : Files.readAllLines(yml.toPath())) {
            if (line.trim().startsWith("password:")) {
                pass = line.trim().substring("password:".length()).trim();
            }
        }

        String dbUrl = "jdbc:kingbase8://127.0.0.1:14321/railwaysystem_integrated_qa_20260916?useUnicode=true&characterEncoding=utf8";
        try (Connection conn = DriverManager.getConnection(dbUrl, "kingbase", pass)) {
            try (Statement st = conn.createStatement()) {
                try (ResultSet rs = st.executeQuery("SELECT current_database(), current_user")) {
                    if (rs.next()) {
                        System.out.println("CURRENT_DATABASE=" + rs.getString(1));
                        System.out.println("CURRENT_USER=" + rs.getString(2));
                    }
                }

                String[] tables = new String[]{
                    "train_class_plan", "train_teacher", "train_student", "train_course",
                    "train_classroom_info", "train_training_site", "train_dormitory_info",
                    "train_lesson", "train_enrollment", "train_accommodation", "train_student_checkin"
                };

                for (String t : tables) {
                    try (ResultSet rs = st.executeQuery("SELECT count(*) FROM public." + t)) {
                        if (rs.next()) {
                            System.out.println("TABLE_" + t.toUpperCase() + "=" + rs.getInt(1));
                        }
                    }
                }
            }
        }
    }
}