Dom4jDemo应用-保存手机信息

---恢复内容开始---

 import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.OutputStreamWriter;
 import java.io.UnsupportedEncodingException;
 import java.util.ArrayList;
 import java.util.Iterator;
 import java.util.List;

 import org.dom4j.Document;
 import org.dom4j.DocumentException;
 import org.dom4j.Element;
 import org.dom4j.io.OutputFormat;
 import org.dom4j.io.SAXReader;
 import org.dom4j.io.XMLWriter;

 public class Dom4jDemo2 {
     private Document document;

     // 获得document对象
     public void loadDocument() {
         SAXReader saxReader = new SAXReader();  //相当于一个解析器
         try {
             document = saxReader.read(new File("newInfo.xml"));  //解析获得document对象
         } catch (DocumentException e) {
             e.printStackTrace();
         }
     }

     // 显示手机的品牌及型号
     public void showPhoneInfo() {
         // 获取XML根节点
         Element root = document.getRootElement();
         // 获取所有<Brand>
         Iterator eleBrands = root.elementIterator();
         while (eleBrands.hasNext()) {
             Element brand = (Element) eleBrands.next();
             System.out.println(brand.attributeValue("name"));
             //获取所有的<Type>
             Iterator eleTypes = brand.elementIterator();
             while (eleTypes.hasNext()) {
                 Element type = (Element) eleTypes.next();
                 System.out.println("\t" + type.attributeValue("name"));
             }
         }
     }

     /**
      * 把xml里手机品牌及型号放入集合里输出
      * @return
      */
     public  List<Brand> queryBrand(){
         //获取XML根节点
         Element root = document.getRootElement();

         List<Element> brands=root.elements();
         List<Brand> outBrands=new ArrayList<Brand>();
         for (Element eleBrand : brands) {
             Brand b=new Brand();
             b.setName(eleBrand.attribute("name").getValue());
             b.setId(Integer.parseInt(eleBrand.attribute("id").getValue()));

             List<String> types=new ArrayList<String>();
             List<Element> eleTypes=eleBrand.elements();
             for(Element elType:eleTypes) {
                 types.add(elType.attribute("name").getValue());
             }
             b.setTypes(types);

             outBrands.add(b);
         }

         return outBrands;
     }

     /**
      * 把集合内容写入XML里
      */
     public boolean saveBrand(List<Brand> brandList) {
         SAXReader reader=new SAXReader();
         document=reader.getDocumentFactory().createDocument();  //创建一个新的xml文档
         Element root=document.addElement("PhoneInfo");   //新增根元素<PhoneInfo>
         for (Brand brand : brandList) {
             Element eleBrand=root.addElement("Brand");
             eleBrand.addAttribute("name", brand.getName());
             eleBrand.addAttribute("id", brand.getId()+"");
             for(String type:brand.getTypes()) {
                 Element eleTypes=eleBrand.addElement("Type");
                 eleTypes.addAttribute("name", type);
             }

         }

         try {
             XMLWriter writer=new XMLWriter(new FileOutputStream(new File("newInfo.xml")),OutputFormat.createPrettyPrint());
             writer.write(document);
             writer.close();
         } catch (UnsupportedEncodingException | FileNotFoundException e) {
             e.printStackTrace();
             return false;
         } catch (IOException e) {
             e.printStackTrace();
             return false;
         }
         return true;
     }

     // 修改节点
     public void updatePhone() {
         Element root = document.getRootElement();
         Iterator eleBrands = root.elementIterator();
         int id = 0;
         while (eleBrands.hasNext()) {
             Element brand = (Element) eleBrands.next();
             id++;
             brand.addAttribute("id", id + "");
         }
         saveXML("newInfo.xml");
     }

     // 增加新的手机品牌信息
     public void addNewPhone() {
         // 获得XML根元素
         Element root = document.getRootElement();
         // 创建<Brand name="三星">
         Element elBrand = root.addElement("Brand");
         elBrand.addAttribute("name", "三星");
         // 创建<Type name="Note 9"/>
         Element elType = elBrand.addElement("Type");
         elType.addAttribute("name", "Note 9");
         Element elType1 = elBrand.addElement("Type");
         elType1.addAttribute("name", "S 10");
         saveXML("newInfo.xml");
     }

     // 保存修改到XML文件
     public void saveXML(String path) {
         OutputFormat format = OutputFormat.createPrettyPrint();
         format.setEncoding("GBK");
         XMLWriter writer = null;
         try {
             writer = new XMLWriter(new OutputStreamWriter(new FileOutputStream(path)), format);
             writer.write(document);
         } catch (FileNotFoundException e) {
             e.printStackTrace();
         } catch (IOException e) {
             e.printStackTrace();
         } finally {
             try {
                 writer.close();
             } catch (IOException e) {
                 e.printStackTrace();
             }
         }
     }

     // 删除节点
     public void deletePhone() {
         // 获取XML根节点
         Element root = document.getRootElement();
         // 获取所有<Brand>
         Iterator eleBrands = root.elementIterator();
         while (eleBrands.hasNext()) {
             Element brand = (Element) eleBrands.next();
             if(brand.attributeValue("name").equals("华为")) {
                 brand.getParent().remove(brand);
             }
         }
         saveXML("newInfo.xml");
     }

     public static void main(String[] args) {
         Dom4jDemo2 domDemo = new Dom4jDemo2();
         domDemo.loadDocument();
 //        domDemo.addNewPhone();
 //        domDemo.updatePhone();
 //        domDemo.deletePhone();
 //        domDemo.showPhoneInfo();

         List<Brand> brandList=domDemo.queryBrand();
         for (Brand brand : brandList) {
             System.out.println(brand);
         }

         /* List<String> typeList=new ArrayList<String>();
          typeList.add("小米Mix 2");
          typeList.add("小米Mix 3");
          typeList.add("小米Mix 2S");
          List<String> typeList2=new ArrayList<String>();
          typeList2.add("Mate 20 S");
          typeList2.add("Mate 20 Pro");
          typeList2.add("Mate 20 ");
          List<String> typeList3=new ArrayList<String>();
          typeList3.add("iPhone XS");
          typeList3.add("iPhone XS Max");
          typeList3.add("iPhone XR ");
         Brand b1=new Brand(01,"小米",typeList);
         Brand b2=new Brand(02,"华为",typeList2);
         Brand b3=new Brand(03,"苹果",typeList3);
         List<Brand> brandList=new ArrayList<Brand>();
         brandList.add(b1);
         brandList.add(b2);
         brandList.add(b3);
         boolean c=domDemo.saveBrand(brandList);
         System.out.println(c);*/
     }
 }

 import java.util.List;

 public class Brand {
     private int id;
     private String name;
     private List<String>types;

     public Brand() {
     }

     public Brand(int id, String name, List<String> types) {
         this.id = id;
         this.name = name;
         this.types = types;
     }

     public int getId() {
         return id;
     }
     public void setId(int id) {
         this.id = id;
     }
     public String getName() {
         return name;
     }
     public void setName(String name) {
         this.name = name;
     }
     public List<String> getTypes() {
         return types;
     }
     public void setTypes(List<String> types) {
         this.types = types;
     }
     @Override
     public String toString() {
         return this.name+this.types;
     }

 }

---恢复内容结束---