import java.sql.*;

public class AddMainMenuRole {
    public static void main(String[] args) {
        String url = "jdbc:dm://192.168.0.145:30236";
        String username = "WATERSHED";
        String password = "WaterShed./1224";
        
        try {
            // 加载DM JDBC驱动
            Class.forName("dm.jdbc.driver.DmDriver");
            
            // 建立数据库连接
            Connection conn = DriverManager.getConnection(url, username, password);
            System.out.println("数据库连接成功！");
            
            // 检查是否已存在主菜单的角色关联
            String checkSql = "SELECT COUNT(*) FROM SYS_ROLE_MENU WHERE role_id = 1 AND menu_id = 2000";
            Statement checkStmt = conn.createStatement();
            ResultSet rs = checkStmt.executeQuery(checkSql);
            rs.next();
            int count = rs.getInt(1);
            
            if (count == 0) {
                // 添加主菜单的角色关联
                String sql = "INSERT INTO SYS_ROLE_MENU (role_id, menu_id) VALUES (?, ?)";
                PreparedStatement pstmt = conn.prepareStatement(sql);
                pstmt.setInt(1, 1);
                pstmt.setInt(2, 2000);
                pstmt.executeUpdate();
                System.out.println("成功添加流域管理主菜单的角色关联");
                pstmt.close();
            } else {
                System.out.println("流域管理主菜单的角色关联已存在");
            }
            
            // 关闭连接
            rs.close();
            checkStmt.close();
            conn.close();
            
            System.out.println("操作完成！");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}