XML利用dom4j进行增删查改等方法

女爷i 2022-06-11 09:22 215阅读 0赞

public class XMLUtil { 
        // dom4j读取XML
        public static Document getDocutment(String inPath) {
            Document doc = null;
            try {
                doc = new SAXReader().read(new File(inPath));
            } catch (DocumentException e) {
                e.printStackTrace();
            }
            return doc;
        }
    
        // dom4j写入XML
        public static boolean WriterXML(String outPath, Document doc) {
            boolean type = true;
            OutputFormat format = new OutputFormat();
            format.setEncoding(System.getProperty("file.encoding"));
            FileWriter fw = null;
            XMLWriter writer = null;
            try {
                fw = new FileWriter(outPath);
                writer = new XMLWriter(fw,format);// 构造方法中两个参数,文件对象和编码格式
                writer.write(doc);
            } catch (IOException e) {
                e.printStackTrace();
                type = false;
            } finally {
                try {
                    if (writer != null) {
                        writer.close();     //无论写入成功还是失败都要执行非空的写入关闭
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                    type = false;
                }
            }
            return type;
        }
    
        //查询所有的信息
        public static List<Team> queryAll(){
            List<Team> al= new ArrayList<>();
            try {
                ///Project07/src/cn/lanou/xml/teaminfo.xml
                //读取文件到document
    // Document doc=new SAXReader().read(new File("src/cn/lanou/xml/team.xml"));
                Document doc=XMLUtil.getDocutment("Soccer.xml");
                //获取根节点
                Element root=doc.getRootElement();
                //获取根节点的子节点
                List<Element> list= root.elements();
                //通过集合找到每一个节点对象
                for (Element elm : list) {
                    //当前节点的子节点内容
                    String no=elm.elementText("no");
                    String name=elm.elementText("name");
                    String address=elm.elementText("address");
                    Team team = new Team(no,name,address);
                    al.add(team);
                }
    // XMLUtil.WriterXML("Soccer2.xml", doc);
            } catch (Exception e) {
                e.printStackTrace();
            }
            return al;
        }
    
    // 添加方法
    // 子节点添加addElement(elementName)
    // 新节点内容 newElement.setText("content");
    
    
        //添加一条队伍信息
        //要知道添加信息的结构,内容 编号 名称 地址
        public static boolean add(Team team){
    
    // Scanner sc = new Scanner(System.in);
    // System.out.println("编号:");
    // String no = sc.next();
    // System.out.println("队名:");
    // String name = sc.next();
    // System.out.println("所在城市:");
    // String address = sc.next();
    
            //操作XML文件
            Document doc = getDocutment("Soccer.xml");
            //获取根节点 每一个XML都只有一个根节点
            Element root = doc.getRootElement();
            //在根节点下添加子节点,叫做team
            Element newElm = root.addElement("team");
    // //新建的子节点添加子节点
    // Element nElm = newElm.addElement("no");
    // //id子节点添加内容
    // nElm.setText(no);
    
            //新建的子节点添加子节点并添加内容
            newElm.addElement("no").setText(team.getNo());
            //添加name节点并添加内容
            newElm.addElement("name").setText(team.getName());
            newElm.addElement("address").setText(team.getAddress());
    
            //将内存中更改的数据写入XML文件中
            return (WriterXML("Soccer.xml",doc));
    
        }
    
    
        //修改
        public static boolean update (Team team){
            //读取XML文件
            Document doc = getDocutment("Soccer.xml");
            //获取根节点
            Element root = doc.getRootElement();
            //获取根节点下的子节点集合
            List<Element> list = root.elements();
            //遍历集合中每一个节点,找到对应no的节点
            //对数据的限制,获得信息中,编号no必须唯一
            for (Element ele : list) {
                //获取每一个节点的no子节点的内容
                String noElmText = ele.elementText("no");
                //判断参数no是否相等
                if(noElmText.equals(team.getNo())){
                    //修改操作
                    //修改name节点的内容为参数内容
                    ele.element("name").setText(team.getName());
                    ele.element("address").setText(team.getAddress());
                    break;
                }
            }
            //写入XML文件
            return WriterXML("Soccer.xml", doc);
        }
    
        //删除
        public static boolean delete(String no){
            //读取XML文件
            Document doc = getDocutment("Soccer.xml");
            Element root = doc.getRootElement();
            List<Element> list = root.elements();
            for (Element ele : list) {
                //判断no
                String noElmText = ele.elementText("no");
                if(no.equals(noElmText)){
                    //删除当前节点下的子节点即team
                    root.remove(ele);
    
    // //尝试删除当前节点下的name节点
    // ele.remove(ele.element("name"));
    
                    break;
                }
            }
            //写入XML文件
            return WriterXML("Soccer.xml", doc);
        }
    
        //根据编号,查询一条队伍信息
        public static List<Team> queryOne(String id){
            List<Team> al= new ArrayList<>();
            try {
                ///Project07/src/cn/lanou/xml/teaminfo.xml
                //读取文件到document
    // Document doc=new SAXReader().read(new File("src/cn/lanou/xml/team.xml"));
                Document doc=XMLUtil.getDocutment("Soccer.xml");
                //获取根节点
                Element root=doc.getRootElement();
                //获取根节点的子节点
                List<Element> list= root.elements();
                //通过集合找到每一个节点对象
                for (Element elm : list) {
                    //当前节点的子节点内容
                    String no=elm.elementText("no");
                    String name=elm.elementText("name");
                    String address=elm.elementText("address");
                    Team team = new Team(no,name,address);
                    if(no.equals(id)){
                        al.add(team);
                    }
                }
    // XMLUtil.WriterXML("Soccer2.xml", doc);
            } catch (Exception e) {
                e.printStackTrace();
            }
            return al;
    
        }
    }